From 7d77e14319d78e1a27e4bdf8f0d492c4333e022b Mon Sep 17 00:00:00 2001
From: Mauricio Siu <47042324+Siumauricio@users.noreply.github.com>
Date: Tue, 25 Feb 2025 23:28:19 -0600
Subject: [PATCH] feat(organization): add logo support for organizations

---
 .../organization/handle-organization.tsx       | 18 ++++++++++++++++--
 apps/dokploy/components/layouts/side.tsx       |  2 ++
 apps/dokploy/components/shared/logo.tsx        | 14 +++++++++++++-
 .../dokploy/server/api/routers/organization.ts |  7 ++++++-
 4 files changed, 37 insertions(+), 4 deletions(-)

diff --git a/apps/dokploy/components/dashboard/organization/handle-organization.tsx b/apps/dokploy/components/dashboard/organization/handle-organization.tsx
index 2a595f43..249b8ccd 100644
--- a/apps/dokploy/components/dashboard/organization/handle-organization.tsx
+++ b/apps/dokploy/components/dashboard/organization/handle-organization.tsx
@@ -35,14 +35,16 @@ export function AddOrganization({ organizationId }: Props) {
 		: api.organization.create.useMutation();
 	const [open, setOpen] = useState(false);
 	const [name, setName] = useState("");
+	const [logo, setLogo] = useState("");
 
 	useEffect(() => {
 		if (organization) {
 			setName(organization.name);
+			setLogo(organization.logo || "");
 		}
 	}, [organization]);
 	const handleSubmit = async () => {
-		await mutateAsync({ name, organizationId: organizationId ?? "" })
+		await mutateAsync({ name, logo, organizationId: organizationId ?? "" })
 			.then(() => {
 				setOpen(false);
 				toast.success(
@@ -91,7 +93,7 @@ export function AddOrganization({ organizationId }: Props) {
 					</DialogTitle>
 					<DialogDescription>
 						{organizationId
-							? "Update the organization name"
+							? "Update the organization name and logo"
 							: "Create a new organization to manage your projects."}
 					</DialogDescription>
 				</DialogHeader>
@@ -107,6 +109,18 @@ export function AddOrganization({ organizationId }: Props) {
 							className="col-span-3"
 						/>
 					</div>
+					<div className="grid grid-cols-4 items-center gap-4">
+						<Label htmlFor="logo" className="text-right">
+							Logo URL
+						</Label>
+						<Input
+							id="logo"
+							value={logo}
+							onChange={(e) => setLogo(e.target.value)}
+							placeholder="https://example.com/logo.png"
+							className="col-span-3"
+						/>
+					</div>
 				</div>
 				<DialogFooter>
 					<Button type="submit" onClick={handleSubmit} isLoading={isLoading}>
diff --git a/apps/dokploy/components/layouts/side.tsx b/apps/dokploy/components/layouts/side.tsx
index 05498871..58638509 100644
--- a/apps/dokploy/components/layouts/side.tsx
+++ b/apps/dokploy/components/layouts/side.tsx
@@ -562,6 +562,7 @@ function SidebarLogo() {
 													"transition-all",
 													state === "collapsed" ? "size-4" : "size-5",
 												)}
+												logoUrl={activeOrganization?.logo}
 											/>
 										</div>
 										<div
@@ -607,6 +608,7 @@ function SidebarLogo() {
 														"transition-all",
 														state === "collapsed" ? "size-6" : "size-10",
 													)}
+													logoUrl={org.logo ?? undefined}
 												/>
 											</div>
 										</DropdownMenuItem>
diff --git a/apps/dokploy/components/shared/logo.tsx b/apps/dokploy/components/shared/logo.tsx
index 086ef3b0..bad6346c 100644
--- a/apps/dokploy/components/shared/logo.tsx
+++ b/apps/dokploy/components/shared/logo.tsx
@@ -1,8 +1,20 @@
 interface Props {
 	className?: string;
+	logoUrl?: string;
 }
 
-export const Logo = ({ className = "size-14" }: Props) => {
+export const Logo = ({ className = "size-14", logoUrl }: Props) => {
+	if (logoUrl) {
+		return (
+			<img
+				src={logoUrl}
+				alt="Organization Logo"
+				className={className}
+				style={{ objectFit: "contain" }}
+			/>
+		);
+	}
+
 	return (
 		<svg
 			xmlns="http://www.w3.org/2000/svg"
diff --git a/apps/dokploy/server/api/routers/organization.ts b/apps/dokploy/server/api/routers/organization.ts
index b98fc26c..6f7a9c67 100644
--- a/apps/dokploy/server/api/routers/organization.ts
+++ b/apps/dokploy/server/api/routers/organization.ts
@@ -11,6 +11,7 @@ export const organizationRouter = createTRPCRouter({
 		.input(
 			z.object({
 				name: z.string(),
+				logo: z.string().optional(),
 			}),
 		)
 		.mutation(async ({ ctx, input }) => {
@@ -81,6 +82,7 @@ export const organizationRouter = createTRPCRouter({
 			z.object({
 				organizationId: z.string(),
 				name: z.string(),
+				logo: z.string().optional(),
 			}),
 		)
 		.mutation(async ({ ctx, input }) => {
@@ -92,7 +94,10 @@ export const organizationRouter = createTRPCRouter({
 			}
 			const result = await db
 				.update(organization)
-				.set({ name: input.name })
+				.set({
+					name: input.name,
+					logo: input.logo,
+				})
 				.where(eq(organization.id, input.organizationId))
 				.returning();
 			return result[0];