Move cp for static files from system to FileSystemWriter

Author: ryukzak

src/api/writer-generator/csharp/csharp.ts

@@ -107,6 +107,7 @@ export class CSharp extends Writer<CSharpGeneratorOptions> {
             tabSize: 4,
             withDebugComment: false,
             commentLinePrefix: "//",
+            resolveAssets: options.resolveAssets ?? resolveCSharpAssets,
             ...options,
         });
     }
@@ -357,9 +358,8 @@ export class CSharp extends Writer<CSharpGeneratorOptions> {
     }
 
     private copyStaticFiles(): void {
-        if (this.opts.inMemoryOnly) return;
-        const sourceDir = resolveCSharpAssets("");
-        fs.cpSync(sourceDir, this.opts.outputDir, { recursive: true });
+        this.cp("Client.cs", "Client.cs");
+        this.cp("Helper.cs", "Helper.cs");
     }
 
     private generateHelperFile(): void {

src/api/writer-generator/python.ts

@@ -150,7 +150,7 @@ export class Python extends Writer<PythonGeneratorOptions> {
     private tsIndex: TypeSchemaIndex | undefined;
 
     constructor(options: PythonGeneratorOptions) {
-        super(options);
+        super({ ...options, resolveAssets: options.resolveAssets ?? resolvePyAssets });
         this.nameFormatFunction = this.getFieldFormatFunction(options.fieldFormat);
     }
 
@@ -166,7 +166,7 @@ export class Python extends Writer<PythonGeneratorOptions> {
 
     private generateRootPackages(groups: TypeSchemaPackageGroups): void {
         this.generateRootInitFile(groups);
-        fs.cpSync(resolvePyAssets("requirements.txt"), Path.resolve(this.opts.outputDir, "requirements.txt"));
+        this.cp(resolvePyAssets("requirements.txt"), "requirements.txt");
     }
 
     private generateSDKPackages(groups: TypeSchemaPackageGroups): void {

src/api/writer-generator/writer.ts

@@ -7,6 +7,7 @@ export type FileSystemWriterOptions = {
     outputDir: string;
     inMemoryOnly?: boolean;
     logger?: CodegenLogger;
+    resolveAssets?: (fn: string) => string;
 };
 
 export type WriterOptions = FileSystemWriterOptions & {
@@ -104,6 +105,19 @@ export abstract class FileSystemWriter<T extends FileSystemWriterOptions = FileS
         buf.tokens.push(str);
     }
 
+    cp(source: string, destination: string) {
+        if (!this.opts.resolveAssets) throw new Error("resolveAssets is not defined");
+        source = Path.resolve(this.opts.resolveAssets(source));
+        destination = Path.normalize(`${this.currentDir ?? this.opts.outputDir}/${destination}`);
+        const content = fs.readFileSync(source, "utf8");
+        this.writtenFilesBuffer[destination] = {
+            relPath: destination,
+            absPath: Path.resolve(destination),
+            tokens: [content],
+        };
+        fs.cpSync(source, destination);
+    }
+
     abstract generate(_tsIndex: TypeSchemaIndex): Promise<void>;
 
     writtenFiles(): FileBuffer[] {