fix: strict() keyword working root interfaces (#134)

* test: failing test case

* fix: missing strict() on root schemas

* test: additional test case

---------

Co-authored-by: tvillaren <tvillaren@users.noreply.github.com>

src/core/generate.test.ts

@@ -478,4 +478,52 @@ describe("generate", () => {
       expect(errors).toEqual([]);
     });
   });
+
+  describe("with @strict tag", () => {
+    it("should generate strict keyword", () => {
+      const sourceText = `
+      /**
+       * @strict
+       */
+      export interface Superman {
+        name: string;
+        age: number;
+      }
+
+      export interface Villain {
+        name: string;
+        /**
+         * @strict
+         */
+        nemesis: {
+          name: string;
+          ref: Superman
+        };
+      }
+      `;
+
+      const { getZodSchemasFile } = generate({
+        sourceText,
+      });
+
+      expect(getZodSchemasFile("./source")).toMatchInlineSnapshot(`
+      "// Generated by ts-to-zod
+      import { z } from \\"zod\\";
+
+      export const supermanSchema = z.object({
+          name: z.string(),
+          age: z.number()
+      }).strict();
+
+      export const villainSchema = z.object({
+          name: z.string(),
+          nemesis: z.object({
+              name: z.string(),
+              ref: supermanSchema
+          }).strict()
+      });
+      "
+      `);
+    });
+  });
 });

src/core/generateZodSchema.ts

@@ -108,6 +108,20 @@ export function generateZodSchemaVariableStatement({
       schemaExtensionClauses,
       skipParseJSDoc,
     });
+
+    if (!skipParseJSDoc) {
+      const jsDocTags = getJSDocTags(node, sourceFile);
+      if (jsDocTags.strict) {
+        schema = f.createCallExpression(
+          f.createPropertyAccessExpression(
+            schema,
+            f.createIdentifier("strict")
+          ),
+          undefined,
+          undefined
+        );
+      }
+    }
   }
 
   if (ts.isTypeAliasDeclaration(node)) {