xertica-ui 2.2.0 → 2.3.0

package/bin/cli.ts

@@ -19,6 +19,34 @@ import {
   generateAppTsx,
 } from './language-config';
 
+// ─────────────────────────────────────────────────────────────────────────────
+// Project config helpers  (.xertica.json)
+// Persists per-project feature flags so `update` can read them later.
+// ─────────────────────────────────────────────────────────────────────────────
+
+const XERTICA_CONFIG_FILE = '.xertica.json';
+
+interface XerticaConfig {
+  version: 1;
+  hasAssistant: boolean;
+}
+
+async function readXerticaConfig(targetDir: string): Promise<XerticaConfig | null> {
+  const configPath = path.join(targetDir, XERTICA_CONFIG_FILE);
+  if (!(await fs.pathExists(configPath))) return null;
+  try {
+    return await fs.readJson(configPath) as XerticaConfig;
+  } catch {
+    return null;
+  }
+}
+
+async function writeXerticaConfig(targetDir: string, config: Partial<XerticaConfig>): Promise<void> {
+  const configPath = path.join(targetDir, XERTICA_CONFIG_FILE);
+  const existing = (await readXerticaConfig(targetDir)) ?? { version: 1 as const, hasAssistant: false };
+  await fs.writeJson(configPath, { ...existing, ...config, version: 1 }, { spaces: 2 });
+}
+
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
 
@@ -28,6 +56,214 @@ const pkgJson = JSON.parse(
   fs.readFileSync(path.resolve(__dirname, '../package.json'), 'utf-8')
 ) as { version: string };
 
