Add option to set import extension

Author: lorefnon

src/cli.ts

@@ -25,6 +25,7 @@ async function main() {
             'schema': { type: 'string', short: 's' },
             'connection-source': { type: 'string', short: 'c' },
             'output-dir': { type: 'string', short: 'o' },
+            'output-import-ext': { type: 'string' },
             'remove-extraneous': { type: 'string' },
             'export-table-instances': { type: 'boolean' },
             'export-row-types': { type: 'boolean' },
@@ -47,6 +48,11 @@ async function main() {
         connectionSourcePath: argv["connection-source"],
         outputDirPath: argv["output-dir"],
         removeExtraneous: argv["remove-extraneous"] ?? "never",
+        output: {
+            import: {
+                extension: argv["output-import-ext"],
+            }
+        },
         export: {
             tableInstances: argv['export-table-instances'],
             tableClasses: argv['export-table-classes'],

src/generator-options.ts

@@ -297,6 +297,14 @@ export const TypeWrapperSchema = z.object({
     wrapper: ImportedItemSchema
 });
 
+export const OutputImportOptionsSchema = z.object({
+    extension: z.string().nullish(),
+});
+
+export const OutputOptionsSchema = z.object({
+    import: OutputImportOptionsSchema.nullish(),
+});
+
 export const GeneratorOptsSchema = z.object({
     /** Simulate the generation and print the outcome without actually modifying any files */
     dryRun: z.boolean().nullish(),
@@ -341,6 +349,11 @@ export const GeneratorOptsSchema = z.object({
      */
     tables: TableInclusionSchema.nullish(),
 
+    /**
+     * Shared options that affect all generated output
+     */
+    output: OutputOptionsSchema.nullish(),
+
     /**
      * Customize what all entities are exported from generated file
      *

src/generator.ts

@@ -180,6 +180,7 @@ export class Generator {
         comment: this.formatComment([table.comment]),
         idPrefix,
       },
+      output: this.opts.output,
       imports: [
         ...utilImports,
         ...adapterImports,

src/template.ts.hbs

@@ -5,8 +5,8 @@
  * This file has been auto-generated from database schema using ts-sql-codegen.
  * Any changes will be overwritten.
  */
-import { {{table.kind}} } from "ts-sql-query/{{table.kind}}";
-import type { DBConnection } from "{{dbConnectionSource}}";
+import { {{table.kind}} } from "ts-sql-query/{{table.kind}}{{output.import.extension}}";
+import type { DBConnection } from "{{dbConnectionSource}}{{output.import.extension}}";
 {{#if importExtraTypes}}
 import {
     {{#if rowTypes.insertable}}
@@ -29,13 +29,13 @@ import {
     {{#if valuesTypes.selected}}
     SelectedValues,
     {{/if}}
-} from "ts-sql-query/extras/types";
+} from "ts-sql-query/extras/types{{output.import.extension}}";
 {{/if}}
 {{#each imports}}
     {{#if isDefault}}
         {{#each imported}}
             {{#dedent-by 4 "level"}}
-                import {{.}} from "{{../importPath}}";
+                import {{.}} from "{{../importPath}}{{output.import.extension}}";
             {{/dedent-by}}
         {{/each}}
     {{else}}
@@ -46,7 +46,7 @@ import {
                         {{.}},
                     {{/each}}
                 {{/dedent-by}}
-            } from "{{importPath}}";
+            } from "{{importPath}}{{output.import.extension}}";
         {{/dedent-by}}
     {{/if}}
 {{/each}}