docit-ai 1.0.0

package/server/src/routes/dashboard.ts

@@ -0,0 +1,243 @@
+import { Router } from "express";
+import { supabase } from "../config/database.js";
+import { authMiddleware, type AuthRequest } from "../middleware/auth.js";
+import { PLANS, getPlanById } from "../services/payment.js";
+
+const router = Router();
+
+// GET /dashboard/stats - Dashboard statistics
+router.get("/stats", authMiddleware, async (req: AuthRequest, res) => {
+  try {
+    const user = req.user!;
+
+    // Get total documents generated count
+    const { count: documentsGenerated } = await supabase
+      .from("document_generations")
+      .select("*", { count: "exact", head: true })
+      .eq("user_id", user.id);
+
+    // Get total files processed
+    const { data: filesData } = await supabase
+      .from("document_generations")
+      .select("files_count")
+      .eq("user_id", user.id);
+
+    const filesProcessed =
+      filesData?.reduce((sum, doc) => sum + (doc.files_count ?? 0), 0) ?? 0;
+
+    // Get last sync (last document generation time)
+    const { data: lastDoc } = await supabase
+      .from("document_generations")
+      .select("created_at")
+      .eq("user_id", user.id)
+      .order("created_at", { ascending: false })
+      .limit(1)
+      .single();
+
+    res.json({
+      creditsAvailable: user.credits,
+      documentsGenerated: documentsGenerated ?? 0,
+      filesProcessed,
+      lastSync: lastDoc?.created_at ?? null,
+    });
+  } catch (error) {
+    console.error("Stats error:", error);
+    res.status(500).json({ error: "Failed to fetch stats" });
+  }
+});
+
+// GET /dashboard/activity - Recent user activity
+router.get("/activity", authMiddleware, async (req: AuthRequest, res) => {
+  try {
+    const user = req.user!;
+    const limit = Math.min(parseInt(String(req.query.limit ?? "10"), 10), 50);
+
+    // Get recent credit transactions
+    const { data: transactions } = await supabase
+      .from("credit_transactions")
+      .select("*")
+      .eq("user_id", user.id)
+      .order("created_at", { ascending: false })
+      .limit(limit);
+
+    const activities = (transactions ?? []).map((tx) => {
+      let activityType:
+        | "document_generated"
+        | "credits_purchased"
+        | "files_processed";
+
+      if (tx.type === "purchase") {
+        activityType = "credits_purchased";
+      } else if (tx.type === "usage") {
+        activityType = tx.description?.includes("file")
+          ? "files_processed"
+          : "document_generated";
+      } else {
+        activityType = "document_generated";
+      }
+
+      return {
+        id: tx.id,
+        type: activityType,
+        description: tx.description,
+        timestamp: tx.created_at,
+        creditsUsed: tx.type === "usage" ? Math.abs(tx.amount) : null,
+      };
+    });
+
+    res.json({ activities });
+  } catch (error) {
+    console.error("Activity error:", error);
+    res.status(500).json({ error: "Failed to fetch activity" });
+  }
+});
+
+// GET /dashboard/usage - Credit usage over time
+router.get("/usage", authMiddleware, async (req: AuthRequest, res) => {
+  try {
+    const user = req.user!;
+    const period = (req.query.period as string) ?? "30d";
+
+    // Calculate date range
+    const days = period === "7d" ? 7 : period === "90d" ? 90 : 30;
+    const startDate = new Date();
+    startDate.setDate(startDate.getDate() - days);
+
+    // Get usage transactions
+    const { data: transactions } = await supabase
+      .from("credit_transactions")
+      .select("amount, created_at")
+      .eq("user_id", user.id)
+      .eq("type", "usage")
+      .gte("created_at", startDate.toISOString())
+      .order("created_at", { ascending: true });
+
+    // Group by date
+    const usageByDate = new Map<string, number>();
+    (transactions ?? []).forEach((tx) => {
+      const date = tx.created_at.split("T")[0];
+      usageByDate.set(date, (usageByDate.get(date) ?? 0) + Math.abs(tx.amount));
+    });
+
+    const data = Array.from(usageByDate.entries()).map(([date, credits]) => ({
+      date,
+      credits,
+    }));
+
+    const totalCreditsUsed = data.reduce((sum, d) => sum + d.credits, 0);
+
+    res.json({
+      period,
+      data,
+      totalCreditsUsed,
+    });
+  } catch (error) {
+    console.error("Usage error:", error);
+    res.status(500).json({ error: "Failed to fetch usage" });
+  }
+});
+
+// GET /dashboard/documents - Document generation history
+router.get("/documents", authMiddleware, async (req: AuthRequest, res) => {
+  try {
+    const user = req.user!;
+    const page = Math.max(parseInt(String(req.query.page ?? "1"), 10), 1);
+    const limit = Math.min(parseInt(String(req.query.limit ?? "20"), 10), 100);
+    const offset = (page - 1) * limit;
+
+    // Get total count
+    const { count: total } = await supabase
+      .from("document_generations")
+      .select("*", { count: "exact", head: true })
+      .eq("user_id", user.id);
+
+    // Get paginated documents
+    const { data: documents } = await supabase
+      .from("document_generations")
+      .select("*")
+      .eq("user_id", user.id)
+      .order("created_at", { ascending: false })
+      .range(offset, offset + limit - 1);
+
+    res.json({
+      documents: (documents ?? []).map((doc) => ({
+        id: doc.id,
+        project: doc.project_name,
+        files: doc.files_count,
+        credits: doc.credits_used,
+        createdAt: doc.created_at,
+      })),
+      pagination: {
+        page,
+        limit,
+        total: total ?? 0,
+        hasMore: offset + limit < (total ?? 0),
+      },
+    });
+  } catch (error) {
+    console.error("Documents error:", error);
+    res.status(500).json({ error: "Failed to fetch documents" });
+  }
+});
+
+// GET /dashboard/billing - Current billing info
+router.get("/billing", authMiddleware, async (req: AuthRequest, res) => {
+  try {
+    const user = req.user!;
+
+    // Get most recent successful payment
+    const { data: lastPayment } = await supabase
+      .from("payment_transactions")
+      .select("*")
+      .eq("user_id", user.id)
+      .eq("status", "successful")
+      .order("created_at", { ascending: false })
+      .limit(1)
+      .single();
+
+    let subscriptionStatus: "active" | "cancelled" | "past_due" | "none" =
+      "none";
+    let currentPlan: string | null = null;
+    let planName: string | null = null;
+    let planAmount: number | null = null;
+    let planCurrency: string | null = null;
+    let nextBillingDate: string | null = null;
+    let maxCredits = 10; // Default for free tier
+
+    if (lastPayment) {
+      const plan = getPlanById(lastPayment.plan_id);
+      if (plan) {
+        currentPlan = plan.id;
+        planName = plan.name;
+        planAmount = plan.amount;
+        planCurrency = plan.currency;
+        maxCredits = plan.credits;
+
+        // Calculate next billing date (30 days from last payment)
+        const lastPaymentDate = new Date(lastPayment.created_at);
+        const nextBilling = new Date(lastPaymentDate);
+        nextBilling.setDate(nextBilling.getDate() + 30);
+        nextBillingDate = nextBilling.toISOString();
+
+        // Check if subscription is still active (within 30 days)
+        subscriptionStatus = nextBilling > new Date() ? "active" : "past_due";
+      }
+    }
+
+    res.json({
+      currentPlan,
+      planName,
+      planAmount,
+      planCurrency,
+      nextBillingDate,
+      creditBalance: user.credits,
+      maxCredits,
+      subscriptionStatus,
+    });
+  } catch (error) {
+    console.error("Billing error:", error);
+    res.status(500).json({ error: "Failed to fetch billing info" });
+  }
+});
+
+export default router;

