From 8c0f67ca351778405279f7e225ee8dae654033f9 Mon Sep 17 00:00:00 2001
From: Blake Matthes <cowchmonkey@gmail.com>
Date: Thu, 20 Nov 2025 19:47:52 -0600
Subject: [PATCH] refactor(forklifts): more refactoring to improve during
 production

---
 .../forklifts/routes/invoices/addInvoice.ts   |  1 +
 .../pkg/db/schema/forkliftLeasesInvoice.ts    |  2 +-
 app/src/pkg/db/schema/leaseInvoices.ts        |  1 +
 .../src/lib/formStuff/components/TextArea.tsx | 28 +++++++++++++++++++
 frontend/src/lib/formStuff/index.tsx          |  2 ++
 .../_forklifts/-components/NewForklift.tsx    |  3 ++
 .../_forklifts/-components/NewInvoice.tsx     | 11 ++++++++
 .../_app/_forklifts/forklifts/invoices.tsx    | 21 ++++++++++++++
 8 files changed, 68 insertions(+), 1 deletion(-)
 create mode 100644 frontend/src/lib/formStuff/components/TextArea.tsx

diff --git a/app/src/internal/forklifts/routes/invoices/addInvoice.ts b/app/src/internal/forklifts/routes/invoices/addInvoice.ts
index ab2217b..5754027 100644
--- a/app/src/internal/forklifts/routes/invoices/addInvoice.ts
+++ b/app/src/internal/forklifts/routes/invoices/addInvoice.ts
@@ -28,6 +28,7 @@ router.post("/", async (req: Request, res: Response) => {
 		companyId: req.body.companyId,
 		invoiceNumber: req.body.invoiceNumber,
 		invoiceDate: req.body.invoiceDate,
+		comment: req.body.comment,
 		uploadedBy: req.body.uploadedBy,
 		totalAmount: req.body.totalAmount,
 	});
diff --git a/app/src/pkg/db/schema/forkliftLeasesInvoice.ts b/app/src/pkg/db/schema/forkliftLeasesInvoice.ts
index cbb4148..0b12e16 100644
--- a/app/src/pkg/db/schema/forkliftLeasesInvoice.ts
+++ b/app/src/pkg/db/schema/forkliftLeasesInvoice.ts
@@ -1,4 +1,4 @@
-import { numeric, pgTable, serial, uuid } from "drizzle-orm/pg-core";
+import { numeric, pgTable, serial, text, uuid } from "drizzle-orm/pg-core";
 import { createInsertSchema } from "drizzle-zod";
 import { forklifts } from "./forklifts.js";
 import { leaseInvoices } from "./leaseInvoices.js";
diff --git a/app/src/pkg/db/schema/leaseInvoices.ts b/app/src/pkg/db/schema/leaseInvoices.ts
index c77d89c..2b9dc0d 100644
--- a/app/src/pkg/db/schema/leaseInvoices.ts
+++ b/app/src/pkg/db/schema/leaseInvoices.ts
@@ -17,6 +17,7 @@ export const leaseInvoices = pgTable("lease_invoices", {
 	invoiceNumber: text("invoice_number").unique().notNull(),
 	invoiceDate: date("invoice_date").notNull(),
 	totalAmount: numeric("total_amount"),
+	comment: text("comment"),
 	add_date: timestamp("add_date"),
 	uploadedBy: text("uploaded_by"),
 });
diff --git a/frontend/src/lib/formStuff/components/TextArea.tsx b/frontend/src/lib/formStuff/components/TextArea.tsx
new file mode 100644
index 0000000..f3cb035
--- /dev/null
+++ b/frontend/src/lib/formStuff/components/TextArea.tsx
@@ -0,0 +1,28 @@
+import { Textarea } from "@/components/ui/textarea";
+import { Label } from "../../../components/ui/label";
+import { useFieldContext } from "..";
+import { FieldErrors } from "./FieldErrors";
+
+type InputFieldProps = {
+	label: string;
+	placeHolder: string;
+	required: boolean;
+};
+export const TextArea = ({ label, placeHolder, required }: InputFieldProps) => {
+	const field = useFieldContext<any>();
+
+	return (
+		<div className="grid gap-3 w-64">
+			<Label htmlFor={field.name}>{label}</Label>
+			<Textarea
+				id={field.name}
+				value={field.state.value}
+				onChange={(e) => field.handleChange(e.target.value)}
+				placeholder={placeHolder}
+				onBlur={field.handleBlur}
+				required={required}
+			/>
+			<FieldErrors meta={field.state.meta} />
+		</div>
+	);
+};
diff --git a/frontend/src/lib/formStuff/index.tsx b/frontend/src/lib/formStuff/index.tsx
index 57f942d..448385e 100644
--- a/frontend/src/lib/formStuff/index.tsx
+++ b/frontend/src/lib/formStuff/index.tsx
@@ -5,6 +5,7 @@ import { InputField } from "./components/InputField";
 import { InputPasswordField } from "./components/InputPasswordField";
 import { SelectField } from "./components/SelectField";
 import { SubmitButton } from "./components/SubmitButton";
