封装ossStore

tone-page-web/lib/oss/OssStore.ts

@@ -0,0 +1,150 @@
+import { useOssStore } from "@/hooks/admin/web/blog/use-oss-store";
+import { Dispatch, SetStateAction, useEffect, useState } from "react";
+
+interface OssObjectItem {
+    id: string;
+    name: string;
+    size: number;// Byte
+    lastModified: Date;
+    isChecked: boolean;
+};
+
+type OssObjectList = null | OssObjectItem[];
+
+export class OssStore {
+
+    private objectList: OssObjectList = null;
+    private setObjectList: Dispatch<SetStateAction<OssObjectList>>;
+
+    public storeMeta: ReturnType<typeof useOssStore>;
+
+    constructor(private options: {
+        prefix?: string;
+        prefixAddUserId?: boolean;
+    } = {}) {
+        const [objectList, setObjectList] = useState<OssObjectList>(null);
+        this.objectList = objectList;
+        this.setObjectList = setObjectList;
+
+        this.storeMeta = useOssStore({
+            region: 'oss-cn-chengdu',
+            bucket: 'tone-personal',
+            onStsTokenDataChanged: (data) => {
+                if (!data) return;
+                this.loadObjectList();
+            }
+        });
+    }
+
+    get useObjectList() {
+        return this.objectList;
+    }
+
+    public async loadObjectList() {
+        const store = await this.getStore();
+        this.setObjectList(null);
+
+        const workDir = this.getWorkDir();
+        const res = await store.listV2({ ...workDir ? { prefix: workDir } : {} }, {});
+        if (!res || !res.objects) throw new Error('文件列表加载失败');
+
+        this.setObjectList(res.objects.map(v => ({
+            id: v.name,
+            name: v.name.replace(`${workDir}/`, ''),
+            size: v.size,
+            lastModified: new Date(v.lastModified),
+            isChecked: false,
+        })))
+    }
+
+    public handleObjectCheckedStateChanged(id: string, value: boolean) {
+        this.setObjectList(current => current ? current.map(objectItem => {
+            if (objectItem.id === id) {
+                return { ...objectItem, isChecked: value }
+            }
+            return objectItem;
+        }) : null)
+    }
+
+    public async deleteObject(id: string) {
+        if (!this.storeMeta.store || !this.storeMeta.stsTokenData) {
+            throw new Error('初始化失败，请刷新界面重试');
+        }
+        const fileItem = (this.objectList || []).find(i => i.id === id);
+        if (!fileItem) throw new Error('文件不存在');
+
+        const objectName = this.getObjectName(fileItem.name);
+        const delRes = await this.storeMeta.store.delete(objectName).catch(() => null);
+        if (!delRes) throw new Error('删除失败');
+    }
+
+    public async deleteCheckedObjects() {
+        if (!this.storeMeta.store || !this.storeMeta.stsTokenData) {
+            throw new Error('初始化失败，请刷新界面重试');
+        }
+
+        const objects = (this.objectList || []).filter(i => i.isChecked);
+        if (objects.length === 0) throw new Error('请选择需要删除的文件');
+
+        let failedCount = 0;
+        for (const objectItem of objects) {
+            await this.deleteObject(objectItem.name).catch(e => failedCount++);
+        }
+
+        return { all: objects.length, failed: failedCount };
+    }
+
+    public async downloadObject(id: string) {
+        if (!this.storeMeta.store || !this.storeMeta.stsTokenData) {
+            throw new Error('初始化失败，请刷新界面重试');
+        }
+
+        const objectItem = this.getObjectItemById(id);
+        const url = this.storeMeta.store.signatureUrl(this.getObjectName(objectItem.name));
+        const a = document.createElement('a');
+        a.href = url;
+        a.download = objectItem.name;
+        a.target = '_blank';
+        document.body.appendChild(a);
+        a.click();
+        document.body.removeChild(a);
+    }
+
+    private async getStore() {
+        if (!this.storeMeta.store || !this.storeMeta.stsTokenData) {
+            throw new Error('初始化失败，请刷新界面重试');
+        }
+        return this.storeMeta.store;
+    }
+
+    private getObjectName(localName: string) {
+        return `${this.getWorkDir()}/${localName}`;
+    }
+
+    private getObjectNameById(id: string) {
+        const objectItem = this.getObjectItemById(id);
+        return this.getObjectName(objectItem.name);
+    }
+
+    private getObjectItemById(id: string) {
+        const objectItem = (this.objectList || []).find(i => i.id === id);
+        if (!objectItem) throw new Error('文件不存在');
+        return objectItem;
+    }
+
+    public getWorkDir() {
+        const { stsTokenData } = this.storeMeta;
+        if (!stsTokenData) return;
+
+        if (this.options.prefixAddUserId) {
+            return `${this.options.prefix ? `${this.options.prefix}/` : ''}${stsTokenData.userId}`;
+        } else {
+            return this.options.prefix;
+        }
+    }
+
+    get isLoading() {
+        return this.storeMeta.isLoading;
+    }
+
+}