+// ─────────────────────────────────────────────────────────────────────────────
+// AuthGuard generator
+// Generates src/app/components/AuthGuard.tsx based on the selected features.
+// ─────────────────────────────────────────────────────────────────────────────
+
+function generateAuthGuard({
+  hasLogin,
+  hasHome,
+  hasTemplate,
+  hasAssistant,
+  firstProtectedPath,
+}: {
+  hasLogin: boolean;
+  hasHome: boolean;
+  hasTemplate: boolean;
+  hasAssistant: boolean;
+  firstProtectedPath: string;
+}): string {
+  return `import React from 'react';
+import { Routes, Route, Navigate } from 'react-router-dom';
+import { useAuth } from '../context/AuthContext';
+
+// ─── Lazy page imports ────────────────────────────────────────────────────────
+
+${hasLogin ? `const LoginPage = React.lazy(() => import('../../pages/LoginPage').then(m => ({ default: m.LoginPage })));
+const ForgotPasswordPage = React.lazy(() => import('../../pages/ForgotPasswordPage').then(m => ({ default: m.ForgotPasswordPage })));
+const VerifyEmailPage = React.lazy(() => import('../../pages/VerifyEmailPage').then(m => ({ default: m.VerifyEmailPage })));
+const ResetPasswordPage = React.lazy(() => import('../../pages/ResetPasswordPage').then(m => ({ default: m.ResetPasswordPage })));` : ''}
+
+${hasHome ? `const HomePage = React.lazy(() => import('../../pages/HomePage').then(m => ({ default: m.HomePage })));` : ''}
+${hasTemplate ? `const TemplatePage = React.lazy(() => import('../../pages/TemplatePage').then(m => ({ default: m.TemplatePage })));` : ''}
+${hasAssistant ? `const AssistantPage = React.lazy(() => import('../../pages/AssistantPage').then(m => ({ default: m.AssistantPage })));` : ''}
+
+// ─── Route guards ─────────────────────────────────────────────────────────────
+
+function ProtectedRoute({ children }: { children: React.ReactNode }) {
+  const { user, isLoading } = useAuth();
+  if (isLoading) return null;
+  if (!user) return <Navigate to="${hasLogin ? '/login' : firstProtectedPath}" replace />;
+  return <>{children}</>;
+}
+
+${hasLogin ? `function GuestRoute({ children }: { children: React.ReactNode }) {
+  const { user, isLoading } = useAuth();
+  if (isLoading) return null;
+  if (user) return <Navigate to="${firstProtectedPath}" replace />;
+  return <>{children}</>;
+}
+
+function LoginPageWithAuth() {
+  const { login } = useAuth();
+  return <LoginPage onLogin={login} />;
+}` : ''}
+
+// ─── Route tree ───────────────────────────────────────────────────────────────
+
+export function AuthGuard() {
+  const { user } = useAuth();
+
+  return (
+    <div className="min-h-screen bg-muted overflow-x-hidden max-w-full">
+      <Routes>
+${hasLogin ? `        <Route path="/login" element={<GuestRoute><LoginPageWithAuth /></GuestRoute>} />
+        <Route path="/forgot-password" element={<GuestRoute><ForgotPasswordPage /></GuestRoute>} />
+        <Route path="/verify-email" element={<GuestRoute><VerifyEmailPage /></GuestRoute>} />
+        <Route path="/reset-password" element={<GuestRoute><ResetPasswordPage /></GuestRoute>} />` : ''}
+
+${hasHome ? `        <Route path="/home" element={<ProtectedRoute><HomePage /></ProtectedRoute>} />` : ''}
+${hasTemplate ? `        <Route path="/template" element={<ProtectedRoute><TemplatePage /></ProtectedRoute>} />` : ''}
+${hasAssistant ? `        <Route path="/assistente" element={<ProtectedRoute><AssistantPage /></ProtectedRoute>} />` : ''}
+
+        <Route path="/" element={<Navigate to={user ? '${firstProtectedPath}' : '${hasLogin ? '/login' : firstProtectedPath}'} replace />} />
+        <Route path="*" element={<Navigate to={user ? '${firstProtectedPath}' : '${hasLogin ? '/login' : firstProtectedPath}'} replace />} />
+      </Routes>
+    </div>
+  );
+}
+`;
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// Page generators
+// Generate page files that vary based on whether the assistant is included.
+// ─────────────────────────────────────────────────────────────────────────────
+
+function generateHomePage(hasAssistant: boolean): string {
+  if (hasAssistant) {
+    return `import React from 'react';
+import { XerticaAssistant, generateDemoResponse } from 'xertica-ui/assistant';
+import { useLayout } from 'xertica-ui/hooks';
+import { useNavigate } from 'react-router-dom';
+import { AppLayout } from '../app/components/AppLayout';
+import { HomeContent } from '../features/home';
+import { useAssistantConfig, getMockRichSuggestions, getMockFeedbackOptions } from '../features/assistant';
+
+/**
+ * Home page — thin layout shell.
+ *
+ * Assistant config (suggestions, feedback options) is fetched via React Query.
+ * To connect to a real API replace \`fetchAssistantConfig\` in
+ * \`features/assistant/data/mock.ts\`.
+ */
+export function HomePage() {
+  const { assistenteExpanded, toggleAssistente } = useLayout();
+  const navigate = useNavigate();
+
+  const { data: assistantConfig } = useAssistantConfig();
+
+  return (
+    <AppLayout
+      assistant={
+        <XerticaAssistant
+          isExpanded={assistenteExpanded}
+          onToggle={toggleAssistente}
+          defaultTab="chat"
+          demoMode={true}
+          userName="Ariel Santos"
+          responseGenerator={generateDemoResponse}
+          suggestions={assistantConfig?.suggestions}
+          richSuggestions={assistantConfig?.richSuggestions ?? getMockRichSuggestions()}
+          feedbackOptions={assistantConfig?.feedbackOptions ?? getMockFeedbackOptions()}
+          showHistory={false}
+          showFavorites={false}
+          onNavigateSettings={() => navigate('/settings')}
+          onNavigateFullPage={() => navigate('/assistente')}
+          onEvaluation={(messageId, type, reason) => {
+            // Wire your feedback persistence logic here
+            console.log(\`Avaliação: \${type} na mensagem \${messageId}. Motivo: \${reason}\`);
+          }}
+        />
+      }
+    >
+      <HomeContent />
+    </AppLayout>
+  );
+}
+`;
+  }
+
+  return `import React from 'react';
+import { AppLayout } from '../app/components/AppLayout';
+import { HomeContent } from '../features/home';
+
+/**
+ * Home page — thin layout shell.
+ */
+export function HomePage() {
+  return (
+    <AppLayout>
+      <HomeContent />
+    </AppLayout>
+  );
+}
+`;
+}
+
+function generateTemplatePage(hasAssistant: boolean): string {
+  if (hasAssistant) {
+    return `import React from 'react';
+import { XerticaAssistant } from 'xertica-ui/assistant';
+import { useLayout } from 'xertica-ui/hooks';
+import { AppLayout } from '../app/components/AppLayout';
+import { TemplateContent } from '../features/template';
+
+/**
+ * Template page — thin layout shell.
+ *
+ * Auth state is consumed from \`AuthContext\` via \`AppLayout\` — no props needed.
+ */
+export function TemplatePage() {
+  const { assistenteExpanded, toggleAssistente } = useLayout();
+
+  return (
+    <AppLayout
+      assistant={
+        <XerticaAssistant
+          isExpanded={assistenteExpanded}
+          onToggle={toggleAssistente}
+          onEvaluation={(id, type, reason) => console.log('Feedback:', id, type, reason)}
+        />
+      }
+    >
+      <TemplateContent />
+    </AppLayout>
+  );
+}
+`;
+  }
+
+  return `import React from 'react';
+import { AppLayout } from '../app/components/AppLayout';
+import { TemplateContent } from '../features/template';
+
+/**
+ * Template page — thin layout shell.
+ *
+ * Auth state is consumed from \`AuthContext\` via \`AppLayout\` — no props needed.
+ */
+export function TemplatePage() {
+  return (
+    <AppLayout>
+      <TemplateContent />
+    </AppLayout>
+  );
+}
+`;
+}
+
 const program = new Command();
 
 program
@@ -43,7 +279,7 @@ program
     const targetDir = path.resolve(process.cwd(), directory);
     const templatesDir = path.resolve(__dirname, '../templates');
 
-    console.log(chalk.blue('🚀 Welcome to Xertica UI CLI!'));
+    console.log(chalk.blue(`🚀 Welcome to Xertica UI CLI! ${chalk.dim(`v${pkgJson.version}`)}`));
 
     const response = await prompts([
       {
@@ -84,6 +320,12 @@ program
         })),
         initial: 0,
       },
