@docyrus/ui-pro-ai-assistant 0.0.1 → 0.0.2

package/src/vite.ts

@@ -2,23 +2,7 @@ import { type IncomingMessage } from 'node:http';
 
 import { type Plugin, loadEnv } from 'vite';
 
-/*
- * ---------------------------------------------------------------------------
- * Shared helpers
- * ---------------------------------------------------------------------------
- */
-
-const COPILOT_SYSTEM_PROMPT = `You are an advanced AI writing assistant, similar to VSCode Copilot but for general text. Your task is to predict and generate the next part of the text based on the given context.
-
-Rules:
-- Continue the text naturally up to the next punctuation mark (., ,, ;, :, ?, or !).
-- Maintain style and tone. Don't repeat given text.
-- For unclear context, provide the most likely continuation.
-- Handle code snippets, lists, or structured text if needed.
-- Don't include """ in your response.
-- CRITICAL: Always end with a punctuation mark.
-- CRITICAL: Avoid starting a new block. Do not use block formatting like >, #, 1., 2., -, etc. The suggestion should continue in the same block as the context.
-- If no context is provided or you can't generate a continuation, return "0" without explanation.`;
+import { handleAiCommand, handleAiCopilot } from './worker';
 
 function readBody(req: IncomingMessage): Promise<Buffer> {
   return new Promise((resolve, reject) => {
@@ -30,210 +14,6 @@ function readBody(req: IncomingMessage): Promise<Buffer> {
   });
 }
 
-function getApiBaseUrl(env: Record<string, string>): string | undefined {
-  return env.VITE_DOCYRUS_API_URL;
-}
-
-function getAgentId(env: Record<string, string>): string | undefined {
-  return env.VITE_DOCYRUS_EDITOR_AGENT_ID || env.VITE_DOCYRUS_AGENT_ID;
-}
-
-function getAuthToken(request: Request): string | undefined {
-  const authHeader = request.headers.get('authorization');
-
-  return authHeader?.replace('Bearer ', '') || undefined;
-}
-
-/*
- * ---------------------------------------------------------------------------
- * Route handlers (also exported for use in Cloudflare Workers)
- * ---------------------------------------------------------------------------
- */
-
-export async function handleAiCommand(
-  request: Request,
-  env: Record<string, string>
-): Promise<Response> {
-  if (request.method !== 'POST') {
-    return new Response(JSON.stringify({ error: 'Method not allowed' }), {
-      status: 405,
-      headers: { 'Content-Type': 'application/json' }
-    });
-  }
-
-  try {
-    const body = await request.json() as {
-      messages?: unknown[];
-      ctx?: { selection?: unknown; children?: unknown; toolName?: string };
-    };
-
-    const token = getAuthToken(request);
-
-    if (!token) {
-      return new Response(
-        JSON.stringify({ error: 'Unauthorized', message: 'No authentication token found. Please log in.' }),
-        { status: 401, headers: { 'Content-Type': 'application/json' } }
-      );
-    }
-
-    const apiBaseUrl = getApiBaseUrl(env);
-
-    if (!apiBaseUrl) {
-      return new Response(
-        JSON.stringify({ error: 'VITE_DOCYRUS_API_URL is not configured' }),
-        { status: 500, headers: { 'Content-Type': 'application/json' } }
-      );
-    }
-
-    const agentId = getAgentId(env);
-
-    if (!agentId) {
-      return new Response(
-        JSON.stringify({ error: 'VITE_DOCYRUS_AGENT_ID is not configured' }),
-        { status: 500, headers: { 'Content-Type': 'application/json' } }
-      );
-    }
-
-    const { messages, ctx } = body;
-
-    let editorContext: string | undefined;
-
-    if (ctx) {
-      const contextData: Record<string, unknown> = { toolName: ctx.toolName };
-
-      if (ctx.selection) contextData.selection = ctx.selection;
-      if (ctx.children) contextData.children = ctx.children;
-      editorContext = JSON.stringify(contextData);
-    }
-
-    const response = await fetch(`${apiBaseUrl}/v1/ai/agents/${agentId}/chat`, {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        Authorization: `Bearer ${token}`
-      },
-      body: JSON.stringify({
-        agentId,
-        messages: messages ?? [],
-        mode: 'agent',
-        stream: true,
-        additionalContext: editorContext
-      })
-    });
-
-    if (!response.ok) {
-      const errorText = await response.text();
-
-      return new Response(
-        errorText || JSON.stringify({ error: 'Backend API error' }),
-        { status: response.status, headers: { 'Content-Type': 'application/json' } }
-      );
-    }
-
-    return new Response(response.body, {
-      headers: {
-        'Content-Type': 'text/event-stream',
-        'Cache-Control': 'no-cache',
-        Connection: 'keep-alive'
-      }
-    });
-  } catch (error) {
-    console.error('[AI Command] Error:', error);
-
-    return new Response(
-      JSON.stringify({ error: 'Internal server error', message: error instanceof Error ? error.message : 'Unknown error' }),
-      { status: 500, headers: { 'Content-Type': 'application/json' } }
-    );
-  }
-}
-
-export async function handleAiCopilot(
-  request: Request,
-  env: Record<string, string>
-): Promise<Response> {
-  if (request.method !== 'POST') {
-    return new Response(JSON.stringify({ error: 'Method not allowed' }), {
-      status: 405,
-      headers: { 'Content-Type': 'application/json' }
-    });
-  }
-
-  try {
-    const body = await request.json() as { prompt?: string };
-
-    const token = getAuthToken(request);
-
-    if (!token) {
-      return new Response(
-        JSON.stringify({ error: 'Unauthorized' }),
-        { status: 401, headers: { 'Content-Type': 'application/json' } }
-      );
-    }
-
-    const apiBaseUrl = getApiBaseUrl(env);
-
-    if (!apiBaseUrl) {
-      return new Response(
-        JSON.stringify({ error: 'VITE_DOCYRUS_API_URL is not configured' }),
-        { status: 500, headers: { 'Content-Type': 'application/json' } }
-      );
-    }
-
-    const agentId = getAgentId(env);
-
-    if (!agentId) {
-      return new Response(
-        JSON.stringify({ error: 'VITE_DOCYRUS_AGENT_ID is not configured' }),
-        { status: 500, headers: { 'Content-Type': 'application/json' } }
-      );
-    }
-
-    const response = await fetch(`${apiBaseUrl}/v1/ai/agents/${agentId}/chat`, {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        Authorization: `Bearer ${token}`
-      },
-      body: JSON.stringify({
-        agentId,
-        messages: [{ role: 'system', content: COPILOT_SYSTEM_PROMPT }, { role: 'user', content: body.prompt }],
-        mode: 'agent',
-        stream: true
-      })
-    });
-
-    if (!response.ok) {
-      const errorText = await response.text();
-
-      return new Response(
-        errorText || JSON.stringify({ error: 'Backend API error' }),
-        { status: response.status, headers: { 'Content-Type': 'application/json' } }
-      );
-    }
-
-    return new Response(response.body, {
-      headers: {
-        'Content-Type': 'text/event-stream',
-        'Cache-Control': 'no-cache',
-        Connection: 'keep-alive'
-      }
-    });
-  } catch (error) {
-    console.error('[AI Copilot] Error:', error);
-
-    return new Response(
-      JSON.stringify({ error: 'Internal server error', message: error instanceof Error ? error.message : 'Unknown error' }),
-      { status: 500, headers: { 'Content-Type': 'application/json' } }
-    );
-  }
-}
-
-/*
- * ---------------------------------------------------------------------------
- * Vite plugin (dev server middleware)
- * ---------------------------------------------------------------------------
- */
-
 export function plateEditorPlugin(): Plugin {
   let env: Record<string, string>;
 

package/src/worker.ts

@@ -0,0 +1,203 @@
+const COPILOT_SYSTEM_PROMPT = `You are an advanced AI writing assistant, similar to VSCode Copilot but for general text. Your task is to predict and generate the next part of the text based on the given context.
+
+Rules:
+- Continue the text naturally up to the next punctuation mark (., ,, ;, :, ?, or !).
+- Maintain style and tone. Don't repeat given text.
+- For unclear context, provide the most likely continuation.
+- Handle code snippets, lists, or structured text if needed.
+- Don't include """ in your response.
+- CRITICAL: Always end with a punctuation mark.
+- CRITICAL: Avoid starting a new block. Do not use block formatting like >, #, 1., 2., -, etc. The suggestion should continue in the same block as the context.
+- If no context is provided or you can't generate a continuation, return "0" without explanation.`;
+
+function getApiBaseUrl(env: Record<string, string>): string | undefined {
+  return env.VITE_DOCYRUS_API_URL;
+}
+
+function getAgentId(env: Record<string, string>): string | undefined {
+  return env.VITE_DOCYRUS_EDITOR_AGENT_ID || env.VITE_DOCYRUS_AGENT_ID;
+}
+
+function getAuthToken(request: Request): string | undefined {
+  const authHeader = request.headers.get('authorization');
+
+  return authHeader?.replace('Bearer ', '') || undefined;
+}
+
+export async function handleAiCommand(
+  request: Request,
+  env: Record<string, string>
+): Promise<Response> {
+  if (request.method !== 'POST') {
+    return new Response(JSON.stringify({ error: 'Method not allowed' }), {
+      status: 405,
+      headers: { 'Content-Type': 'application/json' }
+    });
+  }
+
+  try {
+    const body = await request.json() as {
+      messages?: unknown[];
+      ctx?: { selection?: unknown; children?: unknown; toolName?: string };
+    };
+
+    const token = getAuthToken(request);
+
+    if (!token) {
+      return new Response(
+        JSON.stringify({ error: 'Unauthorized', message: 'No authentication token found. Please log in.' }),
+        { status: 401, headers: { 'Content-Type': 'application/json' } }
+      );
+    }
+
+    const apiBaseUrl = getApiBaseUrl(env);
+
+    if (!apiBaseUrl) {
+      return new Response(
+        JSON.stringify({ error: 'VITE_DOCYRUS_API_URL is not configured' }),
+        { status: 500, headers: { 'Content-Type': 'application/json' } }
+      );
+    }
+
+    const agentId = getAgentId(env);
+
+    if (!agentId) {
+      return new Response(
+        JSON.stringify({ error: 'VITE_DOCYRUS_AGENT_ID is not configured' }),
+        { status: 500, headers: { 'Content-Type': 'application/json' } }
+      );
+    }
+
+    const { messages, ctx } = body;
+
+    let editorContext: string | undefined;
+
+    if (ctx) {
+      const contextData: Record<string, unknown> = { toolName: ctx.toolName };
+
+      if (ctx.selection) contextData.selection = ctx.selection;
+      if (ctx.children) contextData.children = ctx.children;
+      editorContext = JSON.stringify(contextData);
+    }
+
+    const response = await fetch(`${apiBaseUrl}/v1/ai/agents/${agentId}/chat`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        Authorization: `Bearer ${token}`
+      },
+      body: JSON.stringify({
+        agentId,
+        messages: messages ?? [],
+        mode: 'agent',
+        stream: true,
+        additionalContext: editorContext
+      })
+    });
+
+    if (!response.ok) {
+      const errorText = await response.text();
+
+      return new Response(
+        errorText || JSON.stringify({ error: 'Backend API error' }),
+        { status: response.status, headers: { 'Content-Type': 'application/json' } }
+      );
+    }
+
+    return new Response(response.body, {
+      headers: {
+        'Content-Type': 'text/event-stream',
+        'Cache-Control': 'no-cache',
+        Connection: 'keep-alive'
+      }
+    });
+  } catch (error) {
+    console.error('[AI Command] Error:', error);
+
+    return new Response(
+      JSON.stringify({ error: 'Internal server error', message: error instanceof Error ? error.message : 'Unknown error' }),
+      { status: 500, headers: { 'Content-Type': 'application/json' } }
+    );
+  }
+}
+
+export async function handleAiCopilot(
+  request: Request,
+  env: Record<string, string>
+): Promise<Response> {
+  if (request.method !== 'POST') {
+    return new Response(JSON.stringify({ error: 'Method not allowed' }), {
+      status: 405,
+      headers: { 'Content-Type': 'application/json' }
+    });
+  }
+
+  try {
+    const body = await request.json() as { prompt?: string };
+
+    const token = getAuthToken(request);
+
+    if (!token) {
+      return new Response(
+        JSON.stringify({ error: 'Unauthorized' }),
+        { status: 401, headers: { 'Content-Type': 'application/json' } }
+      );
+    }
+
+    const apiBaseUrl = getApiBaseUrl(env);
+
+    if (!apiBaseUrl) {
+      return new Response(
+        JSON.stringify({ error: 'VITE_DOCYRUS_API_URL is not configured' }),
+        { status: 500, headers: { 'Content-Type': 'application/json' } }
+      );
+    }
+
+    const agentId = getAgentId(env);
+
+    if (!agentId) {
+      return new Response(
+        JSON.stringify({ error: 'VITE_DOCYRUS_AGENT_ID is not configured' }),
+        { status: 500, headers: { 'Content-Type': 'application/json' } }
+      );
+    }
+
+    const response = await fetch(`${apiBaseUrl}/v1/ai/agents/${agentId}/chat`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        Authorization: `Bearer ${token}`
+      },
+      body: JSON.stringify({
+        agentId,
+        messages: [{ role: 'system', content: COPILOT_SYSTEM_PROMPT }, { role: 'user', content: body.prompt }],
+        mode: 'agent',
+        stream: true
+      })
+    });
+
+    if (!response.ok) {
+      const errorText = await response.text();
+
+      return new Response(
+        errorText || JSON.stringify({ error: 'Backend API error' }),
+        { status: response.status, headers: { 'Content-Type': 'application/json' } }
+      );
+    }
+
+    return new Response(response.body, {
+      headers: {
+        'Content-Type': 'text/event-stream',
+        'Cache-Control': 'no-cache',
+        Connection: 'keep-alive'
+      }
+    });
+  } catch (error) {
+    console.error('[AI Copilot] Error:', error);
+
+    return new Response(
+      JSON.stringify({ error: 'Internal server error', message: error instanceof Error ? error.message : 'Unknown error' }),
+      { status: 500, headers: { 'Content-Type': 'application/json' } }
+    );
+  }
+}