@jackwener/opencli 1.4.0 → 1.5.0

package/src/clis/douyin/activities.ts

@@ -0,0 +1,23 @@
+import { cli, Strategy } from '../../registry.js';
+import { browserFetch } from './_shared/browser-fetch.js';
+
+cli({
+  site: 'douyin',
+  name: 'activities',
+  description: '官方活动列表',
+  domain: 'creator.douyin.com',
+  strategy: Strategy.COOKIE,
+  args: [],
+  columns: ['activity_id', 'title', 'end_time'],
+  func: async (page, _kwargs) => {
+    const url = 'https://creator.douyin.com/web/api/media/activity/get/?aid=1128';
+    const res = await browserFetch(page, 'GET', url) as {
+      activity_list: Array<{ activity_id: string; title: string; end_time: number }>
+    };
+    return (res.activity_list ?? []).map(a => ({
+      activity_id: a.activity_id,
+      title: a.title,
+      end_time: new Date(a.end_time * 1000).toLocaleString('zh-CN', { timeZone: 'Asia/Tokyo' }),
+    }));
+  },
+});

package/src/clis/douyin/collections.test.ts

@@ -0,0 +1,26 @@
+import { describe, expect, it } from 'vitest';
+import { getRegistry } from '../../registry.js';
+import './collections.js';
+
+describe('douyin collections registration', () => {
+  it('registers the collections command', () => {
+    const registry = getRegistry();
+    const cmd = [...registry.values()].find(c => c.site === 'douyin' && c.name === 'collections');
+    expect(cmd).toBeDefined();
+    expect(cmd?.args.some(a => a.name === 'limit')).toBe(true);
+  });
+
+  it('has expected columns', () => {
+    const registry = getRegistry();
+    const cmd = [...registry.values()].find(c => c.site === 'douyin' && c.name === 'collections');
+    expect(cmd?.columns).toContain('mix_id');
+    expect(cmd?.columns).toContain('name');
+    expect(cmd?.columns).toContain('item_count');
+  });
+
+  it('uses COOKIE strategy', () => {
+    const registry = getRegistry();
+    const cmd = [...registry.values()].find(c => c.site === 'douyin' && c.name === 'collections');
+    expect(cmd?.strategy).toBe('cookie');
+  });
+});

package/src/clis/douyin/collections.ts

@@ -0,0 +1,25 @@
+import { cli, Strategy } from '../../registry.js';
+import { browserFetch } from './_shared/browser-fetch.js';
+
+cli({
+  site: 'douyin',
+  name: 'collections',
+  description: '合集列表',
+  domain: 'creator.douyin.com',
+  strategy: Strategy.COOKIE,
+  args: [
+    { name: 'limit', type: 'int', default: 20 },
+  ],
+  columns: ['mix_id', 'name', 'item_count'],
+  func: async (page, kwargs) => {
+    const url = `https://creator.douyin.com/web/api/mix/list/?aid=1128&count=${kwargs.limit}`;
+    const res = await browserFetch(page, 'GET', url) as {
+      mix_list: Array<{ mix_id: string; mix_name: string; item_count: number }>
+    };
+    return (res.mix_list ?? []).map(m => ({
+      mix_id: m.mix_id,
+      name: m.mix_name,
+      item_count: m.item_count,
+    }));
+  },
+});

package/src/clis/douyin/delete.test.ts

@@ -0,0 +1,12 @@
+import { describe, expect, it } from 'vitest';
+import { getRegistry } from '../../registry.js';
+import './delete.js';
+
+describe('douyin delete registration', () => {
+  it('registers the delete command', () => {
+    const registry = getRegistry();
+    const values = [...registry.values()];
+    const cmd = values.find(c => c.site === 'douyin' && c.name === 'delete');
+    expect(cmd).toBeDefined();
+  });
+});

package/src/clis/douyin/delete.ts

@@ -0,0 +1,20 @@
+import { cli, Strategy } from '../../registry.js';
+import { browserFetch } from './_shared/browser-fetch.js';
+import type { IPage } from '../../types.js';
+
+cli({
+  site: 'douyin',
+  name: 'delete',
+  description: '删除作品',
+  domain: 'creator.douyin.com',
+  strategy: Strategy.COOKIE,
+  args: [
+    { name: 'aweme_id', required: true, positional: true, help: '作品 ID' },
+  ],
+  columns: ['status'],
+  func: async (page: IPage, kwargs) => {
+    const url = 'https://creator.douyin.com/web/api/media/aweme/delete/?aid=1128';
+    await browserFetch(page, 'POST', url, { body: { aweme_id: kwargs.aweme_id } });
+    return [{ status: `✅ 已删除 ${kwargs.aweme_id}` }];
+  },
+});

