-
Mock data generator
+ Mock data generator
+
-
-
-
users generator
+
+
- )
+ );
}
diff --git a/src/components/section.tsx b/src/components/section.tsx
index 1688fe9..0ab0f96 100644
--- a/src/components/section.tsx
+++ b/src/components/section.tsx
@@ -1,13 +1,6 @@
import { ReactNode } from "react";
import { cn } from "@/utilities";
-export default function Section({ children, variant }: {
- children: ReactNode,
- variant?: "secondary" | "disabled"
-}) {
- return (
-
- {children}
-
- )
+export default function Section({ children, variant }: { children: ReactNode; variant?: "secondary" | "disabled" }) {
+ return
{children}
;
}
diff --git a/src/components/sidebar.tsx b/src/components/sidebar.tsx
index 4cb833e..6be3ac6 100644
--- a/src/components/sidebar.tsx
+++ b/src/components/sidebar.tsx
@@ -1,4 +1,4 @@
-"use client"
+"use client";
import { ReactNode, useState, useEffect, useRef } from "react";
import { cn } from "@/utilities";
import Icon from "./ui/icon";
@@ -14,7 +14,12 @@ export default function Sidebar({ children }: { children: ReactNode }) {
useEffect(() => {
const HideSidebarOnClick = (e: MouseEvent) => {
- if (sidebarRef.current && sidebarToggleRef.current && !sidebarRef.current.contains(e.target as Node) && !sidebarToggleRef.current.contains(e.target as Node)) {
+ if (
+ sidebarRef.current &&
+ sidebarToggleRef.current &&
+ !sidebarRef.current.contains(e.target as Node) &&
+ !sidebarToggleRef.current.contains(e.target as Node)
+ ) {
setIsSidebarOpen(false);
}
};
@@ -23,7 +28,7 @@ export default function Sidebar({ children }: { children: ReactNode }) {
return () => {
document.removeEventListener("click", HideSidebarOnClick);
- }
+ };
}, []);
return (
@@ -36,9 +41,9 @@ export default function Sidebar({ children }: { children: ReactNode }) {
-
+
{children}
- )
+ );
}
diff --git a/src/components/switch-section.tsx b/src/components/switch-section.tsx
index 2edf90f..26e6eb7 100644
--- a/src/components/switch-section.tsx
+++ b/src/components/switch-section.tsx
@@ -1,11 +1,8 @@
-"use client"
+"use client";
import { ReactNode, Children, useEffect, useState, isValidElement, createElement } from "react";
import { cn } from "@/utilities";
-export default function SwitchSection({ children, className }: {
- children: ReactNode,
- className?: string
-}) {
+export default function SwitchSection({ children, className }: { children: ReactNode; className?: string }) {
const [childrenArray, setChildrenArray] = useState
([]);
const [activeIndex, setActiveIndex] = useState(0);
@@ -15,7 +12,6 @@ export default function SwitchSection({ children, className }: {
return (
-
{childrenArray.map((el: ReactNode, i: number) => {
if (isValidElement<{ id?: string }>(el)) {
@@ -30,19 +26,19 @@ export default function SwitchSection({ children, className }: {
{childrenArray.map((el: ReactNode, i: number) => {
- if (isValidElement<{ id?: string, children?: ReactNode, className?: string }>(el)) {
+ if (isValidElement<{ id?: string; children?: ReactNode; className?: string }>(el)) {
return createElement(
"div",
{
...el.props,
key: i,
- className: cn(el.props.className, "content-item", { "active": i === activeIndex })
+ className: cn(el.props.className, "content-item", { active: i === activeIndex }),
},
- el.props.children
+ el.props.children,
);
}
})}
- )
+ );
}
diff --git a/src/components/ui/icon.tsx b/src/components/ui/icon.tsx
index 7050963..65aac2f 100644
--- a/src/components/ui/icon.tsx
+++ b/src/components/ui/icon.tsx
@@ -11,6 +11,12 @@ const icons = {
+ ),
+ copy: (
+
+
)
}
diff --git a/src/hooks/useFetch.ts b/src/hooks/useFetch.ts
deleted file mode 100644
index fc63fb4..0000000
--- a/src/hooks/useFetch.ts
+++ /dev/null
@@ -1,36 +0,0 @@
-import { useState } from "react";
-import { GeneratorArgs, GeneratorResponse } from "@/types/generator";
-
-export function useFetch() {
- const [isLoading, setIsLoading] = useState
(false);
- const [data, setData] = useState(null);
- const [error, setError] = useState(null);
-
- const refetch = async ({ count, fields, seed }: GeneratorArgs) => {
- setIsLoading(true);
- setError(null);
- try {
- const response = await fetch("/api/generate", {
- method: "POST",
- headers: {
- "Content-Type": "application/json",
- },
- body: JSON.stringify({ count, fields, seed }),
- });
-
- if (!response.ok) {
- throw new Error(`HTTP error! status: ${response.status}`);
- }
-
- const json = await response.json();
- setData(json);
- } catch (err) {
- setError(err instanceof Error ? err.message : "An error occurred");
- setData(null);
- } finally {
- setIsLoading(false);
- }
- };
-
- return { isLoading, refetch, data, error };
-}
diff --git a/src/hooks/useFetchPosts.ts b/src/hooks/useFetchPosts.ts
new file mode 100644
index 0000000..7ee03e3
--- /dev/null
+++ b/src/hooks/useFetchPosts.ts
@@ -0,0 +1,29 @@
+import { useState } from "react";
+import { PostArgs, GeneratorResponse } from "@/types/generator";
+
+export function useFetchPosts() {
+ const [isLoading, setIsLoading] = useState(false);
+ const [data, setData] = useState(null);
+
+ const refetch = async ({ count, seed }: PostArgs) => {
+ setIsLoading(true);
+ try {
+ const response = await fetch("/api/generator/posts", {
+ method: "POST",
+ headers: {
+ "Content-Type": "application/json",
+ },
+ body: JSON.stringify({ count, seed }),
+ });
+
+ const json = await response.json();
+ setData(json);
+ } catch {
+ setData({ ok: false, error: "Internal server error" });
+ } finally {
+ setIsLoading(false);
+ }
+ };
+
+ return { isLoading, refetch, data };
+}
diff --git a/src/hooks/useFetchUsers.ts b/src/hooks/useFetchUsers.ts
new file mode 100644
index 0000000..cbc289f
--- /dev/null
+++ b/src/hooks/useFetchUsers.ts
@@ -0,0 +1,54 @@
+import { useState } from "react";
+import { UserArgs, GeneratorResponse } from "@/types/generator";
+
+export function useFetchUsers() {
+ const [isLoading, setIsLoading] = useState(false);
+ const [data, setData] = useState(null);
+
+ const validate = ({ count, fields, seed }: UserArgs): { ok: boolean; error?: string } => {
+ if (
+ typeof fields !== "number" ||
+ typeof count !== "number" ||
+ (seed !== undefined && typeof seed !== "number" && typeof seed !== "number")
+ ) {
+ return { ok: false, error: "Invalid arguments" };
+ }
+
+ if (fields < 1 || fields > 63) {
+ return { ok: false, error: "No fields selected" };
+ }
+
+ if (count < 1 || count > 300) {
+ return { ok: false, error: "Count must be between 1 and 300" };
+ }
+
+ return { ok: true };
+ };
+
+ const refetch = async ({ count, fields, seed }: UserArgs) => {
+ const validation = validate({ count, fields });
+ if (!validation.ok) {
+ setData({ ok: false, error: validation.error! });
+ return;
+ }
+ setIsLoading(true);
+ try {
+ const response = await fetch("/api/generator/users", {
+ method: "POST",
+ headers: {
+ "Content-Type": "application/json",
+ },
+ body: JSON.stringify({ count, fields, seed }),
+ });
+
+ const json = await response.json();
+ setData(json);
+ } catch {
+ setData({ ok: false, error: "Internal server error" });
+ } finally {
+ setIsLoading(false);
+ }
+ };
+
+ return { isLoading, refetch, data };
+}
diff --git a/src/types/generator.ts b/src/types/generator.ts
index ebfffd8..938ccc6 100644
--- a/src/types/generator.ts
+++ b/src/types/generator.ts
@@ -1,19 +1,15 @@
-export interface GeneratorResponse {
- meta: {
- count: number;
- generatedAt: string;
- message: string;
- };
- data: Record[];
+export interface UserArgs {
+ count: number;
+ fields: number | string[];
+ seed?: number;
}
-export interface GeneratorArgs {
+export interface PostArgs {
count: number;
- fields: number | string[];
seed?: number;
}
-export interface GeneratorSettings {
+export interface UserSettings {
name: boolean;
surname: boolean;
username: boolean;
@@ -23,3 +19,15 @@ export interface GeneratorSettings {
count: number;
seed: number | undefined;
}
+
+export interface Ok {
+ ok: true;
+ value: Record[];
+}
+
+export interface Err {
+ ok: false;
+ error: string;
+}
+
+export type GeneratorResponse = Ok | Err;
diff --git a/src/utilities/buildQuery.ts b/src/utilities/buildQuery.ts
index 195a267..9d367d1 100644
--- a/src/utilities/buildQuery.ts
+++ b/src/utilities/buildQuery.ts
@@ -1,11 +1,16 @@
-export default function buildQuery(data: Record[]): string {
- if (data.length === 0) {
- return "";
- }
+import { Ok } from "@/types/generator";
- const columns = Object.keys(data[0]).join(", ");
- const values = data
- .map((row) => `(${Object.values(row).map((value) => `"${value}"`).join(", ")})`)
+export default function buildUserQuery(data: Ok): string {
+ const tableName = data.value[0].hasOwnProperty("authorId") ? "posts" : "users";
+
+ const columns = Object.keys(data.value[0]).join(", ");
+ const values = data.value
+ .map(
+ (row) =>
+ `(${Object.values(row)
+ .map((value) => `"${value}"`)
+ .join(", ")})`,
+ )
.join(",\n");
- return `INSERT INTO users (${columns}) VALUES\n${values};`;
+ return `INSERT INTO ${tableName} (${columns}) VALUES\n${values};`;
}
diff --git a/src/utilities/dataGenerator.ts b/src/utilities/dataGenerator.ts
index 0e73c73..0079b2c 100644
--- a/src/utilities/dataGenerator.ts
+++ b/src/utilities/dataGenerator.ts
@@ -1,10 +1,5 @@
-"use server";
import { faker } from "@faker-js/faker";
-import { GeneratorArgs } from "../types/generator";
-// config start
-const errorSuccess = 0; // "Data generated successfully.";
-const errorInvalidFields = 1; // "Invalid fields selected.";
-const errorInvalidCount = 2; // "Invalid count, must be between 1 and 300.";
+import { UserArgs, GeneratorResponse } from "../types/generator";
const availableFields = 6; // 6 fields: name, surname, username, email, password, avatar
@@ -40,17 +35,14 @@ const generators = [
generator: () => faker.internet.password(),
},
];
-// config end
-export async function generateData({ count, fields, seed }: GeneratorArgs): Promise<{
- result: number;
- data: Record[];
-}> {
- let returnCode = errorSuccess;
-
- if (count > 300 || count < 1) {
- console.error("Count is invalid: ", count);
- returnCode = errorInvalidCount;
+export default async function generateData({ count, fields, seed }: UserArgs): Promise {
+ if (typeof count !== "number" || count > 300 || count < 1) {
+ // console.error("Count is invalid: ", count); // dev
+ return {
+ ok: false,
+ error: "Invalid count, must be a number between 1 and 300.",
+ };
}
if (seed !== undefined) {
@@ -61,8 +53,11 @@ export async function generateData({ count, fields, seed }: GeneratorArgs): Prom
if (typeof fields === "number") {
if (fields >= 2 ** availableFields || fields < 1) {
- console.error("Invalid fields number: ", fields);
- returnCode = errorInvalidFields;
+ // console.error("Invalid fields number: ", fields); // dev
+ return {
+ ok: false,
+ error: "Invalid fields number, must be between 1 and " + (2 ** availableFields - 1),
+ };
}
for (let i = 0; i < count; i++) {
@@ -86,12 +81,15 @@ export async function generateData({ count, fields, seed }: GeneratorArgs): Prom
data.push(record);
}
} else {
- console.error("Invalid fields type: ", fields);
- returnCode = errorInvalidFields;
+ // console.error("Invalid fields type: ", fields); // dev
+ return {
+ ok: false,
+ error: "Invalid fields type, must be a number or an array.",
+ };
}
return {
- result: returnCode,
- data: data,
+ ok: true,
+ value: data,
};
}
diff --git a/src/utilities/postsGenerator.ts b/src/utilities/postsGenerator.ts
new file mode 100644
index 0000000..ea1015e
--- /dev/null
+++ b/src/utilities/postsGenerator.ts
@@ -0,0 +1,30 @@
+import { faker } from "@faker-js/faker";
+import { GeneratorResponse, PostArgs } from "@/types/generator";
+
+export default async function generatePosts({ count, seed }: PostArgs): Promise {
+ if (typeof count !== "number" || count > 300 || count < 1) {
+ return {
+ ok: false,
+ error: "Invalid count, must be a number between 1 and 300.",
+ };
+ }
+
+ if (seed !== undefined) {
+ faker.seed(seed);
+ }
+
+ const data: Record[] = [];
+
+ for (let i = 0; i < count; i++) {
+ const record: Record = {};
+ record["title"] = faker.hacker.phrase();
+ record["content"] = faker.lorem.paragraphs();
+ record["authorId"] = (i + 1).toString();
+ data.push(record);
+ }
+
+ return {
+ ok: true,
+ value: data,
+ };
+}
diff --git a/src/utilities/ratelimit/rateLimiter.ts b/src/utilities/ratelimit/rateLimiter.ts
new file mode 100644
index 0000000..133affd
--- /dev/null
+++ b/src/utilities/ratelimit/rateLimiter.ts
@@ -0,0 +1,35 @@
+import { RateLimiterMemory, RateLimiterRes } from "rate-limiter-flexible";
+import { NextRequest } from "next/server";
+
+const rateLimiter = new RateLimiterMemory({
+ points: 10,
+ duration: 1,
+});
+
+export async function checkRateLimit(request: NextRequest): Promise<{ success: boolean; headers: Headers }> {
+ try {
+ const ip = request.headers.get("x-forwarded-for") ?? "127.0.0.1";
+ const res = await rateLimiter.consume(ip, 1);
+
+ const headers = new Headers({
+ "X-RateLimit-Limit": "10",
+ "X-RateLimit-Remaining": String(res.remainingPoints),
+ "X-RateLimit-Reset": new Date(Date.now() + res.msBeforeNext).toISOString(),
+ });
+
+ return { success: true, headers };
+ } catch (err: unknown) {
+ if (err instanceof Error) {
+ return { success: false, headers: new Headers() };
+ }
+ const rejRes = err as RateLimiterRes;
+ const headers = new Headers({
+ "Retry-After": String(Math.round(rejRes.msBeforeNext / 1000) || 1),
+ "X-RateLimit-Limit": "10",
+ "X-RateLimit-Remaining": String(rejRes.remainingPoints),
+ "X-RateLimit-Reset": new Date(Date.now() + rejRes.msBeforeNext).toISOString(),
+ });
+
+ return { success: false, headers };
+ }
+}