feat: support getCurrentApplicationContext API (#981)

* refactor: support getCurrent API

* fix: lint

packages/bootstrap/src/bootstrap.ts

@@ -84,6 +84,7 @@ export class BootstrapStarter {
       applicationContext,
     });
 
+    global['MIDWAY_MAIN_FRAMEWORK'] = this.getMainFramework();
     mainApp = await this.getFirstActions('getApplication');
 
     // 初始化其余的副框架
@@ -148,6 +149,10 @@ export class BootstrapStarter {
     return [];
   }
 
+  protected getMainFramework() {
+    return this.bootstrapItems[0];
+  }
+
   protected refreshBootstrapItems() {
     this.bootstrapItems = this.bootstrapItems.map(bootstrapItem => {
       if (typeof bootstrapItem === 'function') {

packages/bootstrap/test/fixtures/mix-app/src/configuration.ts

@@ -1,7 +1,8 @@
-import { Configuration, Config, ALL, Inject } from '@midwayjs/decorator';
+import { ALL, App, Config, Configuration, Inject, MidwayFrameworkType } from '@midwayjs/decorator';
 import { join } from 'path';
 import * as assert from 'assert';
 import { RemoteConfigService } from './service/remoteConfigService';
+import { getCurrentApplicationContext, getCurrentMainFramework, IMidwayApplication } from '@midwayjs/core';
 
 @Configuration({
   importConfigs: [
@@ -15,9 +16,14 @@ export class AutoConfiguration {
   @Inject()
   configService: RemoteConfigService;
 
+  @App()
+  app: IMidwayApplication;
+
   async onReady() {
     console.log('ready');
     assert(this.prepareConfig['prepare'] === 'remote data');
     assert(this.prepareConfig['id'] === this.configService.innerData);
+    assert(getCurrentApplicationContext() === this.app.getApplicationContext());
+    assert(getCurrentMainFramework().getFrameworkType() === MidwayFrameworkType.WEB);
   }
 }

packages/core/src/index.ts

@@ -51,10 +51,11 @@ export * from './util/webRouterCollector';
 export * from './util/triggerCollector';
 export * from './util/emptyFramework';
 export { plainToClass, classToPlain } from 'class-transformer';
-export * from './logger';
 export { createConfiguration } from './functional/configuration';
 export { MidwayConfigService } from './service/configService';
 export { MidwayEnvironmentService } from './service/environmentService';
+export * from './logger';
+export * from './util/contextUtil';
 /**
  * @deprecated please import from @midwayjs/logger
  */

packages/core/src/util/contextUtil.ts

@@ -0,0 +1,21 @@
+import {
+  IConfigurationOptions,
+  IMidwayApplication,
+  IMidwayContainer,
+  IMidwayFramework,
+} from '../interface';
+
+export const getCurrentApplicationContext = (): IMidwayContainer => {
+  return getCurrentMainFramework().getApplicationContext();
+};
+
+export const getCurrentMainFramework = <
+  APP extends IMidwayApplication,
+  T extends IConfigurationOptions
+>(): IMidwayFramework<APP, T> => {
+  return global['MIDWAY_MAIN_FRAMEWORK'] as IMidwayFramework<APP, T>;
+};
+
+export const getCurrentMainApp = <APP extends IMidwayApplication>(): APP => {
+  return getCurrentMainFramework().getApplication() as APP;
+};

packages/core/test/baseFramework.test.ts

@@ -20,6 +20,7 @@ import {
 import * as mm from 'mm';
 import { LifeCycleTest, LifeCycleTest1, TestBinding } from './fixtures/lifecycle';
 import sinon = require('sinon');
+import { getCurrentApplicationContext, getCurrentMainApp, getCurrentMainFramework } from '../src/util/contextUtil';
 
 @Provide()
 class TestModule {
@@ -709,4 +710,22 @@ describe('/test/baseFramework.test.ts', () => {
       }
     ]);
   });
+
+
+  it('should test global framework', async () => {
+    const framework = new LightFramework();
+    await framework.initialize({
+      baseDir: path.join(__dirname, './fixtures/base-app/src'),
+    });
+
+    mm(global, 'MIDWAY_MAIN_FRAMEWORK', framework);
+
+    const appCtx = framework.getApplicationContext();
+    expect(getCurrentMainFramework()).toEqual(framework);
+    expect(getCurrentApplicationContext()).toEqual(appCtx);
+    expect(getCurrentMainApp()).toEqual(framework.getApplication());
+
+    mm.restore();
+  });
+
 });