package/src/clis/douyin/draft.test.ts

@@ -0,0 +1,12 @@
+import { describe, expect, it } from 'vitest';
+import { getRegistry } from '../../registry.js';
+import './draft.js';
+
+describe('douyin draft registration', () => {
+  it('registers the draft command', () => {
+    const registry = getRegistry();
+    const values = [...registry.values()];
+    const cmd = values.find(c => c.site === 'douyin' && c.name === 'draft');
+    expect(cmd).toBeDefined();
+  });
+});

package/src/clis/douyin/draft.ts

@@ -0,0 +1,282 @@
+/**
+ * Douyin draft — 6-phase pipeline for saving video as draft.
+ *
+ * Phases:
+ *   1. STS2 credentials
+ *   2. Apply TOS upload URL
+ *   3. TOS multipart upload
+ *   4. Cover upload (optional, via ImageX)
+ *   5. Enable video
+ *   6. Poll transcode
+ *   7. (skipped — no safety check for drafts)
+ *   8. create_v2 with is_draft: 1
+ */
+
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import { cli, Strategy } from '../../registry.js';
+import { ArgumentError, CommandExecutionError } from '../../errors.js';
+import type { IPage } from '../../types.js';
+import type { TosUploadInfo } from './_shared/types.js';
+import { getSts2Credentials } from './_shared/sts2.js';
+import { tosUpload } from './_shared/tos-upload.js';
+import { imagexUpload } from './_shared/imagex-upload.js';
+import { pollTranscode } from './_shared/transcode.js';
+import { browserFetch } from './_shared/browser-fetch.js';
+import { generateCreationId } from './_shared/creation-id.js';
+import { parseTextExtra, extractHashtagNames } from './_shared/text-extra.js';
+import type { HashtagInfo } from './_shared/text-extra.js';
+
+const VISIBILITY_MAP: Record<string, number> = {
+  public: 0,
+  friends: 1,
+  private: 2,
+};
+
+const IMAGEX_BASE = 'https://imagex.bytedanceapi.com';
+const IMAGEX_SERVICE_ID = '1147';
+
+const DEVICE_PARAMS =
+  'aid=1128&cookie_enabled=true&screen_width=1512&screen_height=982&browser_language=zh-CN&browser_platform=MacIntel&browser_name=Mozilla&browser_online=true&timezone_name=Asia%2FTokyo&support_h265=1';
+
+const DEFAULT_COVER_TOOLS_INFO = JSON.stringify({
+  video_cover_source: 2,
+  cover_timestamp: 0,
+  recommend_timestamp: 0,
+  is_cover_edit: 0,
+  is_cover_template: 0,
+  cover_template_id: '',
+  is_text_template: 0,
+  text_template_id: '',
+  text_template_content: '',
+  is_text: 0,
+  text_num: 0,
+  text_content: '',
+  is_use_sticker: 0,
+  sticker_id: '',
+  is_use_filter: 0,
+  filter_id: '',
+  is_cover_modify: 0,
+  to_status: 0,
+  cover_type: 0,
+  initial_cover_uri: '',
+  cut_coordinate: '',
+});
+
+cli({
+  site: 'douyin',
+  name: 'draft',
+  description: '上传视频并保存为草稿',
+  domain: 'creator.douyin.com',
+  strategy: Strategy.COOKIE,
+  args: [
+    { name: 'video', required: true, positional: true, help: '视频文件路径' },
+    { name: 'title', required: true, help: '视频标题（≤30字）' },
+    { name: 'caption', default: '', help: '正文内容（≤1000字，支持 #话题）' },
+    { name: 'cover', default: '', help: '封面图片路径' },
+    { name: 'visibility', default: 'public', choices: ['public', 'friends', 'private'] },
+  ],
+  columns: ['status', 'aweme_id'],
+  func: async (page: IPage, kwargs) => {
+    // ── Fail-fast validation ────────────────────────────────────────────
+    const videoPath = path.resolve(kwargs.video as string);
+    if (!fs.existsSync(videoPath)) {
+      throw new ArgumentError(`视频文件不存在: ${videoPath}`);
+    }
+    const ext = path.extname(videoPath).toLowerCase();
+    if (!['.mp4', '.mov', '.avi', '.webm'].includes(ext)) {
+      throw new ArgumentError(`不支持的视频格式: ${ext}（支持 mp4/mov/avi/webm）`);
+    }
+    const fileSize = fs.statSync(videoPath).size;
+
+    const title = kwargs.title as string;
+    if (title.length > 30) {
+      throw new ArgumentError('标题不能超过 30 字');
+    }
+
+    const caption = (kwargs.caption as string) || '';
+    if (caption.length > 1000) {
+      throw new ArgumentError('正文不能超过 1000 字');
+    }
+
+    const visibilityType = VISIBILITY_MAP[kwargs.visibility as string] ?? 0;
+
+    const coverPath = kwargs.cover as string;
+    if (coverPath) {
+      if (!fs.existsSync(path.resolve(coverPath))) {
+        throw new ArgumentError(`封面文件不存在: ${path.resolve(coverPath)}`);
+      }
+    }
+
+    // ── Phase 1: STS2 credentials ───────────────────────────────────────
+    const credentials = await getSts2Credentials(page);
+
+    // ── Phase 2: Apply TOS upload URL ───────────────────────────────────
+    const vodUrl = `https://vod.bytedanceapi.com/?Action=ApplyVideoUpload&ServiceId=1128&Version=2021-01-01&FileType=video&FileSize=${fileSize}`;
+    const vodJs = `fetch(${JSON.stringify(vodUrl)}, { credentials: 'include' }).then(r => r.json())`;
+    const vodRes = (await page.evaluate(vodJs)) as {
+      Result: {
+        UploadAddress: {
+          VideoId: string;
+          UploadHosts: string[];
+          StoreInfos: Array<{ Auth: string; StoreUri: string }>;
+        };
+      };
+    };
+    const { VideoId: videoId, UploadHosts, StoreInfos } = vodRes.Result.UploadAddress;
+    const tosUrl = `https://${UploadHosts[0]}/${StoreInfos[0].StoreUri}`;
+    const tosUploadInfo: TosUploadInfo = {
+      tos_upload_url: tosUrl,
+      auth: StoreInfos[0].Auth,
+      video_id: videoId,
+    };
+
+    // ── Phase 3: TOS upload ─────────────────────────────────────────────
+    await tosUpload({
+      filePath: videoPath,
+      uploadInfo: tosUploadInfo,
+      credentials,
+      onProgress: (uploaded, total) => {
+        const pct = Math.round((uploaded / total) * 100);
+        process.stderr.write(`\r  上传进度: ${pct}%`);
+      },
+    });
+    process.stderr.write('\n');
+
+    // ── Phase 4: Cover upload (optional) ────────────────────────────────
+    let coverUri = '';
+    let coverWidth = 720;
+    let coverHeight = 1280;
+
+    if (kwargs.cover) {
+      const resolvedCoverPath = path.resolve(kwargs.cover as string);
+
+      // 4A: Apply ImageX upload
+      const applyUrl = `${IMAGEX_BASE}/?Action=ApplyImageUpload&ServiceId=${IMAGEX_SERVICE_ID}&Version=2018-08-01&UploadNum=1`;
+      const applyJs = `fetch(${JSON.stringify(applyUrl)}, { credentials: 'include' }).then(r => r.json())`;
+      const applyRes = (await page.evaluate(applyJs)) as {
+        Result: {
+          UploadAddress: {
+            UploadHosts: string[];
+            StoreInfos: Array<{ Auth: string; StoreUri: string; UploadHost: string }>;
+          };
+        };
+      };
+      const { StoreInfos: imgStoreInfos } = applyRes.Result.UploadAddress;
+      const imgUploadUrl = `https://${imgStoreInfos[0].UploadHost}/${imgStoreInfos[0].StoreUri}`;
+
+      // 4B: Upload image
+      const coverStoreUri = await imagexUpload(resolvedCoverPath, {
+        upload_url: imgUploadUrl,
+        store_uri: imgStoreInfos[0].StoreUri,
+      });
+
+      // 4C: Commit ImageX upload
+      const commitUrl = `${IMAGEX_BASE}/?Action=CommitImageUpload&ServiceId=${IMAGEX_SERVICE_ID}&Version=2018-08-01`;
+      const commitBody = JSON.stringify({ SuccessObjKeys: [coverStoreUri] });
+      const commitJs = `
+        fetch(${JSON.stringify(commitUrl)}, {
+          method: 'POST',
+          credentials: 'include',
+          headers: { 'Content-Type': 'application/json' },
+          body: ${JSON.stringify(commitBody)}
+        }).then(r => r.json())
+      `;
+      await page.evaluate(commitJs);
+
+      coverUri = coverStoreUri;
+    }
+
+    // ── Phase 5: Enable video ───────────────────────────────────────────
+    const enableUrl = `https://creator.douyin.com/web/api/media/video/enable/?video_id=${videoId}&aid=1128`;
+    await browserFetch(page, 'GET', enableUrl);
+
+    // ── Phase 6: Poll transcode ─────────────────────────────────────────
+    const transResult = await pollTranscode(page, videoId);
+    coverWidth = transResult.width;
+    coverHeight = transResult.height;
+    if (!coverUri) {
+      coverUri = transResult.poster_uri;
+    }
+
+    // ── Phase 7: SKIP (no safety check for drafts) ──────────────────────
+
+    // ── Phase 8: create_v2 with is_draft: 1 ────────────────────────────
+    const hashtagNames = extractHashtagNames(caption);
+    const hashtags: HashtagInfo[] = [];
+    let searchFrom = 0;
+    for (const name of hashtagNames) {
+      const idx = caption.indexOf(`#${name}`, searchFrom);
+      if (idx === -1) continue;
+      hashtags.push({ name, id: 0, start: idx, end: idx + name.length + 1 });
+      searchFrom = idx + name.length + 1;
+    }
+    const textExtraArr = parseTextExtra(caption, hashtags);
+
+    const publishBody = {
+      item: {
+        common: {
+          text: caption,
+          caption: '',
+          item_title: title,
+          activity: '[]',
+          text_extra: JSON.stringify(textExtraArr),
+          challenges: '[]',
+          mentions: '[]',
+          hashtag_source: '',
+          hot_sentence: '',
+          interaction_stickers: '[]',
+          visibility_type: visibilityType,
+          download: 0,
+          is_draft: 1,
+          creation_id: generateCreationId(),
+          media_type: 4,
+          video_id: videoId,
+          music_source: 0,
+          music_id: null,
+        },
+        cover: {
+          poster: coverUri,
+          custom_cover_image_height: coverHeight,
+          custom_cover_image_width: coverWidth,
+          poster_delay: 0,
+          cover_tools_info: DEFAULT_COVER_TOOLS_INFO,
+          cover_tools_extend_info: '{}',
+        },
+        mix: {},
+        chapter: {
+          chapter: JSON.stringify({
+            chapter_abstract: '',
+            chapter_details: [],
+            chapter_type: 0,
+          }),
+        },
+        anchor: {},
+        sync: {
+          should_sync: false,
+          sync_to_toutiao: 0,
+        },
+        open_platform: {},
+        assistant: { is_preview: 0, is_post_assistant: 1 },
+        declare: { user_declare_info: '{}' },
+      },
+    };
+
+    const publishUrl = `https://creator.douyin.com/web/api/media/aweme/create_v2/?read_aid=2906&${DEVICE_PARAMS}`;
+    const publishRes = (await browserFetch(page, 'POST', publishUrl, {
+      body: publishBody,
+    })) as { status_code: number; aweme_id: string };
+
+    const awemeId = publishRes.aweme_id;
+    if (!awemeId) {
+      throw new CommandExecutionError(`草稿保存成功但未返回 aweme_id: ${JSON.stringify(publishRes)}`);
+    }
+
+    return [
+      {
+        status: '✅ 草稿保存成功！',
+        aweme_id: awemeId,
+      },
+    ];
+  },
+});