+import { TextArea } from "./components/TextArea";
 
 export const { fieldContext, useFieldContext, formContext, useFormContext } =
 	createFormHookContexts();
@@ -16,6 +17,7 @@ export const { useAppForm } = createFormHook({
 		SelectField,
 		CheckboxField,
 		DateField,
+		TextArea,
 	},
 	formComponents: { SubmitButton },
 	fieldContext,
diff --git a/frontend/src/routes/_app/_forklifts/-components/NewForklift.tsx b/frontend/src/routes/_app/_forklifts/-components/NewForklift.tsx
index 777f0af..e9bc6fc 100644
--- a/frontend/src/routes/_app/_forklifts/-components/NewForklift.tsx
+++ b/frontend/src/routes/_app/_forklifts/-components/NewForklift.tsx
@@ -10,6 +10,7 @@ import {
 	DialogTitle,
 } from "@/components/ui/dialog";
 import { getServers } from "@/lib/querys/admin/getServers";
+import { getForklifts } from "@/lib/querys/forklifts/getForklifts";
 import { getLeases } from "@/lib/querys/forklifts/getLeases";
 import { useAppForm } from "../../../../lib/formStuff";
 
@@ -52,6 +53,7 @@ const pfc: any = [
 export default function NewForklift({ setOpenDialog }: { setOpenDialog: any }) {
 	//const search = useSearch({ from: "/_app/(auth)/login" });
 	const { data: s, isLoading: es } = useQuery(getServers());
+	const { refetch: refetchLifts } = useQuery(getForklifts());
 	const {
 		data: leases,
 		isLoading: leaseError,
@@ -91,6 +93,7 @@ export default function NewForklift({ setOpenDialog }: { setOpenDialog: any }) {
 				form.reset();
 				setOpenDialog(false);
 				refetch();
+				refetchLifts();
 				toast.success(`${value.serialNumber} was just created `);
 			} catch (error) {
 				// @ts-ignore
diff --git a/frontend/src/routes/_app/_forklifts/-components/NewInvoice.tsx b/frontend/src/routes/_app/_forklifts/-components/NewInvoice.tsx
index 8f33074..e00e588 100644
--- a/frontend/src/routes/_app/_forklifts/-components/NewInvoice.tsx
+++ b/frontend/src/routes/_app/_forklifts/-components/NewInvoice.tsx
@@ -34,6 +34,7 @@ export default function NewInvoice({
 			invoiceNumber: "",
 			invoiceDate: "",
 			totalAmount: "",
+			comment: "",
 			forklifts: [{ forklift_id: "", serialNumber: "", amount: "" }],
 		},
 		onSubmit: async ({ value }) => {
@@ -198,6 +199,16 @@ export default function NewInvoice({
 						<field.DateField label="Invoice Date" required={true} />
 					)}
 				/>
+				<form.AppField
+					name="comment"
+					children={(field) => (
+						<field.TextArea
+							label="Comment"
+							placeHolder="Enter your comment if needed."
+							required={false}
+						/>
+					)}
+				/>
 
 				<hr className="mt-2 mb-2" />
 				{/* Dynamic forklift section */}
diff --git a/frontend/src/routes/_app/_forklifts/forklifts/invoices.tsx b/frontend/src/routes/_app/_forklifts/forklifts/invoices.tsx
index e95da8b..5d875ba 100644
--- a/frontend/src/routes/_app/_forklifts/forklifts/invoices.tsx
+++ b/frontend/src/routes/_app/_forklifts/forklifts/invoices.tsx
@@ -11,6 +11,7 @@ type Invoices = {
 	id: string;
 	invoiceNumber: string;
 	invoiceDate: Date;
+	comment: string;
 	totalAmount: string;
 	add_date: Date;
 };
@@ -83,6 +84,26 @@ function RouteComponent() {
 				return <span>{format(date, "MM/dd/yyyy")}</span>;
 			},
 		}),
+		columnHelper.accessor("comment", {
+			header: ({ column }) => {
+				return (
+					<Button
+						variant="ghost"
+						onClick={() => column.toggleSorting(column.getIsSorted() === "asc")}
+					>
+						<span className="flex flex-row gap-2">Invoice Date</span>
+						{column.getIsSorted() === "asc" ? (
+							<ArrowUp className="ml-2 h-4 w-4" />
+						) : (
+							<ArrowDown className="ml-2 h-4 w-4" />
+						)}
+					</Button>
+				);
+			},
+			cell: ({ getValue }) => {
+				return <span>{getValue()}</span>;
+			},
+		}),
 		// columnHelper.accessor("add_date", {
 		// 	header: ({ column }) => {
 		// 		return (