mirror of https://github.com/usememos/memos.git
83 lines
3.2 KiB
TypeScript
83 lines
3.2 KiB
TypeScript
import * as api from "@/helpers/api";
|
|
import { useTranslate } from "@/utils/i18n";
|
|
import store, { useAppSelector } from "../";
|
|
import { deleteResource, patchResource, setResources } from "../reducer/resource";
|
|
import { useGlobalStore } from "./global";
|
|
|
|
const convertResponseModelResource = (resource: Resource): Resource => {
|
|
return {
|
|
...resource,
|
|
createdTs: resource.createdTs * 1000,
|
|
updatedTs: resource.updatedTs * 1000,
|
|
};
|
|
};
|
|
|
|
export const useResourceStore = () => {
|
|
const state = useAppSelector((state) => state.resource);
|
|
const t = useTranslate();
|
|
const globalStore = useGlobalStore();
|
|
const maxUploadSizeMiB = globalStore.state.systemStatus.maxUploadSizeMiB;
|
|
|
|
return {
|
|
state,
|
|
getState: () => {
|
|
return store.getState().resource;
|
|
},
|
|
async fetchResourceList(): Promise<Resource[]> {
|
|
const { data } = await api.getResourceList();
|
|
const resourceList = data.map((m) => convertResponseModelResource(m));
|
|
store.dispatch(setResources(resourceList));
|
|
return resourceList;
|
|
},
|
|
async createResource(resourceCreate: ResourceCreate): Promise<Resource> {
|
|
const { data } = await api.createResource(resourceCreate);
|
|
const resource = convertResponseModelResource(data);
|
|
const resourceList = state.resources;
|
|
store.dispatch(setResources([resource, ...resourceList]));
|
|
return resource;
|
|
},
|
|
async createResourceWithBlob(file: File): Promise<Resource> {
|
|
const { name: filename, size } = file;
|
|
if (size > maxUploadSizeMiB * 1024 * 1024) {
|
|
return Promise.reject(t("message.maximum-upload-size-is", { size: maxUploadSizeMiB }));
|
|
}
|
|
|
|
const formData = new FormData();
|
|
formData.append("file", file, filename);
|
|
const { data } = await api.createResourceWithBlob(formData);
|
|
const resource = convertResponseModelResource(data);
|
|
const resourceList = state.resources;
|
|
store.dispatch(setResources([resource, ...resourceList]));
|
|
return resource;
|
|
},
|
|
async createResourcesWithBlob(files: FileList): Promise<Array<Resource>> {
|
|
let newResourceList: Array<Resource> = [];
|
|
for (const file of files) {
|
|
const { name: filename, size } = file;
|
|
if (size > maxUploadSizeMiB * 1024 * 1024) {
|
|
return Promise.reject(t("message.file-exceeds-upload-limit-of", { file: filename, size: maxUploadSizeMiB }));
|
|
}
|
|
|
|
const formData = new FormData();
|
|
formData.append("file", file, filename);
|
|
const { data } = await api.createResourceWithBlob(formData);
|
|
const resource = convertResponseModelResource(data);
|
|
newResourceList = [resource, ...newResourceList];
|
|
}
|
|
const resourceList = state.resources;
|
|
store.dispatch(setResources([...newResourceList, ...resourceList]));
|
|
return newResourceList;
|
|
},
|
|
async deleteResourceById(id: ResourceId) {
|
|
await api.deleteResourceById(id);
|
|
store.dispatch(deleteResource(id));
|
|
},
|
|
async patchResource(resourcePatch: ResourcePatch): Promise<Resource> {
|
|
const { data } = await api.patchResource(resourcePatch);
|
|
const resource = convertResponseModelResource(data);
|
|
store.dispatch(patchResource(resource));
|
|
return resource;
|
|
},
|
|
};
|
|
};
|