package/src/clis/douyin/drafts.test.ts

@@ -0,0 +1,12 @@
+import { describe, expect, it } from 'vitest';
+import { getRegistry } from '../../registry.js';
+import './drafts.js';
+
+describe('douyin drafts registration', () => {
+  it('registers the drafts command', () => {
+    const registry = getRegistry();
+    const values = [...registry.values()];
+    const cmd = values.find(c => c.site === 'douyin' && c.name === 'drafts');
+    expect(cmd).toBeDefined();
+  });
+});

package/src/clis/douyin/drafts.ts

@@ -0,0 +1,27 @@
+import { cli, Strategy } from '../../registry.js';
+import { browserFetch } from './_shared/browser-fetch.js';
+import type { IPage } from '../../types.js';
+
+cli({
+  site: 'douyin',
+  name: 'drafts',
+  description: '获取草稿列表',
+  domain: 'creator.douyin.com',
+  strategy: Strategy.COOKIE,
+  args: [
+    { name: 'limit', type: 'int', default: 20 },
+  ],
+  columns: ['aweme_id', 'title', 'create_time'],
+  func: async (page: IPage, kwargs) => {
+    const url = 'https://creator.douyin.com/web/api/media/aweme/draft/?aid=1128';
+    const res = (await browserFetch(page, 'GET', url)) as {
+      aweme_list: Array<{ aweme_id: string; desc: string; create_time: number }>;
+    };
+    const items = (res.aweme_list ?? []).slice(0, kwargs.limit as number);
+    return items.map((v) => ({
+      aweme_id: v.aweme_id,
+      title: v.desc,
+      create_time: new Date(v.create_time * 1000).toLocaleString('zh-CN', { timeZone: 'Asia/Tokyo' }),
+    }));
+  },
+});

