update

src/jimeng/core.ts

@@ -1,51 +1,121 @@
-import { adapter } from '@kevisual/query/query'
-export type CoreOpts = {
-  baseURL?: string;
-  token?: string;
+import { Result } from '@kevisual/query'
+export interface JimengOptions {
+  /** API密钥，用于认证请求 */
+  apiKey: string;
+  /** API基础URL */
+  baseUrl: string;
+  /** 请求超时时间（毫秒） */
+  timeout: number;
 }
-export class Core {
-  baseURL: string = 'https://jimeng-api.kevisual.cn/v1';
-  token?: string;
-  constructor(opts: CoreOpts = {}) {
-    console.log("Core initialized");
-    if (opts.baseURL) {
-      this.baseURL = opts.baseURL;
-    }
-    if (opts.token) {
-      this.token = opts.token;
-    }
+
+export interface JimengGenerateOptions {
+  /** 图片生成提示词 */
+  prompt: string;
+  /** 使用的模型版本，默认 jimeng-4.0 */
+  model?: string;
+  /** 图片比例，默认 1:1 */
+  ratio?: string;
+  /** 图片分辨率，默认 2k */
+  resolution?: string;
+}
+
+interface JimengResponse {
+  /** 请求创建时间戳 */
+  created: number;
+  /** 生成的图片列表 */
+  data: Array<{
+    /** 图片URL */
+    url: string;
+  }>;
+}
+
+export class JimengService {
+  private apiKey: string;
+  private baseUrl: string;
+  private timeout: number;
+
+  constructor(options: JimengOptions) {
+    this.apiKey = options.apiKey;
+    this.baseUrl = options.baseUrl || 'https://jimeng-api.kevisual.cn/v1';
+    this.timeout = options.timeout;
   }
-  makeHeader() {
-    return {
-      Authorization: this.token ? `Bearer ${this.token}` : undefined,
-      'Content-Type': 'application/json'
-    }
-  }
-  generateImage({ model = 'jimeng-4.0', prompt, resolution = '2k' }: ImageOptions) {
-    const url = `${this.baseURL}/images/generations`;
-    return adapter({
-      url,
-      headers: this.makeHeader(),
-      body: {
-        model,
-        prompt,
-        resolution
+
+  async generateImage(options: JimengGenerateOptions): Promise<Result<JimengResponse>> {
+    const {
+      prompt,
+      model = 'jimeng-4.6',
+      ratio = '1:1',
+      resolution = '2k'
+    } = options;
+
+    try {
+      const controller = new AbortController();
+      const timeoutId = setTimeout(() => controller.abort(), this.timeout);
+
+      const response = await fetch(`${this.baseUrl}/images/generations`, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          'Authorization': `Bearer ${this.apiKey}`,
+        },
+        body: JSON.stringify({
+          model,
+          prompt,
+          ratio,
+          resolution,
+        }),
+        signal: controller.signal,
+      });
+
+      clearTimeout(timeoutId);
+
+      if (!response.ok) {
+        throw new Error(`jimeng API error: ${response.status} ${response.statusText}`);
       }
-    });
+
+      const result = await response.json() as JimengResponse;
+      return { code: 200, data: result };
+    } catch (error: any) {
+      return { code: 500, message: error.message || 'Unknown error' };
+    }
+  }
+
+  async downloadImage(url: string): Promise<Uint8Array> {
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), this.timeout);
+
+    try {
+      const response = await fetch(url, {
+        signal: controller.signal,
+      });
+
+      clearTimeout(timeoutId);
+
+      if (!response.ok) {
+        throw new Error(`Failed to download image: ${response.statusText}`);
+      }
+
+      const arrayBuffer = await response.arrayBuffer();
+      return new Uint8Array(arrayBuffer);
+    } catch (error: any) {
+      clearTimeout(timeoutId);
+      if (error.name === 'AbortError') {
+        throw new Error('Image download timeout');
+      }
+      throw error;
+    }
+  }
+  /** 获取图片过期时间 */
+  async getExpiredTime(url: string): Promise<{ expiredAt: number, expired: boolean }> {
+    // https://p3-dreamina-sign.byteimg.com/tos-cn-i-tb4s082cfz/c018e06ee6654dd78ccacb29eff4744e~tplv-tb4s082cfz-aigc_resize:0:0.png?lk3s=43402efa&x-expires=1767852000&x-signature=34yf37N955BP37eLaYEzKeLQn0Q%3D&format=.png
+    const urlObj = new URL(url);
+    let expires = urlObj.searchParams.get('x-expires');
+    if (!expires) {
+      expires = '0';
+    }
+    const expiredAt = parseInt(expires) * 1000;
+    const expired = Date.now() > expiredAt;
+    return { expiredAt, expired };
   }
 }
 
-export type ImageOptions = {
-  model?: string;
-  prompt: string;
-  /**
-   * 宽高比，如 "16:9", "4:3", "1:1" 等
-   */
-  ratio?: string;
-  /**
-   *  
-   * 图片分辨率，如 "1024x768", "512x512" 等
-   *  4k 2k
-   */
-  resolution?: string;
-}