@lobehub/chat/src/services/file/ClientS3/index.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import { createStore, del, get, set } from 'idb-keyval';

const BROWSER_S3_DB_NAME = 'lobechat-local-s3';

export class BrowserS3Storage {
  private store;

  constructor() {
    // skip server-side rendering
    if (typeof window === 'undefined') return;

    this.store = createStore(BROWSER_S3_DB_NAME, 'objects');
  }

  /**
   * 上传文件
   * @param key 文件 hash
   * @param file File 对象
   */
  putObject = async (key: string, file: File): Promise<void> => {
    try {
      const data = await file.arrayBuffer();
      await set(key, { data, name: file.name, type: file.type }, this.store);
    } catch (e) {
      throw new Error(`Failed to put file ${file.name}: ${(e as Error).message}`);
    }
  };

  /**
   * 获取文件
   * @param key 文件 hash
   * @returns File 对象
   */
  getObject = async (key: string): Promise<File | undefined> => {
    try {
      const res = await get<{ data: ArrayBuffer; name: string; type: string }>(key, this.store);
      if (!res) return;

      return new File([res.data], res!.name, { type: res?.type });
    } catch (e) {
      console.log(`Failed to get object (key=${key}):`, e);
      return undefined;
    }
  };

  /**
   * 删除文件
   * @param key 文件 hash
   */
  deleteObject = async (key: string): Promise<void> => {
    try {
      await del(key, this.store);
    } catch (e) {
      throw new Error(`Failed to delete object (key=${key}): ${(e as Error).message}`);
    }
  };
}

export const clientS3Storage = new BrowserS3Storage();