@agentrhq/webcmd 0.1.2 → 0.2.0

package/clis/bilibili/download.js

@@ -1,178 +0,0 @@
-/**
- * Bilibili download — download videos using yt-dlp.
- *
- * Usage:
- *   webcmd bilibili download --bvid BV1xxx --output ./bilibili
- *
- * Requirements:
- *   - yt-dlp must be installed: pip install yt-dlp
- */
-import { cli, Strategy } from '@agentrhq/webcmd/registry';
-import { CliError, CommandExecutionError, EXIT_CODES } from '@agentrhq/webcmd/errors';
-import { checkYtdlp, sanitizeFilename } from '@agentrhq/webcmd/download';
-import { downloadMedia } from '@agentrhq/webcmd/download/media-download';
-import { apiGet, resolveBvid, parsePageArg, selectVideoPart } from './utils.js';
-
-const PAYMENT_LABELS = {
-    vip: '大会员专享/付费 OGV',
-    ugc_pay: 'UGC 单点付费',
-    upower: '充电专属',
-};
-
-function isObject(value) {
-    return value && typeof value === 'object' && !Array.isArray(value);
-}
-
-/**
- * 下载前付费预检：付费/会员视频 yt-dlp 只能拿到试看流或直接失败，
- * 与其跑一半吐一坨 yt-dlp stderr，不如提前抛结构化 PAID_CONTENT（exit 77）。
- *
- * 大会员专享（vip）会再查一次 nav API：当前账号大会员有效就放行（cookie 喂给
- * yt-dlp 能下完整流）。ugc_pay / upower 的购买/充电状态没有廉价查询端点，保守
- * 拦截，已购用户用 --force 跳过。预检自身的 API 失败不阻塞下载（保持旧行为）。
- */
-async function assertNotPaidContent(page, bvid) {
-    let d;
-    try {
-        const payload = await apiGet(page, '/x/web-interface/view', { params: { bvid } });
-        if (!isObject(payload) || !Object.hasOwn(payload, 'code')) {
-            throw new CommandExecutionError('Bilibili view API returned a malformed payload during paid-content pre-check');
-        }
-        if (payload.code !== 0)
-            return;
-        if (!isObject(payload.data) || !isObject(payload.data.rights)) {
-            throw new CommandExecutionError('Bilibili view API returned malformed paid-content metadata');
-        }
-        d = payload.data;
-    }
-    catch (error) {
-        if (error instanceof CommandExecutionError) {
-            throw error;
-        }
-        return;
-    }
-    const rights = d.rights;
-    const paymentType = rights.pay
-        ? 'vip'
-        : (rights.ugc_pay || rights.arc_pay)
-            ? 'ugc_pay'
-            : d.is_upower_exclusive
-                ? 'upower'
-                : '';
-    if (!paymentType)
-        return;
-    if (paymentType === 'vip') {
-        try {
-            const nav = await apiGet(page, '/x/web-interface/nav');
-            if (nav.code === 0 && Number(nav.data?.vipStatus) === 1)
-                return;
-        }
-        catch {
-            // nav 查询失败按"无会员"保守处理，走下面的拦截
-        }
-    }
-    throw new CliError(
-        'PAID_CONTENT',
-        `该视频为付费内容（${PAYMENT_LABELS[paymentType]}），当前账号无观看权益，无法获取完整视频流`,
-        '若已购买/已充电/已开通会员，加 --force 跳过本检查直接下载',
-        EXIT_CODES.NOPERM,
-    );
-}
-
-async function loadSelectedPart(page, bvid, pageNum) {
-    let payload;
-    try {
-        payload = await apiGet(page, '/x/web-interface/view', { params: { bvid } });
-    }
-    catch (error) {
-        throw new CommandExecutionError(`获取视频分P信息失败: ${error?.message || error}`);
-    }
-    if (!isObject(payload) || payload.code !== 0) {
-        throw new CommandExecutionError(`获取视频分P信息失败: ${payload?.message ?? 'unknown'} (${payload?.code ?? 'malformed'})`);
-    }
-    return selectVideoPart(payload.data, pageNum);
-}
-
-cli({
-    site: 'bilibili',
-    name: 'download',
-    access: 'read',
-    description: '下载B站视频（需要 yt-dlp）',
-    domain: 'www.bilibili.com',
-    strategy: Strategy.COOKIE,
-    args: [
-        { name: 'bvid', required: true, positional: true, help: 'Video BV ID (e.g., BV1xxx)' },
-        { name: 'output', default: './bilibili-downloads', help: 'Output directory' },
-        { name: 'quality', default: 'best', help: 'Video quality (best, 1080p, 720p, 480p)' },
-        { name: 'force', type: 'boolean', default: false, help: '跳过付费内容预检直接下载（已购买/已充电/已开通会员时用）' },
-        { name: 'page', required: false, help: '分P 选集序号（从 1 开始）。多 P 视频下载该集；缺省下载默认 P1' },
-    ],
-    columns: ['bvid', 'title', 'status', 'size'],
-    func: async (page, kwargs) => {
-        const bvid = await resolveBvid(kwargs.bvid);
-        const output = kwargs.output;
-        const quality = kwargs.quality;
-        const selectedPage = parsePageArg(kwargs.page);
-        const selectedPart = selectedPage != null ? await loadSelectedPart(page, bvid, selectedPage) : null;
-        // yt-dlp 原生支持分P URL（?p=N），直接拼到 watch URL 即可定位到该集。
-        const watchUrl = selectedPage != null
-            ? `https://www.bilibili.com/video/${bvid}?p=${selectedPage}`
-            : `https://www.bilibili.com/video/${bvid}`;
-        // Check yt-dlp availability
-        if (!checkYtdlp()) {
-            return [{
-                    bvid,
-                    title: '-',
-                    status: 'failed',
-                    size: 'yt-dlp not installed. Run: pip install yt-dlp',
-                }];
-        }
-        // Navigate to video page to get title and cookies（分P 时定位到该集）
-        await page.goto(watchUrl);
-        await page.wait(3);
-        // 付费内容预检（--force 跳过）
-        if (!kwargs.force) {
-            await assertNotPaidContent(page, bvid);
-        }
-        // Extract video info
-        const data = await page.evaluate(`
-      (() => {
-        const title = document.querySelector('h1.video-title, .video-title')?.textContent?.trim() || 'video';
-        const author = document.querySelector('.up-name, .username')?.textContent?.trim() || 'unknown';
-        return { title, author };
-      })()
-    `);
-        const partTitle = typeof selectedPart?.part === 'string' ? selectedPart.part.trim() : '';
-        const displayTitle = partTitle || data?.title || 'video';
-        const title = sanitizeFilename(displayTitle);
-        // Extract cookies for yt-dlp
-        const browserCookies = await page.getCookies({ domain: 'bilibili.com' });
-        // Build yt-dlp format string based on quality
-        let format = 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best';
-        if (quality === '1080p') {
-            format = 'bestvideo[height<=1080][ext=mp4]+bestaudio[ext=m4a]/best[height<=1080]';
-        }
-        else if (quality === '720p') {
-            format = 'bestvideo[height<=720][ext=mp4]+bestaudio[ext=m4a]/best[height<=720]';
-        }
-        else if (quality === '480p') {
-            format = 'bestvideo[height<=480][ext=mp4]+bestaudio[ext=m4a]/best[height<=480]';
-        }
-        const videoUrl = watchUrl;
-        const filename = selectedPage != null ? `${bvid}_p${selectedPage}_${title}.mp4` : `${bvid}_${title}.mp4`;
-        const results = await downloadMedia([{ type: 'video-ytdlp', url: videoUrl, filename }], {
-            output,
-            browserCookies,
-            filenamePrefix: bvid,
-            ytdlpExtraArgs: ['-f', format, '--merge-output-format', 'mp4', '--embed-thumbnail'],
-        });
-        // Map results to bilibili-specific columns
-        const r = results[0] || { status: 'failed', size: '-' };
-        return [{
-                bvid,
-                title: displayTitle,
-                status: r.status,
-                size: r.size,
-            }];
-    },
-});

