76 lines
2.2 KiB
TypeScript
76 lines
2.2 KiB
TypeScript
import { useInvalidate, useNotification } from "@refinedev/core";
|
|
import { useMutation, useQueryClient } from "@tanstack/react-query";
|
|
import { useCallback, useContext } from "react";
|
|
import { PinningProcess } from "~/data/pinning";
|
|
import { PinningContext } from "~/providers/PinningProvider";
|
|
|
|
export const usePinning = () => {
|
|
const queryClient = useQueryClient();
|
|
const invalidate = useInvalidate();
|
|
const context = useContext(PinningContext);
|
|
const { open } = useNotification();
|
|
|
|
const { status: pinStatus, data: pinData, mutate: pinMutation } = useMutation({
|
|
mutationKey: ["pin-mutation"],
|
|
mutationFn: async (variables: { cid: string }) => {
|
|
const { cid } = variables;
|
|
const response = await PinningProcess.pin(cid);
|
|
|
|
if (!response.success) {
|
|
open?.({
|
|
type: "error",
|
|
message: `Error pinning ${cid}`,
|
|
description: response.message,
|
|
});
|
|
return Promise.reject(response);
|
|
}
|
|
|
|
queryClient.invalidateQueries({ queryKey: ["pin-progress", "file"] });
|
|
invalidate({ resource: "file", invalidates: ["list"] });
|
|
return Promise.resolve(response);
|
|
},
|
|
});
|
|
|
|
const { status: unpinStatus, data: unpinData, mutate: unpinMutation } = useMutation({
|
|
mutationKey: ["unpin-mutation"],
|
|
mutationFn: async (variables: { cid: string }) => {
|
|
const { cid } = variables;
|
|
const response = await PinningProcess.unpin(cid);
|
|
|
|
if (!response.success) {
|
|
open?.({
|
|
type: "error",
|
|
message: `Error removing ${cid}`,
|
|
description: response.message,
|
|
});
|
|
return Promise.reject(response);
|
|
}
|
|
queryClient.invalidateQueries({ queryKey: ["pin-progress"] });
|
|
invalidate({ resource: "file", invalidates: ["list"] });
|
|
return Promise.resolve(response);
|
|
},
|
|
});
|
|
|
|
const bulkPin = useCallback(
|
|
(cids: string[]) => {
|
|
for (const cid of cids) {
|
|
pinMutation({ cid });
|
|
}
|
|
},
|
|
[pinMutation],
|
|
);
|
|
|
|
return {
|
|
progressStatus: context.query.status,
|
|
progressData: context.query.data,
|
|
fetchProgress: context.query.refetch,
|
|
pinStatus,
|
|
pinData,
|
|
unpinStatus,
|
|
unpinData,
|
|
pin: pinMutation,
|
|
unpin: unpinMutation,
|
|
bulkPin,
|
|
};
|
|
};
|