openwriter 0.4.0 → 0.5.1

package/dist/server/mcp.js

@@ -10,10 +10,10 @@ import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { z } from 'zod';
 import { DATA_DIR, ensureDataDir } from './helpers.js';
-import { getDocument, getWordCount, getPendingChangeCount, getTitle, getStatus, getNodesByIds, getMetadata, setMetadata, applyChanges, applyTextEdits, updateDocument, save, markAllNodesAsPending, setAgentLock, updatePendingCacheForActiveDoc, getDocId, getFilePath, } from './state.js';
-import { listDocuments, switchDocument, createDocument, deleteDocument, openFile, getActiveFilename } from './documents.js';
+import { getDocument, getWordCount, getPendingChangeCount, getTitle, getStatus, getNodesByIds, getMetadata, setMetadata, applyChanges, applyTextEdits, updateDocument, save, markAllNodesAsPending, setAgentLock, updatePendingCacheForActiveDoc, populateDocumentFile, applyChangesToFile, applyTextEditsToFile, getDocId, getFilePath, } from './state.js';
+import { listDocuments, switchDocument, createDocument, deleteDocument, openFile, getActiveFilename, updateDocumentTitle } from './documents.js';
 import { broadcastDocumentSwitched, broadcastDocumentsChanged, broadcastWorkspacesChanged, broadcastTitleChanged, broadcastMetadataChanged, broadcastPendingDocsChanged, broadcastWritingStarted, broadcastWritingFinished } from './ws.js';
-import { listWorkspaces, getWorkspace, getDocTitle, getItemContext, addDoc, updateWorkspaceContext, createWorkspace, deleteWorkspace, addContainerToWorkspace, findOrCreateWorkspace, findOrCreateContainer, moveDoc } from './workspaces.js';
+import { listWorkspaces, getWorkspace, getDocTitle, getItemContext, addDoc, updateWorkspaceContext, createWorkspace, deleteWorkspace, addContainerToWorkspace, findOrCreateWorkspace, findOrCreateContainer, moveDoc, renameWorkspace, renameContainer } from './workspaces.js';
 import { addDocTag, removeDocTag, getDocTagsByFilename } from './state.js';
 import { importGoogleDoc } from './gdoc-import.js';
 import { toCompactFormat, compactNodes, parseMarkdownContent } from './compact.js';