package/clis/bilibili/download.test.js

@@ -1,173 +0,0 @@
-import { beforeEach, describe, expect, it, vi } from 'vitest';
-import { ArgumentError, CliError, CommandExecutionError } from '@agentrhq/webcmd/errors';
-
-const { mockApiGet, mockDownloadMedia, mockCheckYtdlp } = vi.hoisted(() => ({
-  mockApiGet: vi.fn(),
-  mockDownloadMedia: vi.fn(),
-  mockCheckYtdlp: vi.fn(),
-}));
-
-vi.mock('./utils.js', async (importOriginal) => ({
-  ...(await importOriginal()),
-  apiGet: mockApiGet,
-}));
-
-vi.mock('@agentrhq/webcmd/download', () => ({
-  checkYtdlp: mockCheckYtdlp,
-  sanitizeFilename: (s) => s,
-}));
-
-vi.mock('@agentrhq/webcmd/download/media-download', () => ({
-  downloadMedia: mockDownloadMedia,
-}));
-
-import { getRegistry } from '@agentrhq/webcmd/registry';
-import './download.js';
-
-/** view API 成功响应的最小骨架 */
-function viewPayload(extra = {}) {
-  return { code: 0, data: { bvid: 'BV1xx411c7mD', rights: {}, ...extra } };
-}
-
-describe('bilibili download paid-content pre-check', () => {
-  const command = getRegistry().get('bilibili/download');
-  const page = {
-    goto: vi.fn().mockResolvedValue(undefined),
-    wait: vi.fn().mockResolvedValue(undefined),
-    evaluate: vi.fn().mockResolvedValue({ title: '标题', author: 'UP主' }),
-    getCookies: vi.fn().mockResolvedValue([]),
-  };
-
-  beforeEach(() => {
-    mockApiGet.mockReset();
-    mockDownloadMedia.mockReset();
-    mockCheckYtdlp.mockReset();
-    mockCheckYtdlp.mockReturnValue(true);
-    mockDownloadMedia.mockResolvedValue([{ status: 'success', size: '10MB' }]);
-    page.goto.mockClear();
-    page.evaluate.mockClear();
-  });
-
-  it('downloads normal (free) video without interference', async () => {
-    mockApiGet.mockResolvedValueOnce(viewPayload());
-
-    const rows = await command.func(page, { bvid: 'BV1xx411c7mD', output: './o', quality: 'best', force: false });
-
-    expect(rows[0].status).toBe('success');
-    expect(mockDownloadMedia).toHaveBeenCalledTimes(1);
-  });
-
-  it('throws PAID_CONTENT for member-only bangumi when account has no vip', async () => {
-    mockApiGet
-      .mockResolvedValueOnce(viewPayload({ rights: { pay: 1 } })) // view
-      .mockResolvedValueOnce({ code: 0, data: { vipStatus: 0 } }); // nav
-
-    await expect(
-      command.func(page, { bvid: 'BV1xx411c7mD', output: './o', quality: 'best', force: false }),
-    ).rejects.toSatisfy((err) => err instanceof CliError && err.code === 'PAID_CONTENT');
-    expect(mockDownloadMedia).not.toHaveBeenCalled();
-  });
-
-  it('allows member-only content when account has active vip', async () => {
-    mockApiGet
-      .mockResolvedValueOnce(viewPayload({ rights: { pay: 1 } }))
-      .mockResolvedValueOnce({ code: 0, data: { vipStatus: 1 } });
-
-    const rows = await command.func(page, { bvid: 'BV1xx411c7mD', output: './o', quality: 'best', force: false });
-
-    expect(rows[0].status).toBe('success');
-    expect(mockDownloadMedia).toHaveBeenCalledTimes(1);
-  });
-
-  it('throws PAID_CONTENT for upower-exclusive video (no entitlement endpoint, conservative block)', async () => {
-    mockApiGet.mockResolvedValueOnce(viewPayload({ is_upower_exclusive: true }));
-
-    await expect(
-      command.func(page, { bvid: 'BV1xx411c7mD', output: './o', quality: 'best', force: false }),
-    ).rejects.toSatisfy((err) => err instanceof CliError && err.code === 'PAID_CONTENT');
-    // upower 没有权益查询端点，不应再打 nav API
-    expect(mockApiGet).toHaveBeenCalledTimes(1);
-    expect(mockDownloadMedia).not.toHaveBeenCalled();
-  });
-
-  it('fails closed when successful view payload lacks paid-content metadata', async () => {
-    mockApiGet.mockResolvedValueOnce({ code: 0, data: { bvid: 'BV1xx411c7mD' } });
-
-    await expect(
-      command.func(page, { bvid: 'BV1xx411c7mD', output: './o', quality: 'best', force: false }),
-    ).rejects.toSatisfy(
-      (err) => err instanceof CommandExecutionError && /paid-content metadata/.test(err.message),
-    );
-    expect(mockDownloadMedia).not.toHaveBeenCalled();
-  });
-
-  it('skips pre-check entirely with --force', async () => {
-    const rows = await command.func(page, { bvid: 'BV1xx411c7mD', output: './o', quality: 'best', force: true });
-
-    expect(rows[0].status).toBe('success');
-    expect(mockApiGet).not.toHaveBeenCalled();
-    expect(mockDownloadMedia).toHaveBeenCalledTimes(1);
-  });
-
-  it('does not block download when the pre-check API itself fails', async () => {
-    mockApiGet.mockRejectedValueOnce(new Error('network down'));
-
-    const rows = await command.func(page, { bvid: 'BV1xx411c7mD', output: './o', quality: 'best', force: false });
-
-    expect(rows[0].status).toBe('success');
-    expect(mockDownloadMedia).toHaveBeenCalledTimes(1);
-  });
-
-  it('targets the selected 分P part URL (?p=N) when --page is given', async () => {
-    mockApiGet
-      .mockResolvedValueOnce(viewPayload({
-        pages: [
-          { cid: 1001, page: 1, part: 'P1' },
-          { cid: 1003, page: 3, part: 'P3 标题' },
-        ],
-      }))
-      .mockResolvedValueOnce(viewPayload());
-
-    await command.func(page, { bvid: 'BV1h6V16SEpg', output: './o', quality: 'best', force: false, page: '3' });
-
-    // goto 与 yt-dlp 下载 URL 都应带 ?p=3
-    expect(page.goto).toHaveBeenCalledWith('https://www.bilibili.com/video/BV1h6V16SEpg?p=3');
-    const job = mockDownloadMedia.mock.calls[0][0][0];
-    expect(job.url).toBe('https://www.bilibili.com/video/BV1h6V16SEpg?p=3');
-    expect(job.filename).toContain('_p3_P3 标题');
-  });
-
-  it('rejects malformed --page before download side effects', async () => {
-    await expect(
-      command.func(page, { bvid: 'BV1h6V16SEpg', output: './o', quality: 'best', force: false, page: '1e2' }),
-    ).rejects.toBeInstanceOf(ArgumentError);
-
-    expect(page.goto).not.toHaveBeenCalled();
-    expect(mockApiGet).not.toHaveBeenCalled();
-    expect(mockDownloadMedia).not.toHaveBeenCalled();
-  });
-
-  it('fails before yt-dlp when selected --page is absent from view API pages', async () => {
-    mockApiGet.mockResolvedValueOnce(viewPayload({
-      pages: [{ cid: 1001, page: 1, part: 'P1' }],
-    }));
-
-    await expect(
-      command.func(page, { bvid: 'BV1h6V16SEpg', output: './o', quality: 'best', force: false, page: '9' }),
-    ).rejects.toBeInstanceOf(CommandExecutionError);
-
-    expect(page.goto).not.toHaveBeenCalled();
-    expect(mockDownloadMedia).not.toHaveBeenCalled();
-  });
-
-  it('downloads default P1 (no ?p=) when --page is omitted', async () => {
-    mockApiGet.mockResolvedValueOnce(viewPayload());
-
-    await command.func(page, { bvid: 'BV1xx411c7mD', output: './o', quality: 'best', force: false });
-
-    expect(page.goto).toHaveBeenCalledWith('https://www.bilibili.com/video/BV1xx411c7mD');
-    const job = mockDownloadMedia.mock.calls[0][0][0];
-    expect(job.url).toBe('https://www.bilibili.com/video/BV1xx411c7mD');
-    expect(job.filename).not.toContain('_p');
-  });
-});

