@idlizer/arktscgen 2.1.10-arktscgen-6 → 2.1.10-arktscgen-9

package/build/libarkts-copy/src/index.ts

@@ -28,13 +28,11 @@ export * from './arkts-api/AbstractVisitor';
 export * from './arkts-api/plugins';
 export * from './arkts-api/ImportStorage';
 export * from './arkts-api/ProgramProvider';
-export * from './arkts-api/node-utilities/Program';
 export * from './arkts-api/node-utilities/ArkTsConfig';
 
 export * from './arkts-api/peers/AstNode';
 export * from './arkts-api/peers/Config';
 export * from './arkts-api/peers/Context';
-export { GlobalContext } from './arkts-api/peers/Context';
 export * from './arkts-api/peers/ExternalSource';
 export * from './arkts-api/peers/ImportPathManager';
 export * from './arkts-api/peers/Options';

package/build/libarkts-copy/src/plugin-utils.ts

@@ -13,6 +13,8 @@
  * limitations under the License.
  */
 
+import * as fs from "node:fs";
+import * as path from "node:path";
 import {
     Es2pandaContextState,
     PluginContext,
@@ -22,49 +24,25 @@ import {
     Program,
     ProgramProvider,
     CompilationOptions,
-    dumpProgramSrcFormatted,
+    PluginContextImpl,
+    Es2pandaCompilationMode,
 } from './arkts-api';
 import { Tracer } from './tracer';
+import { global } from "./arkts-api/static/global";
 
 export interface RunTransformerHooks {
     onProgramTransformStart?(options: CompilationOptions, program: Program): void;
     onProgramTransformEnd?(options: CompilationOptions, program: Program): void;
 }
 
-class ASTCache {
-    processedPrograms = new Set<string>();
-    constructor() {}
-    find(program: Program): boolean {
-        return this.processedPrograms.has(program.absoluteName);
-    }
-    update(program: Program) {
-        this.processedPrograms.add(program.absoluteName);
-    }
-}
-
-export class DumpingHooks implements RunTransformerHooks {
-    constructor(
-        private state: Es2pandaContextState,
-        private pluginName: string,
-        private dumpAst: boolean = false
-    ) {
-        if (process.env.KOALA_DUMP_PLUGIN_AST) {
-            this.dumpAst = true;
-        }
+export class ProfilerHooks implements RunTransformerHooks {
+    constructor() {
     }
     onProgramTransformStart(options: CompilationOptions, program: Program) {
-        if (this.dumpAst && !program.absoluteName.includes("absolute-sdk-patched")) {
-            console.log(`BEFORE ${this.pluginName}:`);
-            dumpProgramSrcFormatted(program, true);
-        }
         if (!options.isProgramForCodegeneration) arktsGlobal.profiler.transformDepStarted();
     }
     onProgramTransformEnd(options: CompilationOptions, program: Program) {
-        if (!options.isProgramForCodegeneration) arktsGlobal.profiler.transformDepEnded(this.state, this.pluginName);
-        if (this.dumpAst && !program.absoluteName.includes("absolute-sdk-patched")) {
-            console.log(`AFTER ${this.pluginName}:`);
-            dumpProgramSrcFormatted(program, true);
-        }
+        if (!options.isProgramForCodegeneration) arktsGlobal.profiler.transformDepEnded();
     }
 }
 
@@ -73,8 +51,8 @@ export function runTransformerOnProgram(
     options: CompilationOptions,
     transform: ProgramTransformer | undefined,
     pluginContext: PluginContext,
-    hooks: RunTransformerHooks = {},
-    stableDeps: boolean = false,
+    hooks: RunTransformerHooks,
+    stableDeps: boolean,
 ) {
     arktsGlobal.filePath = program.absoluteName;
 
@@ -108,9 +86,10 @@ export function runTransformer(
     state: Es2pandaContextState,
     transform: ProgramTransformer | undefined,
     pluginContext: PluginContext,
-    hooks: RunTransformerHooks = {},
     stableDeps: boolean = false,
 ) {
+    const hooks = new ProfilerHooks()
+
     // Program provider used to provide programs to transformer dynamically relative to inserted imports
     const provider = new ProgramProvider(prog, stableDeps);
 
@@ -136,8 +115,55 @@ export function runTransformer(
 }
 
 function isProgramForCodegeneration(program: Program, isMainProgram: boolean): boolean {
-    if (!arktsGlobal.isContextGenerateAbcForExternalSourceFiles) {
+    if (arktsGlobal.configObj?.compilationMode != Es2pandaCompilationMode.COMPILATION_MODE_GEN_ABC_FOR_EXTERNAL_SOURCE) {
         return isMainProgram;
     }
     return program.isGenAbcForExternal;
 }
+
+const isDebugDump: boolean = false;
+
+function checkDebugDump() {
+    return isDebugDump || process.env.KOALA_DUMP_PLUGIN_AST == "1";
+}
+
+export function debugDump(
+    dumpDir: string | undefined,
+    state: Es2pandaContextState,
+    pluginName: string,
+    isAfter: boolean,
+    program: Program,
+) {
+    if (!checkDebugDump()) {
+        return;
+    }
+    if (!dumpDir) {
+        const outDir = global.arktsconfig?.outDir
+        if (outDir) {
+            dumpDir = path.join(outDir, "../dist/cache")
+        } else {
+            dumpDir = "dist/cache"
+        }
+    }
+    const baseDir = path.join(dumpDir,
+        `${state}_${isAfter ? "" : "ORI"}_${pluginName}`
+    );
+    runTransformer(
+        program,
+        state,
+        (program: Program) => {
+            if (program.absoluteName == "") {
+                return
+            }
+            const dumpFilePath = path.join(baseDir, path.sep + program.absoluteName)
+            fs.mkdirSync(path.dirname(dumpFilePath), { recursive: true })
+            fs.writeFileSync(
+                dumpFilePath,
+                program.ast.dumpSrc(),
+                'utf-8',
+            )
+        },
+        new PluginContextImpl(),
+        true,
+    )
+}

package/build/libarkts-copy/src/reexport-for-generated.ts

@@ -13,11 +13,12 @@
  * limitations under the License.
  */
 export { KNativePointer } from '@koalaui/interop';
+export { ArktsObject, isSameNativeObject } from './arkts-api/peers/ArktsObject';
 export { AstNode } from './arkts-api/peers/AstNode';
 export { Config } from './arkts-api/peers/Config';
 export { Context, GlobalContext } from './arkts-api/peers/Context';
-export { ArktsObject, isSameNativeObject } from './arkts-api/peers/ArktsObject';
 export { NodeCache } from './arkts-api/node-cache';
+export { ExternalSource } from './arkts-api/peers/ExternalSource';
 export {
     passNode,
     unpackNonNullableNode,
@@ -26,8 +27,8 @@ export {
     passStringArray,
     unpackNode,
     unpackString,
-    assertValidPeer,
     updateNodeByNode,
+    unpackNativeObjectArray,
 } from './arkts-api/utilities/private';
 export { nodeByType } from './arkts-api/class-by-peer';
 export { global } from './arkts-api/static/global';

package/build/libarkts-copy/tsconfig.json

@@ -10,8 +10,5 @@
     "include": [
         "./src/**/*.ts",
         "./generated/**/*.ts"
-    ],
-    "exclude": [
-        "./src/ts-api/**/*.ts"
     ]
 }