package/src/clis/douyin/hashtag.test.ts

@@ -0,0 +1,28 @@
+import { describe, expect, it } from 'vitest';
+import { getRegistry } from '../../registry.js';
+import './hashtag.js';
+
+describe('douyin hashtag registration', () => {
+  it('registers the hashtag command', () => {
+    const registry = getRegistry();
+    const cmd = [...registry.values()].find(c => c.site === 'douyin' && c.name === 'hashtag');
+    expect(cmd).toBeDefined();
+    expect(cmd?.args.some(a => a.name === 'action')).toBe(true);
+  });
+
+  it('has all expected args', () => {
+    const registry = getRegistry();
+    const cmd = [...registry.values()].find(c => c.site === 'douyin' && c.name === 'hashtag');
+    const argNames = cmd?.args.map(a => a.name) ?? [];
+    expect(argNames).toContain('action');
+    expect(argNames).toContain('keyword');
+    expect(argNames).toContain('cover');
+    expect(argNames).toContain('limit');
+  });
+
+  it('uses COOKIE strategy', () => {
+    const registry = getRegistry();
+    const cmd = [...registry.values()].find(c => c.site === 'douyin' && c.name === 'hashtag');
+    expect(cmd?.strategy).toBe('cookie');
+  });
+});

package/src/clis/douyin/hashtag.ts

