Refactor generator

package.json

@@ -15,7 +15,8 @@
     "rate-limiter-flexible": "^10.0.1",
     "react": "19.2.3",
     "react-dom": "19.2.3",
-    "sass": "^1.98.0"
+    "sass": "^1.98.0",
+    "zod": "^4.3.6"
   },
   "devDependencies": {
     "@faker-js/faker": "^10.3.0",

src/app/users-generator/page.tsx

@@ -82,7 +82,7 @@ export default function UsersGeneratorPage() {
       setTimeout(() => {
         setIsCopied(false);
       }, 2000);
-    } catch {}
+    } catch { }
   };
 
   const copySql = async () => {
@@ -92,7 +92,7 @@ export default function UsersGeneratorPage() {
       setTimeout(() => {
         setIsCopied(false);
       }, 2000);
-    } catch {}
+    } catch { }
   };
 
   return (

src/app/users-generator/template.tsx

@@ -12,18 +12,17 @@ export const UsersGeneratorContext = createContext<GeneratorSettings>({
   isSeedEnabled: false,
 });
 
-export const columnsTable = [
-  "name",
-  "surname",
-  "username",
-  "avatar",
-  "email",
-  "password",
-  "title",
-  "body",
-  "date",
-  "phrase",
-];
+const columnsTable = {
+  "name": "firstName",
+  "surname": "lastName",
+  "full name": "fullName",
+  "username": "username",
+  "avatar": "avatar",
+  "email": "email",
+  "password": "password",
+  "date": "anytime",
+  "phrase": "phrase",
+};
 
 export default function UsersGeneratorTemplate({ children }: { children: ReactNode }) {
   const [count, setCount] = useState<string>("10");
@@ -64,7 +63,7 @@ export default function UsersGeneratorTemplate({ children }: { children: ReactNo
   };
 
   const handleAddColumn = () => {
-    setColumns([...columns, { colName: "", colValue: columnsTable[0] }]);
+    setColumns([...columns, { colName: "", colValue: Object.values(columnsTable)[0] }]);
   };
 
   const changeColumnValue = (e: React.ChangeEvent<HTMLSelectElement>, i: number) => {
@@ -104,10 +103,10 @@ export default function UsersGeneratorTemplate({ children }: { children: ReactNo
                       onChange={(e) => changeColumnValue(e, i)}
                       disabled={!el.colName}
                     >
-                      {columnsTable.map((element, i) => {
+                      {Object.entries(columnsTable).map(([key, value], i) => {
                         return (
-                          <option key={i} value={element}>
-                            {element}
+                          <option key={i} value={value}>
+                            {key}
                           </option>
                         );
                       })}

src/utilities/generator/dataGenerator.ts

@@ -1,7 +1,47 @@
 import { UserArgs, GeneratorResponse } from "../../types/generator";
-import { generators } from "./generators";
+import { FieldCategory } from "./generatorFields";
+import { faker } from "@faker-js/faker";
+import { z } from "zod";
+
+const schema = z.object({
+  count: z.number().int().min(1).max(300),
+  seed: z.number().int().min(1).optional()
+});
 
 export function generateData({ count, fields, seed }: UserArgs): GeneratorResponse {
+  const result = schema.safeParse({ count, seed });
+  if (!result.success) {
+    return {
+      ok: false,
+      error: result.error.message,
+    };
+  }
+
+  faker.seed(seed);
+
+  const data: Record<string, string>[] = [];
+
+  fields.forEach(field => {
+    const category = FieldCategory[field];
+    if (!category) {
+      return;
+    }
+
+    for (let i = 0; i < count; i++) {
+      if (!data[i]) {
+        data[i] = {};
+      }
+      data[i][field] = (faker as unknown as Record<string, Record<string, () => string>>)[category][field]();
+    }
+  });
+
+  return {
+    ok: true,
+    value: data,
+  };
+}
+
+/*export function generateData({ count, fields, seed }: UserArgs): GeneratorResponse {
   if (!/^\d+$/.test(count.toString()) || count > 300 || count < 1) {
     // console.error("Count is invalid: ", count); // dev
     return {
@@ -42,4 +82,4 @@ export function generateData({ count, fields, seed }: UserArgs): GeneratorRespon
     ok: true,
     value: data,
   };
-}
+}*/