推特支持回复图片

src/clis/twitter/reply.test.ts

@@ -0,0 +1,177 @@
+import * as fs from 'node:fs';
+import * as os from 'node:os';
+import * as path from 'node:path';
+
+import { describe, expect, it, vi } from 'vitest';
+
+import { getRegistry } from '../../registry.js';
+import type { IPage } from '../../types.js';
+import { __test__ } from './reply.js';
+
+function createPageMock(evaluateResults: any[], overrides: Partial<IPage> = {}): IPage {
+  const evaluate = vi.fn();
+  for (const result of evaluateResults) {
+    evaluate.mockResolvedValueOnce(result);
+  }
+
+  return {
+    goto: vi.fn().mockResolvedValue(undefined),
+    evaluate,
+    snapshot: vi.fn().mockResolvedValue(undefined),
+    click: vi.fn().mockResolvedValue(undefined),
+    typeText: vi.fn().mockResolvedValue(undefined),
+    pressKey: vi.fn().mockResolvedValue(undefined),
+    scrollTo: vi.fn().mockResolvedValue(undefined),
+    getFormState: vi.fn().mockResolvedValue({ forms: [], orphanFields: [] }),
+    wait: vi.fn().mockResolvedValue(undefined),
+    tabs: vi.fn().mockResolvedValue([]),
+    selectTab: vi.fn().mockResolvedValue(undefined),
+    networkRequests: vi.fn().mockResolvedValue([]),
+    consoleMessages: vi.fn().mockResolvedValue([]),
+    scroll: vi.fn().mockResolvedValue(undefined),
+    autoScroll: vi.fn().mockResolvedValue(undefined),
+    installInterceptor: vi.fn().mockResolvedValue(undefined),
+    getInterceptedRequests: vi.fn().mockResolvedValue([]),
+    getCookies: vi.fn().mockResolvedValue([]),
+    screenshot: vi.fn().mockResolvedValue(''),
+    waitForCapture: vi.fn().mockResolvedValue(undefined),
+    ...overrides,
+  };
+}
+
+describe('twitter reply command', () => {
+  it('keeps the text-only reply flow working', async () => {
+    const cmd = getRegistry().get('twitter/reply');
+    expect(cmd?.func).toBeTypeOf('function');
+
+    const page = createPageMock([
+      { ok: true, message: 'Reply posted successfully.' },
+    ]);
+
+    const result = await cmd!.func!(page, {
+      url: 'https://x.com/_kop6/status/2040254679301718161?s=20',
+      text: 'text-only reply',
+    });
+
+    expect(page.goto).toHaveBeenCalledWith('https://x.com/_kop6/status/2040254679301718161?s=20');
+    expect(page.wait).toHaveBeenCalledWith({ selector: '[data-testid="primaryColumn"]' });
+    expect(result).toEqual([
+      {
+        status: 'success',
+        message: 'Reply posted successfully.',
+        text: 'text-only reply',
+      },
+    ]);
+  });
+
+  it('uploads a local image through the dedicated reply composer when --image is provided', async () => {
+    const cmd = getRegistry().get('twitter/reply');
+    expect(cmd?.func).toBeTypeOf('function');
+
+    const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'opencli-twitter-reply-'));
+    const imagePath = path.join(tempDir, 'qr.png');
+    fs.writeFileSync(imagePath, Buffer.from([0x89, 0x50, 0x4e, 0x47]));
+
+    const setFileInput = vi.fn().mockResolvedValue(undefined);
+    const page = createPageMock([
+      { ok: true, previewCount: 1 },
+      { ok: true, message: 'Reply posted successfully.' },
+    ], {
+      setFileInput,
+    });
+
+    const result = await cmd!.func!(page, {
+      url: 'https://x.com/_kop6/status/2040254679301718161?s=20',
+      text: 'reply with image',
+      image: imagePath,
+    });
+
+    expect(page.goto).toHaveBeenCalledWith('https://x.com/compose/post?in_reply_to=2040254679301718161');
+    expect(page.wait).toHaveBeenNthCalledWith(1, { selector: '[data-testid="tweetTextarea_0"]' });
+    expect(page.wait).toHaveBeenNthCalledWith(2, { selector: 'input[type="file"][data-testid="fileInput"]', timeout: 20 });
+    expect(setFileInput).toHaveBeenCalledWith([imagePath], 'input[type="file"][data-testid="fileInput"]');
+    expect(result).toEqual([
+      {
+        status: 'success',
+        message: 'Reply posted successfully.',
+        text: 'reply with image',
+        image: imagePath,
+      },
+    ]);
+  });
+
+  it('downloads a remote image before uploading when --image-url is provided', async () => {
+    const cmd = getRegistry().get('twitter/reply');
+    expect(cmd?.func).toBeTypeOf('function');
+
+    const fetchMock = vi.fn().mockResolvedValue({
+      ok: true,
+      headers: {
+        get: vi.fn().mockReturnValue('image/png'),
+      },
+      arrayBuffer: vi.fn().mockResolvedValue(Uint8Array.from([0x89, 0x50, 0x4e, 0x47]).buffer),
+    });
+    vi.stubGlobal('fetch', fetchMock);
+
+    const setFileInput = vi.fn().mockResolvedValue(undefined);
+    const page = createPageMock([
+      { ok: true, previewCount: 1 },
+      { ok: true, message: 'Reply posted successfully.' },
+    ], {
+      setFileInput,
+    });
+
+    const result = await cmd!.func!(page, {
+      url: 'https://x.com/_kop6/status/2040254679301718161?s=20',
+      text: 'reply with remote image',
+      'image-url': 'https://example.com/qr',
+    });
+
+    expect(fetchMock).toHaveBeenCalledWith('https://example.com/qr');
+    expect(setFileInput).toHaveBeenCalledTimes(1);
+    const uploadedPath = setFileInput.mock.calls[0][0][0];
+    expect(uploadedPath).toMatch(/opencli-twitter-reply-.*\/image\.png$/);
+    expect(fs.existsSync(uploadedPath)).toBe(false);
+    expect(result).toEqual([
+      {
+        status: 'success',
+        message: 'Reply posted successfully.',
+        text: 'reply with remote image',
+        'image-url': 'https://example.com/qr',
+      },
+    ]);
+
+    vi.unstubAllGlobals();
+  });
+
+  it('rejects invalid image paths early', async () => {
+    await expect(() => __test__.resolveImagePath('/tmp/does-not-exist.png'))
+      .toThrow('Image file not found');
+  });
+
+  it('rejects using --image and --image-url together', async () => {
+    const cmd = getRegistry().get('twitter/reply');
+    expect(cmd?.func).toBeTypeOf('function');
+
+    const page = createPageMock([]);
+
+    await expect(cmd!.func!(page, {
+      url: 'https://x.com/_kop6/status/2040254679301718161?s=20',
+      text: 'nope',
+      image: '/tmp/a.png',
+      'image-url': 'https://example.com/a.png',
+    })).rejects.toThrow('Use either --image or --image-url, not both.');
+  });
+
+  it('extracts tweet ids from both user and i/status URLs', () => {
+    expect(__test__.extractTweetId('https://x.com/_kop6/status/2040254679301718161?s=20')).toBe('2040254679301718161');
+    expect(__test__.extractTweetId('https://x.com/i/status/2040318731105313143')).toBe('2040318731105313143');
+    expect(__test__.buildReplyComposerUrl('https://x.com/i/status/2040318731105313143'))
+      .toBe('https://x.com/compose/post?in_reply_to=2040318731105313143');
+  });
+
+  it('prefers content-type when resolving remote image extensions', () => {
+    expect(__test__.resolveImageExtension('https://example.com/no-ext', 'image/webp')).toBe('.webp');
+    expect(__test__.resolveImageExtension('https://example.com/a.jpeg?x=1', null)).toBe('.jpeg');
+  });
+});