@@ -0,0 +1,56 @@
+import { cli, Strategy } from '../../registry.js';
+import { browserFetch } from './_shared/browser-fetch.js';
+import { ArgumentError } from '../../errors.js';
+
+cli({
+  site: 'douyin',
+  name: 'hashtag',
+  description: '话题搜索 / AI推荐 / 热点词',
+  domain: 'creator.douyin.com',
+  strategy: Strategy.COOKIE,
+  args: [
+    { name: 'action', required: true, positional: true, choices: ['search', 'suggest', 'hot'], help: 'search=关键词搜索 suggest=AI推荐 hot=热点词' },
+    { name: 'keyword', default: '', help: '搜索关键词（search/hot 使用）' },
+    { name: 'cover', default: '', help: '封面 URI（suggest 使用）' },
+    { name: 'limit', type: 'int', default: 10 },
+  ],
+  columns: ['name', 'id', 'view_count'],
+  func: async (page, kwargs) => {
+    const action = kwargs.action as string;
+
+    if (action === 'search') {
+      const url = `https://creator.douyin.com/aweme/v1/challenge/search/?keyword=${encodeURIComponent(kwargs.keyword as string)}&count=${kwargs.limit}&aid=1128`;
+      const res = await browserFetch(page, 'GET', url) as {
+        challenge_list: Array<{ challenge_info: { cid: string; cha_name: string; view_count: number } }>
+      };
+      return (res.challenge_list ?? []).map(c => ({
+        name: c.challenge_info.cha_name,
+        id: c.challenge_info.cid,
+        view_count: c.challenge_info.view_count,
+      }));
+    }
+
+    if (action === 'suggest') {
+      const url = `https://creator.douyin.com/web/api/media/hashtag/rec/?cover_uri=${encodeURIComponent(kwargs.cover as string)}&aid=1128`;
+      const res = await browserFetch(page, 'GET', url) as {
+        hashtag_list: Array<{ name: string; id: string; view_count: number }>
+      };
+      return (res.hashtag_list ?? []).map(h => ({ name: h.name, id: h.id, view_count: h.view_count }));
+    }
+
+    if (action === 'hot') {
+      const kw = kwargs.keyword as string;
+      const url = `https://creator.douyin.com/aweme/v1/hotspot/recommend/?${kw ? `keyword=${encodeURIComponent(kw)}&` : ''}aid=1128`;
+      const res = await browserFetch(page, 'GET', url) as {
+        hotspot_list: Array<{ sentence: string; hot_value: number }>
+      };
+      return (res.hotspot_list ?? []).slice(0, kwargs.limit as number).map(h => ({
+        name: h.sentence,
+        id: '',
+        view_count: h.hot_value,
+      }));
+    }
+
+    throw new ArgumentError(`未知的 action: ${action}`);
+  },
+});