package/clis/bilibili/dynamic.js

@@ -1,34 +0,0 @@
-import { cli, Strategy } from '@agentrhq/webcmd/registry';
-import { apiGet } from './utils.js';
-cli({
-    site: 'bilibili',
-    name: 'dynamic',
-    access: 'read',
-    description: 'Get Bilibili user dynamic feed',
-    domain: 'www.bilibili.com',
-    strategy: Strategy.COOKIE,
-    args: [
-        { name: 'limit', type: 'int', default: 15 },
-    ],
-    columns: ['id', 'author', 'text', 'likes', 'url'],
-    func: async (page, kwargs) => {
-        const payload = await apiGet(page, '/x/polymer/web-dynamic/v1/feed/all', { params: {}, signed: false });
-        const results = payload?.data?.items ?? [];
-        return results.slice(0, Number(kwargs.limit)).map((item) => {
-            let text = '';
-            if (item.modules?.module_dynamic?.desc?.text) {
-                text = item.modules.module_dynamic.desc.text;
-            }
-            else if (item.modules?.module_dynamic?.major?.archive?.title) {
-                text = item.modules.module_dynamic.major.archive.title;
-            }
-            return {
-                id: item.id_str ?? '',
-                author: item.modules?.module_author?.name ?? '',
-                text: text,
-                likes: item.modules?.module_stat?.like?.count ?? 0,
-                url: item.id_str ? `https://t.bilibili.com/${item.id_str}` : ''
-            };
-        });
-    },
-});

