vmcode-cli 1.0.0

package/vmcode-proxy/package.json

@@ -0,0 +1,20 @@
+{
+  "name": "vmcode-proxy",
+  "version": "1.0.0",
+  "description": "Proxy server for vmCode free tier using OpenRouter",
+  "main": "server.js",
+  "scripts": {
+    "start": "node server.js",
+    "dev": "node server.js"
+  },
+  "dependencies": {
+    "dotenv": "^16.3.1",
+    "express": "^4.18.2",
+    "express-rate-limit": "^8.2.1",
+    "cors": "^2.8.5",
+    "compression": "^1.7.4"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}

package/vmcode-proxy/server.js

@@ -0,0 +1,248 @@
+const express = require('express');
+const rateLimit = require('express-rate-limit');
+const cors = require('cors');
+const compression = require('compression');
+require('dotenv').config();
+
+const app = express();
+
+// Middleware
+app.use(cors());
+app.use(compression());
+app.use(express.json({ limit: '1mb' }));
+
+// Rate limiting: 100 requests per 15 minutes per IP
+const limiter = rateLimit({
+  windowMs: 15 * 60 * 1000,
+  max: 100,
+  keyGenerator: (req) => req.ip,
+  message: { error: 'Rate limit exceeded. Try again later.' }
+});
+
+// Daily hard limit: 10,000 requests/day total
+let todayRequestCount = 0;
+let lastResetDate = new Date().toDateString();
+const DAILY_LIMIT = 10000;
+
+// Free tier: per-IP daily limit
+const FREE_TIER_DAILY_LIMIT = 500; // 500 requests per day per IP
+const ipDailyUsage = new Map(); // Map<ip, {count: number, date: string}>
+
+/**
+ * Cleanup function to prevent memory leaks
+ * Removes IP entries older than 2 days
+ */
+function cleanupOldIps() {
+  const today = new Date();
+  const twoDaysAgo = new Date(today);
+  twoDaysAgo.setDate(today.getDate() - 2);
+
+  let cleaned = 0;
+
+  for (const [ip, data] of ipDailyUsage.entries()) {
+    const entryDate = new Date(data.date);
+
+    if (entryDate < twoDaysAgo) {
+      ipDailyUsage.delete(ip);
+      cleaned++;
+    }
+  }
+
+  if (cleaned > 0) {
+    console.log(`[${new Date().toISOString()}] Cleaned up ${cleaned} stale IP entries. Active IPs: ${ipDailyUsage.size}`);
+  }
+}
+
+// Run cleanup every hour
+setInterval(cleanupOldIps, 60 * 60 * 1000);
+
+/**
+ * Get next midnight timestamp for reset time
+ */
+function getNextMidnight() {
+  const tomorrow = new Date();
+  tomorrow.setDate(tomorrow.getDate() + 1);
+  tomorrow.setHours(0, 0, 0, 0);
+  return tomorrow.getTime();
+}
+
+// Health check endpoint
+app.get('/', (req, res) => {
+  res.json({ status: 'ok', service: 'vmcode-proxy' });
+});
+
+app.get('/health', (req, res) => {
+  const clientIp = req.ip;
+  const today = new Date().toDateString();
+  const ipUsage = ipDailyUsage.get(clientIp);
+
+  res.json({
+    status: 'ok',
+    service: 'vmcode-proxy',
+    todayRequests: todayRequestCount,
+    dailyLimit: DAILY_LIMIT,
+    percentUsed: Math.round((todayRequestCount / DAILY_LIMIT) * 100),
+    // Per-IP usage
+    ipUsage: {
+      requests: ipUsage?.count || 0,
+      limit: FREE_TIER_DAILY_LIMIT,
+      percentUsed: ipUsage ? Math.round((ipUsage.count / FREE_TIER_DAILY_LIMIT) * 100) : 0
+    },
+    activeIps: ipDailyUsage.size
+  });
+});
+
+// Stats endpoint
+app.get('/stats', (req, res) => {
+  const clientIp = req.ip;
+  const today = new Date().toDateString();
+  const ipUsage = ipDailyUsage.get(clientIp);
+
+  res.json({
+    todayRequests: todayRequestCount,
+    dailyLimit: DAILY_LIMIT,
+    date: lastResetDate,
+    percentUsed: Math.round((todayRequestCount / DAILY_LIMIT) * 100),
+    status: 'ok',
+    // Per-IP usage
+    ipUsage: {
+      ip: clientIp,
+      requests: ipUsage?.count || 0,
+      limit: FREE_TIER_DAILY_LIMIT,
+      percentUsed: ipUsage ? Math.round((ipUsage.count / FREE_TIER_DAILY_LIMIT) * 100) : 0,
+      date: ipUsage?.date || today
+    },
+    activeIps: ipDailyUsage.size
+  });
+});
+
+// Chat completion proxy endpoint
+app.post('/chat', limiter, async (req, res) => {
+  try {
+    const clientIp = req.ip;
+    const today = new Date().toDateString();
+
+    // Reset global daily counter if it's a new day
+    if (today !== lastResetDate) {
+      todayRequestCount = 0;
+      lastResetDate = today;
+    }
+
+    // Reset per-IP usage if it's a new day for this IP
+    if (!ipDailyUsage.has(clientIp) || ipDailyUsage.get(clientIp).date !== today) {
+      ipDailyUsage.set(clientIp, { count: 0, date: today });
+    }
+
+    const ipUsage = ipDailyUsage.get(clientIp);
+
+    // Check free tier per-IP daily limit
+    if (ipUsage.count >= FREE_TIER_DAILY_LIMIT) {
+      return res.status(429).json({
+        error: 'Free tier daily limit exceeded. Try again tomorrow.',
+        limit: FREE_TIER_DAILY_LIMIT,
+        resetTime: getNextMidnight()
+      });
+    }
+
+    // Check daily hard limit
+    if (todayRequestCount >= DAILY_LIMIT) {
+      return res.status(429).json({
+        error: 'Daily limit exceeded. Try again tomorrow.',
+        resetTime: getNextMidnight()
+      });
+    }
+
+    const apiKey = process.env.OPENROUTER_API_KEY;
+
+    if (!apiKey) {
+      return res.status(500).json({
+        error: 'OPENROUTER_API_KEY not configured'
+      });
+    }
+
+    // Validate request body
+    if (!req.body.messages || !Array.isArray(req.body.messages)) {
+      return res.status(400).json({
+        error: 'Invalid request: messages array required'
+      });
+    }
+
+    // Increment counters and log
+    todayRequestCount++;
+    ipUsage.count++;
+    console.log(`[${new Date().toISOString()}] Request ${todayRequestCount}/${DAILY_LIMIT} from ${req.ip} (IP: ${ipUsage.count}/${FREE_TIER_DAILY_LIMIT})`);
+
+    // Fetch with timeout
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), 60000); // 60 second timeout
+
+    const response = await fetch('https://openrouter.ai/api/v1/chat/completions', {
+      method: 'POST',
+      headers: {
+        'Authorization': `Bearer ${apiKey}`,
+        'Content-Type': 'application/json',
+        'HTTP-Referer': 'https://github.com/vincentm65/vmCode-CLI',
+        'X-Title': 'vmCode'
+      },
+      body: JSON.stringify({
+        model: 'z-ai/glm-4.5-air:free',
+        ...req.body
+      }),
+      signal: controller.signal
+    });
+
+    clearTimeout(timeout);
+
+    const data = await response.json();
+
+    if (!response.ok) {
+      return res.status(response.status).json(data);
+    }
+
+    res.json(data);
+  } catch (error) {
+    console.error('Proxy error:', error);
+
+    if (error.name === 'AbortError') {
+      return res.status(504).json({
+        error: 'Request timeout'
+      });
+    }
+
+    res.status(500).json({
+      error: 'Internal server error',
+      message: error.message
+    });
+  }
+});
+
+// Graceful shutdown
+let isShuttingDown = false;
+
+function gracefulShutdown(signal) {
+  if (isShuttingDown) {
+    console.log('Force shutdown...');
+    process.exit(1);
+  }
+
+  isShuttingDown = true;
+  console.log(`${signal} received. Starting graceful shutdown...`);
+
+  // Give 10 seconds for in-flight requests to complete
+  setTimeout(() => {
+    console.log('Closing server...');
+    server.close(() => {
+      console.log('Server closed.');
+      process.exit(0);
+    });
+  }, 10000);
+}
+
+const PORT = process.env.PORT || 3000;
+const server = app.listen(PORT, () => {
+  console.log(`vmcode-proxy running on port ${PORT}`);
+  console.log(`OPENROUTER_API_KEY configured: ${!!process.env.OPENROUTER_API_KEY}`);
+});
+
+process.on('SIGTERM', () => gracefulShutdown('SIGTERM'));
+process.on('SIGINT', () => gracefulShutdown('SIGINT'));

