protocol-proxy 2.3.4 → 2.4.0

package/lib/config-store.js

@@ -73,6 +73,35 @@ function normalizeModels(models) {
   ));
 }
 
+function normalizeRoutingStrategy(strategy) {
+  return ['primary_fallback', 'round_robin', 'weighted', 'fastest'].includes(strategy)
+    ? strategy
+    : 'primary_fallback';
+}
+
+function normalizeProviderPool(pool) {
+  if (!Array.isArray(pool)) return [];
+  const seen = new Set();
+  const result = [];
+
+  for (const item of pool) {
+    if (!item || typeof item !== 'object') continue;
+    const providerId = typeof item.providerId === 'string' ? item.providerId.trim() : '';
+    if (!providerId) continue;
+    const model = typeof item.model === 'string' ? item.model.trim() : '';
+    const key = `${providerId}\0${model}`;
+    if (seen.has(key)) continue;
+    seen.add(key);
+    result.push({
+      providerId,
+      model,
+      weight: Math.max(1, parseInt(item.weight, 10) || 1),
+    });
+  }
+
+  return result;
+}
+
 function normalizeProvider(provider) {
   if (!provider) return provider;
   return {
@@ -83,7 +112,11 @@ function normalizeProvider(provider) {
 
 function normalizeProxy(proxy) {
   if (!proxy) return proxy;
-  return proxy;
+  return {
+    ...proxy,
+    routingStrategy: normalizeRoutingStrategy(proxy.routingStrategy),
+    providerPool: normalizeProviderPool(proxy.providerPool),
+  };
 }
 
 function normalizeConfig(config) {
@@ -222,4 +255,6 @@ module.exports = {
   addProxy,
   updateProxy,
   removeProxy,
+  normalizeRoutingStrategy,
+  normalizeProviderPool,
 };

package/lib/logger.js

@@ -0,0 +1,58 @@
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+const LOG_DIR = path.join(os.homedir(), '.protocol-proxy');
+const LOG_FILE = path.join(LOG_DIR, 'server.log');
+const MAX_SIZE = 10 * 1024 * 1024; // 10MB
+
+let stream = null;
+
+function init() {
+  try {
+    if (!fs.existsSync(LOG_DIR)) fs.mkdirSync(LOG_DIR, { recursive: true });
+    stream = fs.createWriteStream(LOG_FILE, { flags: 'w' });
+  } catch (err) {
+    console.error('[Logger] 初始化失败:', err.message);
+    stream = null;
+  }
+}
+
+function checkSize() {
+  if (!stream) return;
+  try {
+    const stat = fs.statSync(LOG_FILE);
+    if (stat.size > MAX_SIZE) {
+      // 截断：保留后半部分
+      const content = fs.readFileSync(LOG_FILE, 'utf8');
+      const half = content.slice(Math.floor(content.length / 2));
+      stream.end();
+      fs.writeFileSync(LOG_FILE, half);
+      stream = fs.createWriteStream(LOG_FILE, { flags: 'a' });
+    }
+  } catch {}
+}
+
+function timestamp() {
+  return new Date().toLocaleString('zh-CN', { hour12: false });
+}
+
+function log(...args) {
+  const line = `[${timestamp()}] ${args.join(' ')}`;
+  console.log(line);
+  if (stream) {
+    stream.write(line + '\n');
+    checkSize();
+  }
+}
+
+function error(...args) {
+  const line = `[${timestamp()}] ${args.join(' ')}`;
+  console.error(line);
+  if (stream) {
+    stream.write(line + '\n');
+    checkSize();
+  }
+}
+
+module.exports = { init, log, error, LOG_FILE };

package/lib/proxy-manager.js

@@ -57,6 +57,10 @@ class ProxyManager {
     return true;
   }
 
+  getProxyConfig(id) {
+    return this.servers.get(id)?.config || null;
+  }
+
   isRunning(id) {
     return this.servers.has(id);
   }