package/server/src/routes/generate.d.ts

@@ -0,0 +1,3 @@
+declare const router: import("express-serve-static-core").Router;
+export default router;
+//# sourceMappingURL=generate.d.ts.map

package/server/src/routes/generate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["generate.ts"],"names":[],"mappings":"AAKA,QAAA,MAAM,MAAM,4CAAW,CAAC;AA2DxB,eAAe,MAAM,CAAC"}

package/server/src/routes/generate.js

@@ -0,0 +1,55 @@
+import { Router } from "express";
+import { supabase } from "../config/database.js";
+import { authMiddleware } from "../middleware/auth.js";
+import { generateDocumentation } from "../services/ai.js";
+const router = Router();
+// Generate documentation
+router.post("/", authMiddleware, async (req, res) => {
+    try {
+        const user = req.user;
+        const { codeSkeleton, views, fileCount } = req.body;
+        if (!codeSkeleton) {
+            res.status(400).json({ error: "Code skeleton is required" });
+            return;
+        }
+        // Calculate credits needed
+        const creditsPerFile = parseInt(process.env.CREDITS_PER_FILE ?? "1", 10);
+        const filesProcessed = fileCount ?? 1;
+        const creditsNeeded = filesProcessed * creditsPerFile;
+        // Check if user has enough credits
+        if (user.credits < creditsNeeded) {
+            res.status(402).json({
+                error: "Insufficient credits",
+                required: creditsNeeded,
+                available: user.credits,
+            });
+            return;
+        }
+        // Generate documentation
+        const results = await generateDocumentation(codeSkeleton, views);
+        // Deduct credits
+        const newBalance = user.credits - creditsNeeded;
+        await supabase
+            .from("users")
+            .update({ credits: newBalance })
+            .eq("id", user.id);
+        // Log transaction
+        await supabase.from("credit_transactions").insert({
+            user_id: user.id,
+            amount: -creditsNeeded,
+            type: "usage",
+            description: `Generated documentation for ${filesProcessed} file(s)`,
+        });
+        res.json({
+            results,
+            creditsUsed: creditsNeeded,
+            creditsRemaining: newBalance,
+        });
+    }
+    catch (error) {
+        console.error("Generation error:", error);
+        res.status(500).json({ error: "Failed to generate documentation" });
+    }
+});
+export default router;
+//# sourceMappingURL=generate.js.map