package/vmcode-proxy/server.js.bak

@@ -0,0 +1,157 @@
+const express = require('express');
+const rateLimit = require('express-rate-limit');
+require('dotenv').config();
+
+const app = express();
+
+// Middleware
+app.use(express.json());
+
+// Rate limiting: 100 requests per 15 minutes per IP
+const limiter = rateLimit({
+  windowMs: 15 * 60 * 1000,
+  max: 100,
+  keyGenerator: (req) => req.ip,
+  message: { error: 'Rate limit exceeded. Try again later.' }
+});
+
+// Daily hard limit: 10,000 requests/day total
+let todayRequestCount = 0;
+let lastResetDate = new Date().toDateString();
+const DAILY_LIMIT = 10000;
+
+// Free tier: per-IP daily limit
+const FREE_TIER_DAILY_LIMIT = 500; // 500 requests per day per IP
+const ipDailyUsage = new Map(); // Map<ip, {count: number, date: string}>
+
+// Health check endpoint
+app.get('/', (req, res) => {
+  res.json({ status: 'ok', service: 'vmcode-proxy' });
+});
+
+app.get('/health', (req, res) => {
+  const clientIp = req.ip;
+  const today = new Date().toDateString();
+  const ipUsage = ipDailyUsage.get(clientIp);
+
+  res.json({
+    status: 'ok',
+    service: 'vmcode-proxy',
+    todayRequests: todayRequestCount,
+    dailyLimit: DAILY_LIMIT,
+    percentUsed: Math.round((todayRequestCount / DAILY_LIMIT) * 100),
+    // Per-IP usage
+    ipUsage: {
+      requests: ipUsage?.count || 0,
+      limit: FREE_TIER_DAILY_LIMIT,
+      percentUsed: ipUsage ? Math.round((ipUsage.count / FREE_TIER_DAILY_LIMIT) * 100) : 0
+    }
+  });
+});
+
+// Stats endpoint
+app.get('/stats', (req, res) => {
+  const clientIp = req.ip;
+  const today = new Date().toDateString();
+  const ipUsage = ipDailyUsage.get(clientIp);
+
+  res.json({
+    todayRequests: todayRequestCount,
+    dailyLimit: DAILY_LIMIT,
+    date: lastResetDate,
+    percentUsed: Math.round((todayRequestCount / DAILY_LIMIT) * 100),
+    status: 'ok',
+    // Per-IP usage
+    ipUsage: {
+      ip: clientIp,
+      requests: ipUsage?.count || 0,
+      limit: FREE_TIER_DAILY_LIMIT,
+      percentUsed: ipUsage ? Math.round((ipUsage.count / FREE_TIER_DAILY_LIMIT) * 100) : 0,
+      date: ipUsage?.date || today
+    }
+  });
+});
+
+// Chat completion proxy endpoint
+app.post('/chat', limiter, async (req, res) => {
+  try {
+    const clientIp = req.ip;
+    const today = new Date().toDateString();
+
+    // Reset global daily counter if it's a new day
+    if (today !== lastResetDate) {
+      todayRequestCount = 0;
+      lastResetDate = today;
+    }
+
+    // Reset per-IP usage if it's a new day for this IP
+    if (!ipDailyUsage.has(clientIp) || ipDailyUsage.get(clientIp).date !== today) {
+      ipDailyUsage.set(clientIp, { count: 0, date: today });
+    }
+
+    const ipUsage = ipDailyUsage.get(clientIp);
+
+    // Check free tier per-IP daily limit
+    if (ipUsage.count >= FREE_TIER_DAILY_LIMIT) {
+      return res.status(429).json({
+        error: 'Free tier daily limit exceeded. Try again tomorrow.',
+        limit: FREE_TIER_DAILY_LIMIT,
+        resetTime: new Date().setHours(24, 0, 0, 0)
+      });
+    }
+
+    // Check daily hard limit
+    if (todayRequestCount >= DAILY_LIMIT) {
+      return res.status(429).json({
+        error: 'Daily limit exceeded. Try again tomorrow.'
+      });
+    }
+
+    const apiKey = process.env.OPENROUTER_API_KEY;
+
+    if (!apiKey) {
+      return res.status(500).json({
+        error: 'OPENROUTER_API_KEY not configured'
+      });
+    }
+
+    // Increment counters and log
+    todayRequestCount++;
+    ipUsage.count++;
+    console.log(`[${new Date().toISOString()}] Request ${todayRequestCount}/${DAILY_LIMIT} from ${req.ip} (IP: ${ipUsage.count}/${FREE_TIER_DAILY_LIMIT})`);
+
+    const response = await fetch('https://openrouter.ai/api/v1/chat/completions', {
+      method: 'POST',
+      headers: {
+        'Authorization': `Bearer ${apiKey}`,
+        'Content-Type': 'application/json',
+        'HTTP-Referer': 'https://github.com/vincentm65/vmCode-CLI',
+        'X-Title': 'vmCode'
+      },
+      body: JSON.stringify({
+        model: 'openrouter/free',
+        ...req.body
+      })
+    });
+
+    const data = await response.json();
+
+    if (!response.ok) {
+      return res.status(response.status).json(data);
+    }
+
+    res.json(data);
+  } catch (error) {
+    console.error('Proxy error:', error);
+    res.status(500).json({
+      error: 'Internal server error',
+      message: error.message
+    });
+  }
+});
+
+const PORT = process.env.PORT || 3000;
+app.listen(PORT, () => {
+  console.log(`vmcode-proxy running on port ${PORT}`);
+  console.log(`OPENROUTER_API_KEY configured: ${!!process.env.OPENROUTER_API_KEY}`);
+});