ForgeLM/tauri-app/src/App.tsx

import { useState, useEffect, useCallback } from "react";
import Sidebar from "./components/Sidebar";
import ChatView from "./components/ChatView";
import SettingsPage from "./features/settings/SettingsPage";
import { useDB } from "./hooks/useDB";
import { useModels } from "./hooks/useModels";
import { useRuntimeManager } from "./features/runtime/useRuntimeManager";
import { useRuntimeReadyModelRefresh } from "./features/runtime/useRuntimeReadyModelRefresh";
import type { Chat } from "./lib/types";
import "./App.css";

export default function App() {
  const { db, ready } = useDB();
  const [currentChatId, setCurrentChatId] = useState<number | null>(null);
  const [chats, setChats] = useState<Chat[]>([]);
  const [sidebarOpen, setSidebarOpen] = useState(false);
  const [view, setView] = useState<"chat" | "settings">("chat");

  const runtime = useRuntimeManager();
  const { models, selectedModel, setSelectedModel, status, refresh: refreshModels } = useModels(
    runtime.connection.serverUrl,
    runtime.connection.apiKey,
    Boolean(runtime.connection.serverUrl),
  );

  useRuntimeReadyModelRefresh(runtime.snapshot.status, refreshModels);

  const loadChats = useCallback(() => {
    if (!db) return;
    const rows = db.query<Chat>("SELECT * FROM chats ORDER BY updated_at DESC");
    setChats(rows);
  }, [db]);

  useEffect(() => { loadChats(); }, [loadChats]);

  const newChat = useCallback(() => {
    if (!db) return;
    const id = db.insert("INSERT INTO chats (title) VALUES ('New Chat')");
    if (id > 0) {
      setCurrentChatId(id);
      setSidebarOpen(false);
      loadChats();
    }
  }, [db, loadChats]);

  const deleteChat = useCallback((id: number) => {
    if (!db) return;
    if (!window.confirm("Delete this conversation and all of its messages?")) return;
    db.run("DELETE FROM chats WHERE id = ?", [id]);
    if (currentChatId === id) setCurrentChatId(null);
    setSidebarOpen(false);
    loadChats();
  }, [db, currentChatId, loadChats]);

  const handleSelectChat = useCallback((id: number) => {
    setCurrentChatId(id);
    setSidebarOpen(false);
  }, []);

  if (!ready) {
    return (
      <div className="app-loading">
        <div className="loading-spinner" />
        <span>Initializing…</span>
      </div>
    );
  }

  return (
    <div className={`app-layout${sidebarOpen && view !== "settings" ? " sidebar-open" : ""}`}>
      {view !== "settings" && (
        <Sidebar
          chats={chats}
          currentChatId={currentChatId}
          onSelectChat={handleSelectChat}
          onNewChat={newChat}
          onDeleteChat={deleteChat}
          models={models}
          selectedModel={selectedModel}
          onModelChange={setSelectedModel}
          status={status}
          onRefreshModels={refreshModels}
          onOpenSettings={() => setView("settings")}
          onCloseMobile={() => setSidebarOpen(false)}
        />
      )}
      {view !== "settings" && sidebarOpen && (
        <button
          type="button"
          className="app-sidebar-backdrop"
          aria-label="Close navigation"
          onClick={() => setSidebarOpen(false)}
        />
      )}
      <div className="app-main">
        {view === "settings" ? (
          <SettingsPage runtime={runtime} onClose={() => setView("chat")} />
        ) : (
          <ChatView
            db={db!}
            chatId={currentChatId}
            server={runtime.connection.serverUrl}
            apiKey={runtime.connection.apiKey}
            model={selectedModel}
            onChatCreated={(id) => { setCurrentChatId(id); loadChats(); setSidebarOpen(false); }}
            onChatUpdated={loadChats}
            onOpenSidebar={() => setSidebarOpen(true)}
          />
        )}
      </div>
    </div>
  );
}