quackstack 1.0.23 → 1.0.25

package/dist/commands/search.js

@@ -2,21 +2,48 @@
 import { client } from "../lib/database.js";
 import { localEmbeddings } from "../lib/local-embeddings.js";
 import { getAIClient } from "../lib/ai-provider.js";
-export async function search(query, projectName, provider, model) {
+export async function search(query, projectName, provider, model, options = {}) {
+    const { boostRecent = false, boostFrequent = false, filterAuthor, recentDays = 30, } = options;
+    const whereClause = { projectName };
+    if (filterAuthor) {
+        whereClause.OR = [
+            { lastCommitEmail: filterAuthor },
+            { primaryAuthorEmail: filterAuthor },
+        ];
+    }
     const snippets = await client.codeSnippet.findMany({
-        where: { projectName },
+        where: whereClause,
     });
     const allContent = snippets.map(s => s.content);
     localEmbeddings.addDocuments(allContent);
     const queryVector = localEmbeddings.getVector(query);
     const ranked = snippets
-        .map(snippet => ({
-        id: snippet.id,
-        content: snippet.content,
-        filePath: snippet.filePath,
-        functionName: snippet.functionName,
-        score: localEmbeddings.cosineSimilarity(queryVector, snippet.embedding),
-    }))
+        .map(snippet => {
+        let score = localEmbeddings.cosineSimilarity(queryVector, snippet.embedding);
+        if (boostRecent && snippet.lastCommitDate) {
+            const daysSinceCommit = (Date.now() - snippet.lastCommitDate.getTime()) / (1000 * 60 * 60 * 24);
+            if (daysSinceCommit <= recentDays) {
+                const recencyBoost = 1 + (0.2 * (1 - daysSinceCommit / recentDays));
+                score *= recencyBoost;
+            }
+        }
+        if (boostFrequent && snippet.totalCommits) {
+            const commitBoost = 1 + (Math.min(snippet.totalCommits, 50) / 50) * 0.15;
+            score *= commitBoost;
+        }
+        return {
+            id: snippet.id,
+            content: snippet.content,
+            filePath: snippet.filePath,
+            functionName: snippet.functionName,
+            score,
+            lastCommitAuthor: snippet.lastCommitAuthor,
+            lastCommitDate: snippet.lastCommitDate,
+            lastCommitMessage: snippet.lastCommitMessage,
+            totalCommits: snippet.totalCommits,
+            primaryAuthor: snippet.primaryAuthor,
+        };
+    })
         .sort((a, b) => b.score - a.score);
     const seenFiles = new Set();
     const uniqueResults = ranked.filter(item => {
@@ -26,9 +53,86 @@ export async function search(query, projectName, provider, model) {
         return true;
     }).slice(0, 5);
     const context = uniqueResults
-        .map((r, i) => `[${i + 1}] ${r.filePath}${r.functionName ? ` (${r.functionName})` : ""}\n${r.content}`)
+        .map((r, i) => {
+        let entry = `[${i + 1}] ${r.filePath}${r.functionName ? ` (${r.functionName})` : ""}`;
+        if (r.lastCommitAuthor || r.primaryAuthor) {
+            const author = r.primaryAuthor || r.lastCommitAuthor;
+            entry += `\nPrimary Author: ${author}`;
+            if (r.totalCommits) {
+                entry += ` (${r.totalCommits} commits)`;
+            }
+        }
+        if (r.lastCommitDate) {
+            const daysAgo = Math.floor((Date.now() - r.lastCommitDate.getTime()) / (1000 * 60 * 60 * 24));
+            entry += `\nLast Modified: ${daysAgo} days ago`;
+            if (r.lastCommitMessage) {
+                entry += `\nLast Commit: "${r.lastCommitMessage.substring(0, 60)}${r.lastCommitMessage.length > 60 ? '...' : ''}"`;
+            }
+        }
+        entry += `\n\n${r.content}`;
+        return entry;
+    })
         .join("\n\n---\n\n");
     const aiClient = getAIClient(provider, model);
-    const answer = await aiClient.generateAnswer(query, context);
+    const enhancedPrompt = query +
+        "\n\nNote: Code snippets include git history metadata (authors, commit counts, last modified dates). " +
+        "Use this information to provide context about code ownership and recency when relevant.";
+    const answer = await aiClient.generateAnswer(enhancedPrompt, context);
     return { answer, sources: uniqueResults };
 }
+export async function searchByAuthor(authorEmail, projectName) {
+    const snippets = await client.codeSnippet.findMany({
+        where: {
+            projectName,
+            OR: [
+                { lastCommitEmail: authorEmail },
+                { primaryAuthorEmail: authorEmail },
+            ],
+        },
+        select: {
+            filePath: true,
+            functionName: true,
+            totalCommits: true,
+            lastCommitDate: true,
+        },
+        orderBy: {
+            totalCommits: "desc",
+        },
+    });
+    const fileMap = new Map();
+    snippets.forEach(s => {
+        if (!fileMap.has(s.filePath)) {
+            fileMap.set(s.filePath, s);
+        }
+    });
+    return Array.from(fileMap.values());
+}
+export async function getRecentActivity(projectName, days = 7) {
+    const since = new Date();
+    since.setDate(since.getDate() - days);
+    const snippets = await client.codeSnippet.findMany({
+        where: {
+            projectName,
+            lastCommitDate: {
+                gte: since,
+            },
+        },
+        select: {
+            filePath: true,
+            lastCommitAuthor: true,
+            lastCommitDate: true,
+            lastCommitMessage: true,
+        },
+        orderBy: {
+            lastCommitDate: "desc",
+        },
+        take: 20,
+    });
+    const fileMap = new Map();
+    snippets.forEach(s => {
+        if (!fileMap.has(s.filePath)) {
+            fileMap.set(s.filePath, s);
+        }
+    });
+    return Array.from(fileMap.values());
+}

package/dist/lib/context-generator.js

@@ -1,9 +1,11 @@
 import fs from "fs";
 import path from "path";
 import os from "os";
-import { search } from "../commands/search.js";
+import { search, getRecentActivity } from "../commands/search.js";
+import { getProjectAuthors } from "../lib/database.js";
+import { gitHistory } from "../lib/git-history.js";
 export async function generateCodebaseDoc(projectName) {
-    console.log("🔍 Analyzing your codebase...\n");
+    console.log("Analyzing your codebase...\n");
     const queries = [
         "What is the overall architecture and design patterns used?",
         "What are the main entry points and how does the application start?",
@@ -15,8 +17,24 @@ export async function generateCodebaseDoc(projectName) {
     let doc = `# ${projectName} - Codebase Documentation\n\n`;
     doc += `**Auto-generated by QuackStack** | Last updated: ${new Date().toLocaleString()}\n\n`;
     doc += `This document provides a high-level overview of the codebase architecture, key components, and design decisions.\n\n`;
-    doc += `---\n\n`;
-    doc += `## 📋 Table of Contents\n\n`;
+    if (gitHistory.isRepository()) {
+        doc += `## Repository Information\n\n`;
+        const currentBranch = gitHistory.getCurrentBranch();
+        if (currentBranch) {
+            doc += `**Current Branch:** ${currentBranch}\n\n`;
+        }
+        const recentCommits = gitHistory.getRecentCommits(10);
+        if (recentCommits.length > 0) {
+            doc += `**Recent Activity:**\n`;
+            recentCommits.slice(0, 5).forEach(commit => {
+                const date = new Date(commit.date).toLocaleDateString();
+                doc += `- ${date} - ${commit.author}: ${commit.message.substring(0, 60)}${commit.message.length > 60 ? '...' : ''}\n`;
+            });
+            doc += `\n`;
+        }
+        doc += `---\n\n`;
+    }
+    doc += `## Table of Contents\n\n`;
     queries.forEach((q, i) => {
         const anchor = q.toLowerCase().replace(/[^a-z0-9]+/g, '-');
         doc += `${i + 1}. [${q}](#${anchor})\n`;
@@ -28,12 +46,15 @@ export async function generateCodebaseDoc(projectName) {
             doc += `## ${query}\n\n`;
             doc += `${answer}\n\n`;
             if (sources.length > 0) {
-                doc += `### 📁 Key Files\n\n`;
+                doc += `### Key Files\n\n`;
                 sources.slice(0, 5).forEach(s => {
                     doc += `- \`${s.filePath}\``;
                     if (s.functionName) {
                         doc += ` - ${s.functionName}`;
                     }
+                    if (s.primaryAuthor) {
+                        doc += ` (maintained by ${s.primaryAuthor})`;
+                    }
                     doc += `\n`;
                 });
                 doc += `\n`;
@@ -44,7 +65,28 @@ export async function generateCodebaseDoc(projectName) {
             console.error(`Error analyzing: ${query}`);
         }
     }
-    doc += `## 📂 Project Structure\n\n`;
+    if (gitHistory.isRepository()) {
+        doc += `## Contributors\n\n`;
+        const authors = await getProjectAuthors(projectName);
+        if (authors.length > 0) {
+            doc += `This project has ${authors.length} contributor${authors.length > 1 ? 's' : ''}:\n\n`;
+            authors.slice(0, 10).forEach((author, i) => {
+                doc += `${i + 1}. **${author.author}** (${author.email})\n`;
+                doc += `   - ${author.totalCommits} commits\n`;
+                doc += `   - +${author.linesAdded} / -${author.linesRemoved} lines\n`;
+                if (author.recentActivity) {
+                    const daysAgo = Math.floor((Date.now() - author.recentActivity.getTime()) / (1000 * 60 * 60 * 24));
+                    doc += `   - Last active ${daysAgo} days ago\n`;
+                }
+                if (author.filesOwned.length > 0) {
+                    doc += `   - Primary owner of ${author.filesOwned.length} files\n`;
+                }
+                doc += `\n`;
+            });
+            doc += `---\n\n`;
+        }
+    }
+    doc += `## Project Structure\n\n`;
     doc += '```\n';
     doc += await getProjectStructure(process.cwd());
     doc += '```\n\n';
@@ -59,7 +101,7 @@ export async function generateCodebaseDoc(projectName) {
     doc += `### Running the Project\n`;
     doc += `Refer to \`package.json\` scripts section for available commands.\n\n`;
     doc += `---\n\n`;
-    doc += `## 🔄 Updating This Document\n\n`;
+    doc += `## Updating This Document\n\n`;
     doc += `This documentation is auto-generated. To regenerate:\n\n`;
     doc += '```bash\n';
     doc += 'quack --docs\n';
@@ -81,6 +123,34 @@ export async function generateContextFiles(projectName) {
     let baseContext = `# ${projectName} - Codebase Context\n\n`;
     baseContext += `Generated: ${new Date().toISOString()}\n\n`;
     baseContext += "This file is auto-generated by QuackStack to provide AI assistants with codebase context.\n\n";
+    if (gitHistory.isRepository()) {
+        baseContext += `## Git Repository Info\n\n`;
+        const branch = gitHistory.getCurrentBranch();
+        if (branch) {
+            baseContext += `**Branch:** ${branch}\n\n`;
+        }
+        const recentActivity = await getRecentActivity(projectName, 7);
+        if (recentActivity.length > 0) {
+            baseContext += `**Recent Changes (Last 7 days):**\n\n`;
+            recentActivity.slice(0, 5).forEach(item => {
+                const daysAgo = Math.floor((Date.now() - item.lastCommitDate.getTime()) / (1000 * 60 * 60 * 24));
+                baseContext += `- ${item.filePath}\n`;
+                baseContext += `  - Modified by ${item.lastCommitAuthor} ${daysAgo} days ago\n`;
+                if (item.lastCommitMessage) {
+                    baseContext += `  - "${item.lastCommitMessage.substring(0, 60)}${item.lastCommitMessage.length > 60 ? '...' : ''}"\n`;
+                }
+            });
+            baseContext += `\n`;
+        }
+        const authors = await getProjectAuthors(projectName);
+        if (authors.length > 0) {
+            baseContext += `**Top Contributors:**\n\n`;
+            authors.slice(0, 5).forEach(author => {
+                baseContext += `- ${author.author} (${author.totalCommits} commits, owns ${author.filesOwned.length} files)\n`;
+            });
+            baseContext += `\n`;
+        }
+    }
     baseContext += "---\n\n";
     for (const query of queries) {
         try {
@@ -89,7 +159,11 @@ export async function generateContextFiles(projectName) {
             if (sources.length > 0) {
                 baseContext += "**Key files:**\n";
                 sources.slice(0, 3).forEach(s => {
-                    baseContext += `- ${s.filePath}\n`;
+                    baseContext += `- ${s.filePath}`;
+                    if (s.primaryAuthor) {
+                        baseContext += ` (${s.primaryAuthor})`;
+                    }
+                    baseContext += `\n`;
                 });
                 baseContext += "\n";
             }
@@ -100,12 +174,28 @@ export async function generateContextFiles(projectName) {
     }
     baseContext += "---\n\n## Project Structure\n\n";
     baseContext += await getProjectStructure(process.cwd());
+    if (gitHistory.isRepository()) {
+        const authors = await getProjectAuthors(projectName);
+        if (authors.length > 0) {
+            baseContext += "\n---\n\n## Code Ownership & Expertise\n\n";
+            baseContext += "When making changes, consider consulting these experts:\n\n";
+            for (const author of authors.slice(0, 5)) {
+                if (author.filesOwned.length > 0) {
+                    baseContext += `**${author.author}** - Expert in:\n`;
+                    author.filesOwned.slice(0, 5).forEach(file => {
+                        baseContext += `- ${file}\n`;
+                    });
+                    baseContext += `\n`;
+                }
+            }
+        }
+    }
     await generateCursorRules(baseContext);
     await generateWindsurfContext(baseContext);
     await generateClineContext(baseContext);
     await generateContinueContext(baseContext);
     await generateAiderContext(baseContext);
-    console.log("\n✅ Context files generated for:");
+    console.log("\n Context files generated for:");
     console.log("  - Cursor (.cursorrules)");
     console.log("  - Windsurf (.windsurfrules)");
     console.log("  - Cline (.clinerules)");
@@ -138,11 +228,9 @@ async function generateAiderContext(context) {
     const aiderConfig = `# Aider configuration with QuackStack context
 # Project: ${path.basename(process.cwd())}
 
-# Context file
 read:
   - .aider.context.md
 
-# Model settings
 model: gpt-4o-mini
 edit-format: whole
 `;
@@ -166,6 +254,12 @@ export async function updateGlobalContext(projectName) {
         topFiles: sources.slice(0, 5).map(s => s.filePath),
         lastUpdated: new Date().toISOString(),
     };
+    if (gitHistory.isRepository()) {
+        const branch = gitHistory.getCurrentBranch();
+        const authors = await getProjectAuthors(projectName);
+        contexts[projectName].gitBranch = branch;
+        contexts[projectName].contributors = authors.length;
+    }
     fs.writeFileSync(contextPath, JSON.stringify(contexts, null, 2), "utf-8");
 }
 export function watchAndUpdateContext(projectName) {
@@ -180,7 +274,7 @@ export function watchAndUpdateContext(projectName) {
             await generateContextFiles(projectName);
         }, 5000);
     });
-    console.log("👀 Watching for file changes...");
+    console.log("Watching for file changes...");
 }
 async function getProjectStructure(dir, prefix = "", maxDepth = 3, currentDepth = 0) {
     if (currentDepth >= maxDepth)

package/dist/lib/database.js

@@ -1,41 +1,109 @@
 import { PrismaClient } from "@prisma/client";
 export const client = new PrismaClient();
-export const saveToDB = async (data) => {
-    try {
-        const result = await client.codeSnippet.create({
-            data: {
-                ...data,
-                embedding: data.embedding,
+export async function saveToDB(data) {
+    await client.codeSnippet.create({
+        data: {
+            content: data.content,
+            embedding: data.embedding,
+            filePath: data.filePath,
+            projectName: data.projectName,
+            language: data.language,
+            functionName: data.functionName,
+            lineStart: data.lineStart,
+            lineEnd: data.lineEnd,
+            lastCommitHash: data.lastCommitHash,
+            lastCommitAuthor: data.lastCommitAuthor,
+            lastCommitEmail: data.lastCommitEmail,
+            lastCommitDate: data.lastCommitDate,
+            lastCommitMessage: data.lastCommitMessage,
+            totalCommits: data.totalCommits,
+            primaryAuthor: data.primaryAuthor,
+            primaryAuthorEmail: data.primaryAuthorEmail,
+            fileOwnerCommits: data.fileOwnerCommits,
+        },
+    });
+}
+export async function saveAuthorToDB(data) {
+    await client.gitAuthor.upsert({
+        where: {
+            projectName_email: {
+                projectName: data.projectName,
+                email: data.email,
             },
-        });
-        return result;
-    }
-    catch (e) {
-        if (e instanceof Error) {
-            console.error(`Error saving to DB: ${e.message}`);
+        },
+        create: data,
+        update: {
+            author: data.author,
+            totalCommits: data.totalCommits,
+            linesAdded: data.linesAdded,
+            linesRemoved: data.linesRemoved,
+            recentActivity: data.recentActivity,
+            filesOwned: data.filesOwned,
+        },
+    });
+}
+export async function clearProject(projectName) {
+    await client.codeSnippet.deleteMany({
+        where: { projectName },
+    });
+    await client.gitAuthor.deleteMany({
+        where: { projectName },
+    });
+}
+export async function getProjectAuthors(projectName) {
+    return await client.gitAuthor.findMany({
+        where: { projectName },
+        orderBy: { totalCommits: "desc" },
+    });
+}
+export async function getAuthorFiles(projectName, authorEmail) {
+    const snippets = await client.codeSnippet.findMany({
+        where: {
+            projectName,
+            primaryAuthorEmail: authorEmail,
+        },
+        select: {
+            filePath: true,
+            functionName: true,
+            totalCommits: true,
+        },
+        orderBy: {
+            totalCommits: "desc",
+        },
+    });
+    const fileMap = new Map();
+    snippets.forEach(s => {
+        if (!fileMap.has(s.filePath)) {
+            fileMap.set(s.filePath, s);
         }
-        else {
-            console.error(`Unknown error saving to DB:`, e);
-        }
-        throw e;
-    }
-};
-export const getFromDB = async (projectName) => {
-    try {
-        const results = await client.codeSnippet.findMany({
-            where: {
-                projectName: projectName
-            }
-        });
-        return results;
-    }
-    catch (e) {
-        if (e instanceof Error) {
-            console.error(`Error fetching from DB: ${e.message}`);
-        }
-        else {
-            console.error(`Unknown error fetching from DB:`, e);
+    });
+    return Array.from(fileMap.values());
+}
+export async function getRecentlyModifiedFiles(projectName, days = 7) {
+    const since = new Date();
+    since.setDate(since.getDate() - days);
+    const snippets = await client.codeSnippet.findMany({
+        where: {
+            projectName,
+            lastCommitDate: {
+                gte: since,
+            },
+        },
+        select: {
+            filePath: true,
+            lastCommitAuthor: true,
+            lastCommitDate: true,
+            lastCommitMessage: true,
+        },
+        orderBy: {
+            lastCommitDate: "desc",
+        },
+    });
+    const fileMap = new Map();
+    snippets.forEach(s => {
+        if (!fileMap.has(s.filePath)) {
+            fileMap.set(s.filePath, s);
         }
-        throw e;
-    }
-};
+    });
+    return Array.from(fileMap.values());
+}