+      {
+        type: 'confirm',
+        name: 'hasAssistant',
+        message: 'Include AI Assistant? (XerticaAssistant chat page + sidebar variant)',
+        initial: true,
+      },
       {
         type: 'confirm',
         name: 'install',
@@ -104,6 +346,7 @@ program
       const hasLogin = pages.includes('login');
       const hasHome = pages.includes('home');
       const hasTemplate = pages.includes('template');
+      const hasAssistant = response.hasAssistant ?? true;
 
       // Resolve selected languages — fall back to all defaults if the user
       // somehow ended up with an empty array (the prompt's min:1 should prevent
@@ -182,6 +425,9 @@ program
       // 6.3 Persist the language selection so `update` can remember it
       await writeLanguagesConfig(targetDir, selectedLanguages);
 
+      // 6.5 Persist project feature flags (.xertica.json)
+      await writeXerticaConfig(targetDir, { hasAssistant });
+
       // 6.4 Copy context
       await fs.ensureDir(path.join(targetDir, 'src', 'app', 'context'));
       await fs.copy(
@@ -208,11 +454,13 @@ program
           path.join(targetDir, 'src', 'features', 'template')
         );
       }
-      // Always copy assistant as AppLayout depends on it
-      await fs.copy(
-        path.join(templatesDir, 'src', 'features', 'assistant'),
-        path.join(targetDir, 'src', 'features', 'assistant')
-      );
+      // Copy assistant feature only if selected
+      if (hasAssistant) {
+        await fs.copy(
+          path.join(templatesDir, 'src', 'features', 'assistant'),
+          path.join(targetDir, 'src', 'features', 'assistant')
+        );
+      }
 
       // 8. Copy pages based on selections
       await fs.ensureDir(path.join(targetDir, 'src', 'pages'));
