1import { spawnSync } from 'node:child_process';
2import fs from 'node:fs';
3import path from 'node:path';
4import process from 'node:process';
5 
6const WECHAT_URL = 'https://mp.weixin.qq.com/';
7const SESSION = 'wechat-post';
8 
9function sleep(ms: number): Promise<void> {
10  return new Promise((resolve) => setTimeout(resolve, ms));
11}
12 
13function quoteForLog(arg: string): string {
14  return /[\s"'\\]/.test(arg) ? JSON.stringify(arg) : arg;
15}
16 
17function toSafeJsStringLiteral(value: string): string {
18  return JSON.stringify(value)
19    .replace(/\u2028/g, '\\u2028')
20    .replace(/\u2029/g, '\\u2029');
21}
22 
23function runAgentBrowser(args: string[]): {
24  success: boolean;
25  output: string;
26  spawnError?: string;
27} {
28  const result = spawnSync('agent-browser', ['--session', SESSION, ...args], {
29    encoding: 'utf8',
30    stdio: ['pipe', 'pipe', 'pipe']
31  });
32  const spawnError = result.error?.message?.trim();
33  const output = result.stdout || result.stderr || '';
34  return {
35    success: result.status === 0,
36    output: output || spawnError || '',
37    spawnError
38  };
39}
40 
41function ab(args: string[], json = false): string {
42  const fullArgs = json ? [...args, '--json'] : args;
43  console.log(`[ab] agent-browser --session ${SESSION} ${fullArgs.map(quoteForLog).join(' ')}`);
44  const result = runAgentBrowser(fullArgs);
45  if (result.spawnError) {
46    throw new Error(`agent-browser failed to start: ${result.spawnError}`);
47  }
48  if (!result.success) {
49    console.error(`[ab] Error: ${result.output.trim()}`);
50  }
51  return result.output.trim();
52}
53 
54function abRaw(args: string[]): { success: boolean; output: string } {
55  return runAgentBrowser(args);
56}
57 
58interface SnapshotElement {
59  ref: string;
60  role: string;
61  name: string;
62}
63 
64function parseSnapshot(output: string): SnapshotElement[] {
65  const elements: SnapshotElement[] = [];
66  const refPattern = /\[ref=(@?\w+)\]/g;
67  const lines = output.split('\n');
68 
69  for (const line of lines) {
70    const match = line.match(/\[ref=([@\w]+)\]/);
71    if (match) {
72      const ref = match[1].startsWith('@') ? match[1] : `@${match[1]}`;
73      const roleMatch = line.match(/^-\s+(\w+)/);
74      const nameMatch = line.match(/"([^"]+)"/);
75      elements.push({
76        ref,
77        role: roleMatch?.[1] || 'unknown',
78        name: nameMatch?.[1] || ''
79      });
80    }
81  }
82  return elements;
83}
84 
85function findElementByText(snapshot: string, text: string): string | null {
86  const lines = snapshot.split('\n');
87  for (const line of lines) {
88    if (line.includes(`"${text}"`) || line.includes(text)) {
89      const match = line.match(/\[ref=([@\w]+)\]/);
90      if (match) {
91        return match[1].startsWith('@') ? match[1] : `@${match[1]}`;
92      }
93    }
94  }
95  return null;
96}
97 
98function findElementBySelector(snapshot: string, selector: string): string | null {
99  return null;
100}
101 
102interface WeChatOptions {
103  title: string;
104  content: string;
105  images: string[];
106  submit?: boolean;
107  keepOpen?: boolean;
108}
109 
110async function postToWeChat(options: WeChatOptions): Promise<void> {
111  const { title, content, images, submit = false, keepOpen = true } = options;
112 
113  if (title.length > 20) throw new Error(`Title too long: ${title.length} chars (max 20)`);
114  if (content.length > 1000) throw new Error(`Content too long: ${content.length} chars (max 1000)`);
115  if (images.length === 0) throw new Error('At least one image is required');
116 
117  const absoluteImages = images.map(p => path.isAbsolute(p) ? p : path.resolve(process.cwd(), p));
118  for (const img of absoluteImages) {
119    if (!fs.existsSync(img)) throw new Error(`Image not found: ${img}`);
120  }
121 
122  console.log('[wechat] Opening WeChat Official Account...');
123  ab(['open', WECHAT_URL, '--headed']);
124  await sleep(5000);
125 
126  console.log('[wechat] Checking login status...');
127  let url = ab(['get', 'url']);
128  console.log(`[wechat] Current URL: ${url}`);
129 
130  const waitForLogin = async (timeoutMs = 120_000): Promise<boolean> => {
131    const start = Date.now();
132    while (Date.now() - start < timeoutMs) {
133      url = ab(['get', 'url']);
134      if (url.includes('/cgi-bin/home')) return true;
135      console.log('[wechat] Waiting for login...');
136      await sleep(3000);
137    }
138    return false;
139  };
140 
141  if (!url.includes('/cgi-bin/home')) {
142    console.log('[wechat] Not logged in. Please scan QR code...');
143    const loggedIn = await waitForLogin();
144    if (!loggedIn) throw new Error('Login timeout');
145  }
146  console.log('[wechat] Logged in.');
147  await sleep(2000);
148 
149  console.log('[wechat] Getting page snapshot...');
150  let snapshot = ab(['snapshot']);
151  console.log(snapshot);
152 
153  console.log('[wechat] Looking for "图文" menu...');
154  const tuWenRef = findElementByText(snapshot, '图文');
155 
156  if (!tuWenRef) {
157    console.log('[wechat] Using eval to find and click menu...');
158    ab(['eval', "document.querySelectorAll('.new-creation__menu .new-creation__menu-item')[2].click()"]);
159  } else {
160    console.log(`[wechat] Clicking menu ref: ${tuWenRef}`);
161    ab(['click', tuWenRef]);
162  }
163 
164  await sleep(4000);
165 
166  console.log('[wechat] Checking for new tab...');
167  const tabsOutput = ab(['tab']);
168  console.log(`[wechat] Tabs: ${tabsOutput}`);
169 
170  const tabLines = tabsOutput.split('\n');
171  const editorTabLine = tabLines.find(l => l.includes('appmsg') || (!l.includes('cgi-bin/home') && l.includes('mp.weixin.qq.com')));
172 
173  if (tabLines.length > 1) {
174    const tabMatch = tabsOutput.match(/\[(\d+)\].*(?:appmsg|edit)/i);
175    if (tabMatch) {
176      console.log(`[wechat] Switching to editor tab ${tabMatch[1]}...`);
177      ab(['tab', tabMatch[1]]);
178    } else {
179      const lastTabMatch = tabsOutput.match(/\[(\d+)\]/g);
180      if (lastTabMatch && lastTabMatch.length > 1) {
181        const lastTab = lastTabMatch[lastTabMatch.length - 1].match(/\d+/)?.[0];
182        if (lastTab) {
183          console.log(`[wechat] Switching to last tab ${lastTab}...`);
184          ab(['tab', lastTab]);
185        }
186      }
187    }
188  }
189 
190  await sleep(3000);
191 
192  url = ab(['get', 'url']);
193  console.log(`[wechat] Editor URL: ${url}`);
194 
195  console.log('[wechat] Getting editor snapshot...');
196  snapshot = ab(['snapshot']);
197  console.log(snapshot.substring(0, 2000));
198 
199  console.log('[wechat] Uploading images...');
200  const fileInputSelector = '.js_upload_btn_container input[type=file]';
201  const fileInputSelectorJs = toSafeJsStringLiteral(fileInputSelector);
202 
203  ab(['eval', `{
204    const input = document.querySelector(${fileInputSelectorJs});
205    if (input) input.style.display = 'block';
206  }`]);
207  await sleep(500);
208 
209  const uploadResult = abRaw(['upload', fileInputSelector, ...absoluteImages]);
210  console.log(`[wechat] Upload result: ${uploadResult.output}`);
211 
212  if (!uploadResult.success) {
213    console.log('[wechat] Using alternative upload method...');
214    for (const img of absoluteImages) {
215      console.log(`[wechat] Uploading: ${img}`);
216      const imgUrlJs = toSafeJsStringLiteral(`file://${img}`);
217      const imgFileNameJs = toSafeJsStringLiteral(path.basename(img));
218      ab(['eval', `
219        const input = document.querySelector(${fileInputSelectorJs});
220        if (input) {
221          const dt = new DataTransfer();
222          fetch(${imgUrlJs}).then(r => r.blob()).then(b => {
223            const file = new File([b], ${imgFileNameJs}, { type: 'image/png' });
224            dt.items.add(file);
225            input.files = dt.files;
226            input.dispatchEvent(new Event('change', { bubbles: true }));
227          });
228        }
229      `]);
230      await sleep(2000);
231    }
232  }
233 
234  console.log('[wechat] Waiting for uploads to complete...');
235  await sleep(10000);
236 
237  console.log('[wechat] Filling title...');
238  snapshot = ab(['snapshot', '-i']);
239  const titleRef = findElementByText(snapshot, 'title') || findElementByText(snapshot, '标题');
240 
241  if (titleRef) {
242    ab(['fill', titleRef, title]);
243  } else {
244    const titleJs = toSafeJsStringLiteral(title);
245    ab(['eval', `const t = document.querySelector('#title'); if(t) { t.value = ${titleJs}; t.dispatchEvent(new Event('input', {bubbles: true})); }`]);
246  }
247  await sleep(500);
248 
249  console.log('[wechat] Clicking on content editor...');
250  const editorRef = findElementByText(snapshot, 'js_pmEditorArea') || findElementByText(snapshot, 'textbox');
251 
252  if (editorRef) {
253    ab(['click', editorRef]);
254  } else {
255    ab(['eval', "document.querySelector('.js_pmEditorArea')?.click()"]);
256  }
257  await sleep(500);
258 
259  console.log('[wechat] Typing content...');
260  const lines = content.split('\n');
261  for (let i = 0; i < lines.length; i++) {
262    const line = lines[i];
263    if (line.length > 0) {
264      const lineJs = toSafeJsStringLiteral(line);
265      ab(['eval', `document.execCommand('insertText', false, ${lineJs})`]);
266    }
267    if (i < lines.length - 1) {
268      ab(['press', 'Enter']);
269    }
270    await sleep(100);
271  }
272 
273  console.log('[wechat] Content typed.');
274  await sleep(1000);
275 
276  if (submit) {
277    console.log('[wechat] Saving as draft...');
278    const submitRef = findElementByText(snapshot, 'js_submit') || findElementByText(snapshot, '保存');
279    if (submitRef) {
280      ab(['click', submitRef]);
281    } else {
282      ab(['eval', "document.querySelector('#js_submit')?.click()"]);
283    }
284    await sleep(3000);
285    console.log('[wechat] Draft saved!');
286  } else {
287    console.log('[wechat] Article composed (preview mode). Add --submit to save as draft.');
288  }
289 
290  if (!keepOpen) {
291    console.log('[wechat] Closing browser...');
292    ab(['close']);
293  } else {
294    console.log('[wechat] Done. Browser window left open.');
295  }
296}
297 
298function printUsage(): never {
299  console.log(`Post to WeChat Official Account using agent-browser
300 
301Usage:
302  npx -y bun wechat-agent-browser.ts [options]
303 
304Options:
305  --title <text>   Article title (max 20 chars, required)
306  --content <text> Article content (max 1000 chars, required)
307  --image <path>   Add image (can be repeated, 1+ images, required)
308  --submit         Save as draft (default: preview only)
309  --close          Close browser after operation (default: keep open)
310  --help           Show this help
311 
312Examples:
313  npx -y bun wechat-agent-browser.ts --title "测试" --content "内容" --image ./photo.png
314  npx -y bun wechat-agent-browser.ts --title "测试" --content "内容" --image a.png --image b.png --submit
315`);
316  process.exit(0);
317}
318 
319async function main(): Promise<void> {
320  const args = process.argv.slice(2);
321  if (args.includes('--help') || args.includes('-h')) printUsage();
322 
323  const images: string[] = [];
324  let submit = false;
325  let keepOpen = true;
326  let title: string | undefined;
327  let content: string | undefined;
328 
329  for (let i = 0; i < args.length; i++) {
330    const arg = args[i]!;
331    if (arg === '--image' && args[i + 1]) {
332      images.push(args[++i]!);
333    } else if (arg === '--title' && args[i + 1]) {
334      title = args[++i];
335    } else if (arg === '--content' && args[i + 1]) {
336      content = args[++i];
337    } else if (arg === '--submit') {
338      submit = true;
339    } else if (arg === '--close') {
340      keepOpen = false;
341    }
342  }
343 
344  if (!title) {
345    console.error('Error: --title is required');
346    process.exit(1);
347  }
348  if (!content) {
349    console.error('Error: --content is required');
350    process.exit(1);
351  }
352  if (images.length === 0) {
353    console.error('Error: At least one --image is required');
354    process.exit(1);
355  }
356 
357  await postToWeChat({ title, content, images, submit, keepOpen });
358}
359 
360await main().catch((err) => {
361  console.error(`Error: ${err instanceof Error ? err.message : String(err)}`);
362  process.exit(1);
363});
364