fix: mainThread case

Author: yulingtianxia

lib/main.js

@@ -124,8 +124,9 @@ function generateDartWithWorker(content, path, script) {
                 resolve({ result: result, error: error })
             })
         } else {
+            isPartOfFile = path.endsWith(dnPartExtension) && !path.endsWith(`0.${dnPartExtension}`)
             const worker = new Worker(script, {
-                workerData: { content: content },
+                workerData: { content: content, isPartOfFile: isPartOfFile },
                 resourceLimits: { maxOldGenerationSizeMb: 8 * 1024 }
             })
             worker.on("message", resolve)
@@ -151,7 +152,7 @@ function generateDartWithWorker(content, path, script) {
 
 function splitCodeFileIfNeed(separator, maxLength, path) {
     const content = rf.readFileSync(path, "utf-8")
-    if (content.length > maxLength) {
+    if (!this.isMainThread && isSupportWorkerThreads && content.length > maxLength) {
         let components = content.split(separator)
         return components.map((c, i) => {
             if (i == components.length - 1) {

lib/objc/DNObjectiveCContext.js

@@ -576,15 +576,19 @@ class DNImportContext extends DNContext {
 }
 
 class DNRootContext extends DNContext {
-    constructor(internal, needExport) {
+    constructor(internal, needExport, isPartOfFile = false) {
         super(internal)
         this.needExport = needExport
+        this.isPartOfFile = isPartOfFile
     }
 
     parse() {
-        var result = '// Generated by @dartnative/codegen:\n// https://www.npmjs.com/package/@dartnative/codegen\n\n'
+        var result = ''
+        if (!this.isPartOfFile) {
+            result += '// Generated by @dartnative/codegen:\n// https://www.npmjs.com/package/@dartnative/codegen\n\n'
+        }
         var packageSet = new Set()
-        if (!this.needExport) {
+        if (!this.needExport && !this.isPartOfFile) {
             result += "import 'dart:ffi';\n\n"
             result += "import 'package:dart_native/dart_native.dart';\n"
             result += "import 'package:dart_native_gen/dart_native_gen.dart';\n"

lib/objc/DNObjectiveCConverter.js

@@ -26,6 +26,7 @@ function convert(content, cb) {
     if (!cb) {
         cb = callback
     }
+    
     try {
         const chars = new antlr4.InputStream(content)
         const lexer = new ObjectiveCLexer(chars)
@@ -36,7 +37,8 @@ function convert(content, cb) {
         const parser = new ObjectiveCParser(tokens)
         parser.addErrorListener(errorListener)
         const tree = parser.translationUnit()
-        const listener = new DNObjectiveCParserListener(cb)
+        let isPartOfFile = workerData ? workerData.isPartOfFile : false
+        const listener = new DNObjectiveCParserListener(cb, isPartOfFile)
         antlr4.tree.ParseTreeWalker.DEFAULT.walk(listener, tree)
     } catch (e) {
         cb(null, e)

lib/objc/DNObjectiveCParserListener.js

@@ -20,11 +20,12 @@ let DNRootContext = c.DNRootContext,
 
 // This class defines a complete listener for a parse tree produced by ObjectiveCParser.
 class DNObjectiveCParserListener extends ObjectiveCParserListener {
-    constructor(cb) {
+    constructor(cb, isPartOfFile) {
         super()
         ObjectiveCParserListener.call(this)
         //success callack
         this.cb = cb
+        this.isPartOfFile = isPartOfFile
     }
 
     buildDart() {
@@ -45,7 +46,7 @@ class DNObjectiveCParserListener extends ObjectiveCParserListener {
                 }
             }
         }
-        this.rootContext = new DNRootContext(ctx, this.needExport)
+        this.rootContext = new DNRootContext(ctx, this.needExport, this.isPartOfFile)
         this.currentContext = this.rootContext
     }
     // Exit a parse tree produced by ObjectiveCParser#translationUnit.