1import fs from "node:fs";
2import path from "node:path";
3import os from "node:os";
4 
5export interface WechatAccount {
6  name: string;
7  alias: string;
8  default?: boolean;
9  default_publish_method?: string;
10  default_author?: string;
11  need_open_comment?: number;
12  only_fans_can_comment?: number;
13  app_id?: string;
14  app_secret?: string;
15  chrome_profile_path?: string;
16}
17 
18export interface WechatExtendConfig {
19  default_theme?: string;
20  default_color?: string;
21  default_publish_method?: string;
22  default_author?: string;
23  need_open_comment?: number;
24  only_fans_can_comment?: number;
25  chrome_profile_path?: string;
26  accounts?: WechatAccount[];
27}
28 
29export interface ResolvedAccount {
30  name?: string;
31  alias?: string;
32  default_publish_method?: string;
33  default_author?: string;
34  need_open_comment: number;
35  only_fans_can_comment: number;
36  app_id?: string;
37  app_secret?: string;
38  chrome_profile_path?: string;
39}
40 
41function stripQuotes(s: string): string {
42  return s.replace(/^['"]|['"]$/g, "");
43}
44 
45function toBool01(v: string): number {
46  return v === "1" || v === "true" ? 1 : 0;
47}
48 
49function parseWechatExtend(content: string): WechatExtendConfig {
50  const config: WechatExtendConfig = {};
51  const lines = content.split("\n");
52  let inAccounts = false;
53  let current: Record<string, string> | null = null;
54  const rawAccounts: Record<string, string>[] = [];
55 
56  for (const raw of lines) {
57    const trimmed = raw.trim();
58    if (!trimmed || trimmed.startsWith("#")) continue;
59 
60    if (trimmed === "accounts:") {
61      inAccounts = true;
62      continue;
63    }
64 
65    if (inAccounts) {
66      const listMatch = raw.match(/^\s+-\s+(.+)$/);
67      if (listMatch) {
68        if (current) rawAccounts.push(current);
69        current = {};
70        const kv = listMatch[1]!;
71        const ci = kv.indexOf(":");
72        if (ci > 0) {
73          current[kv.slice(0, ci).trim()] = stripQuotes(kv.slice(ci + 1).trim());
74        }
75        continue;
76      }
77 
78      if (current && /^\s{2,}/.test(raw) && !trimmed.startsWith("-")) {
79        const ci = trimmed.indexOf(":");
80        if (ci > 0) {
81          current[trimmed.slice(0, ci).trim()] = stripQuotes(trimmed.slice(ci + 1).trim());
82        }
83        continue;
84      }
85 
86      if (!/^\s/.test(raw)) {
87        if (current) rawAccounts.push(current);
88        current = null;
89        inAccounts = false;
90      } else {
91        continue;
92      }
93    }
94 
95    const ci = trimmed.indexOf(":");
96    if (ci < 0) continue;
97    const key = trimmed.slice(0, ci).trim();
98    const val = stripQuotes(trimmed.slice(ci + 1).trim());
99    if (val === "null" || val === "") continue;
100 
101    switch (key) {
102      case "default_theme": config.default_theme = val; break;
103      case "default_color": config.default_color = val; break;
104      case "default_publish_method": config.default_publish_method = val; break;
105      case "default_author": config.default_author = val; break;
106      case "need_open_comment": config.need_open_comment = toBool01(val); break;
107      case "only_fans_can_comment": config.only_fans_can_comment = toBool01(val); break;
108      case "chrome_profile_path": config.chrome_profile_path = val; break;
109    }
110  }
111 
112  if (current) rawAccounts.push(current);
113 
114  if (rawAccounts.length > 0) {
115    config.accounts = rawAccounts.map(a => ({
116      name: a.name || "",
117      alias: a.alias || "",
118      default: a.default === "true" || a.default === "1",
119      default_publish_method: a.default_publish_method || undefined,
120      default_author: a.default_author || undefined,
121      need_open_comment: a.need_open_comment ? toBool01(a.need_open_comment) : undefined,
122      only_fans_can_comment: a.only_fans_can_comment ? toBool01(a.only_fans_can_comment) : undefined,
123      app_id: a.app_id || undefined,
124      app_secret: a.app_secret || undefined,
125      chrome_profile_path: a.chrome_profile_path || undefined,
126    }));
127  }
128 
129  return config;
130}
131 
132export function loadWechatExtendConfig(): WechatExtendConfig {
133  const paths = [
134    path.join(process.cwd(), ".baoyu-skills", "baoyu-post-to-wechat", "EXTEND.md"),
135    path.join(
136      process.env.XDG_CONFIG_HOME || path.join(os.homedir(), ".config"),
137      "baoyu-skills", "baoyu-post-to-wechat", "EXTEND.md"
138    ),
139    path.join(os.homedir(), ".baoyu-skills", "baoyu-post-to-wechat", "EXTEND.md"),
140  ];
141  for (const p of paths) {
142    try {
143      const content = fs.readFileSync(p, "utf-8");
144      return parseWechatExtend(content);
145    } catch {
146      continue;
147    }
148  }
149  return {};
150}
151 
152function selectAccount(config: WechatExtendConfig, alias?: string): WechatAccount | undefined {
153  if (!config.accounts || config.accounts.length === 0) return undefined;
154  if (alias) return config.accounts.find(a => a.alias === alias);
155  if (config.accounts.length === 1) return config.accounts[0];
156  return config.accounts.find(a => a.default);
157}
158 
159export function resolveAccount(config: WechatExtendConfig, alias?: string): ResolvedAccount {
160  const acct = selectAccount(config, alias);
161  return {
162    name: acct?.name,
163    alias: acct?.alias,
164    default_publish_method: acct?.default_publish_method ?? config.default_publish_method,
165    default_author: acct?.default_author ?? config.default_author,
166    need_open_comment: acct?.need_open_comment ?? config.need_open_comment ?? 1,
167    only_fans_can_comment: acct?.only_fans_can_comment ?? config.only_fans_can_comment ?? 0,
168    app_id: acct?.app_id,
169    app_secret: acct?.app_secret,
170    chrome_profile_path: acct?.chrome_profile_path ?? config.chrome_profile_path,
171  };
172}
173 
174function loadEnvFile(envPath: string): Record<string, string> {
175  const env: Record<string, string> = {};
176  if (!fs.existsSync(envPath)) return env;
177  const content = fs.readFileSync(envPath, "utf-8");
178  for (const line of content.split("\n")) {
179    const trimmed = line.trim();
180    if (!trimmed || trimmed.startsWith("#")) continue;
181    const eqIdx = trimmed.indexOf("=");
182    if (eqIdx > 0) {
183      const key = trimmed.slice(0, eqIdx).trim();
184      let value = trimmed.slice(eqIdx + 1).trim();
185      if ((value.startsWith('"') && value.endsWith('"')) ||
186          (value.startsWith("'") && value.endsWith("'"))) {
187        value = value.slice(1, -1);
188      }
189      env[key] = value;
190    }
191  }
192  return env;
193}
194 
195function aliasToEnvKey(alias: string): string {
196  return alias.toUpperCase().replace(/-/g, "_");
197}
198 
199interface CredentialSource {
200  name: string;
201  appIdKey: string;
202  appSecretKey: string;
203  appId?: string;
204  appSecret?: string;
205}
206 
207export interface LoadedCredentials {
208  appId: string;
209  appSecret: string;
210  source: string;
211  skippedSources: string[];
212}
213 
214function normalizeCredentialValue(value?: string): string | undefined {
215  const trimmed = value?.trim();
216  return trimmed ? trimmed : undefined;
217}
218 
219function describeMissingKeys(source: CredentialSource): string {
220  const missingKeys: string[] = [];
221  if (!source.appId) missingKeys.push(source.appIdKey);
222  if (!source.appSecret) missingKeys.push(source.appSecretKey);
223  return `${source.name} missing ${missingKeys.join(" and ")}`;
224}
225 
226function buildCredentialSource(
227  name: string,
228  values: Record<string, string | undefined>,
229  appIdKey: string,
230  appSecretKey: string,
231): CredentialSource {
232  return {
233    name,
234    appIdKey,
235    appSecretKey,
236    appId: normalizeCredentialValue(values[appIdKey]),
237    appSecret: normalizeCredentialValue(values[appSecretKey]),
238  };
239}
240 
241function resolveCredentialSource(
242  sources: CredentialSource[],
243  account?: ResolvedAccount,
244): LoadedCredentials {
245  const skippedSources: string[] = [];
246 
247  for (const source of sources) {
248    if (source.appId && source.appSecret) {
249      return {
250        appId: source.appId,
251        appSecret: source.appSecret,
252        source: source.name,
253        skippedSources,
254      };
255    }
256 
257    if (source.appId || source.appSecret) {
258      skippedSources.push(describeMissingKeys(source));
259    }
260  }
261 
262  const hint = account?.alias ? ` (account: ${account.alias})` : "";
263  const partialHint = skippedSources.length > 0
264    ? `\nIncomplete credential sources skipped:\n- ${skippedSources.join("\n- ")}`
265    : "";
266 
267  throw new Error(
268    `Missing WECHAT_APP_ID or WECHAT_APP_SECRET${hint}.\n` +
269    "Set via EXTEND.md account config, environment variables, or .baoyu-skills/.env file." +
270    partialHint
271  );
272}
273 
274export function loadCredentials(account?: ResolvedAccount): LoadedCredentials {
275  const cwdEnvPath = path.join(process.cwd(), ".baoyu-skills", ".env");
276  const homeEnvPath = path.join(os.homedir(), ".baoyu-skills", ".env");
277  const cwdEnv = loadEnvFile(cwdEnvPath);
278  const homeEnv = loadEnvFile(homeEnvPath);
279 
280  const sources: CredentialSource[] = [];
281 
282  if (account?.app_id || account?.app_secret) {
283    sources.push({
284      name: account.alias ? `EXTEND.md account "${account.alias}"` : "EXTEND.md account config",
285      appIdKey: "app_id",
286      appSecretKey: "app_secret",
287      appId: normalizeCredentialValue(account.app_id),
288      appSecret: normalizeCredentialValue(account.app_secret),
289    });
290  }
291 
292  const prefix = account?.alias ? `WECHAT_${aliasToEnvKey(account.alias)}_` : "";
293  if (prefix) {
294    const prefixedKeyLabel = `${prefix}APP_ID/${prefix}APP_SECRET`;
295    sources.push(
296      buildCredentialSource(`process.env (${prefixedKeyLabel})`, process.env, `${prefix}APP_ID`, `${prefix}APP_SECRET`),
297      buildCredentialSource(`<cwd>/.baoyu-skills/.env (${prefixedKeyLabel})`, cwdEnv, `${prefix}APP_ID`, `${prefix}APP_SECRET`),
298      buildCredentialSource(`~/.baoyu-skills/.env (${prefixedKeyLabel})`, homeEnv, `${prefix}APP_ID`, `${prefix}APP_SECRET`),
299    );
300  }
301 
302  sources.push(
303    buildCredentialSource("process.env", process.env, "WECHAT_APP_ID", "WECHAT_APP_SECRET"),
304    buildCredentialSource("<cwd>/.baoyu-skills/.env", cwdEnv, "WECHAT_APP_ID", "WECHAT_APP_SECRET"),
305    buildCredentialSource("~/.baoyu-skills/.env", homeEnv, "WECHAT_APP_ID", "WECHAT_APP_SECRET"),
306  );
307 
308  return resolveCredentialSource(sources, account);
309}
310 
311export function listAccounts(config: WechatExtendConfig): string[] {
312  return (config.accounts || []).map(a => a.alias);
313}
314