feat: add bundlerOutput api

shuvijs · Jan 14, 2022 · e417db0 · e417db0
1 parent be8bf9a
commit e417db0
Show file tree

Hide file tree

Showing 5 changed files with 60 additions and 3 deletions.
diff --git a/packages/service/src/core/_types.ts b/packages/service/src/core/_types.ts
@@ -133,6 +133,7 @@ export interface IApi {
   addEntryCode: typeof ProjectBuilder.prototype.addEntryCode;
   addAppFile: typeof ProjectBuilder.prototype.addFile;
   addRuntimeService: typeof ProjectBuilder.prototype.addRuntimeService;
+  addResources: typeof ProjectBuilder.prototype.addResources;
   addAppPolyfill: typeof ProjectBuilder.prototype.addPolyfill;
 
   setPlatformModule: typeof ProjectBuilder.prototype.setPlatformModule;

diff --git a/packages/service/src/core/api.ts b/packages/service/src/core/api.ts
@@ -148,6 +148,7 @@ class Api {
       serverPlugins: this.serverPlugins,
       // resources
       resources: this.resources,
+      addResources: this.addResources.bind(this),
       assetPublicPath: this.assetPublicPath,
       getAssetPublicUrl: this.getAssetPublicUrl.bind(this),
       resolveAppFile: this.resolveAppFile.bind(this),
@@ -355,6 +356,10 @@ class Api {
     this._projectBuilder.addRuntimeService(source, exported, filepath);
   }
 
+  addResources(source: string, exported: string, filepath?: string): void {
+    this._projectBuilder.addResources(source, exported, filepath);
+  }
+
   addAppPolyfill(file: string): void {
     this._projectBuilder.addPolyfill(file);
   }

diff --git a/packages/service/src/core/apiTypes.ts b/packages/service/src/core/apiTypes.ts
@@ -53,10 +53,10 @@ export interface IPaths {
   // dir to store shuvi generated src files
   appDir: string;
 
-  privateDir: string
+  privateDir: string;
 
   // dir to runtime libraries
-  runtimeDir: string
+  runtimeDir: string;
 
   // user src dir
   srcDir: string;
@@ -154,6 +154,7 @@ export type IPluginContext = {
   // resources
   assetPublicPath: string;
   resources: IResources;
+  addResources: (source: string, exported: string, filepath?: string) => void;
   resolveAppFile(...paths: string[]): string;
   resolveUserFile(...paths: string[]): string;
   resolveBuildFile(...paths: string[]): string;

diff --git a/packages/service/src/project/projectBuilder.ts b/packages/service/src/project/projectBuilder.ts
@@ -39,6 +39,10 @@ const contextValidatingRuleMap: ContextValidatingRuleMap = {
     ignore: true,
     method: 'addRuntimeService'
   },
+  resources: {
+    ignore: true,
+    method: 'addResources'
+  },
   runtimePlugins: {
     ignore: true,
     method: 'addRuntimePlugin'
@@ -139,7 +143,11 @@ class ProjectBuilder {
     this._projectContext.clientModule = module;
   }
 
-  addRuntimeService(source: string, exported: string, filepath: string = 'index.js'): void {
+  addRuntimeService(
+    source: string,
+    exported: string,
+    filepath: string = 'index.js'
+  ): void {
     const services = this._projectContext.runtimeServices;
     filepath = path.join('runtime', path.resolve('/', filepath));
     const service = services.get(filepath);
@@ -176,6 +184,46 @@ class ProjectBuilder {
     }
   }
 
+  addResources(
+    source: string,
+    exported: string,
+    filepath: string = 'index.js'
+  ): void {
+    const services = this._projectContext.resources;
+    filepath = path.join('resources', path.resolve('/', filepath));
+    const service = services.get(filepath);
+    if (service) {
+      const targetSource = service.get(source);
+      if (targetSource) {
+        targetSource.add(exported);
+      } else {
+        const exportedSet: Set<string> = new Set();
+        exportedSet.add(exported);
+        service.set(source, exportedSet);
+      }
+    } else {
+      const exportedSet: Set<string> = new Set();
+      exportedSet.add(exported);
+      const service: Map<string, Set<string>> = new Map();
+      service.set(source, exportedSet);
+      services.set(filepath, service);
+      this.addFile({
+        name: filepath,
+        content: (context: ProjectContext) => {
+          const exportsConfig: { [key: string]: string[] } = {};
+          const service = context.runtimeServices.get(filepath);
+          if (!service) {
+            return null;
+          }
+          for (const [s, e] of service) {
+            exportsConfig[s] = Array.from(e);
+          }
+          return exportsFromObject(exportsConfig);
+        }
+      });
+    }
+  }
+
   /**
    * default path is the root path
    */

diff --git a/packages/service/src/project/projectContext.ts b/packages/service/src/project/projectContext.ts
@@ -37,6 +37,7 @@ export interface ProjectContext {
    * }
    */
   runtimeServices: Map<string, Map<string, Set<string>>>;
+  resources: Map<string, Map<string, Set<string>>>;
   runtimePlugins: IRuntimeOrServerPlugin[];
   runtimeConfigContent: string | null;
   platformModule: string;
@@ -53,6 +54,7 @@ export const createProjectContext = () =>
     entryWrapperContent: '',
     polyfills: [],
     runtimeServices: new Map(),
+    resources: new Map(),
     runtimePlugins: [],
     runtimeConfigContent: null,
     clientModule: {