@@ -227,9 +475,24 @@ program
         );
       if (hasHome) pagesToCopy.push('HomePage.tsx');
       if (hasTemplate) pagesToCopy.push('TemplatePage.tsx');
-      pagesToCopy.push('AssistantPage.tsx');
+      if (hasAssistant) pagesToCopy.push('AssistantPage.tsx');
 
       for (const pageFile of pagesToCopy) {
+        // HomePage and TemplatePage are generated (they vary by hasAssistant)
+        if (pageFile === 'HomePage.tsx') {
+          await fs.writeFile(
+            path.join(targetDir, 'src', 'pages', 'HomePage.tsx'),
+            generateHomePage(hasAssistant)
+          );
+          continue;
+        }
+        if (pageFile === 'TemplatePage.tsx') {
+          await fs.writeFile(
+            path.join(targetDir, 'src', 'pages', 'TemplatePage.tsx'),
+            generateTemplatePage(hasAssistant)
+          );
+          continue;
+        }
         const src = path.join(templatesDir, 'src', 'pages', pageFile);
         if (await fs.pathExists(src)) {
           await fs.copy(src, path.join(targetDir, 'src', 'pages', pageFile));
@@ -238,67 +501,7 @@ program
 
       // 9. Generate AuthGuard.tsx based on selected pages
       const firstProtectedPath = hasHome ? '/home' : hasTemplate ? '/template' : '/login';
-
-      const authGuardContent = `import React from 'react';
-import { Routes, Route, Navigate } from 'react-router-dom';
-import { useAuth } from '../context/AuthContext';
-
-// ─── Lazy page imports ────────────────────────────────────────────────────────
-
-${hasLogin ? `const LoginPage = React.lazy(() => import('../../pages/LoginPage').then(m => ({ default: m.LoginPage })));
-const ForgotPasswordPage = React.lazy(() => import('../../pages/ForgotPasswordPage').then(m => ({ default: m.ForgotPasswordPage })));
-const VerifyEmailPage = React.lazy(() => import('../../pages/VerifyEmailPage').then(m => ({ default: m.VerifyEmailPage })));
-const ResetPasswordPage = React.lazy(() => import('../../pages/ResetPasswordPage').then(m => ({ default: m.ResetPasswordPage })));` : ''}
-
-${hasHome ? `const HomePage = React.lazy(() => import('../../pages/HomePage').then(m => ({ default: m.HomePage })));` : ''}
-${hasTemplate ? `const TemplatePage = React.lazy(() => import('../../pages/TemplatePage').then(m => ({ default: m.TemplatePage })));` : ''}
-const AssistantPage = React.lazy(() => import('../../pages/AssistantPage').then(m => ({ default: m.AssistantPage })));
-
-// ─── Route guards ─────────────────────────────────────────────────────────────
-
-function ProtectedRoute({ children }: { children: React.ReactNode }) {
-  const { user, isLoading } = useAuth();
-  if (isLoading) return null;
-  if (!user) return <Navigate to="${hasLogin ? '/login' : firstProtectedPath}" replace />;
-  return <>{children}</>;
-}
-
-${hasLogin ? `function GuestRoute({ children }: { children: React.ReactNode }) {
-  const { user, isLoading } = useAuth();
-  if (isLoading) return null;
-  if (user) return <Navigate to="${firstProtectedPath}" replace />;
-  return <>{children}</>;
-}
-
-function LoginPageWithAuth() {
-  const { login } = useAuth();
-  return <LoginPage onLogin={login} />;
-}` : ''}
-
-// ─── Route tree ───────────────────────────────────────────────────────────────
-
-export function AuthGuard() {
-  const { user } = useAuth();
-
-  return (
-    <div className="min-h-screen bg-muted overflow-x-hidden max-w-full">
-      <Routes>
-${hasLogin ? `        <Route path="/login" element={<GuestRoute><LoginPageWithAuth /></GuestRoute>} />
-        <Route path="/forgot-password" element={<GuestRoute><ForgotPasswordPage /></GuestRoute>} />
-        <Route path="/verify-email" element={<GuestRoute><VerifyEmailPage /></GuestRoute>} />
-        <Route path="/reset-password" element={<GuestRoute><ResetPasswordPage /></GuestRoute>} />` : ''}
-
-${hasHome ? `        <Route path="/home" element={<ProtectedRoute><HomePage /></ProtectedRoute>} />` : ''}
-${hasTemplate ? `        <Route path="/template" element={<ProtectedRoute><TemplatePage /></ProtectedRoute>} />` : ''}
-        <Route path="/assistente" element={<ProtectedRoute><AssistantPage /></ProtectedRoute>} />
-
-        <Route path="/" element={<Navigate to={user ? '${firstProtectedPath}' : '${hasLogin ? '/login' : firstProtectedPath}'} replace />} />
-        <Route path="*" element={<Navigate to={user ? '${firstProtectedPath}' : '${hasLogin ? '/login' : firstProtectedPath}'} replace />} />
-      </Routes>
-    </div>
-  );
-}
-`;
+      const authGuardContent = generateAuthGuard({ hasLogin, hasHome, hasTemplate, hasAssistant, firstProtectedPath });
 
       await fs.writeFile(
         path.join(targetDir, 'src', 'app', 'components', 'AuthGuard.tsx'),
@@ -343,6 +546,12 @@ ${hasTemplate ? `        <Route path="/template" element={<ProtectedRoute><Templ
       console.log(
         chalk.gray('  To add/remove languages later: npx xertica-ui update → Languages')
       );
+      console.log(
+        chalk.gray(`  AI Assistant: ${hasAssistant ? 'included (/assistente)' : 'not included'}`)
+      );
+      if (!hasAssistant) {
+        console.log(chalk.gray('  To add the assistant later: npx xertica-ui update → Assistant'));
+      }
     } catch (error) {
       spinner.fail('Failed to initialize project');
       console.error(error);
@@ -356,6 +565,10 @@ program
   .action(async () => {
     const targetDir = process.cwd();
 
+    console.log(chalk.blue(`🔧 Xertica UI CLI ${chalk.dim(`v${pkgJson.version}`)}`));
+
+    const currentConfig = await readXerticaConfig(targetDir);
+
     const { updateType } = await prompts({
       type: 'select',
       name: 'updateType',
@@ -371,6 +584,13 @@ program
           description: 'Add or remove supported languages (pt-BR, en, es, …)',
           value: 'languages',
         },
+        {
+          title: 'Assistant',
+          description: currentConfig?.hasAssistant
+            ? 'Remove the AI Assistant from your project'
+            : 'Add the AI Assistant to your project',
+          value: 'assistant',
+        },
         {
           title: 'Project files',
           description: 'Update app shell, shared, features and pages to a specific version',
@@ -548,6 +768,155 @@ program
       return;
     }
 
+    // ── Assistant add / remove ────────────────────────────────────────────────
+    if (updateType === 'assistant') {
+      // Determine current state: prefer persisted config, fall back to file presence
+      // (handles projects scaffolded before .xertica.json existed).
+      let currentlyHas: boolean;
+      if (currentConfig !== null) {
+        currentlyHas = currentConfig.hasAssistant;
+      } else {
+        const assistantFeatureDir = path.join(targetDir, 'src', 'features', 'assistant');
+        const assistantPage = path.join(targetDir, 'src', 'pages', 'AssistantPage.tsx');
+        currentlyHas =
+          (await fs.pathExists(assistantFeatureDir)) || (await fs.pathExists(assistantPage));
+        // Persist the inferred state so future runs don't need to infer again
+        await writeXerticaConfig(targetDir, { hasAssistant: currentlyHas });
+      }
+
+      console.log(
+        chalk.cyan(
+          `\nAI Assistant is currently: ${currentlyHas ? chalk.green('enabled') : chalk.red('disabled')}\n`
+        )
+      );
+
+      const { action } = await prompts({
+        type: 'select',
+        name: 'action',
+        message: currentlyHas
+          ? 'Remove the AI Assistant from your project?'
+          : 'Add the AI Assistant to your project?',
+        choices: currentlyHas
+          ? [
+              { title: 'Remove assistant', description: 'Deletes AssistantPage and assistant feature files', value: 'remove' },
+              { title: 'Cancel', value: 'cancel' },
+            ]
+          : [
+              { title: 'Add assistant', description: 'Copies AssistantPage and assistant feature files', value: 'add' },
+              { title: 'Cancel', value: 'cancel' },
+            ],
+      });
+
+      if (!action || action === 'cancel') {
+        console.log(chalk.gray('Update cancelled.'));
+        return;
+      }
+
+      const { confirmed } = await prompts({
+        type: 'confirm',
+        name: 'confirmed',
+        message: chalk.yellow(
+          action === 'remove'
+            ? '⚠️  This will delete src/features/assistant/ and src/pages/AssistantPage.tsx and regenerate AuthGuard.tsx. Continue?'
+            : '⚠️  This will copy src/features/assistant/ and src/pages/AssistantPage.tsx and regenerate AuthGuard.tsx. Continue?'
+        ),
+        initial: action === 'add',
+      });
+
+      if (!confirmed) {
+        console.log(chalk.gray('Update cancelled.'));
+        return;
+      }
+
+      const spinner = ora(action === 'add' ? 'Adding assistant...' : 'Removing assistant...').start();
+
+      try {
+        const installedTemplatesDir = path.join(targetDir, 'node_modules', 'xertica-ui', 'templates');
+        const templatesSourceDir = (await fs.pathExists(installedTemplatesDir))
+          ? installedTemplatesDir
+          : path.resolve(__dirname, '../templates');
+
+        // Infer current page set from the pages directory
+        const pagesDir = path.join(targetDir, 'src', 'pages');
+        const existingPages = (await fs.pathExists(pagesDir)) ? await fs.readdir(pagesDir) : [];
+        const hasLogin = existingPages.includes('LoginPage.tsx');
+        const hasHome = existingPages.includes('HomePage.tsx');
+        const hasTemplate = existingPages.includes('TemplatePage.tsx');
+        const firstProtectedPath = hasHome ? '/home' : hasTemplate ? '/template' : '/login';
+
+        // Read persisted language selection so AuthGuard can be regenerated correctly
+        const persistedCodes = await readLanguagesConfig(targetDir);
+        const selectedCodes =
+          persistedCodes && persistedCodes.length > 0 ? persistedCodes : DEFAULT_SELECTION;
+
+        if (action === 'add') {
+          // Copy assistant feature
+          await fs.copy(
+            path.join(templatesSourceDir, 'src', 'features', 'assistant'),
+            path.join(targetDir, 'src', 'features', 'assistant'),
+            { overwrite: true }
+          );
+          // Copy AssistantPage
+          await fs.copy(
+            path.join(templatesSourceDir, 'src', 'pages', 'AssistantPage.tsx'),
+            path.join(targetDir, 'src', 'pages', 'AssistantPage.tsx'),
+            { overwrite: true }
+          );
+        } else {
+          // Remove assistant feature
+          await fs.remove(path.join(targetDir, 'src', 'features', 'assistant'));
+          await fs.remove(path.join(targetDir, 'src', 'pages', 'AssistantPage.tsx'));
+        }
+
+        // Regenerate pages and AuthGuard reflecting the new assistant state
+        const newHasAssistant = action === 'add';
+
+        if (hasHome) {
+          await fs.writeFile(
+            path.join(targetDir, 'src', 'pages', 'HomePage.tsx'),
+            generateHomePage(newHasAssistant)
+          );
+        }
+        if (hasTemplate) {
+          await fs.writeFile(
+            path.join(targetDir, 'src', 'pages', 'TemplatePage.tsx'),
+            generateTemplatePage(newHasAssistant)
+          );
+        }
+
+        await fs.writeFile(
+          path.join(targetDir, 'src', 'app', 'components', 'AuthGuard.tsx'),
+          generateAuthGuard({
+            hasLogin,
+            hasHome,
+            hasTemplate,
+            hasAssistant: newHasAssistant,
+            firstProtectedPath,
+          })
+        );
+
+        // Persist the updated flag
+        await writeXerticaConfig(targetDir, { hasAssistant: newHasAssistant });
+
+        spinner.succeed(
+          action === 'add'
+            ? 'AI Assistant added successfully!'
+            : 'AI Assistant removed successfully!'
+        );
+
+        if (action === 'add') {
+          console.log(chalk.gray('\n  Route /assistente is now available.'));
+          console.log(chalk.gray('  Configure your Gemini API key in VITE_GEMINI_API_KEY.'));
+        } else {
+          console.log(chalk.gray('\n  Assistant files removed and AuthGuard updated.'));
+        }
+      } catch (error) {
+        spinner.fail('Failed to update assistant');
+        console.error(error);
+      }
+      return;
+    }
+
     // ── Project files update ──────────────────────────────────────────────────
     const { versionType } = await prompts({
       type: 'select',
@@ -684,6 +1053,26 @@ program
         await syncLocaleFiles(updatedTemplatesDir, targetDir, selectedCodes, {
           pruneOthers: true,
         });
+
+        // Regenerate AuthGuard preserving the current page set and assistant flag
+        const projectConfig = await readXerticaConfig(targetDir);
+        const pagesDir = path.join(targetDir, 'src', 'pages');
+        const existingPages = (await fs.pathExists(pagesDir)) ? await fs.readdir(pagesDir) : [];
+        const hasLoginP = existingPages.includes('LoginPage.tsx');
+        const hasHomeP = existingPages.includes('HomePage.tsx');
+        const hasTemplateP = existingPages.includes('TemplatePage.tsx');
+        const hasAssistantP = projectConfig?.hasAssistant ?? existingPages.includes('AssistantPage.tsx');
+        const firstProtectedP = hasHomeP ? '/home' : hasTemplateP ? '/template' : '/login';
+        await fs.writeFile(
+          path.join(targetDir, 'src', 'app', 'components', 'AuthGuard.tsx'),
+          generateAuthGuard({
+            hasLogin: hasLoginP,
+            hasHome: hasHomeP,
+            hasTemplate: hasTemplateP,
+            hasAssistant: hasAssistantP,
+            firstProtectedPath: firstProtectedP,
+          })
+        );
       }
 
       if (filesToUpdate.includes('shared')) {
@@ -696,7 +1085,7 @@ program
 
       if (filesToUpdate.includes('features')) {
         // Only update feature directories that already exist in the project.
-        // 'assistant' is always included because AppLayout depends on it.
+        // 'assistant' is only present if it was included during init (or added via update → Assistant).
         for (const feature of ['auth', 'home', 'template', 'assistant']) {
           const destFeature = path.join(targetDir, 'src', 'features', feature);
           const srcFeature = path.join(updatedTemplatesDir, 'src', 'features', feature);
@@ -710,9 +1099,27 @@ program
         const pagesDir = path.join(targetDir, 'src', 'pages');
         const srcPagesDir = path.join(updatedTemplatesDir, 'src', 'pages');
         if ((await fs.pathExists(pagesDir)) && (await fs.pathExists(srcPagesDir))) {
-          // Only overwrite pages that already exist in the project
+          const projectCfg = await readXerticaConfig(targetDir);
           const existingPages = await fs.readdir(pagesDir);
+          const assistantEnabled =
+            projectCfg?.hasAssistant ?? existingPages.includes('AssistantPage.tsx');
+
           for (const pageFile of existingPages) {
+            // Generated pages: regenerate instead of copy so assistant imports stay correct
+            if (pageFile === 'HomePage.tsx') {
+              await fs.writeFile(
+                path.join(pagesDir, 'HomePage.tsx'),
+                generateHomePage(assistantEnabled)
+              );
+              continue;
+            }
+            if (pageFile === 'TemplatePage.tsx') {
+              await fs.writeFile(
+                path.join(pagesDir, 'TemplatePage.tsx'),
+                generateTemplatePage(assistantEnabled)
+              );
+              continue;
+            }
             const src = path.join(srcPagesDir, pageFile);
             if (await fs.pathExists(src)) {
               await fs.copy(src, path.join(pagesDir, pageFile), { overwrite: true });