Presupuestador_web/client/src/app/quotes/hooks/useQuotes.tsx

import { useDownloader } from "@/lib/hooks";
import { UseListQueryResult, useCustom, useList, useOne, useSave } from "@/lib/hooks/useDataSource";
import {
  IFilterItemDataProviderParam,
  IGetListDataProviderParams,
} from "@/lib/hooks/useDataSource/DataSource";
import { TDataSourceError } from "@/lib/hooks/useDataSource/types";
import { useDataSource } from "@/lib/hooks/useDataSource/useDataSource";
import { useQueryKey } from "@/lib/hooks/useQueryKey";
import {
  ICreateQuote_Request_DTO,
  ICreateQuote_Response_DTO,
  IGetQuote_Response_DTO,
  IListQuotes_Response_DTO,
  IListResponse_DTO,
  IReportQuote_Response_DTO,
  IUpdateQuote_Request_DTO,
  IUpdateQuote_Response_DTO,
  UniqueID,
} from "@shared/contexts";
import { useQueryClient } from "@tanstack/react-query";
import { useCallback, useMemo, useState } from "react";

export type UseQuotesListParams = Omit<IGetListDataProviderParams, "filters" | "resource"> & {
  status?: string;
  enabled?: boolean;
  queryOptions?: Record<string, unknown>;
};

export type UseQuotesListResponse = UseListQueryResult<
  IListResponse_DTO<IListQuotes_Response_DTO>,
  unknown
>;

export type UseQuotesGetParamsType = {
  enabled?: boolean;
  queryOptions?: Record<string, unknown>;
};

export type UseQuotesReportParamsType = {
  enabled?: boolean;
  queryOptions?: Record<string, unknown>;
};

const quoteStatusFilter: Record<string, IFilterItemDataProviderParam> = {
  draft: {
    field: "status",
    operator: "eq",
    value: "draft",
  },
  archived: {
    field: "status",
    operator: "eq",
    value: "archived",
  },
  active: {
    field: "status",
    operator: "eq",
    value: "active",
  },
};

export const useQuotes = () => {
  const dataSource = useDataSource();
  const keys = useQueryKey();

  const actions = {
    useList: (params: UseQuotesListParams): UseQuotesListResponse => {
      const dataSource = useDataSource();
      const keys = useQueryKey();

      const {
        pagination,
        status = "draft",
        quickSearchTerm = undefined,
        enabled = true,
        queryOptions,
      } = params;

      return useList({
        queryKey: keys().data().resource("quotes").action("list").params(params).get(),
        queryFn: () => {
          return dataSource.getList({
            resource: "quotes",
            quickSearchTerm,
            filters: status !== "all" ? Array.of(quoteStatusFilter[status]) : undefined,
            pagination,
          });
        },
        enabled,
        queryOptions,
      });
    },

    useCreate: () =>
      useSave<ICreateQuote_Response_DTO, TDataSourceError, ICreateQuote_Request_DTO>({
        //mutationKey: keys().data().resource("quotes").action("one").id("").params().get(),
        mutationFn: (data) => {
          const { date } = data;

          const id = UniqueID.generateNewID().object.toString();
          const status = "draft";

          return dataSource.createOne({
            resource: "quotes",
            data: {
              ...data,
              date: new Date(date).toISOString().slice(0, 10),
              status,
              id,
            },
          });
        },
      }),

    useUpdate: (id: string) =>
      useSave<IUpdateQuote_Response_DTO, TDataSourceError, IUpdateQuote_Request_DTO>({
        mutationKey: keys().data().resource("quotes").action("one").id(id).params().get(),
        mutationFn: (data) => {
          const { date } = data;

          return dataSource.updateOne({
            resource: "quotes",
            id,
            data: {
              ...data,
              date: new Date(date).toISOString().slice(0, 10),
            },
          });
        },
      }),

    useOne: (id?: string, params?: UseQuotesGetParamsType) =>
      useOne<IGetQuote_Response_DTO>({
        queryKey: keys().data().resource("quotes").action("one").id(id).params().get(),
        queryFn: () =>
          dataSource.getOne({
            resource: "quotes",
            id: String(id),
          }),
        enabled: !!id,
        ...params,
      }),

    useReport2: (id?: string, params?: UseQuotesReportParamsType) => {
      const queryClient = useQueryClient();
      const queryKey = useMemo(
        () => keys().data().resource("quotes").action("report").id(id).params().get(),
        [id]
      );

      return {
        ...useCustom<ArrayBuffer, TDataSourceError, IReportQuote_Response_DTO>({
          queryKey,
          queryFn: ({ signal }) =>
            dataSource.custom({
              url: `${dataSource.getApiUrl()}/quotes/${id}/report`,
              method: "get",
              responseType: "arraybuffer",
              signal,
            }),

          enabled: !!id,
          select: useCallback(
            (data: ArrayBuffer) => ({
              original: data,
              data: new Uint8Array(data),
            }),
            []
          ),
          ...params,
        }),
        cancelQuery: () => queryClient.cancelQueries({ queryKey }),
      };
    },

    getQuotePDFDownloadURL: (id: string) => `${dataSource.getApiUrl()}/quotes/${id}/report`,

    getQuotePDFFilename: (quote: IListQuotes_Response_DTO | IGetQuote_Response_DTO) =>
      `filename-quote.pdf`,

    /*useDownload2: (id?: string, params?: UseQuotesReportParamsType) => {
      const queryKey = useMemo(
        () => keys().data().resource("quotes").action("report").id(id).params().get(),
        [id]
      );

      const { data, error, refetch, isFetching, isError } = useCustom<
        IDownloadPDFDataProviderResponse,
        TDataSourceError
      >({
        queryKey,
        queryFn: () =>
          dataSource.downloadPDF({
            url: `${dataSource.getApiUrl()}/quotes/${id}/report`,
          }),

        enabled: false,
        refetchInterval: false,
        ...params,
      });

      const download = () => {
        console.log("pido");
        refetch().then((result) => {
          console.log("termino");
          if (result.isSuccess) {
            const blob = data!.filedata;
            const link = document.createElement("a");
            const url = window.URL.createObjectURL(blob);
            link.href = url;
            link.setAttribute("download", data!.filename); // Nombre del archivo
            document.body.appendChild(link);
            link.click();
            link.remove();
            window.URL.revokeObjectURL(url);
          }
        });
      };

      return { download, error, isFetching, isError };
    },*/

    useReport: () => {
      const auth = dataSource.getApiAuthorization();
      const [reportBlob, setReportBlob] = useState<Blob>();

      const downloader = useDownloader({
        headers: {
          Authorization: auth,
        },
        customHandleDownload: (data: Blob) => {
          const blobData = [data];
          const blob = new Blob(blobData, {
            type: "application/pdf",
          });
          setReportBlob(blob);

          return true;
        },
      });

      const download = (id: string) => {
        const url = actions.getQuotePDFDownloadURL(id);
        downloader.download(url, "");
        return reportBlob;
      };

      return {
        ...downloader,
        download,
      };
    },

    useDownloader: () => {
      const auth = dataSource.getApiAuthorization();
      const downloader = useDownloader({
        headers: {
          Authorization: auth,
        },
      });

      const download = (id: string, filename?: string) => {
        const url = actions.getQuotePDFDownloadURL(id);
        return downloader.download(url, filename ?? "ssaas");
      };

      return {
        ...downloader,
        download,
      };
    },
  };
  return actions;
};