package/src/clis/douyin/location.test.ts

@@ -0,0 +1,26 @@
+import { describe, expect, it } from 'vitest';
+import { getRegistry } from '../../registry.js';
+import './location.js';
+
+describe('douyin location registration', () => {
+  it('registers the location command', () => {
+    const registry = getRegistry();
+    const cmd = [...registry.values()].find(c => c.site === 'douyin' && c.name === 'location');
+    expect(cmd).toBeDefined();
+    expect(cmd?.args.some(a => a.name === 'query')).toBe(true);
+  });
+
+  it('has all expected args', () => {
+    const registry = getRegistry();
+    const cmd = [...registry.values()].find(c => c.site === 'douyin' && c.name === 'location');
+    const argNames = cmd?.args.map(a => a.name) ?? [];
+    expect(argNames).toContain('query');
+    expect(argNames).toContain('limit');
+  });
+
+  it('uses COOKIE strategy', () => {
+    const registry = getRegistry();
+    const cmd = [...registry.values()].find(c => c.site === 'douyin' && c.name === 'location');
+    expect(cmd?.strategy).toBe('cookie');
+  });
+});

package/src/clis/douyin/location.ts

@@ -0,0 +1,27 @@
+import { cli, Strategy } from '../../registry.js';
+import { browserFetch } from './_shared/browser-fetch.js';
+
+cli({
+  site: 'douyin',
+  name: 'location',
+  description: '地理位置 POI 搜索',
+  domain: 'creator.douyin.com',
+  strategy: Strategy.COOKIE,
+  args: [
+    { name: 'query', required: true, positional: true, help: '地名关键词' },
+    { name: 'limit', type: 'int', default: 20 },
+  ],
+  columns: ['poi_id', 'name', 'address', 'city'],
+  func: async (page, kwargs) => {
+    const url = `https://creator.douyin.com/aweme/v1/life/video_api/search/poi/?keyword=${encodeURIComponent(kwargs.query as string)}&count=${kwargs.limit}&aid=1128`;
+    const res = await browserFetch(page, 'GET', url) as {
+      poi_list: Array<{ poi_id: string; poi_name: string; address: string; city_name: string }>
+    };
+    return (res.poi_list ?? []).map(p => ({
+      poi_id: p.poi_id,
+      name: p.poi_name,
+      address: p.address,
+      city: p.city_name,
+    }));
+  },
+});

package/src/clis/douyin/profile.test.ts

@@ -0,0 +1,12 @@
+import { describe, expect, it } from 'vitest';
+import { getRegistry } from '../../registry.js';
+import './profile.js';
+
+describe('douyin profile registration', () => {
+  it('registers the profile command', () => {
+    const registry = getRegistry();
+    const values = [...registry.values()];
+    const cmd = values.find(c => c.site === 'douyin' && c.name === 'profile');
+    expect(cmd).toBeDefined();
+  });
+});

package/src/clis/douyin/profile.ts

@@ -0,0 +1,37 @@
+import { cli, Strategy } from '../../registry.js';
+import { browserFetch } from './_shared/browser-fetch.js';
+import { CommandExecutionError } from '../../errors.js';
+import type { IPage } from '../../types.js';
+
+cli({
+  site: 'douyin',
+  name: 'profile',
+  description: '获取账号信息',
+  domain: 'creator.douyin.com',
+  strategy: Strategy.COOKIE,
+  args: [],
+  columns: ['uid', 'nickname', 'follower_count', 'following_count', 'aweme_count'],
+  func: async (page: IPage, _kwargs) => {
+    const url = 'https://creator.douyin.com/web/api/media/user/info/?aid=1128';
+    const res = (await browserFetch(page, 'GET', url)) as {
+      user_info: {
+        uid: string;
+        nickname: string;
+        follower_count: number;
+        following_count: number;
+        aweme_count: number;
+      };
+    };
+    const u = res.user_info;
+    if (!u) throw new CommandExecutionError('用户信息获取失败，请确认已登录 creator.douyin.com');
+    return [
+      {
+        uid: u.uid,
+        nickname: u.nickname,
+        follower_count: u.follower_count,
+        following_count: u.following_count,
+        aweme_count: u.aweme_count,
+      },
+    ];
+  },
+});