opencode-pollinations-plugin 6.0.0 → 6.1.0-beta.10

package/dist/server/index.js

@@ -4,7 +4,9 @@ import * as path from 'path';
 import { getAggregatedModels } from './pollinations-api.js';
 import { loadConfig, saveConfig } from './config.js';
 import { handleChatCompletion } from './proxy.js';
-const LOG_FILE = path.join(process.env.HOME || '/tmp', '.config/opencode/plugins/pollinations-v3.log');
+import { createCommandHooks, setClientForCommands, checkKeyPermissions } from './commands.js';
+import { fileURLToPath } from 'url';
+const LOG_FILE = path.join(process.env.HOME || '/tmp', '.config/opencode/plugins/pollinations-v6.log');
 // Simple file logger
 function log(msg) {
     const ts = new Date().toISOString();
@@ -14,170 +16,143 @@ function log(msg) {
         }
         fs.appendFileSync(LOG_FILE, `[${ts}] ${msg}\n`);
     }
-    catch (e) {
-        // silent fail
-    }
-}
-// CRASH GUARD
-const CRASH_LOG = '/tmp/opencode_pollinations_crash.log';
-process.on('uncaughtException', (err) => {
-    try {
-        const msg = `[CRASH] Uncaught Exception: ${err.message}\n${err.stack}\n`;
-        fs.appendFileSync(CRASH_LOG, msg);
-        console.error(msg);
-    }
     catch (e) { }
-    process.exit(1);
-});
-process.on('unhandledRejection', (reason, promise) => {
-    try {
-        const msg = `[CRASH] Unhandled Rejection: ${reason}\n`;
-        fs.appendFileSync(CRASH_LOG, msg);
-    }
-    catch (e) { }
-});
-const server = http.createServer(async (req, res) => {
-    log(`${req.method} ${req.url}`);
-    // CORS Headers
-    res.setHeader('Access-Control-Allow-Origin', '*');
-    res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
-    res.setHeader('Access-Control-Allow-Headers', '*');
-    if (req.method === 'OPTIONS') {
-        res.writeHead(204);
-        res.end();
+}
+const PORT = parseInt(process.env.POLLINATIONS_PORT || '10001', 10);
+let serverInstance = null;
+// --- SERVER LOGIC ---
+function startServer() {
+    if (serverInstance)
         return;
-    }
-    // AUTH ENDPOINT (Kept for compatibility, though Native Auth is preferred)
-    if (req.method === 'POST' && req.url === '/v1/auth') {
-        const chunks = [];
-        req.on('data', chunk => chunks.push(chunk));
-        req.on('end', async () => {
-            try {
-                const body = JSON.parse(Buffer.concat(chunks).toString());
-                if (body && body.apiKey) {
-                    saveConfig({ apiKey: body.apiKey, mode: 'pro' });
-                    log(`[AUTH] Key saved via Server Endpoint`);
-                    res.writeHead(200, { 'Content-Type': 'application/json' });
-                    res.end(JSON.stringify({ status: "ok" }));
+    serverInstance = http.createServer(async (req, res) => {
+        log(`${req.method} ${req.url}`);
+        // CORS Headers
+        res.setHeader('Access-Control-Allow-Origin', '*');
+        res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
+        res.setHeader('Access-Control-Allow-Headers', '*');
+        if (req.method === 'OPTIONS') {
+            res.writeHead(204);
+            res.end();
+            return;
+        }
+        // AUTH ENDPOINT
+        if (req.method === 'POST' && req.url === '/v1/auth') {
+            const chunks = [];
+            req.on('data', chunk => chunks.push(chunk));
+            req.on('end', async () => {
+                try {
+                    const body = JSON.parse(Buffer.concat(chunks).toString());
+                    if (body && body.apiKey) {
+                        saveConfig({ apiKey: body.apiKey, mode: 'pro' });
+                        log(`[AUTH] Key saved via Server Endpoint`);
+                        res.writeHead(200, { 'Content-Type': 'application/json' });
+                        res.end(JSON.stringify({ status: "ok" }));
+                    }
+                    else {
+                        res.writeHead(400);
+                        res.end(JSON.stringify({ error: "Missing apiKey" }));
+                    }
                 }
-                else {
-                    res.writeHead(400);
-                    res.end(JSON.stringify({ error: "Missing apiKey" }));
+                catch (e) {
+                    log(`[AUTH] Error: ${e}`);
+                    res.writeHead(500);
+                    res.end(JSON.stringify({ error: String(e) }));
                 }
-            }
-            catch (e) {
-                log(`[AUTH] Error: ${e}`);
-                res.writeHead(500);
-                res.end(JSON.stringify({ error: String(e) }));
-            }
-        });
-        return;
-    }
-    if (req.method === 'GET' && req.url === '/health') {
-        const config = loadConfig();
-        res.writeHead(200, { 'Content-Type': 'application/json' });
-        res.end(JSON.stringify({
-            status: "ok",
-            version: "v3.0.0-phase3",
-            mode: config.mode,
-            hasKey: !!config.apiKey
-        }));
-        return;
-    }
-    if (req.method === 'GET' && req.url === '/v1/models') {
-        try {
-            const models = await getAggregatedModels();
-            res.writeHead(200, { 'Content-Type': 'application/json' });
-            res.end(JSON.stringify(models));
+            });
+            return;
         }
-        catch (e) {
-            log(`Error fetching models: ${e}`);
-            res.writeHead(500, { 'Content-Type': 'application/json' });
-            res.end(JSON.stringify({ error: "Failed to fetch models" }));
+        if (req.method === 'GET' && req.url === '/health') {
+            const config = loadConfig();
+            res.writeHead(200, { 'Content-Type': 'application/json' });
+            res.end(JSON.stringify({
+                status: "ok",
+                version: "v6.0.0-beta.99",
+                mode: config.mode,
+                hasKey: !!config.apiKey
+            }));
+            return;
         }
-        return;
-    }
-    if (req.method === 'POST' && req.url === '/v1/chat/completions') {
-        // Accumulate body for the proxy
-        const chunks = [];
-        req.on('data', chunk => chunks.push(chunk));
-        req.on('end', async () => {
+        if (req.method === 'GET' && req.url === '/v1/models') {
             try {
-                const bodyRaw = Buffer.concat(chunks).toString();
-                await handleChatCompletion(req, res, bodyRaw);
+                const models = await getAggregatedModels();
+                res.writeHead(200, { 'Content-Type': 'application/json' });
+                res.end(JSON.stringify(models));
             }
             catch (e) {
-                log(`Error in chat handler: ${e}`);
+                log(`Error fetching models: ${e}`);
                 res.writeHead(500, { 'Content-Type': 'application/json' });
-                res.end(JSON.stringify({ error: "Internal Server Error in Chat Handler" }));
+                res.end(JSON.stringify({ error: "Failed to fetch models" }));
             }
-        });
-        return;
-    }
-    res.writeHead(404);
-    res.end("Not Found");
-});
-const PORT = parseInt(process.env.POLLINATIONS_PORT || '10001', 10);
-// ANTI-ZOMBIE (Visible Logs Restored)
-try {
-    const { execSync } = require('child_process');
-    try {
-        console.log(`[POLLINATIONS] Checking port ${PORT}...`);
-        execSync(`fuser -k ${PORT}/tcp || true`);
-        console.log(`[POLLINATIONS] Port ${PORT} cleared.`);
-    }
-    catch (e) {
-        console.log(`[POLLINATIONS] Port check skipped (cmd missing?)`);
-    }
-}
-catch (e) { }
-// LIFECYCLE DEBUG (Sync Write)
-const LIFE_LOG = '/tmp/POLLI_LIFECYCLE.log';
-const LOC_LOG = '/tmp/POLLI_LOCATION.log'; // NEW: Track source location
-try {
-    fs.appendFileSync(LIFE_LOG, `[${new Date().toISOString()}] [STARTUP] PID:${process.pid} Initializing...\n`);
-    fs.writeFileSync(LOC_LOG, `[${new Date().toISOString()}] RUNNING FROM: ${__filename}\n`);
-}
-catch (e) { }
-process.on('exit', (code) => {
+            return;
+        }
+        if (req.method === 'POST' && req.url === '/v1/chat/completions') {
+            const chunks = [];
+            req.on('data', chunk => chunks.push(chunk));
+            req.on('end', async () => {
+                try {
+                    const bodyRaw = Buffer.concat(chunks).toString();
+                    await handleChatCompletion(req, res, bodyRaw);
+                }
+                catch (e) {
+                    log(`Error in chat handler: ${e}`);
+                    res.writeHead(500, { 'Content-Type': 'application/json' });
+                    res.end(JSON.stringify({ error: "Internal Server Error in Chat Handler" }));
+                }
+            });
+            return;
+        }
+        res.writeHead(404);
+        res.end("Not Found");
+    });
+    // ANTI-ZOMBIE
     try {
-        fs.appendFileSync(LIFE_LOG, `[${new Date().toISOString()}] [EXIT] PID:${process.pid} Exiting with code ${code}\n`);
+        const { execSync } = require('child_process');
+        try {
+            console.log(`[POLLINATIONS] Checking port ${PORT}...`);
+            execSync(`fuser -k ${PORT}/tcp || true`);
+            console.log(`[POLLINATIONS] Port ${PORT} cleared.`);
+        }
+        catch (e) { }
     }
     catch (e) { }
-});
-// STARTUP CHECK: Re-validate Key (in case of upgrade/config drift)
-import { checkKeyPermissions } from './commands.js';
-(async () => {
-    const config = loadConfig();
-    if (config.apiKey) {
-        try {
-            console.log('Pollinations Plugin: Verifying API Key on startup...');
-            const check = await checkKeyPermissions(config.apiKey);
-            if (!check.ok) {
-                console.warn(`Pollinations Plugin: Limited Key Detected on Startup (${check.reason}). Enforcing Manual Mode.`);
-                saveConfig({
-                    apiKey: config.apiKey,
-                    mode: 'manual',
-                    keyHasAccessToProfile: false
-                });
-            }
-            else {
-                if (config.keyHasAccessToProfile === false) {
-                    saveConfig({ apiKey: config.apiKey, keyHasAccessToProfile: true });
+    // STARTUP CHECK
+    (async () => {
+        const config = loadConfig();
+        if (config.apiKey) {
+            try {
+                console.log('Pollinations Plugin: Verifying API Key on startup...');
+                const check = await checkKeyPermissions(config.apiKey);
+                if (!check.ok) {
+                    console.warn(`Pollinations Plugin: Limited Key Detected on Startup (${check.reason}). Enforcing Manual Mode.`);
+                    saveConfig({ apiKey: config.apiKey, mode: 'manual', keyHasAccessToProfile: false });
+                }
+                else {
+                    if (config.keyHasAccessToProfile === false)
+                        saveConfig({ apiKey: config.apiKey, keyHasAccessToProfile: true });
                 }
             }
+            catch (e) {
+                console.error('Pollinations Plugin: Startup Check Failed:', e);
+            }
         }
-        catch (e) {
-            console.error('Pollinations Plugin: Startup Check Failed:', e);
-        }
-    }
-    server.listen(PORT, '127.0.0.1', () => {
+    })();
+    serverInstance.listen(PORT, '127.0.0.1', () => {
         const url = `http://127.0.0.1:${PORT}`;
-        log(`[SERVER] Started V3 Phase 3 (Auth Enabled) on port ${PORT}`);
-        try {
-            fs.appendFileSync(LIFE_LOG, `[${new Date().toISOString()}] [LISTEN] PID:${process.pid} Listening on ${PORT}\n`);
-        }
-        catch (e) { }
-        console.log(`POLLINATIONS_V3_URL=${url}`);
+        log(`[SERVER] Started V6 (Plugin Mode) on port ${PORT}`);
+        console.log(`POLLINATIONS_V6_URL=${url}`);
     });
-})();
+}
+// --- OPENCODE PLUGIN EXPORT ---
+export const plugin = async ({ client }) => {
+    // 1. Inject Client for Command Handling
+    setClientForCommands(client);
+    // 2. Start Local Proxy Server
+    startServer();
+    // 3. Register Hooks (Commands, TUI, etc.)
+    return createCommandHooks();
+};
+// --- STANDALONE SUPPORT ---
+// If run directly via `node dist/index.js`, start server immediately
+if (process.argv[1] === fileURLToPath(import.meta.url)) {
+    startServer();
+}

package/dist/server/pollinations-api.d.ts

@@ -41,6 +41,17 @@ export interface DetailedUsageResponse {
     count: number;
 }
 export declare function getDetailedUsage(apiKey: string): Promise<DetailedUsageResponse | null>;
+export interface DailyUsageEntry {
+    date: string;
+    model: string;
+    meter_source: 'tier' | 'pack' | 'combined';
+    requests: number;
+    cost_usd: number;
+}
+export interface DailyUsageResponse {
+    usage: DailyUsageEntry[];
+}
+export declare function getDailyUsage(apiKey: string): Promise<DailyUsageResponse | null>;
 export declare function getAggregatedModels(): Promise<{
     object: string;
     data: OpenAIModel[];

package/dist/server/pollinations-api.js

@@ -136,6 +136,26 @@ export async function getDetailedUsage(apiKey) {
         return null;
     }
 }
+export async function getDailyUsage(apiKey) {
+    if (!apiKey || apiKey.length < 10)
+        return null;
+    try {
+        logDebug("Fetching Daily Usage (aggregated)...");
+        const response = await fetch('https://gen.pollinations.ai/account/usage/daily', {
+            headers: { 'Authorization': `Bearer ${apiKey}` }
+        });
+        if (!response.ok) {
+            logDebug(`Daily Usage API Error: ${response.status}`);
+            return null;
+        }
+        const data = await response.json();
+        return data;
+    }
+    catch (e) {
+        logDebug(`Error Daily Usage: ${e}`);
+        return null;
+    }
+}
 export async function getAggregatedModels() {
     const config = loadConfig();
     const [free, enter] = await Promise.all([