UIKit/src/api/makeRequest.ts

import { create } from "zustand";
import { persist } from "zustand/middleware";
import axios, { AxiosResponse, Method } from "axios";


interface AuthStore {
    token: string;
}

export const useAuthStore = create<AuthStore>()(
    persist(
        (set, get) => ({
            token: "",
        }),
        {
            name: "token",
        }
    )
);

export const getAuthToken = () => useAuthStore.getState().token;

export const setAuthToken = (token: string) => useAuthStore.setState({ token });

export const clearAuthToken = () => useAuthStore.setState({ token: "" });

export function useToken() {
    return useAuthStore(state => state.token);
}

export async function makeRequest<TRequest = unknown, TResponse = unknown>({
    method = "post",
    url,
    body,
    useToken = true,
    contentType = false,
    signal,
    withCredentials,
}: {
    method?: Method;
    url: string;
    body?: TRequest;
    /** Send access token */
    useToken?: boolean;
    contentType?: boolean;
    signal?: AbortSignal;
    /** Send refresh token */
    withCredentials?: boolean;
}): Promise<TResponse> {
    const headers: Record<string, string> = {};
    if (useToken) headers["Authorization"] = `Bearer ${getAuthToken()}`;
    if (contentType) headers["Content-Type"] = "application/json";

    try {
        const response = await axios<TRequest, AxiosResponse<TResponse & { accessToken?: string; }>>({
            url,
            method,
            headers,
            data: body,
            signal,
            withCredentials,
        });

        if (response.data?.accessToken) {
            setAuthToken(response.data.accessToken);
        }

        return response.data;
    } catch (error) {
        if (axios.isAxiosError(error) && error.response?.status === 401 && !withCredentials) {
            const refreshResponse = await refresh(getAuthToken());
            if (refreshResponse.data?.accessToken) setAuthToken(refreshResponse.data.accessToken);

            headers["Authorization"] = refreshResponse.data.accessToken;
            const response = await axios.request<TRequest, AxiosResponse<TResponse>>({
                url,
                method,
                headers,
                data: body,
                signal,
            });

            return response.data;
        }

        throw error;
    }
};

function refresh(token: string) {
    return axios<never, AxiosResponse<{ accessToken: string; }>>("https://admin.pena.digital/auth/refresh", {
        headers: {
            "Authorization": token,
            "Content-Type": "application/json",
        },
    });
}