jackwener--opencli/clis/zhihu/favorite.js

import { CliError, CommandExecutionError } from '@jackwener/opencli/errors';
import { cli, Strategy } from '@jackwener/opencli/registry';
import { assertAllowedKinds, parseTarget } from './target.js';
import { buildResultRow, requireExecute } from './write-shared.js';
function normalizeCollectionName(value) {
    return value
        .replace(/\s+/g, ' ')
        .replace(/\s+\d+\s*(条内容|个内容|items?)$/i, '')
        .replace(/\s+(公开|私密|默认)$/i, '')
        .trim();
}
cli({
    site: 'zhihu',
    name: 'favorite',
    access: 'write',
    description: 'Favorite a Zhihu answer or article into a specific collection',
    domain: 'zhihu.com',
    strategy: Strategy.COOKIE,
    browser: true,
    args: [
        { name: 'target', positional: true, required: true, help: 'Zhihu target URL or typed target' },
        { name: 'collection', help: 'Collection name' },
        { name: 'collection-id', help: 'Stable collection id' },
        { name: 'execute', type: 'boolean', help: 'Actually perform the write action' },
    ],
    columns: ['status', 'outcome', 'message', 'target_type', 'target', 'collection_name', 'collection_id'],
    func: async (page, kwargs) => {
        if (!page)
            throw new CommandExecutionError('Browser session required for zhihu favorite');
        requireExecute(kwargs);
        const rawTarget = String(kwargs.target);
        const target = assertAllowedKinds('favorite', parseTarget(rawTarget));
        const collectionName = typeof kwargs.collection === 'string' ? kwargs.collection : undefined;
        const collectionId = typeof kwargs['collection-id'] === 'string' ? kwargs['collection-id'] : undefined;
        if ((collectionName ? 1 : 0) + (collectionId ? 1 : 0) !== 1) {
            throw new CliError('INVALID_INPUT', 'Use exactly one of --collection or --collection-id');
        }
        await page.goto('https://www.zhihu.com');
        await page.wait(2);
        const apiResult = await page.evaluate(`(async () => {
            var collectionId = ${JSON.stringify(collectionId || null)};
            var collectionName = ${JSON.stringify(collectionName || null)};
            var targetKind = ${JSON.stringify(target.kind)};
            var targetId = ${JSON.stringify(target.id)};
            var normalizeCollectionName = function(value) {
                return String(value || '')
                    .replace(/\\s+/g, ' ')
                    .replace(/\\s+\\d+\\s*(条内容|个内容|items?)$/i, '')
                    .replace(/\\s+(公开|私密|默认)$/i, '')
                    .trim()
                    .toLowerCase();
            };

            if (!collectionId && collectionName) {
                var listResp = await fetch('https://www.zhihu.com/api/v4/people/self/collections?limit=50', { credentials: 'include' });
                if (!listResp.ok) return { ok: false, message: 'Failed to list collections: HTTP ' + listResp.status };
                var listData = {};
                try { listData = await listResp.json(); } catch(e) {
                    return { ok: false, message: 'Failed to parse collection list' };
                }
                var needle = normalizeCollectionName(collectionName);
                var matches = (listData.data || []).filter(function(c) { return normalizeCollectionName(c.title) === needle; });
                if (matches.length === 0) return { ok: false, message: 'Collection not found: ' + collectionName };
                if (matches.length > 1) return { ok: false, message: 'Collection name is ambiguous: ' + collectionName };
                collectionId = String(matches[0].id);
            }

            var resp = await fetch('https://www.zhihu.com/api/v4/favlists/' + collectionId + '/items', {
                method: 'POST',
                credentials: 'include',
                headers: { 'Content-Type': 'application/json' },
                body: JSON.stringify({ item_id: targetId, item_type: targetKind }),
            });
            if (resp.ok || resp.status === 204) return { ok: true, collectionId: collectionId };
            var data = {};
            try { data = await resp.json(); } catch(e) {}
            return { ok: false, message: data.error ? data.error.message : 'HTTP ' + resp.status };
        })()`);
        if (!apiResult?.ok) {
            throw new CliError('COMMAND_EXEC', apiResult?.message || 'Failed to favorite');
        }
        return buildResultRow(`Favorited ${target.kind} ${target.id}`, target.kind, rawTarget, 'applied', {
            collection_name: collectionName || '',
            collection_id: apiResult.collectionId || collectionId || '',
        });
    },
});