package/server/src/routes/generate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.js","sourceRoot":"","sources":["generate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAoB,MAAM,uBAAuB,CAAC;AACzE,OAAO,EAAE,qBAAqB,EAAwB,MAAM,mBAAmB,CAAC;AAEhF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;AAExB,yBAAyB;AACzB,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,EAAE,KAAK,EAAE,GAAgB,EAAE,GAAG,EAAE,EAAE;IAC/D,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,GAAG,CAAC,IAAK,CAAC;QACvB,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,IAE9C,CAAC;QAEF,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,2BAA2B,EAAE,CAAC,CAAC;YAC7D,OAAO;QACT,CAAC;QAED,2BAA2B;QAC3B,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;QACzE,MAAM,cAAc,GAAG,SAAS,IAAI,CAAC,CAAC;QACtC,MAAM,aAAa,GAAG,cAAc,GAAG,cAAc,CAAC;QAEtD,mCAAmC;QACnC,IAAI,IAAI,CAAC,OAAO,GAAG,aAAa,EAAE,CAAC;YACjC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACnB,KAAK,EAAE,sBAAsB;gBAC7B,QAAQ,EAAE,aAAa;gBACvB,SAAS,EAAE,IAAI,CAAC,OAAO;aACxB,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,yBAAyB;QACzB,MAAM,OAAO,GAAG,MAAM,qBAAqB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;QAEjE,iBAAiB;QACjB,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,GAAG,aAAa,CAAC;QAChD,MAAM,QAAQ;aACX,IAAI,CAAC,OAAO,CAAC;aACb,MAAM,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;aAC/B,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QAErB,kBAAkB;QAClB,MAAM,QAAQ,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,MAAM,CAAC;YAChD,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,MAAM,EAAE,CAAC,aAAa;YACtB,IAAI,EAAE,OAAO;YACb,WAAW,EAAE,+BAA+B,cAAc,UAAU;SACrE,CAAC,CAAC;QAEH,GAAG,CAAC,IAAI,CAAC;YACP,OAAO;YACP,WAAW,EAAE,aAAa;YAC1B,gBAAgB,EAAE,UAAU;SAC7B,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;QAC1C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,kCAAkC,EAAE,CAAC,CAAC;IACtE,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,eAAe,MAAM,CAAC"}

package/server/src/routes/generate.ts

