@9000ai/cli 0.3.0 → 0.5.0

package/dist/client.d.ts

@@ -7,4 +7,13 @@ export interface RequestOptions {
     timeout?: number;
 }
 export declare function request(opts: RequestOptions): Promise<unknown>;
-export declare function printJson(data: unknown): void;
+/** Poll a task/batch until terminal status. Returns final response data. */
+export declare function pollUntilDone(path: string, opts?: {
+    interval?: number;
+    timeout?: number;
+    quiet?: boolean;
+}): Promise<unknown>;
+export declare function printJson(data: unknown, opts?: {
+    fields?: string;
+    compact?: boolean;
+}): void;

package/dist/client.js

@@ -40,6 +40,115 @@ export async function request(opts) {
         clearTimeout(timer);
     }
 }
-export function printJson(data) {
-    console.log(JSON.stringify(data, null, 2));
+/** Poll a task/batch until terminal status. Returns final response data. */
+export async function pollUntilDone(path, opts) {
+    const interval = opts?.interval ?? 3_000;
+    const timeout = opts?.timeout ?? 600_000;
+    const start = Date.now();
+    while (Date.now() - start < timeout) {
+        const data = await request({ method: "GET", path });
+        const resp = data;
+        const inner = (resp.data ?? resp);
+        const status = (inner.status ?? "").toUpperCase();
+        if (["SUCCESS", "COMPLETED", "FAILED", "ERROR"].includes(status)) {
+            return data;
+        }
+        if (!opts?.quiet) {
+            const elapsed = Math.round((Date.now() - start) / 1000);
+            console.error(`[${elapsed}s] status: ${status || "PENDING"} — polling...`);
+        }
+        await new Promise((r) => setTimeout(r, interval));
+    }
+    console.error("Error: poll timed out");
+    process.exit(1);
+}
+export function printJson(data, opts) {
+    let output = data;
+    // --fields: extract specified fields from items/data
+    if (opts?.fields) {
+        const fieldList = opts.fields.split(",").map((f) => f.trim());
+        output = pickFields(output, fieldList);
+    }
+    // --compact: one-line-per-item for arrays
+    if (opts?.compact) {
+        const arr = extractArray(output);
+        if (arr) {
+            arr.forEach((item) => console.log(JSON.stringify(item)));
+            return;
+        }
+    }
+    console.log(JSON.stringify(output, null, 2));
+}
+/** Deep-pick fields from API response. Handles nested data.items / data.tasks structures. */
+function pickFields(data, fields) {
+    if (!data || typeof data !== "object")
+        return data;
+    const obj = data;
+    // If response has code/message/data wrapper, dig into data
+    if ("code" in obj && "data" in obj) {
+        const inner = obj.data;
+        if (!inner)
+            return obj;
+        return pickFieldsFromInner(inner, fields);
+    }
+    // Already unwrapped (e.g. inner passed directly)
+    return pickFieldsFromInner(obj, fields);
+}
+function pickFieldsFromInner(inner, fields) {
+    // If has items array, pick from each item
+    if (Array.isArray(inner.items)) {
+        return {
+            ...stripArrays(inner),
+            items: inner.items.map((item) => pick(item, fields)),
+        };
+    }
+    // If has tasks array with nested items
+    if (Array.isArray(inner.tasks)) {
+        return {
+            ...stripArrays(inner),
+            tasks: inner.tasks.map((task) => {
+                const t = { ...task };
+                if (Array.isArray(t.items)) {
+                    t.items = t.items.map((item) => pick(item, fields));
+                }
+                return t;
+            }),
+        };
+    }
+    // Flat data object
+    return pick(inner, fields);
+}
+function pick(obj, fields) {
+    const result = {};
+    for (const field of fields) {
+        if (field in obj) {
+            result[field] = obj[field];
+        }
+    }
+    return result;
+}
+function stripArrays(obj) {
+    const result = {};
+    for (const [k, v] of Object.entries(obj)) {
+        if (!Array.isArray(v))
+            result[k] = v;
+    }
+    return result;
+}
+function extractArray(data) {
+    if (Array.isArray(data))
+        return data;
+    if (!data || typeof data !== "object")
+        return null;
+    const obj = data;
+    // Wrapped response: { code, data: { items: [...] } }
+    if ("code" in obj && "data" in obj) {
+        const inner = obj.data;
+        if (inner && Array.isArray(inner.items))
+            return inner.items;
+    }
+    // Already unwrapped: { items: [...] }
+    if (Array.isArray(obj.items))
+        return obj.items;
+    return null;
 }

package/dist/commands/init.d.ts

@@ -0,0 +1,2 @@
+import { Command } from "commander";
+export declare function registerInitCommand(program: Command): void;

package/dist/commands/init.js

@@ -0,0 +1,83 @@
+import { existsSync, mkdirSync, copyFileSync, readdirSync, statSync } from "fs";
+import { homedir } from "os";
+import { join, relative } from "path";
+const TEMPLATES_DIR = join(homedir(), ".9000ai", "skills", "9000AI-hub", "init", "templates");
+/** Files under profile/ are never overwritten — user may have edited them. */
+const NEVER_OVERWRITE_DIRS = ["profile"];
+function copyTemplates(src, dest, relBase) {
+    const created = [];
+    const skipped = [];
+    if (!existsSync(src))
+        return { created, skipped };
+    const entries = readdirSync(src);
+    for (const entry of entries) {
+        const srcPath = join(src, entry);
+        const destPath = join(dest, entry);
+        const rel = relative(relBase, destPath);
+        const stat = statSync(srcPath);
+        if (stat.isDirectory()) {
+            if (!existsSync(destPath)) {
+                mkdirSync(destPath, { recursive: true });
+            }
+            const sub = copyTemplates(srcPath, destPath, relBase);
+            created.push(...sub.created);
+            skipped.push(...sub.skipped);
+        }
+        else {
+            if (existsSync(destPath)) {
+                // Never overwrite profile files
+                const topDir = rel.split(/[/\\]/)[0];
+                if (NEVER_OVERWRITE_DIRS.includes(topDir)) {
+                    skipped.push(rel);
+                    continue;
+                }
+                // Other existing files: also skip (idempotent)
+                skipped.push(rel);
+            }
+            else {
+                mkdirSync(join(destPath, ".."), { recursive: true });
+                copyFileSync(srcPath, destPath);
+                created.push(rel);
+            }
+        }
+    }
+    return { created, skipped };
+}
+export function registerInitCommand(program) {
+    program
+        .command("init")
+        .description("Initialize content workspace — copy templates to project directory")
+        .requiredOption("--path <dir>", "Project directory to initialize")
+        .action((opts) => {
+        const dest = opts.path;
+        // Check templates exist
+        if (!existsSync(TEMPLATES_DIR)) {
+            console.error(`Error: Templates not found at ${TEMPLATES_DIR}\n` +
+                `Run: npm install -g @9000ai/cli`);
+            process.exit(1);
+        }
+        // Create dest if needed
+        if (!existsSync(dest)) {
+            mkdirSync(dest, { recursive: true });
+        }
+        const { created, skipped } = copyTemplates(TEMPLATES_DIR, dest, dest);
+        // Output summary
+        if (created.length > 0) {
+            console.log(`Created ${created.length} files:`);
+            created.forEach((f) => console.log(`  + ${f}`));
+        }
+        if (skipped.length > 0) {
+            console.log(`Skipped ${skipped.length} existing files:`);
+            skipped.forEach((f) => console.log(`  - ${f}`));
+        }
+        if (created.length === 0 && skipped.length === 0) {
+            console.log("No template files found.");
+            return;
+        }
+        console.log(`\nWorkspace ready: ${dest}`);
+        console.log(`\nNext steps:`);
+        console.log(`  1. Fill in profile/ (identity.md, voice.md, topics.md, product.md)`);
+        console.log(`  2. cd ${dest}`);
+        console.log(`  3. Open Claude Code and start working`);
+    });
+}

package/dist/commands/monitor.js

@@ -5,9 +5,11 @@ export function registerMonitorCommands(parent) {
     cmd
         .command("list-creators")
         .description("List all monitoring targets")
-        .action(async () => {
+        .option("--fields <list>", "Comma-separated fields to extract")
+        .option("--compact", "One JSON object per line")
+        .action(async (opts) => {
         const data = await request({ method: "GET", path: "/api/v1/douyin/monitor/creators" });
-        printJson(data);
+        printJson(data, { fields: opts.fields, compact: opts.compact });
     });
     cmd
         .command("create-creator")
@@ -83,19 +85,40 @@ export function registerMonitorCommands(parent) {
         .option("--page <n>", "Page number", "1")
         .option("--page-size <n>", "Items per page", "20")
         .option("--status <status>", "Filter by status")
+        .option("--fields <list>", "Comma-separated fields to extract")
+        .option("--compact", "One JSON object per line")
         .action(async (opts) => {
         const params = new URLSearchParams({ page: opts.page, page_size: opts.pageSize });
         if (opts.status)
             params.set("status", opts.status);
         const data = await request({ method: "GET", path: `/api/v1/douyin/monitor/runs?${params}` });
-        printJson(data);
+        printJson(data, { fields: opts.fields, compact: opts.compact });
     });
     cmd
         .command("run-detail")
         .description("View single run details")
         .requiredOption("--run-id <id>", "Run ID")
+        .option("--fields <list>", "Comma-separated fields to extract")
+        .option("--compact", "One JSON object per line")
         .action(async (opts) => {
         const data = await request({ method: "GET", path: `/api/v1/douyin/monitor/runs/${opts.runId}` });
-        printJson(data);
+        printJson(data, { fields: opts.fields, compact: opts.compact });
+    });
+    cmd
+        .command("fetch")
+        .description("Fetch user homepage videos directly (sync, no queue)")
+        .requiredOption("--sec-user <id>", "User sec_user_id or homepage URL")
+        .option("--count <n>", "Number of videos to fetch", "20")
+        .option("--sort <n>", "Sort: 0=newest 1=hottest", "0")
+        .option("--fields <list>", "Comma-separated fields to extract")
+        .option("--compact", "One JSON object per line")
+        .action(async (opts) => {
+        const params = new URLSearchParams({
+            sec_user_id: opts.secUser,
+            count: opts.count,
+            sort_type: opts.sort,
+        });
+        const data = await request({ method: "GET", path: `/api/v1/douyin/monitor/user-posts?${params}` });
+        printJson(data, { fields: opts.fields, compact: opts.compact });
     });
 }

package/dist/commands/search.js

@@ -1,4 +1,4 @@
-import { request, printJson } from "../client.js";
+import { request, printJson, pollUntilDone } from "../client.js";
 import { writeJson, writeTsv, listOutputFiles, readOutputJson, timestampSlug } from "../output.js";
 export function registerSearchCommands(parent) {
     const cmd = parent.command("search").description("Douyin topic discovery — trending boards and keyword search");
@@ -36,6 +36,9 @@ export function registerSearchCommands(parent) {
         .option("--filter-duration <v>", "Duration filter", "")
         .option("--min-likes <n>", "Minimum likes", "0")
         .option("--min-comments <n>", "Minimum comments", "0")
+        .option("--wait", "Poll until batch completes then print results")
+        .option("--fields <list>", "Comma-separated fields to extract (used with --wait)")
+        .option("--compact", "One JSON object per line (used with --wait)")
         .action(async (keywords, opts) => {
         const payload = {
             keywords,
@@ -55,14 +58,39 @@ export function registerSearchCommands(parent) {
         });
         const resp = data;
         const inner = resp.data;
-        console.log(`Batch submitted → batch_id: ${inner?.batch_id}`);
-        console.log(`Use: 9000ai search batch-result --batch-id ${inner?.batch_id}`);
-        printJson(inner);
+        const batchId = inner?.batch_id;
+        if (!opts.wait || !batchId) {
+            console.log(`Batch submitted → batch_id: ${batchId}`);
+            console.log(`Use: 9000ai search batch-result --batch-id ${batchId}`);
+            printJson(inner);
+            return;
+        }
+        // --wait: poll then fetch results
+        console.error(`Batch submitted → ${batchId}, waiting...`);
+        await pollUntilDone(`/api/v1/tasks/batch/${batchId}`);
+        const resultData = await request({
+            method: "GET",
+            path: `/api/v1/douyin/discovery/search-videos/batch/${batchId}/results`,
+        });
+        const resultResp = resultData;
+        const resultInner = resultResp.data;
+        const items = (resultInner?.items ?? []);
+        if (items.length > 0) {
+            const slug = timestampSlug();
+            writeJson("latest_search.json", resultInner);
+            writeJson(`latest_search_${slug}.json`, resultInner);
+            writeTsv("latest_search.tsv", items);
+            writeTsv(`latest_search_${slug}.tsv`, items);
+            console.error(`${items.length} items → output/latest_search.json`);
+        }
+        printJson(resultInner, { fields: opts.fields, compact: opts.compact });
     });
     cmd
         .command("batch-result")
         .description("Query async search batch results")
         .requiredOption("--batch-id <id>", "Batch ID")
+        .option("--fields <list>", "Comma-separated fields to extract (e.g. desc,author_name,likes,video_url)")
+        .option("--compact", "One JSON object per line, no indentation")
         .action(async (opts) => {
         const data = await request({
             method: "GET",
@@ -79,7 +107,7 @@ export function registerSearchCommands(parent) {
             writeTsv(`latest_search_${slug}.tsv`, items);
             console.log(`${items.length} items → output/latest_search.json`);
         }
-        printJson(inner);
+        printJson(inner, { fields: opts.fields, compact: opts.compact });
     });
     cmd
         .command("list-output")

package/dist/commands/task.js

@@ -5,16 +5,20 @@ export function registerTaskCommands(parent) {
         .command("status")
         .description("Check task status")
         .requiredOption("--task-id <id>", "Task ID")
+        .option("--fields <list>", "Comma-separated fields to extract (e.g. desc,video_url,likes)")
+        .option("--compact", "One JSON object per line, no indentation")
         .action(async (opts) => {
         const data = await request({ method: "GET", path: `/api/v1/tasks/${opts.taskId}` });
-        printJson(data);
+        printJson(data, { fields: opts.fields, compact: opts.compact });
     });
     cmd
         .command("results")
         .description("Get task results")
         .requiredOption("--task-id <id>", "Task ID")
+        .option("--fields <list>", "Comma-separated fields to extract (e.g. desc,video_url,likes)")
+        .option("--compact", "One JSON object per line, no indentation")
         .action(async (opts) => {
         const data = await request({ method: "GET", path: `/api/v1/tasks/${opts.taskId}/results` });
-        printJson(data);
+        printJson(data, { fields: opts.fields, compact: opts.compact });
     });
 }

package/dist/commands/transcribe.js

@@ -1,4 +1,4 @@
-import { request, printJson } from "../client.js";
+import { request, printJson, pollUntilDone } from "../client.js";
 import { loadJsonFile } from "../utils/format.js";
 export function registerTranscribeCommands(parent) {
     const cmd = parent.command("transcribe").description("Video / audio transcription");
@@ -6,6 +6,9 @@ export function registerTranscribeCommands(parent) {
         .command("submit")
         .description("Submit batch video-to-text task")
         .requiredOption("--json-file <path>", "JSON file with video task details")
+        .option("--wait", "Poll until task completes then print results")
+        .option("--fields <list>", "Comma-separated fields to extract (used with --wait)")
+        .option("--compact", "One JSON object per line (used with --wait)")
         .action(async (opts) => {
         const payload = loadJsonFile(opts.jsonFile);
         const data = await request({
@@ -13,7 +16,21 @@ export function registerTranscribeCommands(parent) {
             path: "/api/v1/media/batch-video-to-text",
             payload,
         });
-        printJson(data);
+        const resp = data;
+        const inner = resp.data;
+        const batchId = inner?.batch_id;
+        if (!opts.wait || !batchId) {
+            printJson(data);
+            return;
+        }
+        // --wait: poll then fetch results
+        console.error(`Batch submitted → ${batchId}, waiting...`);
+        await pollUntilDone(`/api/v1/tasks/batch/${batchId}`);
+        const resultData = await request({
+            method: "GET",
+            path: `/api/v1/tasks/batch/${batchId}`,
+        });
+        printJson(resultData, { fields: opts.fields, compact: opts.compact });
     });
     cmd
         .command("text")

package/dist/index.d.ts

@@ -1 +1,2 @@
+#!/usr/bin/env node
 export {};

package/dist/index.js

@@ -1,3 +1,4 @@
+#!/usr/bin/env node
 import { Command } from "commander";
 import { registerConfigCommands } from "./commands/config.js";
 import { registerAuthCommands } from "./commands/auth.js";
@@ -7,11 +8,12 @@ import { registerTranscribeCommands } from "./commands/transcribe.js";
 import { registerTaskCommands } from "./commands/task.js";
 import { registerFeedbackCommands } from "./commands/feedback.js";
 import { registerSkillCommands } from "./commands/skill.js";
+import { registerInitCommand } from "./commands/init.js";
 const program = new Command();
 program
     .name("9000ai")
     .description("9000AI Toolbox CLI — unified interface for 9000AI platform")
-    .version("0.1.0");
+    .version("0.5.0");
 registerConfigCommands(program);
 registerAuthCommands(program);
 registerSearchCommands(program);
@@ -20,6 +22,7 @@ registerTranscribeCommands(program);
 registerTaskCommands(program);
 registerFeedbackCommands(program);
 registerSkillCommands(program);
+registerInitCommand(program);
 program.parseAsync(process.argv).catch((err) => {
     console.error(err);
     process.exit(1);

package/dist/postinstall.js

@@ -14,4 +14,9 @@ if (!existsSync(src)) {
 }
 mkdirSync(dest, { recursive: true });
 cpSync(src, dest, { recursive: true, force: true });
-console.log(`9000AI skills installed → ${dest}`);
+console.log(`\n9000AI CLI installed successfully!\n`);
+console.log(`Skills installed: ${dest}\n`);
+console.log(`Get started:`);
+console.log(`  1. 9000ai config set --base-url <server-url> --api-key <key>`);
+console.log(`  2. 9000ai init --path <your-project-dir>`);
+console.log(`  3. cd <your-project-dir> → open Claude Code → start working\n`);

package/package.json

@@ -1,10 +1,10 @@
 {
   "name": "@9000ai/cli",
-  "version": "0.3.0",
+  "version": "0.5.0",
   "description": "9000AI Toolbox CLI — unified command-line interface for 9000AI platform",
   "type": "module",
   "bin": {
-    "9000ai": "./dist/index"
+    "9000ai": "./dist/index.js"
   },
   "scripts": {
     "build": "tsc",