@@ -33,7 +33,7 @@ export const TOOL_REGISTRY = [
     },
     {
         name: 'write_to_pad',
-        description: 'Preferred tool for all document edits. Send 3-8 changes per call for responsive feel. Multiple rapid calls better than one monolithic call. Content can be a markdown string (preferred) or TipTap JSON. Markdown strings are auto-converted. Changes appear as pending decorations the user accepts or rejects. Use afterNodeId: "end" to append to the document without knowing node IDs. Response includes lastNodeId for chaining subsequent inserts.',
+        description: 'Preferred tool for all document edits. Send 3-8 changes per call for responsive feel. Multiple rapid calls better than one monolithic call. Content can be a markdown string (preferred) or TipTap JSON. Markdown strings are auto-converted. Changes appear as pending decorations the user accepts or rejects. Use afterNodeId: "end" to append to the document without knowing node IDs. Response includes lastNodeId for chaining subsequent inserts. Always specify filename — edits target that file directly without switching the user\'s view.',
         schema: {
             changes: z.array(z.object({
                 operation: z.enum(['rewrite', 'insert', 'delete']),
@@ -41,8 +41,9 @@ export const TOOL_REGISTRY = [
                 afterNodeId: z.string().optional(),
                 content: z.any().optional(),
             })).describe('Array of node changes. Content accepts markdown strings or TipTap JSON.'),
+            filename: z.string().describe('Target filename (e.g. "My Essay.md"). Required — identifies which document to edit.'),
         },
-        handler: async ({ changes }) => {
+        handler: async ({ changes, filename }) => {
             const processed = changes.map((change) => {
                 const resolved = { ...change };
                 if (typeof resolved.content === 'string') {
@@ -50,6 +51,22 @@ export const TOOL_REGISTRY = [
                 }
                 return resolved;
             });
+            const targetIsNonActive = filename && filename !== getActiveFilename();
+            if (targetIsNonActive) {
+                const { count: appliedCount, lastNodeId } = applyChangesToFile(filename, processed);
+                broadcastPendingDocsChanged();
+                return {
+                    content: [{
+                            type: 'text',
+                            text: JSON.stringify({
+                                success: appliedCount > 0,
+                                appliedCount,
+                                ...(lastNodeId ? { lastNodeId } : {}),
+                                ...(appliedCount < processed.length ? { skipped: processed.length - appliedCount } : {}),
+                            }),
+                        }],
+                };
+            }
             const { count: appliedCount, lastNodeId } = applyChanges(processed);
             // broadcastPendingDocsChanged() already fires via onChanges listener in ws.ts
             return {
@@ -187,11 +204,12 @@ export const TOOL_REGISTRY = [
     },
     {
         name: 'populate_document',
-        description: 'Populate the active document with content. Use after create_document (without content) to complete the two-step creation flow. Content appears as pending decorations for user review. Clears the sidebar creation spinner and shows the document.',
+        description: 'Populate a document with content. Use after create_document (without content) to complete the two-step creation flow. Content appears as pending decorations for user review. Clears the sidebar creation spinner and shows the document. Pass the filename from create_document\'s response to ensure content goes to the right doc even if the user switched away.',
         schema: {
             content: z.any().describe('Document content: markdown string (preferred) or TipTap JSON doc object.'),
+            filename: z.string().optional().describe('Target filename (e.g. "My Essay.md"). If provided and differs from the active doc, writes directly to disk without switching the user\'s view. Recommended — prevents race conditions when the user navigates during content generation.'),
         },
-        handler: async ({ content }) => {
+        handler: async ({ content, filename }) => {
             try {
                 let doc;
                 if (typeof content === 'string') {
@@ -206,6 +224,22 @@ export const TOOL_REGISTRY = [
                         content: [{ type: 'text', text: 'Error: content must be a markdown string or TipTap JSON { type: "doc", content: [...] }' }],
                     };
                 }
+                // Non-active target: write directly to disk without disrupting the user's view
+                const targetIsNonActive = filename && filename !== getActiveFilename();
+                if (targetIsNonActive) {
+                    const result = populateDocumentFile(filename, doc);
+                    broadcastDocumentsChanged();
+                    broadcastWorkspacesChanged();
+                    broadcastPendingDocsChanged();
+                    broadcastWritingFinished();
+                    return {
+                        content: [{
+                                type: 'text',
+                                text: `Populated "${result.title}" — ${result.wordCount.toLocaleString()} words`,
+                            }],
+                    };
+                }
+                // Active target (or no filename): existing flow
                 setAgentLock(); // Block browser doc-updates during population
                 markAllNodesAsPending(doc, 'insert');
                 updateDocument(doc);
@@ -484,9 +518,44 @@ export const TOOL_REGISTRY = [
             return { content: [{ type: 'text', text: `Moved "${docFile}"${targetContainerId ? ` to container ${targetContainerId}` : ' to root'}` }] };
         },
     },
+    {
+        name: 'rename_item',
+        description: 'Rename a workspace, container, or document. For workspaces: updates the manifest title. For containers: updates the container name in the workspace tree. For documents: updates the title in frontmatter.',
+        schema: {
+            type: z.enum(['workspace', 'container', 'document']).describe('What to rename'),
+            filename: z.string().describe('Workspace manifest filename (for workspace/container) or document filename (for document)'),
+            newName: z.string().describe('The new name/title'),
+            containerId: z.string().optional().describe('Container ID (required for container renames)'),
+            workspaceFile: z.string().optional().describe('Parent workspace filename (required for container renames)'),
+        },
+        handler: async ({ type, filename, newName, containerId, workspaceFile }) => {
+            if (type === 'workspace') {
+                renameWorkspace(filename, newName);
+                broadcastWorkspacesChanged();
+                return { content: [{ type: 'text', text: `Renamed workspace to "${newName}"` }] };
+            }
+            if (type === 'container') {
+                const wsFile = workspaceFile || filename;
+                if (!containerId)
+                    return { content: [{ type: 'text', text: 'Error: containerId is required for container renames' }] };
+                renameContainer(wsFile, containerId, newName);
+                broadcastWorkspacesChanged();
+                return { content: [{ type: 'text', text: `Renamed container ${containerId} to "${newName}"` }] };
+            }
+            if (type === 'document') {
+                updateDocumentTitle(filename, newName);
+                broadcastDocumentsChanged();
+                if (filename === getActiveFilename()) {
+                    broadcastTitleChanged(newName);
+                }
+                return { content: [{ type: 'text', text: `Renamed document "${filename}" to "${newName}"` }] };
+            }
+            return { content: [{ type: 'text', text: `Error: unknown type "${type}"` }] };
+        },
+    },
     {
         name: 'edit_text',
-        description: 'Apply fine-grained text edits within a node. Find text by exact match and replace it, or add/remove marks on matched text. More precise than rewriting the whole node.',
+        description: 'Apply fine-grained text edits within a node. Find text by exact match and replace it, or add/remove marks on matched text. More precise than rewriting the whole node. Always specify filename — edits target that file directly without switching the user\'s view.',
         schema: {
             nodeId: z.string().describe('ID of the node to edit'),
             edits: z.array(z.object({
@@ -498,8 +567,16 @@ export const TOOL_REGISTRY = [
                 }).optional().describe('Mark to add to the matched text (e.g. link, bold)'),
                 removeMark: z.string().optional().describe('Mark type to remove from matched text'),
             })).describe('Array of text edits to apply'),
-        },
-        handler: async ({ nodeId, edits }) => {
+            filename: z.string().describe('Target filename (e.g. "My Essay.md"). Required — identifies which document to edit.'),
+        },
+        handler: async ({ nodeId, edits, filename }) => {
+            const targetIsNonActive = filename && filename !== getActiveFilename();
+            if (targetIsNonActive) {
+                const result = applyTextEditsToFile(filename, nodeId, edits);
+                if (result.success)
+                    broadcastPendingDocsChanged();
+                return { content: [{ type: 'text', text: JSON.stringify(result) }] };
+            }
             return { content: [{ type: 'text', text: JSON.stringify(applyTextEdits(nodeId, edits)) }] };
         },
     },
@@ -524,7 +601,7 @@ export const TOOL_REGISTRY = [
     },
     {
         name: 'generate_image',
-        description: 'Generate an image using Gemini Imagen 4. Saves to ~/.openwriter/_images/. Optionally sets it as the active article\'s cover image atomically. Requires GEMINI_API_KEY env var.',
+        description: 'Generate an image using Gemini Nano Banana 2. Saves to ~/.openwriter/_images/. Optionally sets it as the active article\'s cover image atomically. Requires GEMINI_API_KEY env var.',
         schema: {
             prompt: z.string().max(1000).describe('Image generation prompt (max 1000 chars)'),
             aspect_ratio: z.string().optional().describe('Aspect ratio (default "16:9"). Supported: 1:1, 9:16, 16:9, 4:3, 3:4.'),
@@ -537,16 +614,16 @@ export const TOOL_REGISTRY = [
             }
             const { GoogleGenAI } = await import('@google/genai');
             const ai = new GoogleGenAI({ apiKey });
-            const response = await ai.models.generateImages({
-                model: 'imagen-4.0-generate-001',
-                prompt,
+            const response = await ai.models.generateContent({
+                model: 'gemini-3.1-flash-image-preview',
+                contents: `Generate a ${aspect_ratio || '16:9'} aspect ratio image: ${prompt}`,
                 config: {
-                    numberOfImages: 1,
-                    aspectRatio: (aspect_ratio || '16:9'),
+                    responseModalities: ['IMAGE'],
                 },
             });
-            const image = response.generatedImages?.[0];
-            if (!image?.image?.imageBytes) {
+            const parts = response.candidates?.[0]?.content?.parts;
+            const imagePart = parts?.find((p) => p.inlineData);
+            if (!imagePart?.inlineData?.data) {
                 return { content: [{ type: 'text', text: 'Error: Gemini returned no image data.' }] };
             }
             // Save to ~/.openwriter/_images/
@@ -556,7 +633,7 @@ export const TOOL_REGISTRY = [
                 mkdirSync(imagesDir, { recursive: true });
             const filename = `${randomUUID().slice(0, 8)}.png`;
             const filePath = join(imagesDir, filename);
-            writeFileSync(filePath, Buffer.from(image.image.imageBytes, 'base64'));
+            writeFileSync(filePath, Buffer.from(imagePart.inlineData.data, 'base64'));
             const src = `/_images/${filename}`;
             // Optionally set as article cover + append to carousel history
             if (set_cover) {

package/dist/server/plugin-manager.js

@@ -127,6 +127,7 @@ export class PluginManager {
                 continue;
             results.push({
                 name: managed.plugin.name,
+                displayName: managed.discovered.displayName,
                 contextMenuItems: managed.plugin.contextMenuItems?.() || [],
                 sidebarMenuItems: managed.plugin.sidebarMenuItems?.() || [],
             });

package/dist/server/prompt-debug.js

@@ -0,0 +1,58 @@
+/**
+ * File: prompt-debug.ts
+ * Purpose: Write AV prompt debug data to timestamped .md files for inspection.
+ * Each enhance creates a new file in DATA_DIR, visible in the sidebar.
+ */
+import { DATA_DIR, ensureDataDir, atomicWriteFileSync } from './helpers.js';
+import { join } from 'path';
+/**
+ * Write prompt debug info to a timestamped markdown file.
+ * Returns the filename created.
+ */
+export function writePromptDebug(action, debug, metadata) {
+    ensureDataDir();
+    const now = new Date();
+    const ts = now.toISOString().replace(/[:.]/g, '-').slice(0, 19);
+    const filename = `_prompt-${action || 'debug'}-${ts}.md`;
+    const filePath = join(DATA_DIR, filename);
+    const timeStr = now.toLocaleTimeString('en-US', { hour12: true, hour: '2-digit', minute: '2-digit', second: '2-digit' });
+    let md = `---\ntitle: "Prompt Debug: ${action} @ ${timeStr}"\n---\n\n`;
+    // Metadata summary
+    if (metadata) {
+        md += `## Metadata\n\n`;
+        md += `| Key | Value |\n|-----|-------|\n`;
+        if (metadata.action)
+            md += `| Action | ${metadata.action} |\n`;
+        if (metadata.profileUsed)
+            md += `| Profile | ${metadata.profileUsed} |\n`;
+        if (metadata.nodesIn != null)
+            md += `| Nodes In | ${metadata.nodesIn} |\n`;
+        if (metadata.nodesOut != null)
+            md += `| Nodes Out | ${metadata.nodesOut} |\n`;
+        if (metadata.ragExamples != null)
+            md += `| RAG Examples | ${metadata.ragExamples} |\n`;
+        if (metadata.ragTotalWords != null)
+            md += `| RAG Total Words | ${metadata.ragTotalWords} |\n`;
+        if (metadata.processingTimeMs != null)
+            md += `| Processing Time | ${metadata.processingTimeMs}ms |\n`;
+        if (metadata.estimatedCost != null)
+            md += `| Estimated Cost | $${metadata.estimatedCost.toFixed(4)} |\n`;
+        md += `\n`;
+    }
+    // System prompt
+    if (debug.systemPrompt) {
+        md += `## System Prompt\n\n`;
+        md += debug.systemPrompt + '\n\n';
+    }
+    // User prompt
+    if (debug.userPrompt) {
+        md += `---\n\n## User Prompt\n\n`;
+        md += debug.userPrompt + '\n\n';
+    }
+    // Raw LLM response (when available)
+    if (debug.rawResponse) {
+        md += `---\n\n## Raw LLM Output\n\n\`\`\`json\n${debug.rawResponse}\n\`\`\`\n\n`;
+    }
+    atomicWriteFileSync(filePath, md);
+    return filename;
+}