CoreControl/hooks/useSite.ts

import { useState, useEffect, useCallback } from "react";
import axios from "axios";
import { Site } from "@/app/types";

const useSite = () => {
    const [site, setSite] = useState<Site>({
        id: "",
        name: "",
        description: "",
        networks: []
    });
    const [siteId, setSiteId] = useState("");
    const [loading, setLoading] = useState(false);

    const loadSite = useCallback(() => {
        if (!siteId) return;
        setLoading(true);
        axios.get('/api/sites/get', {
            params: { siteId }
        }).then((response) => {
            setSite(response.data.site);
            setLoading(false);
        });
    }, [siteId]);

    useEffect(() => {
        loadSite();
    }, [loadSite]);

    const editSite = (site: Site): Promise<string> | string => {
        if (!site.id) {
            return 'Site ID is required';
        }

        if(site.name.length < 3) {
            return 'Site name must be at least 3 characters long';
        }
        return axios.post('/api/sites/edit', site)
        .then(() => {
            return "Site edited successfully";
        })
        .catch(err => {
            throw err.response?.data?.error || 'An error occurred';
        });
    };

    const addSite = (site: Site): Promise<Site> | string => {
        if(site.name.length < 3) {
            return 'Site name must be at least 3 characters long';
        }
        
        return axios.post('/api/sites/add', site)
        .then((response) => {
            return response.data.site;
        })
        .catch(err => {
            throw err.response?.data?.error || 'An error occurred';
        });
    };

    const deleteSite = (siteId: string) => {
        axios.delete('/api/sites/delete', {
            params: { siteId }
        })
        .then(() => {
            setSiteId('');
        })
        .catch(err => {
            console.error(err);
        })
    };
    return {
        site,
        loadSite,
        setSiteId,
        editSite,
        addSite,
        deleteSite,
        loading
    };
};

export default useSite;