@lancewu602/mihomo-dash 0.0.0-stage → 0.6.1

package/server/src/http/providers.js

@@ -0,0 +1,251 @@
+/**
+ * 订阅（proxy-providers）增删改：以文本方式增删 config.yaml 中的条目，保留原有注释与排版，
+ * 校验「解析结果与预期一致」后写盘 → 热重载（复用 config-files）。
+ */
+import fs from 'node:fs';
+import path from 'node:path';
+import yaml from 'js-yaml';
+import { detectMihomoDir } from '../shared/paths.js';
+import { deleteFile, saveConfigWithReload } from './config-files.js';
+import {
+  assertName,
+  findMapEntryLine,
+  findSection,
+  indentOf,
+  keyOf,
+  loadConfig,
+  mapEntryEnd,
+  mapSection,
+  normalize,
+  removeRange,
+  sectionEnd,
+  verifyRewrite,
+} from './config-sections.js';
+
+const SECTION = 'proxy-providers';
+const DEFAULT_HEALTH_URL = 'https://www.gstatic.com/generate_204';
+const DEFAULT_INTERVAL = 3600;
+const DEFAULT_HEALTH_INTERVAL = 300;
+
+/** proxy-providers 段（非映射时视作空） */
+function providerMap(doc) {
+  return mapSection(doc, SECTION);
+}
+
+/**
+ * 「拉取用代理」取值：策略组名 / 节点名 / `DIRECT`，空串表示不设（= 内核直连拉取）。
+ * 名称里出现 `:` `#` 或换行时一律拒绝（文本级写行时会被当成结构）。
+ * @param {unknown} raw
+ * @returns {string}
+ */
+function cleanProxy(raw) {
+  const v = String(raw ?? '').trim();
+  if (!v) return '';
+  if (v.length > 128) throw new Error('拉取用代理的名称过长（≤128 字符）');
+  if (/[\n\r:#]/.test(v)) throw new Error('拉取用代理的名称不能包含 : # 或换行');
+  return v;
+}
+
+/** 订阅缓存文件的相对路径：优先 config 中的 path，其次 providers/<name>.yaml；不安全则返回 null */
+function cacheRelativePath(rawPath, name) {
+  const rel =
+    typeof rawPath === 'string' && rawPath.trim() ? rawPath.trim() : `./providers/${name}.yaml`;
+  const clean = rel.replace(/^\.\//, '');
+  if (path.isAbsolute(clean) || clean.split(/[\\/]/).includes('..')) return null;
+  if (/\.ya?ml$/i.test(clean) === false) return null;
+  if (path.basename(clean) === 'config.yaml') return null;
+  return clean;
+}
+
+/**
+ * 读取 config.yaml 中全部订阅条目（原样返回）。
+ * 内核 `/providers/proxies` 不回传 interval / health-check 等配置项，面板展示时以此为准。
+ */
+export function listProviders() {
+  const { doc } = loadConfig();
+  return providerMap(doc);
+}
+
+/**
+ * 新增订阅：写入 `proxy-providers.<name>`（type: http）并热重载。
+ * `proxy` 不填就不写这个键（内核按直连拉取，墙内常因此拉不动 —— 面板表单默认给一个策略组）。
+ * @param {{ name: string, url: string, interval?: number, healthCheckUrl?: string,
+ *   healthCheckInterval?: number, proxy?: string }} input
+ */
+export async function addProvider(input) {
+  const name = assertName(input?.name, '订阅名称');
+  const url = String(input?.url ?? '').trim();
+  if (!/^https?:\/\/\S+$/i.test(url)) throw new Error('订阅链接需为 http(s) 开头的 URL');
+
+  const { text, doc } = loadConfig();
+  const existing = providerMap(doc);
+  if (Object.prototype.hasOwnProperty.call(existing, name)) throw new Error(`订阅 ${name} 已存在`);
+  const dir = detectMihomoDir();
+  const relPath = `providers/${name}.yaml`;
+  if (fs.existsSync(path.join(dir, relPath))) {
+    throw new Error(`${relPath} 已存在，请换一个名称`);
+  }
+
+  const proxy = cleanProxy(input?.proxy);
+  const entry = {
+    type: 'http',
+    url,
+    interval: Number(input?.interval) > 0 ? Number(input.interval) : DEFAULT_INTERVAL,
+    path: `./${relPath}`,
+    ...(proxy ? { proxy } : {}),
+    'health-check': {
+      enable: true,
+      interval:
+        Number(input?.healthCheckInterval) > 0
+          ? Number(input.healthCheckInterval)
+          : DEFAULT_HEALTH_INTERVAL,
+      url: String(input?.healthCheckUrl ?? '').trim() || DEFAULT_HEALTH_URL,
+    },
+  };
+
+  const block = yaml
+    .dump({ [name]: entry }, { indent: 2, lineWidth: 100, noRefs: true })
+    .trimEnd()
+    .split('\n')
+    .map((l) => (l ? `  ${l}` : l));
+
+  const lines = text.split('\n');
+  const at = findSection(lines, SECTION);
+  const out =
+    at >= 0
+      ? [...lines.slice(0, at + 1), ...block, ...lines.slice(at + 1)]
+      : [...lines, `${SECTION}:`, ...block];
+
+  const next = normalize(out);
+  verifyRewrite(next, { ...doc, [SECTION]: { ...existing, [name]: entry } });
+
+  const reload = await saveConfigWithReload(next);
+  return { ok: true, name, path: relPath, proxy, ...reload };
+}
+
+/**
+ * 改订阅的「拉取用代理」（`proxy` 字段）：走哪个策略组 / 节点去拉这份订阅。
+ * 空串 = 删掉该键（回退内核默认的直连拉取）。
+ * 文本级改写：只替换 / 插入 `proxy:` 这一行，条目里其它键、注释与排版都不动。
+ * @param {string} name
+ * @param {string} [rawProxy]
+ */
+export async function setProviderProxy(name, rawProxy) {
+  const providerName = assertName(name, '订阅名称');
+  const proxy = cleanProxy(rawProxy);
+  const { text, doc } = loadConfig();
+  const existing = providerMap(doc);
+  const entry = existing[providerName];
+  if (!entry || typeof entry !== 'object' || Array.isArray(entry)) {
+    throw new Error(`config.yaml 中未找到订阅 ${providerName}`);
+  }
+  const current = typeof entry.proxy === 'string' ? entry.proxy.trim() : '';
+  if (current === proxy) {
+    return { ok: true, name: providerName, proxy, unchanged: true, reloaded: false };
+  }
+
+  const lines = text.split('\n');
+  const at = findSection(lines, SECTION);
+  if (at < 0) throw new Error(`config.yaml 中未找到 ${SECTION} 段`);
+  const start = findMapEntryLine(lines, at + 1, providerName);
+  if (start < 0) throw new Error(`config.yaml 中未找到订阅 ${providerName} 的条目`);
+  if (/^\s*[^:#]+:\s*\{/.test(lines[start])) {
+    throw new Error(`订阅 ${providerName} 是行内写法（{…}），请先去「配置文件」页改成块状写法`);
+  }
+
+  const end = mapEntryEnd(lines, start + 1);
+  /** 已有 `proxy:` 子键所在行（没有则 -1）与子键缩进（默认 4） */
+  let proxyLine = -1;
+  let childIndent = 0;
+  for (let i = start + 1; i < end; i++) {
+    const t = lines[i].trim();
+    if (!t || t.startsWith('#')) continue;
+    if (!childIndent) childIndent = indentOf(lines[i]);
+    if (keyOf(lines[i]) === 'proxy') proxyLine = i;
+  }
+  if (!childIndent) childIndent = 4;
+
+  const expectedEntry = { ...entry };
+  let out;
+  if (!proxy) {
+    delete expectedEntry.proxy;
+    out = proxyLine >= 0 ? removeRange(lines, proxyLine, proxyLine + 1, { blank: false }) : lines;
+  } else {
+    expectedEntry.proxy = proxy;
+    // 用 js-yaml 渲染这一行：名字里带空格 / 特殊字符时它会自己加引号
+    const line = ' '.repeat(childIndent) + yaml.dump({ proxy }, { lineWidth: 100 }).trimEnd();
+    if (proxyLine >= 0) {
+      out = [...lines.slice(0, proxyLine), line, ...lines.slice(proxyLine + 1)];
+    } else {
+      // 插在 `path:` 之后（没有 path 就紧跟条目首行），让键序与面板新增的条目一致
+      let after = start;
+      for (let i = start + 1; i < end; i++) {
+        if (keyOf(lines[i]) === 'path') {
+          after = i;
+          break;
+        }
+      }
+      out = [...lines.slice(0, after + 1), line, ...lines.slice(after + 1)];
+    }
+  }
+
+  const next = normalize(out);
+  verifyRewrite(next, { ...doc, [SECTION]: { ...existing, [providerName]: expectedEntry } });
+
+  const reload = await saveConfigWithReload(next);
+  return { ok: true, name: providerName, proxy, ...reload };
+}
+
+/**
+ * 删除订阅：从 config.yaml 移除条目并热重载，随后删除本地缓存文件。
+ * @param {string} name
+ * @param {{ deleteCache?: boolean }} [opts]
+ */
+export async function removeProvider(name, opts = {}) {
+  const providerName = assertName(name, '订阅名称');
+  const { text, doc } = loadConfig();
+  const existing = providerMap(doc);
+  if (!Object.prototype.hasOwnProperty.call(existing, providerName)) {
+    throw new Error(`config.yaml 中未找到订阅 ${providerName}`);
+  }
+
+  const entry = existing[providerName] ?? {};
+  const rest = { ...existing };
+  delete rest[providerName];
+
+  const lines = text.split('\n');
+  const at = findSection(lines, SECTION);
+  if (at < 0) throw new Error(`config.yaml 中未找到 ${SECTION} 段`);
+
+  let out;
+  let expected;
+  if (Object.keys(rest).length === 0) {
+    // 最后一个订阅：连同 proxy-providers 标题行与其下所有条目一起移除
+    out = removeRange(lines, at, sectionEnd(lines, at + 1));
+    expected = { ...doc };
+    delete expected[SECTION];
+  } else {
+    const start = findMapEntryLine(lines, at + 1, providerName);
+    if (start < 0) throw new Error(`config.yaml 中未找到订阅 ${providerName} 的条目`);
+    out = removeRange(lines, start, mapEntryEnd(lines, start + 1));
+    expected = { ...doc, [SECTION]: rest };
+  }
+
+  const next = normalize(out);
+  verifyRewrite(next, expected);
+
+  const reload = await saveConfigWithReload(next);
+
+  // 配置已生效后再清理本地缓存；删除失败不影响结果，仅回报
+  let cache = { removed: false, path: null };
+  const rel = cacheRelativePath(entry.path, providerName);
+  if (opts.deleteCache !== false && rel) {
+    try {
+      cache = { ...deleteFile(rel), path: rel };
+    } catch (e) {
+      cache = { removed: false, path: rel, error: String(e?.message || e) };
+    }
+  }
+
+  return { ok: true, name: providerName, cache, ...reload };
+}