@@ -0,0 +1,75 @@
+import { Router } from "express";
+import { supabase } from "../config/database.js";
+import { authMiddleware, type AuthRequest } from "../middleware/auth.js";
+import { generateDocumentation, type GenerateRequest } from "../services/ai.js";
+
+const router = Router();
+
+// Generate documentation
+router.post("/", authMiddleware, async (req: AuthRequest, res) => {
+  try {
+    const user = req.user!;
+    const { codeSkeleton, views, fileCount, projectName } =
+      req.body as GenerateRequest & {
+        fileCount?: number;
+        projectName?: string;
+      };
+
+    if (!codeSkeleton) {
+      res.status(400).json({ error: "Code skeleton is required" });
+      return;
+    }
+
+    // Calculate credits needed
+    const creditsPerFile = parseInt(process.env.CREDITS_PER_FILE ?? "1", 10);
+    const filesProcessed = fileCount ?? 1;
+    const creditsNeeded = filesProcessed * creditsPerFile;
+
+    // Check if user has enough credits
+    if (user.credits < creditsNeeded) {
+      res.status(402).json({
+        error: "Insufficient credits",
+        required: creditsNeeded,
+        available: user.credits,
+      });
+      return;
+    }
+
+    // Generate documentation
+    const results = await generateDocumentation(codeSkeleton, views);
+
+    // Deduct credits
+    const newBalance = user.credits - creditsNeeded;
+    await supabase
+      .from("users")
+      .update({ credits: newBalance })
+      .eq("id", user.id);
+
+    // Log credit transaction
+    await supabase.from("credit_transactions").insert({
+      user_id: user.id,
+      amount: -creditsNeeded,
+      type: "usage",
+      description: `Generated documentation for ${filesProcessed} file(s)`,
+    });
+
+    // Log document generation
+    await supabase.from("document_generations").insert({
+      user_id: user.id,
+      project_name: projectName ?? "Untitled Project",
+      files_count: filesProcessed,
+      credits_used: creditsNeeded,
+    });
+
+    res.json({
+      results,
+      creditsUsed: creditsNeeded,
+      creditsRemaining: newBalance,
+    });
+  } catch (error) {
+    console.error("Generation error:", error);
+    res.status(500).json({ error: "Failed to generate documentation" });
+  }
+});
+
+export default router;

package/server/src/routes/payment.ts

