specific-ai-cli 0.0.0-stage → 0.1.1

package/dist/platforms/tiktok/web/profile.js

@@ -0,0 +1,169 @@
+import { quote, quotePlus } from '../../../core/py.js';
+/**
+ * 浏览器画像、请求头与 query 构造（上游 builder/auth.py 的常量、builder/header.py、builder/params.py）。
+ * UA 是上游取证用的 Chrome 153；TLS 指纹用 wreq-js 最新的 Chrome 画像。
+ */
+export const PROFILE = {
+    ua: 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/153.0.0.0 Safari/537.36',
+    secChUa: '"Google Chrome";v="153", "Not_A Brand";v="8", "Chromium";v="153"',
+    secChUaPlatform: '"Windows"',
+    acceptLanguage: 'zh-CN,zh;q=0.9,en;q=0.8,zh-TW;q=0.7,ja;q=0.6',
+    clientAbVersions: '70508271,72437276,73720540,75360573,75657507,75843653,75878361,' +
+        '76074824,76122410,76179551,76403724,76499369,76626930,76651230,' +
+        '76669176,76713103,76740102,76792575,76805936,76813520,76815102,' +
+        '76826749,76839930,76840792,76887350,76889673,76907405,76933422,' +
+        '76945954,76950455,77005875,77011497,77014490,77056281,77064650,' +
+        '77071444,77093404,77120364,77141446,77150562,77155134,70138197,' +
+        '70156809,70405643,71057832,71200802,71381811,71803300,72360691,' +
+        '72408100,72854054,72892778,73171280,73208420,73989921,74276218,' +
+        '74844724,75330961',
+};
+export const BROWSER = 'chrome';
+export const ORIGIN = 'https://www.tiktok.com';
+export const WEBCAST = 'https://webcast.tiktok.com';
+export const SHOP = 'https://shop.tiktok.com';
+export const COOKIE_DOMAIN = '.tiktok.com';
+/** 有序请求头，对应上游的 OrderedDict：改值不动位置，新键追加到末尾。 */
+export class Headers {
+    map = new Map();
+    set(key, value) {
+        if (value != null)
+            this.map.set(key, String(value));
+        return this;
+    }
+    get(key) {
+        return this.map.get(key);
+    }
+    has(key) {
+        return this.map.has(key);
+    }
+    delete(key) {
+        this.map.delete(key);
+        return this;
+    }
+    update(values) {
+        for (const [k, v] of values instanceof Map ? values : Object.entries(values ?? {}))
+            this.set(k, v);
+        return this;
+    }
+    /** 按给定顺序重排；不在表里的键保持原顺序排在后面。drop 为 true 时丢掉表外的键。 */
+    reorder(order, drop = false) {
+        const next = new Map();
+        for (const k of order)
+            if (this.map.has(k))
+                next.set(k, this.map.get(k));
+        if (!drop)
+            for (const [k, v] of this.map)
+                if (!next.has(k))
+                    next.set(k, v);
+        this.map = next;
+        return this;
+    }
+    /** 发送用的键值对。cookie 由 cookie 罐按 URL 补上（位置由 core 决定），这里去掉。 */
+    pairs() {
+        return [...this.map].filter(([k]) => k !== 'cookie');
+    }
+}
+/** 上游 HeaderBuilder.build：只放浏览器在该类请求里实际发出的字段。cookie 占位，保持与上游相同的相对顺序。 */
+export function buildHeaders(type, b, o) {
+    const h = new Headers();
+    if (type === 'DOC') {
+        h.set('accept', 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7');
+        h.set('accept-language', b.acceptLanguage);
+        h.set('cache-control', 'no-cache');
+        h.set('pragma', 'no-cache');
+        h.set('priority', 'u=0, i');
+        h.set('referer', o.referer);
+        h.set('sec-ch-ua', b.secChUa);
+        h.set('sec-ch-ua-mobile', '?0');
+        h.set('sec-ch-ua-platform', b.secChUaPlatform);
+        h.set('sec-fetch-dest', 'document');
+        h.set('sec-fetch-mode', 'navigate');
+        h.set('sec-fetch-site', 'same-origin');
+        h.set('sec-fetch-user', '?1');
+        h.set('upgrade-insecure-requests', '1');
+        h.set('user-agent', b.ua);
+        h.set('cookie', '');
+        return h;
+    }
+    h.set('sec-ch-ua-platform', b.secChUaPlatform);
+    h.set('referer', o.referer);
+    h.set('user-agent', b.ua);
+    h.set('sec-ch-ua', b.secChUa);
+    h.set('sec-ch-ua-mobile', '?0');
+    if (type === 'POST')
+        h.set('content-type', 'application/json');
+    else if (type === 'FORM')
+        h.set('content-type', 'application/x-www-form-urlencoded');
+    h.set('accept', '*/*');
+    h.set('accept-encoding', 'gzip, deflate, br, zstd');
+    h.set('accept-language', b.acceptLanguage);
+    h.set('cookie', '');
+    if (o.origin)
+        h.set('origin', o.origin);
+    if (o.contentLength != null)
+        h.set('content-length', o.contentLength);
+    h.set('priority', 'u=1, i');
+    h.set('sec-fetch-dest', 'empty');
+    h.set('sec-fetch-mode', 'cors');
+    h.set('sec-fetch-site', o.secFetchSite ?? 'same-origin');
+    return h;
+}
+// ================================================================ query（builder/params.py）
+const SIGNATURE_SAFE = new Set(['X-Dynosaur', 'X-Gnarly', 'msToken']);
+/** 保持插入顺序、按 TikTok 浏览器规则转义的 query。 */
+export class Params {
+    spacePlusKeys;
+    pairs = [];
+    constructor(values = [], spacePlusKeys = new Set()) {
+        this.spacePlusKeys = spacePlusKeys;
+        for (const [k, v] of values)
+            this.add(k, v);
+    }
+    /** add_param：同名键只保留最新的一个，追加到末尾。 */
+    add(key, value) {
+        const v = pyValue(value);
+        this.pairs = this.pairs.filter(([k]) => k !== key);
+        this.pairs.push([key, v]);
+        return this;
+    }
+    /** add_pair：允许重复键（直播 im/fetch 的两个 version_code）。 */
+    addPair(key, value) {
+        this.pairs.push([key, pyValue(value)]);
+        return this;
+    }
+    update(values) {
+        const entries = Symbol.iterator in Object(values) ? values : Object.entries(values);
+        for (const [k, v] of entries)
+            this.add(k, v);
+        return this;
+    }
+    /** 原样复制（保留重复键）。 */
+    clone() {
+        const p = new Params([], this.spacePlusKeys);
+        p.pairs = this.pairs.map(([k, v]) => [k, v]);
+        return p;
+    }
+    get(key) {
+        return this.pairs.findLast(([k]) => k === key)?.[1];
+    }
+    toQuery() {
+        return this.pairs
+            .map(([key, value]) => {
+            const safe = SIGNATURE_SAFE.has(key) ? '/=+.-_~' : key === 'next' ? ':' : '-_.~';
+            const enc = this.spacePlusKeys.has(key) ? quotePlus : quote;
+            return `${quote(key, '-_.~')}=${enc(value, safe)}`;
+        })
+            .join('&');
+    }
+}
+/** Python 的 `"" if value is None else str(value)`。 */
+function pyValue(value) {
+    if (value == null)
+        return '';
+    if (value === true)
+        return 'True';
+    if (value === false)
+        return 'False';
+    return String(value);
+}

package/dist/platforms/tiktok/web/resolve.js

@@ -0,0 +1,156 @@
+import { SpecificAIError } from '../../../core/errors.js';
+import { jsonLoads } from '../../../core/py.js';
+import * as api from './api.js';
+import { hydration } from './client.js';
+import { ORIGIN } from './profile.js';
+/** 参数归一化（AGENTS 4.8）：ID、URL、分享短链、@用户名、secUid、`me`。 */
+const SHORT_RE = /^https?:\/\/(vt\.tiktok\.com|vm\.tiktok\.com|www\.tiktok\.com\/t)\//i;
+/** 分享短链跟一次跳转，取 Location。 */
+export async function expand(t, input) {
+    const s = input.trim();
+    if (!SHORT_RE.test(s))
+        return s;
+    const res = await t.send({ url: s, redirect: 'manual', cookies: false, headers: [['user-agent', t.ua]] });
+    return res.headers.get('location') ?? s;
+}
+/** 作品：纯数字 ID、视频 / 图文页 URL、分享短链。 */
+export async function resolveItem(t, input) {
+    const s = await expand(t, input);
+    const m = /tiktok\.com\/@([^/?#]+)\/(?:video|photo)\/(\d+)/.exec(s);
+    if (m)
+        return { id: m[2], handle: decodeURIComponent(m[1]), url: `${ORIGIN}/@${m[1]}/video/${m[2]}` };
+    const id = /(?:^|\/(?:video|photo|v)\/)(\d{15,21})(?:\.html)?(?:[/?#]|$)/.exec(s)?.[1];
+    if (id)
+        return { id, handle: null, url: `${ORIGIN}/@_/video/${id}` };
+    throw new SpecificAIError('USAGE', `无法识别的作品：${input}`, { hint: '传作品 ID 或视频链接，例如 https://www.tiktok.com/@tiktok/video/7300000000000000000' });
+}
+const SEC_UID_RE = /^MS4wLj[A-Za-z0-9_-]{20,}$/;
+/** 用户：@用户名、主页 URL、secUid、数字 uid、`me`。 */
+export async function parseUser(t, input) {
+    const s = (await expand(t, input)).trim();
+    if (s === 'me') {
+        if (!t.loggedIn)
+            throw new SpecificAIError('AUTH_REQUIRED', 'me 表示当前账号，需要先登录', { hint: 'specific-ai tiktok auth login' });
+        const me = await currentUser(t);
+        return { handle: me.uniqueId ?? null, secUid: me.secUid ?? null, id: me.uid ?? t.uid };
+    }
+    const url = /tiktok\.com\/@([^/?#]+)/.exec(s)?.[1];
+    if (url)
+        return { handle: decodeURIComponent(url), secUid: null, id: null };
+    if (SEC_UID_RE.test(s))
+        return { handle: null, secUid: s, id: null };
+    if (/^\d{5,21}$/.test(s))
+        return { handle: null, secUid: null, id: s };
+    const handle = /^@?([A-Za-z0-9_.]{1,64})$/.exec(s)?.[1];
+    if (handle)
+        return { handle, secUid: null, id: null };
+    throw new SpecificAIError('USAGE', `无法识别的用户：${input}`, { hint: '传用户名（@tiktok）、主页链接、secUid 或 me' });
+}
+/** 登录用户（首页 hydration 的 app-context.user）。 */
+export async function currentUser(t) {
+    const html = await t.document(`${ORIGIN}/`, `${ORIGIN}/`);
+    const user = hydration(html)?.['webapp.app-context']?.user;
+    if (!user?.uid && !user?.uniqueId)
+        throw new SpecificAIError('AUTH_EXPIRED', 'TikTok 首页没有返回当前登录用户，登录态可能已失效', { hint: 'specific-ai tiktok auth login' });
+    return user;
+}
+/** 主页 SSR 里的 userInfo（上游 get_user_info：webapp.user-detail）。 */
+export async function userDetail(t, handle) {
+    const html = await api.userHtml(t, `${ORIGIN}/@${encodeURIComponent(handle).replace(/%40/g, '@')}`);
+    const detail = hydration(html)?.['webapp.user-detail'] ?? matchUserDetail(html);
+    if (!detail)
+        throw new SpecificAIError('UPSTREAM', '页面中没有 webapp.user-detail', { hint: '确认用户名是否正确' });
+    if (detail.statusCode && detail.statusCode !== 0) {
+        throw new SpecificAIError('UPSTREAM', `用户不存在或不可见：@${handle}`, { detail: { statusCode: detail.statusCode, statusMsg: detail.statusMsg } });
+    }
+    return detail.userInfo;
+}
+/** 上游的正则取法：`"webapp.user-detail":(.*?),"webapp.a-b"`。 */
+function matchUserDetail(html) {
+    const m = /"webapp\.user-detail":(.*?),"webapp\.a-b"/.exec(html);
+    if (!m)
+        return null;
+    try {
+        return jsonLoads(m[1]);
+    }
+    catch {
+        return null;
+    }
+}
+/** 数字 uid → 用户名（私信页的用户卡片接口，需要登录）。 */
+async function handleById(t, id) {
+    const r = await api.imUserProfile(t, [id]);
+    const handle = r.users?.[0]?.im_user_profile?.unique_id;
+    if (!handle)
+        throw new SpecificAIError('UPSTREAM', `找不到用户 ${id}`);
+    return handle;
+}
+/** 需要完整资料时：返回 userInfo（含 user.id / secUid / uniqueId）。 */
+export async function resolveUserInfo(t, input) {
+    const u = await parseUser(t, input);
+    if (u.handle)
+        return userDetail(t, u.handle);
+    if (u.id)
+        return userDetail(t, await handleById(t, u.id));
+    throw new SpecificAIError('USAGE', '只给了 secUid，无法取用户资料', { hint: '传用户名（@tiktok）或主页链接' });
+}
+/** 只需要 secUid 的列表类接口。 */
+export async function resolveSecUid(t, input) {
+    const u = await parseUser(t, input);
+    if (u.secUid)
+        return u.secUid;
+    const info = await resolveUserInfo(t, input);
+    const sec = info?.user?.secUid;
+    if (!sec)
+        throw new SpecificAIError('UPSTREAM', '用户资料里没有 secUid');
+    return sec;
+}
+const ROOM_ID_RE = /^\d{10,21}$/;
+export const isRoomId = (input) => ROOM_ID_RE.test(input.trim());
+/** 只有房间号时取主播信息：进房（上游 enter_live_room）拿房间对象里的 owner。 */
+export async function enterRoom(t, roomId) {
+    const r = await api.enterLiveRoom(t, roomId);
+    const room = r.data ?? {};
+    const owner = room.owner ?? {};
+    const hostId = owner.id_str ?? room.owner_user_id_str ?? owner.id ?? room.owner_user_id;
+    if (!hostId)
+        throw new SpecificAIError('UPSTREAM', `直播间 ${roomId} 的进房结果里没有主播信息`, { detail: { status: room.status } });
+    return { roomId: String(room.id_str ?? roomId), handle: owner.display_id ?? null, hostId: String(hostId), data: null, webcast: room };
+}
+/**
+ * 直播间：@主播、主播主页或直播间 URL（查 /api-live/user/room 得到房间号）、纯数字房间号。
+ * 只给房间号、又需要主播信息（点赞、排行榜、发弹幕、直播间信息）时，先进房取房间对象。
+ */
+export async function resolveRoom(t, input, o = {}) {
+    const s = (await expand(t, input)).trim();
+    if (ROOM_ID_RE.test(s)) {
+        if (o.needHost)
+            return enterRoom(t, s);
+        return { roomId: s, handle: null, hostId: null, data: null };
+    }
+    const handle = /tiktok\.com\/@([^/?#]+)/.exec(s)?.[1] ?? /^@?([A-Za-z0-9_.]{1,64})$/.exec(s)?.[1];
+    if (!handle)
+        throw new SpecificAIError('USAGE', `无法识别的直播间：${input}`, { hint: '传主播用户名（@tiktok）、直播间链接或房间号' });
+    const r = await api.liveUserRoom(t, decodeURIComponent(handle));
+    const u = r.data?.user ?? {};
+    const roomId = String(u.roomId || r.data?.liveRoom?.roomId || '');
+    if (!roomId)
+        throw new SpecificAIError('UPSTREAM', `@${handle} 当前没有直播间`, { detail: { status: u.status } });
+    return { roomId, handle: u.uniqueId ?? decodeURIComponent(handle), hostId: u.id ? String(u.id) : null, data: r.data };
+}
+/** Shop 商品：PDP URL（/pdp/<slug>/<id>、/view/product/<id>）或纯数字 ID。 */
+export function resolveProduct(input) {
+    const s = input.trim();
+    const m = /shop\.tiktok\.com(?:\/[^?#]*?)?(?:\/pdp\/(?:[^/?#]+\/)?|\/view\/product\/)(\d+)/.exec(s);
+    if (m) {
+        const url = s.split('#')[0].split('?')[0];
+        return { id: m[1], url: /\/pdp\//.test(url) ? url : `https://shop.tiktok.com/view/product/${m[1]}` };
+    }
+    if (/^\d{10,21}$/.test(s))
+        return { id: s, url: `https://shop.tiktok.com/view/product/${s}` };
+    throw new SpecificAIError('USAGE', `无法识别的商品：${input}`, { hint: '传商品 ID 或 https://shop.tiktok.com/.../pdp/<商品ID> 链接' });
+}
+/** 是否像商品参数（`comment list` 自动识别商品 URL）。 */
+export function isProductInput(input) {
+    return /shop\.tiktok\.com\//.test(input);
+}