package/clis/bilibili/dynamic.test.js

@@ -1,68 +0,0 @@
-import { beforeEach, describe, expect, it, vi } from 'vitest';
-const { mockApiGet } = vi.hoisted(() => ({
-    mockApiGet: vi.fn(),
-}));
-vi.mock('./utils.js', () => ({
-    apiGet: mockApiGet,
-}));
-import { getRegistry } from '@agentrhq/webcmd/registry';
-import './dynamic.js';
-describe('bilibili dynamic adapter', () => {
-    const command = getRegistry().get('bilibili/dynamic');
-    beforeEach(() => {
-        mockApiGet.mockReset();
-    });
-    it('maps desc text rows from the dynamic feed payload', async () => {
-        mockApiGet.mockResolvedValue({
-            data: {
-                items: [
-                    {
-                        id_str: '123',
-                        modules: {
-                            module_author: { name: 'Alice' },
-                            module_dynamic: { desc: { text: 'hello world' } },
-                            module_stat: { like: { count: 9 } },
-                        },
-                    },
-                ],
-            },
-        });
-        const result = await command.func({}, { limit: 5 });
-        expect(mockApiGet).toHaveBeenCalledWith({}, '/x/polymer/web-dynamic/v1/feed/all', { params: {}, signed: false });
-        expect(result).toEqual([
-            {
-                id: '123',
-                author: 'Alice',
-                text: 'hello world',
-                likes: 9,
-                url: 'https://t.bilibili.com/123',
-            },
-        ]);
-    });
-    it('falls back to archive title when desc text is absent', async () => {
-        mockApiGet.mockResolvedValue({
-            data: {
-                items: [
-                    {
-                        id_str: '456',
-                        modules: {
-                            module_author: { name: 'Bob' },
-                            module_dynamic: { major: { archive: { title: 'Video title' } } },
-                            module_stat: { like: { count: 3 } },
-                        },
-                    },
-                ],
-            },
-        });
-        const result = await command.func({}, { limit: 5 });
-        expect(result).toEqual([
-            {
-                id: '456',
-                author: 'Bob',
-                text: 'Video title',
-                likes: 3,
-                url: 'https://t.bilibili.com/456',
-            },
-        ]);
-    });
-});