@@ -0,0 +1,192 @@
+import { Router } from "express";
+import { supabase } from "../config/database.js";
+import { authMiddleware, type AuthRequest } from "../middleware/auth.js";
+import {
+  PLANS,
+  getPlanById,
+  initiatePayment,
+  verifyTransaction,
+} from "../services/payment.js";
+
+const router = Router();
+
+// Get available plans
+router.get("/plans", (_req, res) => {
+  res.json({
+    plans: PLANS.map((p) => ({
+      id: p.id,
+      name: p.name,
+      amount: p.amount,
+      currency: p.currency,
+      credits: p.credits,
+      description: p.description,
+    })),
+  });
+});
+
+// Initiate payment
+router.post("/initiate", authMiddleware, async (req: AuthRequest, res) => {
+  try {
+    const user = req.user!;
+    const { planId, redirectUrl } = req.body as {
+      planId?: string;
+      redirectUrl?: string;
+    };
+
+    if (!planId) {
+      res.status(400).json({ error: "Plan ID is required" });
+      return;
+    }
+
+    const plan = getPlanById(planId);
+    if (!plan) {
+      res.status(400).json({ error: "Invalid plan ID" });
+      return;
+    }
+
+    const result = await initiatePayment({
+      email: user.email,
+      userId: user.id,
+      planId,
+      redirectUrl:
+        redirectUrl ??
+        `${process.env.FRONTEND_URL ?? "http://localhost:3000"}/payment/callback`,
+    });
+
+    // Store pending transaction
+    await supabase.from("payment_transactions").insert({
+      user_id: user.id,
+      tx_ref: result.tx_ref,
+      plan_id: planId,
+      amount: plan.amount,
+      currency: plan.currency,
+      credits: plan.credits,
+      status: "pending",
+    });
+
+    res.json({
+      paymentLink: result.link,
+      txRef: result.tx_ref,
+    });
+  } catch (error) {
+    console.error("Payment initiation error:", error);
+    res.status(500).json({ error: "Failed to initiate payment" });
+  }
+});
+
+// Webhook for Flutterwave
+router.post("/webhook", async (req, res) => {
+  try {
+    // Verify webhook signature
+    const secretHash = process.env.FLUTTERWAVE_SECRET_HASH;
+    const signature = req.headers["verif-hash"];
+
+    if (secretHash && signature !== secretHash) {
+      res.status(401).json({ error: "Invalid signature" });
+      return;
+    }
+
+    const { event, data } = req.body as {
+      event?: string;
+      data?: { id?: number; tx_ref?: string; status?: string };
+    };
+
+    if (event === "charge.completed" && data?.status === "successful") {
+      const transactionId = data.id;
+      if (!transactionId) {
+        res.status(400).json({ error: "Missing transaction ID" });
+        return;
+      }
+
+      // Verify the transaction
+      const verification = await verifyTransaction(String(transactionId));
+
+      if (verification.status === "successful" && verification.userId) {
+        // Update transaction status
+        await supabase
+          .from("payment_transactions")
+          .update({ status: "successful", flutterwave_id: transactionId })
+          .eq("tx_ref", data.tx_ref);
+
+        // Add credits to user
+        const { data: user } = await supabase
+          .from("users")
+          .select("credits")
+          .eq("id", verification.userId)
+          .single();
+
+        if (user) {
+          const newCredits = (user.credits ?? 0) + (verification.credits ?? 0);
+
+          await supabase
+            .from("users")
+            .update({ credits: newCredits })
+            .eq("id", verification.userId);
+
+          // Log credit transaction
+          await supabase.from("credit_transactions").insert({
+            user_id: verification.userId,
+            amount: verification.credits,
+            type: "purchase",
+            description: `Subscription payment - ${verification.credits} credits`,
+            reference_id: data.tx_ref,
+          });
+        }
+      }
+    }
+
+    res.json({ status: "ok" });
+  } catch (error) {
+    console.error("Webhook error:", error);
+    res.status(500).json({ error: "Webhook processing failed" });
+  }
+});
+
+// Verify payment (for redirect callback)
+router.get("/verify/:txRef", authMiddleware, async (req: AuthRequest, res) => {
+  try {
+    const { txRef } = req.params;
+    const user = req.user!;
+
+    // Get transaction from database
+    const { data: transaction } = await supabase
+      .from("payment_transactions")
+      .select("*")
+      .eq("tx_ref", txRef)
+      .eq("user_id", user.id)
+      .single();
+
+    if (!transaction) {
+      res.status(404).json({ error: "Transaction not found" });
+      return;
+    }
+
+    res.json({
+      status: transaction.status,
+      credits: transaction.credits,
+      amount: transaction.amount,
+      currency: transaction.currency,
+    });
+  } catch (error) {
+    console.error("Verify error:", error);
+    res.status(500).json({ error: "Verification failed" });
+  }
+});
+
+// Get payment history
+router.get("/history", authMiddleware, async (req: AuthRequest, res) => {
+  const user = req.user!;
+
+  const { data: transactions } = await supabase
+    .from("payment_transactions")
+    .select("*")
+    .eq("user_id", user.id)
+    .order("created_at", { ascending: false })
+    .limit(20);
+
+  res.json({
+    transactions: transactions ?? [],
+  });
+});
+
+export default router;

package/server/src/services/ai.d.ts

@@ -0,0 +1,10 @@
+export interface GenerateRequest {
+    codeSkeleton: string;
+    views?: string[];
+}
+export interface GenerateResult {
+    view: string;
+    content: string;
+}
+export declare function generateDocumentation(codeSkeleton: string, views?: string[]): Promise<GenerateResult[]>;
+//# sourceMappingURL=ai.d.ts.map

package/server/src/services/ai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.d.ts","sourceRoot":"","sources":["ai.ts"],"names":[],"mappings":"AA2BA,MAAM,WAAW,eAAe;IAC9B,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;CACjB;AAMD,wBAAsB,qBAAqB,CACzC,YAAY,EAAE,MAAM,EACpB,KAAK,GAAE,MAAM,EAA2C,GACvD,OAAO,CAAC,cAAc,EAAE,CAAC,CAsD3B"}

package/server/src/services/ai.js

