/**
 * This file is part of the NocoBase (R) project.
 * Copyright (c) 2020-2024 NocoBase Co., Ltd.
 * Authors: NocoBase Team.
 *
 * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
 * For more information, please refer to: https://www.nocobase.com/agreement.
 */

import { Database } from '@nocobase/database';
import { AppSupervisor, Gateway } from '@nocobase/server';
import { createMockServer, MockServer } from '@nocobase/test';
import { uid } from '@nocobase/utils';
import { vi } from 'vitest';
import { PluginMultiAppManagerServer } from '../server';

const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));

describe('multiple apps', () => {
  let app: MockServer;
  let db: Database;

  beforeEach(async () => {
    app = await createMockServer({
      plugins: ['nocobase', 'multi-app-manager'],
    });
    db = app.db;
  });

  afterEach(async () => {
    await app.destroy();
  });

  it('should merge database options', async () => {
    const name = `td_${uid()}`;

    await db.getRepository('applications').create({
      values: {
        name,
        options: {
          plugins: [],
          database: {
            underscored: true,
          },
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    const subApp = await AppSupervisor.getInstance().getApp(name);

    expect(subApp.db.options.underscored).toBeTruthy();
  });

  it('should register db creator', async () => {
    const fn = vi.fn();

    const appPlugin = app.getPlugin<PluginMultiAppManagerServer>(PluginMultiAppManagerServer);
    const defaultDbCreator = appPlugin.appDbCreator;

    appPlugin.setAppDbCreator(async (app) => {
      fn();
      await defaultDbCreator(app);
    });

    const name = `td_${uid()}`;

    await db.getRepository('applications').create({
      values: {
        name,
        options: {
          plugins: [],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    await AppSupervisor.getInstance().removeApp(name);

    expect(fn).toBeCalled();
  });

  it('should create application', async () => {
    const name = `td_${uid()}`;

    await db.getRepository('applications').create({
      values: {
        name,
        options: {
          plugins: [],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    const subAppStatus = AppSupervisor.getInstance().getAppStatus(name);
    expect(subAppStatus).toEqual('running');
  });

  it('should not create application named with main', async () => {
    const name = 'main';

    let err;

    try {
      await db.getRepository('applications').create({
        values: {
          name,
          options: {
            plugins: [],
          },
        },
        context: {
          waitSubAppInstall: true,
        },
      });
    } catch (e) {
      err = e;
    }

    expect(err).toBeDefined();

    expect(await db.getRepository('applications').count()).toBe(0);
  });

  it('should upgrade sub app', async () => {
    await db.getRepository('applications').create({
      values: {
        name: 'test1',
        options: {
          plugins: ['nocobase'],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    await db.getRepository('applications').create({
      values: {
        name: 'test2',
        options: {
          plugins: ['nocobase'],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    await app.runCommand('restart');
    await app.runCommand('upgrade');
    // const subAppStatus = AppSupervisor.getInstance().getAppStatus(name);
    // expect(subAppStatus).toEqual('running');
    //
    // const subApp = await AppSupervisor.getInstance().getApp(name);
    // await subApp.runCommand('upgrade');
    //
    // await AppSupervisor.getInstance().removeApp(name);
    // expect(await db.getRepository('applications').count()).toBe(1);
  });

  it('should list application with status', async () => {
    const sub1 = `td_${uid()}`;
    await db.getRepository('applications').create({
      values: {
        name: sub1,
        options: {
          plugins: [],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    await AppSupervisor.getInstance().removeApp(sub1);

    await sleep(1000);

    const expectStatus = async (appName, status) => {
      const { body } = await app.agent().resource('applications').list();
      const { data } = body;

      const subApp = data.find((item) => item.name === appName);
      expect(subApp.status).toEqual(status);
    };

    await expectStatus(sub1, 'stopped');

    // start sub1
    // const startResponse = await app.agent().resource('applications').send({
    //   action: 'start',
    //   appName: sub1,
    // });

    // expect(startResponse.statusCode).toEqual(200);

    // await expectStatus(sub1, 'started');

    // const stopResponse = await app.agent().resource('applications').send({
    //   action: 'stop',
    //   appName: 'sub1',
    // });
    //
    // await expectStatus('sub1', 'stopped');
  });

  it('should remove application', async () => {
    const name = `td_${uid()}`;
    await db.getRepository('applications').create({
      values: {
        name,
        options: {
          plugins: [],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    expect(AppSupervisor.getInstance().hasApp(name)).toBeTruthy();

    await db.getRepository('applications').destroy({
      filter: {
        name,
      },
    });

    expect(AppSupervisor.getInstance().hasApp(name)).toBeFalsy();
  });

  it('should create with plugins', async () => {
    const name = `td_${uid()}`;
    await db.getRepository('applications').create({
      values: {
        name,
        options: {
          plugins: [['ui-schema-storage', { test: 'B' }]],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    const miniApp = await AppSupervisor.getInstance().getApp(name);
    expect(miniApp).toBeDefined();

    const plugin = miniApp.pm.get('ui-schema-storage');

    expect(plugin).toBeDefined();
    expect(plugin.options).toMatchObject({
      test: 'B',
    });
  });

  it('should lazy load applications', async () => {
    const name = `td_${uid()}`;

    // create app instance
    await db.getRepository('applications').create({
      values: {
        name,
        options: {
          plugins: ['ui-schema-storage'],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    // remove it from supervisor
    await AppSupervisor.getInstance().removeApp(name);

    expect(AppSupervisor.getInstance().hasApp(name)).toBeFalsy();

    Gateway.getInstance().addAppSelectorMiddleware((ctx) => (ctx.resolvedAppName = name));

    await AppSupervisor.getInstance().getApp(name);

    expect(AppSupervisor.getInstance().hasApp(name)).toBeTruthy();
  });

  it('should upgrade sub apps when main app upgrade', async () => {
    const subAppName = `t_${uid()}`;

    await app.db.getRepository('applications').create({
      values: {
        name: subAppName,
        options: {
          plugins: [],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    await AppSupervisor.getInstance().removeApp(subAppName);

    const jestFn = vi.fn();

    AppSupervisor.getInstance().on('afterAppAdded', (subApp) => {
      subApp.on('afterUpgrade', () => {
        jestFn();
      });
    });

    await app.runCommand('upgrade');

    expect(jestFn).toBeCalled();

    // sub app should remove after upgrade
    expect(AppSupervisor.getInstance().hasApp(subAppName)).toBeFalsy();
  });

  it('should start automatically', async () => {
    const subAppName = `t_${uid()}`;

    const subApp = await app.db.getRepository('applications').create({
      values: {
        name: subAppName,
        options: {
          plugins: [],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    await AppSupervisor.getInstance().removeApp(subAppName);

    await app.start();

    expect(AppSupervisor.getInstance().hasApp(subAppName)).toBeFalsy();

    await subApp.update({
      options: {
        autoStart: true,
      },
    });

    await AppSupervisor.getInstance().removeApp(subAppName);

    expect(AppSupervisor.getInstance().hasApp(subAppName)).toBeFalsy();
    await app.stop();

    await app.start();

    expect(AppSupervisor.getInstance().hasApp(subAppName)).toBeTruthy();
  });

  it('should start automatically with quick start', async () => {
    const subAppName = `t_${uid()}`;

    const subApp = await app.db.getRepository('applications').create({
      values: {
        name: subAppName,
        options: {
          plugins: ['nocobase'],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    await AppSupervisor.getInstance().removeApp(subAppName);

    await app.start();

    expect(AppSupervisor.getInstance().hasApp(subAppName)).toBeFalsy();

    await subApp.update({
      options: {
        autoStart: true,
      },
    });

    await AppSupervisor.getInstance().removeApp(subAppName);

    expect(AppSupervisor.getInstance().hasApp(subAppName)).toBeFalsy();

    await app.stop();

    await app.db.reconnect();
    await AppSupervisor.getInstance().getApp(subAppName, {
      upgrading: true,
    });

    await app.start();
    await sleep(10000);
    expect(AppSupervisor.getInstance().hasApp(subAppName)).toBeTruthy();
    const appStatus = AppSupervisor.getInstance().getAppStatus(subAppName);
    expect(appStatus).toEqual('running');
  });

  it('should get same obj ref when asynchronously access with same sub app name', async () => {
    const subAppName = `t_${uid()}`;

    await app.db.getRepository('applications').create({
      values: {
        name: subAppName,
        options: {
          plugins: [],
        },
      },
      context: {
        waitSubAppInstall: true,
      },
    });

    await AppSupervisor.getInstance().removeApp(subAppName);

    expect(AppSupervisor.getInstance().hasApp(subAppName)).toBeFalsy();

    const instances = [];

    AppSupervisor.getInstance().on('afterAppAdded', (subApp) => {
      instances.push(subApp);
    });

    const promises = [];
    for (let i = 0; i < 3; i++) {
      promises.push(
        (async () => {
          await AppSupervisor.getInstance().getApp(subAppName);
        })(),
      );
    }
    await Promise.all(promises);

    expect(instances.length).toBe(1);
    expect(instances[0]).toBeDefined();
    expect(instances[0].name).toEqual(subAppName);
  });
});