package/clis/bilibili/favorite.js

@@ -1,47 +0,0 @@
-import { cli, Strategy } from '@agentrhq/webcmd/registry';
-import { apiGet, payloadData, getSelfUid } from './utils.js';
-cli({
-    site: 'bilibili',
-    name: 'favorite',
-    access: 'write',
-    description: '我的收藏夹',
-    domain: 'www.bilibili.com',
-    strategy: Strategy.COOKIE,
-    args: [
-        { name: 'fid', type: 'int', required: false, help: 'Favorite folder ID (defaults to first folder)' },
-        { name: 'limit', type: 'int', default: 20, help: 'Number of results' },
-        { name: 'page', type: 'int', default: 1, help: 'Page number' },
-    ],
-    columns: ['rank', 'title', 'author', 'plays', 'url'],
-    func: async (page, kwargs) => {
-        const { fid: favoriteId, limit = 20, page: pageNum = 1 } = kwargs;
-        let fid;
-        if (favoriteId) {
-            fid = Number(favoriteId);
-        } else {
-            // Fall back to the default (first) favorite folder
-            const uid = await getSelfUid(page);
-            const foldersPayload = await apiGet(page, '/x/v3/fav/folder/created/list-all', {
-                params: { up_mid: uid },
-                signed: true,
-            });
-            const folders = payloadData(foldersPayload)?.list ?? [];
-            if (!folders.length)
-                return [];
-            fid = folders[0].id;
-        }
-        // Fetch favorite items
-        const payload = await apiGet(page, '/x/v3/fav/resource/list', {
-            params: { media_id: fid, pn: pageNum, ps: Math.min(Number(limit), 40) },
-            signed: true,
-        });
-        const medias = payloadData(payload)?.medias ?? [];
-        return medias.slice(0, Number(limit)).map((item, i) => ({
-            rank: i + 1,
-            title: item.title ?? '',
-            author: item.upper?.name ?? '',
-            plays: item.cnt_info?.play ?? 0,
-            url: item.bvid ? `https://www.bilibili.com/video/${item.bvid}` : '',
-        }));
-    },
-});