schemock 0.0.4-alpha.1 → 0.0.4-alpha.2

package/dist/cli/index.js

@@ -480,6 +480,12 @@ function toCamelCase(str) {
 function toSnakeCase(str) {
   return str.replace(/([A-Z])/g, "_$1").toLowerCase().replace(/^_/, "").replace(/[-\s]+/g, "_");
 }
+function toSafePropertyName(str) {
+  if (/^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(str)) {
+    return str;
+  }
+  return toCamelCase(str);
+}
 
 // src/cli/utils/faker-mapping.ts
 function escapeJsString(value) {
@@ -1251,11 +1257,12 @@ function deriveEndpointName(path) {
   const parts = cleaned.split("/");
   const nameParts = [];
   for (let i = 0; i < parts.length; i++) {
-    const part = parts[i];
+    let part = parts[i];
     if (part.startsWith(":")) {
-      const paramName = part.slice(1);
+      const paramName = toCamelCaseFromHyphen(part.slice(1));
       nameParts.push("By" + capitalize(paramName));
     } else if (part) {
+      part = toCamelCaseFromHyphen(part);
       if (i === 0) {
         nameParts.push(part);
       } else {
@@ -1275,6 +1282,9 @@ function toPascalCase2(str) {
 function capitalize(str) {
   return str.charAt(0).toUpperCase() + str.slice(1);
 }
+function toCamelCaseFromHyphen(str) {
+  return str.split("-").map((part, index) => index === 0 ? part : capitalize(part)).join("");
+}
 function analyzeFields(fields) {
   return Object.entries(fields).map(([name, field]) => analyzeField2(name, field));
 }
@@ -1587,7 +1597,7 @@ function generateCommonTypes(code) {
 // src/cli/generators/mock/db.ts
 function generateMockDb(schemas, config) {
   const code = new CodeBuilder();
-  const entityNames = schemas.map((s) => s.name);
+  const entityNames = schemas.map((s) => toSafePropertyName(s.name));
   const persist = config.persist !== false;
   code.comment("GENERATED BY SCHEMOCK - DO NOT EDIT");
   code.line("import { factory, primaryKey, nullable } from '@mswjs/data';");
@@ -1743,7 +1753,8 @@ function generatePersistenceLayer(code, entityNames, storageKey) {
   code.line("wrapDbMethods();");
 }
 function generateEntityFactory(code, schema) {
-  code.block(`${schema.name}: {`, () => {
+  const safeName = toSafePropertyName(schema.name);
+  code.block(`${safeName}: {`, () => {
     for (const field of schema.fields) {
       if (field.name === "id") {
         code.line("id: primaryKey(faker.string.uuid),");
@@ -2465,15 +2476,17 @@ function generateSeed(schemas, config) {
   code.block("export interface SeedCounts {", () => {
     for (const schema of schemas) {
       if (schema.isJunctionTable) continue;
-      code.line(`${schema.name}?: number;`);
+      const safeName = toSafePropertyName(schema.name);
+      code.line(`${safeName}?: number;`);
     }
   });
   code.line();
   code.block("const defaultCounts: Required<SeedCounts> = {", () => {
     for (const schema of schemas) {
       if (schema.isJunctionTable) continue;
+      const safeName = toSafePropertyName(schema.name);
       const count = config.seed?.[schema.name] ?? 10;
-      code.line(`${schema.name}: ${count},`);
+      code.line(`${safeName}: ${count},`);
     }
   }, "};");
   code.line();
@@ -2490,16 +2503,17 @@ function generateSeed(schemas, config) {
     code.line();
     for (const schema of schemas) {
       if (schema.isJunctionTable) continue;
+      const safeName = toSafePropertyName(schema.name);
       const belongsToRels = schema.relations.filter((r) => r.type === "belongsTo");
       const fkFields = belongsToRels.map((r) => ({
         fieldName: r.localField || r.foreignKey,
-        target: r.target,
+        target: toSafePropertyName(r.target),
         nullable: schema.fields.find((f) => f.name === (r.localField || r.foreignKey))?.nullable ?? false
       }));
-      code.line(`ids.${schema.name} = [];`);
-      code.block(`for (let i = 0; i < merged.${schema.name}; i++) {`, () => {
+      code.line(`ids.${safeName} = [];`);
+      code.block(`for (let i = 0; i < merged.${safeName}; i++) {`, () => {
         if (fkFields.length > 0) {
-          code.line(`const item = db.${schema.name}.create({`);
+          code.line(`const item = db.${safeName}.create({`);
           code.indent();
           for (const fk of fkFields) {
             if (fk.nullable) {
@@ -2512,9 +2526,9 @@ function generateSeed(schemas, config) {
           code.line(`// eslint-disable-next-line @typescript-eslint/no-explicit-any`);
           code.line("} as any);");
         } else {
-          code.line(`const item = db.${schema.name}.create({});`);
+          code.line(`const item = db.${safeName}.create({});`);
         }
-        code.line(`ids.${schema.name}.push(item.id);`);
+        code.line(`ids.${safeName}.push(item.id);`);
       });
       code.line();
     }
@@ -2522,14 +2536,16 @@ function generateSeed(schemas, config) {
   code.line();
   code.block("export function reset(): void {", () => {
     for (const schema of [...schemas].reverse()) {
-      code.line(`db.${schema.name}.deleteMany({ where: {} });`);
+      const safeName = toSafePropertyName(schema.name);
+      code.line(`db.${safeName}.deleteMany({ where: {} });`);
     }
   });
   code.line();
   code.block("export function getAll(): Record<string, unknown[]> {", () => {
     code.block("return {", () => {
       for (const schema of schemas) {
-        code.line(`${schema.name}: db.${schema.name}.getAll(),`);
+        const safeName = toSafePropertyName(schema.name);
+        code.line(`${safeName}: db.${safeName}.getAll(),`);
       }
     }, "};");
   });