ai-world-sdk 1.2.1 → 1.2.3

package/dist/index.d.ts

@@ -28,8 +28,9 @@ export { VideoGenerationClient, type VideoGenerationConfig, type VideoGeneration
 export { OpenAIVideoGenerationClient, type OpenAIVideoGenerationConfig, type OpenAIVideoGenerationRequest, type OpenAIVideoTaskResponse, };
 export { DownloadClient, type DownloadConfig, type DownloadOptions, type StreamDownloadOptions, };
 export { MinioStorageClient, type MinioStorageConfig, type UploadResponse, type ObjectInfo, type PresignedUrlResponse, type UploadOptions, type ListOptions, } from "./minio";
+export { ResourceClient, type ResourceClientConfig, type ResourceInfo, type AccessLevel, type UploadResourceOptions, type ListResourceOptions, } from "./resource";
 export { AuthClient, getCurrentUserInfo, type AuthConfig, type UserInfo, };
-export { sdkConfig, VersionCompatibilityError, SDK_SIGNATURE } from "./config";
+export { sdkConfig, VersionCompatibilityError, SDK_SIGNATURE, type AuthenticatedUser } from "./config";
 /**
  * Create a chat model instance based on model name
  */

package/dist/index.js

@@ -20,7 +20,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SDK_SIGNATURE = exports.VersionCompatibilityError = exports.sdkConfig = exports.getCurrentUserInfo = exports.AuthClient = exports.MinioStorageClient = exports.DownloadClient = exports.OpenAIVideoGenerationClient = exports.VideoGenerationClient = exports.GeminiImageGenerationClient = exports.DoubaoImageGenerationClient = exports.ChatAnthropic = exports.ChatGoogleGenerativeAI = exports.ChatOpenAI = exports.BaseChatModel = exports.AIMessageChunk = exports.SystemMessage = exports.AIMessage = exports.HumanMessage = void 0;
+exports.SDK_SIGNATURE = exports.VersionCompatibilityError = exports.sdkConfig = exports.getCurrentUserInfo = exports.AuthClient = exports.ResourceClient = exports.MinioStorageClient = exports.DownloadClient = exports.OpenAIVideoGenerationClient = exports.VideoGenerationClient = exports.GeminiImageGenerationClient = exports.DoubaoImageGenerationClient = exports.ChatAnthropic = exports.ChatGoogleGenerativeAI = exports.ChatOpenAI = exports.BaseChatModel = exports.AIMessageChunk = exports.SystemMessage = exports.AIMessage = exports.HumanMessage = void 0;
 exports.createChatModel = createChatModel;
 const openai_1 = require("./chat_models/openai");
 const google_1 = require("./chat_models/google");
@@ -56,6 +56,9 @@ Object.defineProperty(exports, "ChatAnthropic", { enumerable: true, get: functio
 // Export MinIO storage client
 var minio_1 = require("./minio");
 Object.defineProperty(exports, "MinioStorageClient", { enumerable: true, get: function () { return minio_1.MinioStorageClient; } });
+// Export resource management client
+var resource_1 = require("./resource");
+Object.defineProperty(exports, "ResourceClient", { enumerable: true, get: function () { return resource_1.ResourceClient; } });
 // Export global configuration
 var config_2 = require("./config");
 Object.defineProperty(exports, "sdkConfig", { enumerable: true, get: function () { return config_2.sdkConfig; } });

package/dist/login.d.ts

@@ -0,0 +1,32 @@
+/**
+ * AI World SDK - Login utilities (Node.js only)
+ * 登录工具函数，仅在 Node.js 环境使用（Vite Plugin / CLI）
+ */
+/**
+ * 从项目根目录推断 pluginId：plugin.json plugin_id → package.json name → 目录名
+ */
+export declare function resolvePluginId(projectRoot: string): string;
+export interface LoginOptions {
+    baseUrl?: string;
+    timeout?: number;
+}
+export interface LoginResult {
+    token: string;
+    baseUrl: string;
+}
+/**
+ * 验证 token 是否有效
+ */
+export declare function validateToken(baseUrl: string, token: string): Promise<boolean>;
+/**
+ * 读取 .env.local 中的环境变量
+ */
+export declare function readEnvLocal(projectRoot: string): Record<string, string>;
+/**
+ * 写入/合并 .env.local，只更新指定的 key，保留其他内容
+ */
+export declare function writeEnvLocal(projectRoot: string, updates: Record<string, string>): void;
+/**
+ * 执行登录流程：启动本地 HTTP 服务器 → 打开浏览器 → 等待回调
+ */
+export declare function performLogin(options?: LoginOptions): Promise<LoginResult>;

package/dist/login.js

@@ -0,0 +1,263 @@
+"use strict";
+/**
+ * AI World SDK - Login utilities (Node.js only)
+ * 登录工具函数，仅在 Node.js 环境使用（Vite Plugin / CLI）
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolvePluginId = resolvePluginId;
+exports.validateToken = validateToken;
+exports.readEnvLocal = readEnvLocal;
+exports.writeEnvLocal = writeEnvLocal;
+exports.performLogin = performLogin;
+const http = __importStar(require("http"));
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const child_process_1 = require("child_process");
+/**
+ * 从项目根目录推断 pluginId：plugin.json plugin_id → package.json name → 目录名
+ */
+function resolvePluginId(projectRoot) {
+    try {
+        const pluginJsonPath = path.join(projectRoot, "plugin.json");
+        if (fs.existsSync(pluginJsonPath)) {
+            const data = JSON.parse(fs.readFileSync(pluginJsonPath, "utf-8"));
+            if (data.plugin_id)
+                return data.plugin_id;
+        }
+    }
+    catch {
+        // ignore parse errors
+    }
+    try {
+        const pkgJsonPath = path.join(projectRoot, "package.json");
+        if (fs.existsSync(pkgJsonPath)) {
+            const data = JSON.parse(fs.readFileSync(pkgJsonPath, "utf-8"));
+            if (data.name)
+                return data.name;
+        }
+    }
+    catch {
+        // ignore parse errors
+    }
+    return path.basename(projectRoot);
+}
+const CALLBACK_PORT = 18000;
+const LOGIN_TIMEOUT_MS = 60000;
+const DEFAULT_BASE_URL = "https://aiworld.local:8000";
+/**
+ * 验证 token 是否有效
+ */
+async function validateToken(baseUrl, token) {
+    try {
+        const response = await fetch(`${baseUrl}/api/auth/me`, {
+            method: "GET",
+            headers: {
+                Authorization: `Bearer ${token}`,
+                "Content-Type": "application/json",
+            },
+        });
+        return response.ok;
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * 强制确保端口可用：如果被占用则 kill 占用进程
+ */
+function ensurePort(port) {
+    try {
+        const result = (0, child_process_1.execSync)(`lsof -ti:${port}`, { encoding: "utf-8" }).trim();
+        if (result) {
+            const pids = result.split("\n").filter(Boolean);
+            for (const pid of pids) {
+                try {
+                    (0, child_process_1.execSync)(`kill -9 ${pid}`, { stdio: "ignore" });
+                }
+                catch {
+                    // pid may have already exited
+                }
+            }
+            // Brief wait for OS to release the port
+            (0, child_process_1.execSync)("sleep 0.3");
+        }
+    }
+    catch {
+        // lsof returns non-zero when no process found — port is free
+    }
+}
+/**
+ * 打开浏览器
+ */
+function openBrowser(url) {
+    const platform = process.platform;
+    const cmd = platform === "darwin"
+        ? "open"
+        : platform === "win32"
+            ? "start"
+            : "xdg-open";
+    (0, child_process_1.exec)(`${cmd} "${url}"`, (err) => {
+        if (err) {
+            console.log(`[ai-world] 无法自动打开浏览器，请手动访问: ${url}`);
+        }
+    });
+}
+/**
+ * 读取 .env.local 中的环境变量
+ */
+function readEnvLocal(projectRoot) {
+    const envPath = path.join(projectRoot, ".env.local");
+    const vars = {};
+    if (!fs.existsSync(envPath))
+        return vars;
+    const content = fs.readFileSync(envPath, "utf-8");
+    for (const line of content.split("\n")) {
+        const trimmed = line.trim();
+        if (!trimmed || trimmed.startsWith("#"))
+            continue;
+        const eqIdx = trimmed.indexOf("=");
+        if (eqIdx === -1)
+            continue;
+        const key = trimmed.slice(0, eqIdx).trim();
+        const value = trimmed.slice(eqIdx + 1).trim();
+        vars[key] = value;
+    }
+    return vars;
+}
+/**
+ * 写入/合并 .env.local，只更新指定的 key，保留其他内容
+ */
+function writeEnvLocal(projectRoot, updates) {
+    const envPath = path.join(projectRoot, ".env.local");
+    let lines = [];
+    if (fs.existsSync(envPath)) {
+        lines = fs.readFileSync(envPath, "utf-8").split("\n");
+    }
+    const updatedKeys = new Set();
+    for (let i = 0; i < lines.length; i++) {
+        const trimmed = lines[i].trim();
+        if (!trimmed || trimmed.startsWith("#"))
+            continue;
+        const eqIdx = trimmed.indexOf("=");
+        if (eqIdx === -1)
+            continue;
+        const key = trimmed.slice(0, eqIdx).trim();
+        if (key in updates) {
+            lines[i] = `${key}=${updates[key]}`;
+            updatedKeys.add(key);
+        }
+    }
+    for (const [key, value] of Object.entries(updates)) {
+        if (!updatedKeys.has(key)) {
+            lines.push(`${key}=${value}`);
+        }
+    }
+    // Remove trailing empty lines then add single newline at end
+    while (lines.length > 0 && lines[lines.length - 1].trim() === "") {
+        lines.pop();
+    }
+    lines.push("");
+    fs.writeFileSync(envPath, lines.join("\n"), "utf-8");
+}
+/**
+ * 执行登录流程：启动本地 HTTP 服务器 → 打开浏览器 → 等待回调
+ */
+function performLogin(options = {}) {
+    const baseUrl = options.baseUrl || DEFAULT_BASE_URL;
+    const timeout = options.timeout || LOGIN_TIMEOUT_MS;
+    return new Promise((resolve, reject) => {
+        ensurePort(CALLBACK_PORT);
+        let settled = false;
+        let timeoutId;
+        const server = http.createServer((req, res) => {
+            const url = new URL(req.url || "/", `http://localhost:${CALLBACK_PORT}`);
+            if (url.pathname === "/callback") {
+                const token = url.searchParams.get("token");
+                res.writeHead(200, { "Content-Type": "text/html; charset=utf-8" });
+                if (token) {
+                    res.end(`
+<!DOCTYPE html>
+<html><head><meta charset="utf-8"><title>AI World 登录成功</title></head>
+<body style="display:flex;justify-content:center;align-items:center;height:100vh;font-family:system-ui;background:#f5f5f5">
+<div style="text-align:center;padding:40px;background:white;border-radius:12px;box-shadow:0 2px 12px rgba(0,0,0,.1)">
+<h2 style="color:#52c41a">✓ 登录成功</h2>
+<p style="color:#666">Token 已保存，可以关闭此页面</p>
+</div></body></html>`);
+                    if (!settled) {
+                        settled = true;
+                        clearTimeout(timeoutId);
+                        server.close();
+                        resolve({ token, baseUrl });
+                    }
+                }
+                else {
+                    res.end(`
+<!DOCTYPE html>
+<html><head><meta charset="utf-8"><title>AI World 登录失败</title></head>
+<body style="display:flex;justify-content:center;align-items:center;height:100vh;font-family:system-ui;background:#f5f5f5">
+<div style="text-align:center;padding:40px;background:white;border-radius:12px;box-shadow:0 2px 12px rgba(0,0,0,.1)">
+<h2 style="color:#ff4d4f">✗ 登录失败</h2>
+<p style="color:#666">未收到 token，请重试</p>
+</div></body></html>`);
+                }
+                return;
+            }
+            res.writeHead(404);
+            res.end("Not Found");
+        });
+        server.on("error", (err) => {
+            if (!settled) {
+                settled = true;
+                reject(new Error(`无法启动本地回调服务器: ${err.message}`));
+            }
+        });
+        server.listen(CALLBACK_PORT, () => {
+            const callbackUrl = `http://localhost:${CALLBACK_PORT}/callback`;
+            const loginUrl = `${baseUrl}/api/auth/login?redirect=${encodeURIComponent(callbackUrl)}`;
+            console.log(`[ai-world] 正在打开浏览器登录...`);
+            console.log(`[ai-world] 如果浏览器未自动打开，请手动访问:`);
+            console.log(`[ai-world] ${loginUrl}`);
+            openBrowser(loginUrl);
+            timeoutId = setTimeout(() => {
+                if (!settled) {
+                    settled = true;
+                    server.close();
+                    reject(new Error(`登录超时（${timeout / 1000}s），请重新启动 dev server`));
+                }
+            }, timeout);
+        });
+    });
+}

package/dist/resource.d.ts

@@ -0,0 +1,137 @@
+/**
+ * Resource Client
+ * 资源管理客户端
+ * 通过后端 /api/resources 代理操作资源管理系统
+ * 支持 private / specific_users / public 三级访问控制，默认按 plugin_id 隔离
+ */
+export interface ResourceClientConfig {
+    baseUrl?: string;
+    token?: string;
+    headers?: Record<string, string>;
+    pluginId?: string;
+}
+export type AccessLevel = "private" | "specific_users" | "public";
+export interface ResourceInfo {
+    id: number;
+    owner_id: number;
+    plugin_id: string;
+    path: string;
+    object_path: string;
+    filename: string;
+    size: number;
+    content_type: string;
+    access_level: AccessLevel;
+    description?: string;
+    created_at: string;
+    updated_at: string;
+    shared_users?: number[];
+}
+export interface UploadResourceOptions {
+    accessLevel?: AccessLevel;
+    description?: string;
+    contentType?: string;
+}
+export interface ListResourceOptions {
+    crossPlugin?: boolean;
+}
+/**
+ * Resource Client
+ * 资源管理客户端类
+ *
+ * @example
+ * ```typescript
+ * import { ResourceClient, sdkConfig } from 'ai-world-sdk';
+ *
+ * sdkConfig.setPluginId('my-plugin');
+ * const resources = new ResourceClient();
+ *
+ * // 上传并设为公开
+ * const file = new File(['hello'], 'hello.txt', { type: 'text/plain' });
+ * const res = await resources.upload('docs/hello.txt', file, { accessLevel: 'public' });
+ *
+ * // 列出我的资源
+ * const myFiles = await resources.listMy();
+ *
+ * // 查看某用户的公开资源
+ * const userFiles = await resources.listByUser(42);
+ *
+ * // 分享给指定用户
+ * await resources.addShare(res.id, 99);
+ *
+ * // 通过 userId + path 下载
+ * const blob = await resources.downloadByUser(42, 'docs/hello.txt');
+ * ```
+ */
+export declare class ResourceClient {
+    private baseUrl;
+    private headers;
+    constructor(config?: ResourceClientConfig);
+    private ensurePluginId;
+    private handleErrorResponse;
+    /**
+     * 上传文件并创建资源记录
+     */
+    upload(path: string, file: File | Blob, options?: UploadResourceOptions): Promise<ResourceInfo>;
+    /**
+     * 列出当前用户的资源
+     */
+    listMy(options?: ListResourceOptions): Promise<ResourceInfo[]>;
+    /**
+     * 获取资源详情
+     */
+    getInfo(resourceId: number): Promise<ResourceInfo>;
+    /**
+     * 修改资源访问级别
+     */
+    updateAccess(resourceId: number, accessLevel: AccessLevel): Promise<ResourceInfo>;
+    /**
+     * 批量添加用户的访问授权
+     *
+     * @param resourceId - 资源 ID
+     * @param userIds - 被授权用户 ID 列表（支持单个或多个）
+     */
+    addShare(resourceId: number, userIds: number | number[]): Promise<{
+        resource_id: number;
+        added_user_ids: number[];
+        total_requested: number;
+    }>;
+    /**
+     * 批量移除用户的访问授权
+     *
+     * @param resourceId - 资源 ID
+     * @param userIds - 要移除授权的用户 ID 列表（支持单个或多个）
+     */
+    removeShare(resourceId: number, userIds: number | number[]): Promise<{
+        resource_id: number;
+        removed_user_ids: number[];
+        total_requested: number;
+    }>;
+    /**
+     * 通过资源 ID 下载
+     */
+    download(resourceId: number): Promise<Blob>;
+    /**
+     * 删除资源
+     */
+    delete(resourceId: number): Promise<void>;
+    /**
+     * 列出指定用户的可访问资源
+     */
+    listByUser(userId: number, options?: ListResourceOptions): Promise<ResourceInfo[]>;
+    /**
+     * 通过 userId + path 下载资源
+     */
+    downloadByUser(userId: number, path: string, pluginId?: string): Promise<Blob>;
+    /**
+     * 列出被分享给当前用户的资源
+     */
+    listSharedWithMe(options?: ListResourceOptions): Promise<ResourceInfo[]>;
+    /**
+     * 列出所有 public 资源
+     */
+    listPublic(options?: ListResourceOptions): Promise<ResourceInfo[]>;
+    /**
+     * 下载资源并触发浏览器下载
+     */
+    downloadAsFile(resourceId: number, filename?: string): Promise<void>;
+}