Cloud-Code-AI · sauravpanda · Dec 2, 2024 · Dec 2, 2024
diff --git a/packages/editor/src/app/editMode/page.tsx b/packages/editor/src/app/editMode/page.tsx
@@ -7,6 +7,7 @@ import { Input } from "@/components/ui/input"
 import { Folder, File, Plus, X, ChevronRight, ChevronDown, Trash2 } from "lucide-react"
 import { motion, AnimatePresence } from "framer-motion"
 import { fetchAllContent } from '@/src/lib/getContents'
+import { ThemeToggle } from "@/components/ui/ThemeToggle"
 
 const API_URL = process.env.NEXT_PUBLIC_BACKEND_API_URL || 'http://localhost:3001'
 
@@ -408,9 +409,12 @@ export default function ImprovedFileTreeUI() {
 
   return (
     <div className="p-6 bg-background min-h-screen text-foreground">
-      <h1 className="text-3xl font-bold mb-6 text-foreground">
-        Project Explorer
-      </h1>
+      <div className="flex items-center justify-between mb-6">
+        <h1 className="text-3xl font-bold text-foreground">
+          Project Explorer
+        </h1>
+        <ThemeToggle />
+      </div>
       <div className="bg-card rounded-lg shadow-xl p-6 border">
         <motion.div
           initial={{ opacity: 0, y: 20 }}

diff --git a/packages/editor/src/app/layout.tsx b/packages/editor/src/app/layout.tsx
@@ -1,6 +1,8 @@
 import type { Metadata } from "next";
 import localFont from "next/font/local";
 import "./globals.css";
+import { ThemeProvider } from "next-themes";
+import { Toaster } from "sonner";
 
 const geistSans = localFont({
   src: "./fonts/GeistVF.woff",
@@ -24,11 +26,19 @@ export default function RootLayout({
   children: React.ReactNode;
 }>) {
   return (
-    <html lang="en">
+    <html lang="en" suppressHydrationWarning>
       <body
         className={`${geistSans.variable} ${geistMono.variable} antialiased`}
       >
-        {children}
+        <ThemeProvider
+          attribute="class"
+          defaultTheme="system"
+          enableSystem
+          disableTransitionOnChange
+        >
+          {children}
+          <Toaster position="bottom-right" />
+        </ThemeProvider>
       </body>
     </html>
   );

diff --git a/packages/editor/src/components/layout/TitleBar.tsx b/packages/editor/src/components/layout/TitleBar.tsx
@@ -4,6 +4,7 @@ import { ArrowLeft, Save } from 'lucide-react'
 import { Button } from "@/components/ui/button"
 import Link from "next/link"
 import { usePathname } from 'next/navigation'
+import { ThemeToggle } from "@/components/ui/ThemeToggle"
 
 interface TitleBarProps {
   onSave: () => void
@@ -34,14 +35,17 @@ export function TitleBar({ onSave, isSaving = false }: TitleBarProps) {
         <h1 className="text-lg font-medium">{filename}</h1>
       </div>
 
-      <Button 
-        onClick={onSave}
-        disabled={isSaving}
-        className="flex items-center space-x-2"
-      >
-        <Save className="w-4 h-4" />
-        <span>{isSaving ? 'Saving...' : 'Save'}</span>
-      </Button>
+      <div className="flex items-center space-x-4">
+        <Button 
+          onClick={onSave} 
+          disabled={isSaving}
+          className="flex items-center gap-2"
+        >
+          <Save className="w-4 h-4" />
+          {isSaving ? 'Saving...' : 'Save'}
+        </Button>
+        <ThemeToggle />
+      </div>
     </div>
   )
 }
diff --git a/packages/editor/src/components/ui/ThemeToggle.tsx b/packages/editor/src/components/ui/ThemeToggle.tsx
@@ -0,0 +1,35 @@
+'use client'
+
+import { Moon, Sun } from "lucide-react"
+import { useTheme } from "next-themes"
+import { Button } from "./button"
+import { useEffect, useState } from "react"
+
+export function ThemeToggle() {
+  const [mounted, setMounted] = useState(false)
+  const { theme, setTheme } = useTheme()
+
+  useEffect(() => {
+    setMounted(true)
+  }, [])
+
+  if (!mounted) {
+    return null
+  }
+
+  return (
+    <Button
+      variant="ghost"
+      size="icon"
+      onClick={() => setTheme(theme === "light" ? "dark" : "light")}
+      className="rounded-full hover:bg-muted"
+    >
+      {theme === "light" ? (
+        <Moon className="h-5 w-5" />
+      ) : (
+        <Sun className="h-5 w-5" />
+      )}
+      <span className="sr-only">Toggle theme</span>
+    </Button>
+  )
+}