@agentrhq/webcmd 0.1.2 → 0.2.0

package/clis/xiaohongshu/download.js

@@ -1,249 +0,0 @@
-/**
- * Xiaohongshu download — download images and videos from a note.
- *
- * Usage:
- *   webcmd xiaohongshu download <signed-note-url-or-shortlink> --output ./xhs
- *
- * Accepts a full xiaohongshu.com URL with xsec_token or an xhslink short link.
- */
-import { cli, Strategy } from '@agentrhq/webcmd/registry';
-import { formatCookieHeader } from '@agentrhq/webcmd/download';
-import { downloadMedia } from '@agentrhq/webcmd/download/media-download';
-import { CliError, CommandExecutionError, EmptyResultError } from '@agentrhq/webcmd/errors';
-import { buildNoteUrl, parseNoteId } from './note-helpers.js';
-/**
- * Build the media-extraction IIFE. The note id is interpolated as a default
- * since the IIFE may also resolve it from `location.pathname`. The CDN
- * substring allowlist includes `rednote` so the rednote adapter can reuse
- * this script unchanged — image / video URLs on both sites are served from
- * the same xhscdn family per #1136.
- */
-export function buildDownloadExtractJs(noteId) {
-    return `
-      (() => {
-        const bodyText = document.body?.innerText || '';
-        const result = {
-          noteId: '${noteId}',
-          pageUrl: location.href,
-          securityBlock: /安全限制|访问链接异常/.test(bodyText)
-            || /website-login\\/error|error_code=300017|error_code=300031/.test(location.href),
-          title: '',
-          author: '',
-          media: []
-        };
-        const seenMedia = new Set();
-        const pushMedia = (type, url) => {
-          if (!url) return;
-          const key = type + ':' + url;
-          if (seenMedia.has(key)) return;
-          seenMedia.add(key);
-          result.media.push({ type, url });
-        };
-        const locationMatch = (location.pathname || '').match(/\\/(?:explore|note|search_result|discovery\\/item)\\/([a-f0-9]+)|\\/user\\/profile\\/[^/?#]+\\/([a-f0-9]+)/i);
-        if (locationMatch) {
-          result.noteId = locationMatch[1] || locationMatch[2];
-        }
-
-        // Get title
-        const titleEl = document.querySelector('.title, #detail-title, .note-content .title');
-        result.title = titleEl?.textContent?.trim() || 'untitled';
-
-        // Get author
-        const authorEl = document.querySelector('.username, .author-name, .name');
-        result.author = authorEl?.textContent?.trim() || 'unknown';
-
-        // Get images: prefer canonical carousel order from __INITIAL_STATE__
-        // so the saved order matches what the user sees on the platform (#1514).
-        // DOM extraction is used only as a fallback because multiple selectors,
-        // hidden / duplicated / preloaded slides, and lazy rendering can reorder
-        // the discovered nodes away from the platform's display order.
-
-        const normalizeImageUrl = (raw) => {
-          if (!raw || typeof raw !== 'string') return '';
-          let src = raw.split('?')[0];
-          src = src.replace(/\\/imageView\\d+\\/\\d+\\/w\\/\\d+/, '');
-          return src;
-        };
-        const orderedImageUrls = [];
-        const seenImageUrls = new Set();
-        const pushImage = (url) => {
-          if (!url || seenImageUrls.has(url)) return;
-          seenImageUrls.add(url);
-          orderedImageUrls.push(url);
-        };
-
-        const getStructuredNotes = () => {
-          const state = window.__INITIAL_STATE__;
-          const noteData = state?.note?.noteDetailMap || state?.note?.note || {};
-          if (!noteData || typeof noteData !== 'object') return [];
-          const currentIds = [...new Set([result.noteId, '${noteId}'].filter(Boolean))];
-          const notes = [];
-          for (const id of currentIds) {
-            const entry = noteData[id];
-            const note = entry?.note || entry;
-            if (note && typeof note === 'object') notes.push(note);
-          }
-          // Compatibility fallback for legacy single-note stores. Do not use this
-          // when keyed detail maps contain multiple notes, or carousel order can
-          // be polluted by preloaded/previous note entries.
-          const keys = Object.keys(noteData);
-          if (notes.length === 0 && keys.length === 1) {
-            const entry = noteData[keys[0]];
-            const note = entry?.note || entry;
-            if (note && typeof note === 'object') notes.push(note);
-          }
-          return notes;
-        };
-
-        // Method 1: walk __INITIAL_STATE__.note.noteDetailMap[id].note.imageList
-        // in array order. Each entry exposes urlDefault as the canonical CDN URL.
-        let imageInitialStateUsed = false;
-        try {
-          for (const note of getStructuredNotes()) {
-            const list = Array.isArray(note?.imageList) ? note.imageList : [];
-            for (const item of list) {
-              const candidate = item?.urlDefault || item?.urlPre || item?.url
-                || item?.infoList?.find(i => i?.imageScene === 'WB_DFT')?.url
-                || item?.infoList?.[0]?.url
-                || '';
-              const src = normalizeImageUrl(candidate);
-              if (src && (src.includes('xhscdn') || src.includes('xiaohongshu') || src.includes('rednote'))) {
-                pushImage(src);
-                imageInitialStateUsed = true;
-              }
-            }
-          }
-        } catch(e) {}
-
-        // Method 2: fallback to DOM scraping when the structured state is missing
-        // (e.g. preview pages without full SSR hydration). Order may differ from
-        // the carousel; surface it anyway rather than returning zero images.
-        if (!imageInitialStateUsed) {
-          const imageSelectors = [
-            '.swiper-slide img',
-            '.carousel-image img',
-            '.note-slider img',
-            '.note-image img',
-            '.image-wrapper img',
-            '#noteContainer .media-container img[src*="xhscdn"]',
-            'img[src*="ci.xiaohongshu.com"]'
-          ];
-          for (const selector of imageSelectors) {
-            document.querySelectorAll(selector).forEach(img => {
-              const raw = img.src || img.getAttribute('data-src') || '';
-              const src = normalizeImageUrl(raw);
-              if (src && (src.includes('xhscdn') || src.includes('xiaohongshu') || src.includes('rednote'))) {
-                pushImage(src);
-              }
-            });
-          }
-        }
-
-        // Get video — prefer real URL from page state over blob: URLs
-
-        // Method 1: Extract from __INITIAL_STATE__ (SSR hydration data)
-        try {
-          for (const note of getStructuredNotes()) {
-            const video = note?.video;
-            if (video) {
-              const vUrl = video.url || video.originVideoKey || video.consumer?.originVideoKey;
-              if (vUrl) {
-                const fullUrl = vUrl.startsWith('http') ? vUrl : 'https://sns-video-bd.xhscdn.com/' + vUrl;
-                pushMedia('video', fullUrl);
-              }
-              const streams = video.media?.stream?.h264 || [];
-              for (const stream of streams) {
-                if (stream.masterUrl) pushMedia('video', stream.masterUrl);
-              }
-            }
-          }
-        } catch(e) {}
-
-        // Method 2: Extract video URLs from inline script JSON
-        if (result.media.filter(m => m.type === 'video').length === 0) {
-          try {
-            const scripts = document.querySelectorAll('script');
-            for (const s of scripts) {
-              const text = s.textContent || '';
-              const videoMatches = text.match(/https?:\\/\\/sns-video[^"'\\s]+\\.mp4[^"'\\s]*/g)
-                || text.match(/https?:\\/\\/[^"'\\s]*xhscdn[^"'\\s]*\\.mp4[^"'\\s]*/g);
-              if (videoMatches) {
-                videoMatches.forEach(url => {
-                  pushMedia('video', url.replace(/\\\\u002F/g, '/'));
-                });
-              }
-            }
-          } catch(e) {}
-        }
-
-        // Method 3: Fallback to DOM video elements, skip blob: URLs
-        if (result.media.filter(m => m.type === 'video').length === 0) {
-          const videoSelectors = [
-            'video source',
-            'video[src]',
-            '.player video',
-            '.video-player video'
-          ];
-          for (const selector of videoSelectors) {
-            document.querySelectorAll(selector).forEach(v => {
-              const src = v.src || v.getAttribute('src') || '';
-              if (src && !src.startsWith('blob:')) {
-                pushMedia('video', src);
-              }
-            });
-          }
-        }
-
-        // Preserve the pre-existing media type order (videos first, then images)
-        // while keeping image carousel order stable within the image batch.
-        orderedImageUrls.forEach(url => pushMedia('image', url));
-
-        return result;
-      })()
-    `;
-}
-export const command = cli({
-    site: 'xiaohongshu',
-    name: 'download',
-    access: 'read',
-    description: '下载小红书笔记中的图片和视频',
-    domain: 'www.xiaohongshu.com',
-    strategy: Strategy.COOKIE,
-    navigateBefore: false,
-    args: [
-        { name: 'note-id', positional: true, required: true, help: 'Full Xiaohongshu note URL with xsec_token, or xhslink short link' },
-        { name: 'output', default: './xiaohongshu-downloads', help: 'Output directory' },
-    ],
-    columns: ['index', 'type', 'status', 'size'],
-    func: async (page, kwargs) => {
-        const rawInput = String(kwargs['note-id']);
-        const output = kwargs.output;
-        const noteId = parseNoteId(rawInput);
-        await page.goto(buildNoteUrl(rawInput, { allowShortLink: true, commandName: 'xiaohongshu download' }));
-        await page.wait({ time: 1 + Math.random() * 2 });
-        const data = await page.evaluate(buildDownloadExtractJs(noteId));
-        if (data?.securityBlock) {
-            throw new CliError('SECURITY_BLOCK', 'Xiaohongshu security block: the note detail page was blocked by risk control.', /^https?:\/\//.test(rawInput)
-                ? 'The page may be temporarily restricted. Try again later or from a different session.'
-                : 'Try using a full URL from search results (with xsec_token) instead of a bare note ID.');
-        }
-        if (!data || typeof data !== 'object' || !Array.isArray(data.media)) {
-            throw new CommandExecutionError('Xiaohongshu media extraction returned malformed payload.');
-        }
-        if (data.media.length === 0) {
-            throw new EmptyResultError('xiaohongshu download', 'No downloadable media found on this note.');
-        }
-        // Extract cookies for authenticated downloads
-        const cookies = formatCookieHeader(await page.getCookies({ domain: 'xiaohongshu.com' }));
-        const resolvedNoteId = typeof data.noteId === 'string' && data.noteId.trim()
-            ? data.noteId.trim()
-            : noteId;
-        return downloadMedia(data.media, {
-            output,
-            subdir: resolvedNoteId,
-            cookies,
-            filenamePrefix: resolvedNoteId,
-            timeout: 60000,
-        });
-    },
-});

