Add options.suffix (#7)

kriasoft · Sep 1, 2021 · 589178d · 589178d
1 parent 1dddd09
commit 589178d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/main.test.ts b/main.test.ts
@@ -88,6 +88,7 @@ test("updateTypes", async function () {
     output,
     overrides,
     prefix: 'import { PostgresInterval} from "postgres-interval";',
+    suffix: '// user supplied suffix',
     schema: ["public", "log", "!secret"],
     exclude: ["login"],
   });
@@ -159,6 +160,7 @@ test("updateTypes", async function () {
       interval: PostgresInterval;
     };
 
+    // user supplied suffix
     "
   `);
 });

diff --git a/main.ts b/main.ts
@@ -23,6 +23,7 @@ export type Options = {
   overrides?: Record<string, string>;
 
   prefix?: string;
+  suffix?: string;
 
   /**
    * Schemas that should be included/excluded when generating types.
@@ -183,6 +184,12 @@ export async function updateTypes(db: Knex, options: Options): Promise<void> {
         output.write("};\n\n");
       }
     });
+
+    if (options.suffix) {
+      output.write(options.suffix);
+      output.write("\n")
+    }
+
   } finally {
     output.end();
     db.destroy();