@@ -0,0 +1,75 @@
+import { GoogleGenAI } from "@google/genai";
+const STRATEGIC_VIEW_PROMPT = `You are a technical writer creating documentation for Project Managers.
+Focus on business value and project status. NO code snippets. Use bullet points and bold text.
+Include "What this feature enables" and a "Completeness Score" (0-100%).`;
+const TECHNICAL_VIEW_PROMPT = `You are a technical writer creating documentation for Developers.
+Focus on architecture and integration. Include function signatures, dependency maps, and code examples.
+Use strict Markdown with syntax highlighting.`;
+const PERSONAL_VIEW_PROMPT = `You are a technical writer creating documentation for Solo Developers.
+Focus on productivity. Generate To-Do lists from TODO comments and Technical Debt summaries.
+Use concise, checklist-style formatting.`;
+function getPromptForView(view) {
+    switch (view) {
+        case "strategic":
+            return STRATEGIC_VIEW_PROMPT;
+        case "technical":
+            return TECHNICAL_VIEW_PROMPT;
+        case "personal":
+            return PERSONAL_VIEW_PROMPT;
+        default:
+            return TECHNICAL_VIEW_PROMPT;
+    }
+}
+async function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+export async function generateDocumentation(codeSkeleton, views = ["strategic", "technical", "personal"]) {
+    const apiKey = process.env.GEMINI_API_KEY;
+    if (!apiKey) {
+        throw new Error("Gemini API key not configured");
+    }
+    const ai = new GoogleGenAI({ apiKey });
+    const model = "gemini-2.0-flash";
+    const results = [];
+    for (const view of views) {
+        const systemPrompt = getPromptForView(view);
+        let lastError;
+        for (let attempt = 0; attempt < 5; attempt++) {
+            try {
+                const response = await ai.models.generateContent({
+                    model,
+                    contents: [
+                        {
+                            role: "user",
+                            parts: [
+                                {
+                                    text: `${systemPrompt}\n\n---\n\nAnalyze the following code skeleton and generate documentation:\n\n${codeSkeleton}`,
+                                },
+                            ],
+                        },
+                    ],
+                });
+                results.push({
+                    view,
+                    content: response.text ?? "",
+                });
+                break;
+            }
+            catch (error) {
+                lastError = error;
+                const status = error.status;
+                if (status === 503 || status === 429) {
+                    const waitTime = Math.pow(2, attempt) * 1000;
+                    await sleep(waitTime);
+                    continue;
+                }
+                throw error;
+            }
+        }
+        if (results.length <= views.indexOf(view)) {
+            throw lastError ?? new Error("Failed to generate documentation");
+        }
+    }
+    return results;
+}
+//# sourceMappingURL=ai.js.map

package/server/src/services/ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.js","sourceRoot":"","sources":["ai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,MAAM,qBAAqB,GAAG;;yEAE2C,CAAC;AAE1E,MAAM,qBAAqB,GAAG;;8CAEgB,CAAC;AAE/C,MAAM,oBAAoB,GAAG;;yCAEY,CAAC;AAE1C,SAAS,gBAAgB,CAAC,IAAY;IACpC,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,WAAW;YACd,OAAO,qBAAqB,CAAC;QAC/B,KAAK,WAAW;YACd,OAAO,qBAAqB,CAAC;QAC/B,KAAK,UAAU;YACb,OAAO,oBAAoB,CAAC;QAC9B;YACE,OAAO,qBAAqB,CAAC;IACjC,CAAC;AACH,CAAC;AAYD,KAAK,UAAU,KAAK,CAAC,EAAU;IAC7B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,YAAoB,EACpB,QAAkB,CAAC,WAAW,EAAE,WAAW,EAAE,UAAU,CAAC;IAExD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IAED,MAAM,EAAE,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IACvC,MAAM,KAAK,GAAG,kBAAkB,CAAC;IACjC,MAAM,OAAO,GAAqB,EAAE,CAAC;IAErC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,YAAY,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAE5C,IAAI,SAA4B,CAAC;QACjC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC;oBAC/C,KAAK;oBACL,QAAQ,EAAE;wBACR;4BACE,IAAI,EAAE,MAAM;4BACZ,KAAK,EAAE;gCACL;oCACE,IAAI,EAAE,GAAG,YAAY,iFAAiF,YAAY,EAAE;iCACrH;6BACF;yBACF;qBACF;iBACF,CAAC,CAAC;gBAEH,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI;oBACJ,OAAO,EAAE,QAAQ,CAAC,IAAI,IAAI,EAAE;iBAC7B,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,SAAS,GAAG,KAAc,CAAC;gBAC3B,MAAM,MAAM,GAAI,KAA6B,CAAC,MAAM,CAAC;gBAErD,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;oBACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;oBAC7C,MAAM,KAAK,CAAC,QAAQ,CAAC,CAAC;oBACtB,SAAS;gBACX,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1C,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC"}