package/clis/xiaohongshu/download.test.js

@@ -1,339 +0,0 @@
-import { beforeEach, describe, expect, it, vi } from 'vitest';
-const { mockDownloadMedia, mockFormatCookieHeader } = vi.hoisted(() => ({
-    mockDownloadMedia: vi.fn(),
-    mockFormatCookieHeader: vi.fn(() => 'a=b'),
-}));
-vi.mock('@agentrhq/webcmd/download/media-download', () => ({
-    downloadMedia: mockDownloadMedia,
-}));
-vi.mock('@agentrhq/webcmd/download', () => ({
-    formatCookieHeader: mockFormatCookieHeader,
-}));
-import { getRegistry } from '@agentrhq/webcmd/registry';
-import { CommandExecutionError, EmptyResultError } from '@agentrhq/webcmd/errors';
-import { JSDOM } from 'jsdom';
-import './download.js';
-import { buildDownloadExtractJs } from './download.js';
-function createPageMock(evaluateResult) {
-    return {
-        goto: vi.fn().mockResolvedValue(undefined),
-        evaluate: vi.fn().mockResolvedValue(evaluateResult),
-        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([{ name: 'sid', value: 'secret', domain: '.xiaohongshu.com' }]),
-        screenshot: vi.fn().mockResolvedValue(''),
-        waitForCapture: vi.fn().mockResolvedValue(undefined),
-    };
-}
-describe('xiaohongshu download', () => {
-    const command = getRegistry().get('xiaohongshu/download');
-    beforeEach(() => {
-        mockDownloadMedia.mockReset();
-        mockFormatCookieHeader.mockClear();
-        mockDownloadMedia.mockResolvedValue([{ index: 1, type: 'video', status: 'success', size: '1 MB' }]);
-    });
-    it('preserves short links for navigation but uses canonical note id for output naming', async () => {
-        const page = createPageMock({
-            noteId: '69bc166f000000001a02069a',
-            media: [{ type: 'video', url: 'https://sns-video-hw.xhscdn.com/example.mp4' }],
-        });
-        const shortUrl = 'http://xhslink.com/o/4MKEjsZnhCz';
-        await command.func(page, { 'note-id': shortUrl, output: './out' });
-        expect(page.goto.mock.calls[0][0]).toBe(shortUrl);
-        expect(mockDownloadMedia).toHaveBeenCalledWith([{ type: 'video', url: 'https://sns-video-hw.xhscdn.com/example.mp4' }], expect.objectContaining({
-            output: './out',
-            subdir: '69bc166f000000001a02069a',
-            filenamePrefix: '69bc166f000000001a02069a',
-            cookies: 'a=b',
-        }));
-    });
-    it('preserves full note URL with xsec_token for navigation', async () => {
-        const page = createPageMock({
-            noteId: '69bc166f000000001a02069a',
-            media: [{ type: 'image', url: 'https://ci.xiaohongshu.com/example.jpg' }],
-        });
-        const fullUrl = 'https://www.xiaohongshu.com/explore/69bc166f000000001a02069a?xsec_token=abc&xsec_source=pc_search';
-        await command.func(page, { 'note-id': fullUrl, output: './out' });
-        expect(page.goto.mock.calls[0][0]).toBe(fullUrl);
-        expect(mockDownloadMedia).toHaveBeenCalledWith([{ type: 'image', url: 'https://ci.xiaohongshu.com/example.jpg' }], expect.objectContaining({
-            subdir: '69bc166f000000001a02069a',
-            filenamePrefix: '69bc166f000000001a02069a',
-        }));
-    });
-    it('uses canonical note id for signed user profile note URLs', async () => {
-        const page = createPageMock({
-            noteId: '',
-            media: [{ type: 'image', url: 'https://ci.xiaohongshu.com/example.jpg' }],
-        });
-        const fullUrl = 'https://www.xiaohongshu.com/user/profile/user123/69bc166f000000001a02069a?xsec_token=abc&xsec_source=pc_user';
-        await command.func(page, { 'note-id': fullUrl, output: './out' });
-        expect(page.goto.mock.calls[0][0]).toBe(fullUrl);
-        expect(mockDownloadMedia).toHaveBeenCalledWith([{ type: 'image', url: 'https://ci.xiaohongshu.com/example.jpg' }], expect.objectContaining({
-            subdir: '69bc166f000000001a02069a',
-            filenamePrefix: '69bc166f000000001a02069a',
-        }));
-    });
-    it('rejects bare note IDs before browser navigation', async () => {
-        const page = createPageMock({
-            noteId: '69bc166f000000001a02069a',
-            media: [],
-        });
-        await expect(command.func(page, { 'note-id': '69bc166f000000001a02069a', output: './out' })).rejects.toMatchObject({
-            code: 'ARGUMENT',
-            message: expect.stringContaining('signed URL'),
-            hint: expect.stringContaining('xsec_token'),
-        });
-        expect(page.goto).not.toHaveBeenCalled();
-        expect(mockDownloadMedia).not.toHaveBeenCalled();
-    });
-    it('throws SECURITY_BLOCK with retry guidance for blocked full URLs', async () => {
-        const page = createPageMock({
-            pageUrl: 'https://www.xiaohongshu.com/website-login/error?error_code=300031',
-            securityBlock: true,
-            noteId: '69bc166f000000001a02069a',
-            media: [],
-        });
-        await expect(command.func(page, {
-            'note-id': 'https://www.xiaohongshu.com/explore/69bc166f000000001a02069a?xsec_token=abc&xsec_source=pc_search',
-            output: './out',
-        })).rejects.toMatchObject({
-            code: 'SECURITY_BLOCK',
-            hint: expect.stringContaining('Try again later'),
-        });
-        expect(mockDownloadMedia).not.toHaveBeenCalled();
-    });
-    it('throws EmptyResultError when extraction returns an explicit empty media array', async () => {
-        const page = createPageMock({
-            noteId: '69bc166f000000001a02069a',
-            media: [],
-        });
-        await expect(command.func(page, {
-            'note-id': 'https://www.xiaohongshu.com/explore/69bc166f000000001a02069a?xsec_token=abc',
-            output: './out',
-        })).rejects.toBeInstanceOf(EmptyResultError);
-        expect(mockDownloadMedia).not.toHaveBeenCalled();
-    });
-    it('throws CommandExecutionError when extraction media shape is malformed', async () => {
-        const page = createPageMock({
-            noteId: '69bc166f000000001a02069a',
-            media: { url: 'https://ci.xiaohongshu.com/example.jpg' },
-        });
-        await expect(command.func(page, {
-            'note-id': 'https://www.xiaohongshu.com/explore/69bc166f000000001a02069a?xsec_token=abc',
-            output: './out',
-        })).rejects.toBeInstanceOf(CommandExecutionError);
-        expect(mockDownloadMedia).not.toHaveBeenCalled();
-    });
-});
-
-describe('xiaohongshu download buildDownloadExtractJs carousel ordering (JSDOM)', () => {
-    function runExtract({ html = '', initialState = null, url = 'https://www.xiaohongshu.com/explore/69f9716c000000003601f90e' } = {}) {
-        const dom = new JSDOM(`<!doctype html><html><body>${html}</body></html>`, { url, runScripts: 'outside-only' });
-        if (initialState) {
-            dom.window.__INITIAL_STATE__ = initialState;
-        }
-        const js = buildDownloadExtractJs('69f9716c000000003601f90e');
-        return dom.window.eval(js);
-    }
-
-    it('preserves carousel order from __INITIAL_STATE__ imageList over DOM discovery order', () => {
-        const initialState = {
-            note: {
-                noteDetailMap: {
-                    '69f9716c000000003601f90e': {
-                        note: {
-                            imageList: [
-                                { urlDefault: 'https://sns-img-bd.xhscdn.com/canonical-cover.jpg' },
-                                { urlDefault: 'https://sns-img-bd.xhscdn.com/canonical-second.jpg' },
-                                { urlDefault: 'https://sns-img-bd.xhscdn.com/canonical-third.jpg' },
-                            ],
-                        },
-                    },
-                },
-            },
-        };
-        // DOM has the same images but in a DIFFERENT order: repro for #1514.
-        const html = `
-          <div class="swiper-slide"><img src="https://sns-img-bd.xhscdn.com/canonical-second.jpg" /></div>
-          <div class="swiper-slide"><img src="https://sns-img-bd.xhscdn.com/canonical-cover.jpg" /></div>
-          <div class="swiper-slide"><img src="https://sns-img-bd.xhscdn.com/canonical-third.jpg" /></div>
-        `;
-        const result = runExtract({ html, initialState });
-        const images = result.media.filter((m) => m.type === 'image');
-        expect(images.map((m) => m.url)).toEqual([
-            'https://sns-img-bd.xhscdn.com/canonical-cover.jpg',
-            'https://sns-img-bd.xhscdn.com/canonical-second.jpg',
-            'https://sns-img-bd.xhscdn.com/canonical-third.jpg',
-        ]);
-    });
-
-    it('prefers urlDefault but falls back to urlPre / url / infoList.WB_DFT / infoList[0]', () => {
-        const initialState = {
-            note: {
-                noteDetailMap: {
-                    '69f9716c000000003601f90e': {
-                        note: {
-                            imageList: [
-                                { urlDefault: 'https://sns-img-bd.xhscdn.com/a.jpg' },
-                                { urlPre: 'https://sns-img-bd.xhscdn.com/b.jpg' },
-                                { url: 'https://sns-img-bd.xhscdn.com/c.jpg' },
-                                { infoList: [{ imageScene: 'WB_PRV', url: 'https://sns-img-bd.xhscdn.com/d-low.jpg' }, { imageScene: 'WB_DFT', url: 'https://sns-img-bd.xhscdn.com/d.jpg' }] },
-                                { infoList: [{ url: 'https://sns-img-bd.xhscdn.com/e.jpg' }] },
-                            ],
-                        },
-                    },
-                },
-            },
-        };
-        const result = runExtract({ initialState });
-        const urls = result.media.filter((m) => m.type === 'image').map((m) => m.url);
-        expect(urls).toEqual([
-            'https://sns-img-bd.xhscdn.com/a.jpg',
-            'https://sns-img-bd.xhscdn.com/b.jpg',
-            'https://sns-img-bd.xhscdn.com/c.jpg',
-            'https://sns-img-bd.xhscdn.com/d.jpg',
-            'https://sns-img-bd.xhscdn.com/e.jpg',
-        ]);
-    });
-
-    it('strips imageView resize params + query strings from canonical urls', () => {
-        const initialState = {
-            note: {
-                noteDetailMap: {
-                    '69f9716c000000003601f90e': {
-                        note: {
-                            imageList: [
-                                { urlDefault: 'https://sns-img-bd.xhscdn.com/raw/imageView2/2/w/1080/cover.jpg?expires=123' },
-                            ],
-                        },
-                    },
-                },
-            },
-        };
-        const result = runExtract({ initialState });
-        const urls = result.media.filter((m) => m.type === 'image').map((m) => m.url);
-        expect(urls).toEqual(['https://sns-img-bd.xhscdn.com/raw/cover.jpg']);
-    });
-
-    it('falls back to DOM extraction when __INITIAL_STATE__ omits imageList', () => {
-        const html = `
-          <div class="swiper-slide"><img src="https://sns-img-bd.xhscdn.com/dom-1.jpg" /></div>
-          <div class="swiper-slide"><img src="https://sns-img-bd.xhscdn.com/dom-2.jpg" /></div>
-        `;
-        const result = runExtract({ html });
-        const urls = result.media.filter((m) => m.type === 'image').map((m) => m.url);
-        expect(urls).toEqual([
-            'https://sns-img-bd.xhscdn.com/dom-1.jpg',
-            'https://sns-img-bd.xhscdn.com/dom-2.jpg',
-        ]);
-    });
-
-    it('skips non-xhscdn / non-xiaohongshu / non-rednote urls in initial state', () => {
-        const initialState = {
-            note: {
-                noteDetailMap: {
-                    '69f9716c000000003601f90e': {
-                        note: {
-                            imageList: [
-                                { urlDefault: 'https://sns-img-bd.xhscdn.com/keep.jpg' },
-                                { urlDefault: 'https://imgur.com/drop.jpg' },
-                                { urlDefault: '' },
-                            ],
-                        },
-                    },
-                },
-            },
-        };
-        const result = runExtract({ initialState });
-        const urls = result.media.filter((m) => m.type === 'image').map((m) => m.url);
-        expect(urls).toEqual(['https://sns-img-bd.xhscdn.com/keep.jpg']);
-    });
-
-    it('does not run DOM fallback when initial state yielded any image (preserves canonical order)', () => {
-        const initialState = {
-            note: {
-                noteDetailMap: {
-                    '69f9716c000000003601f90e': {
-                        note: {
-                            imageList: [
-                                { urlDefault: 'https://sns-img-bd.xhscdn.com/canonical-only.jpg' },
-                            ],
-                        },
-                    },
-                },
-            },
-        };
-        const html = `
-          <div class="swiper-slide"><img src="https://sns-img-bd.xhscdn.com/dom-extra.jpg" /></div>
-        `;
-        const result = runExtract({ html, initialState });
-        const urls = result.media.filter((m) => m.type === 'image').map((m) => m.url);
-        expect(urls).toEqual(['https://sns-img-bd.xhscdn.com/canonical-only.jpg']);
-    });
-
-    it('uses only the current note entry from multi-note initial state maps', () => {
-        const initialState = {
-            note: {
-                noteDetailMap: {
-                    othernote0000000000000001: {
-                        note: {
-                            imageList: [{ urlDefault: 'https://sns-img-bd.xhscdn.com/other.jpg' }],
-                            video: { url: 'https://sns-video-bd.xhscdn.com/other.mp4' },
-                        },
-                    },
-                    '69f9716c000000003601f90e': {
-                        note: {
-                            imageList: [
-                                { urlDefault: 'https://sns-img-bd.xhscdn.com/current-1.jpg' },
-                                { urlDefault: 'https://sns-img-bd.xhscdn.com/current-2.jpg' },
-                            ],
-                            video: { url: 'https://sns-video-bd.xhscdn.com/current.mp4' },
-                        },
-                    },
-                },
-            },
-        };
-        const result = runExtract({ initialState });
-        const images = result.media.filter((m) => m.type === 'image').map((m) => m.url);
-        const videos = result.media.filter((m) => m.type === 'video').map((m) => m.url);
-        expect(images).toEqual([
-            'https://sns-img-bd.xhscdn.com/current-1.jpg',
-            'https://sns-img-bd.xhscdn.com/current-2.jpg',
-        ]);
-        expect(videos).toEqual(['https://sns-video-bd.xhscdn.com/current.mp4']);
-    });
-
-    it('still resolves videos from __INITIAL_STATE__ alongside the image fix', () => {
-        const initialState = {
-            note: {
-                noteDetailMap: {
-                    '69f9716c000000003601f90e': {
-                        note: {
-                            imageList: [{ urlDefault: 'https://sns-img-bd.xhscdn.com/cover.jpg' }],
-                            video: { url: 'https://sns-video-bd.xhscdn.com/test.mp4' },
-                        },
-                    },
-                },
-            },
-        };
-        const result = runExtract({ initialState });
-        const images = result.media.filter((m) => m.type === 'image').map((m) => m.url);
-        const videos = result.media.filter((m) => m.type === 'video').map((m) => m.url);
-        expect(images).toEqual(['https://sns-img-bd.xhscdn.com/cover.jpg']);
-        expect(videos).toEqual(['https://sns-video-bd.xhscdn.com/test.mp4']);
-        expect(result.media.map((m) => m.type)).toEqual(['video', 'image']);
-    });
-});

package/clis/xiaohongshu/draft-clear.js

@@ -1,91 +0,0 @@
-import { CommandExecutionError } from '@agentrhq/webcmd/errors';
-import { cli, Strategy } from '@agentrhq/webcmd/registry';
-import {
-    DRAFT_DB_NAME,
-    STORE_NAME_MAP,
-    ensureDraftDbPage,
-    normalizeDraftType,
-    unwrapBrowserResult,
-} from './draft-utils.js';
-
-function clearDraftsScript(storeNames, execute) {
-    return `
-    (async () => {
-      const openDb = () => new Promise((resolve, reject) => {
-        const req = indexedDB.open(${JSON.stringify(DRAFT_DB_NAME)});
-        req.onsuccess = () => resolve(req.result);
-        req.onerror = () => reject(req.error || new Error('failed to open IndexedDB'));
-      });
-      const countStore = (db, storeName) => new Promise((resolve, reject) => {
-        if (!db.objectStoreNames.contains(storeName)) return resolve(0);
-        const tx = db.transaction(storeName, 'readonly');
-        const store = tx.objectStore(storeName);
-        const req = store.count();
-        req.onsuccess = () => resolve(req.result || 0);
-        req.onerror = () => reject(req.error || new Error('failed to count drafts'));
-        tx.onerror = () => reject(tx.error || new Error('failed to count drafts'));
-      });
-      const clearStore = (db, storeName) => new Promise((resolve, reject) => {
-        if (!db.objectStoreNames.contains(storeName)) return resolve(true);
-        const tx = db.transaction(storeName, 'readwrite');
-        const store = tx.objectStore(storeName);
-        const req = store.clear();
-        req.onsuccess = () => resolve(true);
-        req.onerror = () => reject(req.error || new Error('failed to clear drafts'));
-        tx.onerror = () => reject(tx.error || new Error('failed to clear drafts'));
-      });
-      try {
-        const db = await openDb();
-        const storeNames = ${JSON.stringify(storeNames)};
-        let before = 0;
-        for (const storeName of storeNames) before += await countStore(db, storeName);
-        if (${JSON.stringify(execute)}) {
-          for (const storeName of storeNames) await clearStore(db, storeName);
-          let after = 0;
-          for (const storeName of storeNames) after += await countStore(db, storeName);
-          db.close();
-          return { ok: true, before, after, cleared: before - after };
-        }
-        db.close();
-        return { ok: true, before, after: before, cleared: 0 };
-      } catch (error) {
-        return { ok: false, error: String(error && error.message || error) };
-      }
-    })()
-  `;
-}
-
-cli({
-    site: 'xiaohongshu',
-    name: 'draft-clear',
-    access: 'write',
-    description: '清空小红书本地草稿',
-    domain: 'creator.xiaohongshu.com',
-    strategy: Strategy.COOKIE,
-    browser: true,
-    navigateBefore: false,
-    args: [
-        { name: 'type', default: 'image', help: 'Draft type: image, video, article, audio, all' },
-        { name: 'execute', type: 'bool', default: false, help: 'Actually clear local drafts. Default is dry-run count only.' },
-    ],
-    columns: ['status', 'type', 'count', 'message'],
-    func: async (page, kwargs) => {
-        const draftType = normalizeDraftType(kwargs.type, { allowAll: true });
-        const execute = kwargs.execute === true;
-        const storeNames = draftType === 'all' ? Object.values(STORE_NAME_MAP) : [STORE_NAME_MAP[draftType]];
-        await ensureDraftDbPage(page);
-        const result = unwrapBrowserResult(await page.evaluate(clearDraftsScript(storeNames, execute)));
-        if (!result?.ok) {
-            throw new CommandExecutionError(result?.error || `Failed to ${execute ? 'clear' : 'count'} ${draftType} drafts`);
-        }
-        if (execute && Number(result.after) !== 0) {
-            throw new CommandExecutionError(`${result.after} ${draftType} drafts still exist after clear`);
-        }
-        return [{
-            status: execute ? 'cleared' : 'dry-run',
-            type: draftType,
-            count: execute ? Number(result.cleared || 0) : Number(result.before || 0),
-            message: execute ? 'Drafts cleared.' : 'Drafts counted. Re-run with --execute to clear.',
-        }];
-    },
-});