1Panel/frontend/src/utils/xpack.ts

import {
    getLicenseStatus,
    getMasterLicenseStatus,
    getSettingBaseInfo,
    getEnterpriseLicenseStatus,
} from '@/api/modules/setting';
import { useTheme } from '@/global/use-theme';
import {
    searchXpackSetting,
    updateXpackSettingByKey as updateXpackSettingByKeyFromExtension,
} from '@/extensions/xpack';
import { GlobalStore } from '@/store';
import faviconUrl from '@/assets/images/favicon.svg';

let switchThemeFn: (() => void) | undefined;

const switchTheme = () => {
    if (!switchThemeFn) {
        switchThemeFn = useTheme().switchTheme;
    }
    switchThemeFn();
};

export function resetXSetting() {
    const globalStore = GlobalStore();
    globalStore.themeConfig.title = '';
    globalStore.themeConfig.logo = '';
    globalStore.themeConfig.logoWithText = '';
    globalStore.themeConfig.favicon = '';
    globalStore.watermark = null;
    globalStore.watermarkShow = false;
    globalStore.masterAlias = '';
}

async function getColoredFavicon(url: string, color: string) {
    const res = await fetch(url);
    let svgText = await res.text();
    svgText = svgText.replace(/fill=(["'])(.*?)\1/g, `fill="${color}"`);
    return `data:image/svg+xml,${encodeURIComponent(svgText)}`;
}

export async function initFavicon() {
    const globalStore = GlobalStore();
    document.title = globalStore.themeConfig.panelName;
    const favicon = globalStore.themeConfig.favicon;
    const isPro = globalStore.isXpackOrEE;
    const themeColor = globalStore.themeConfig.primary;
    const customFaviconUrl = `/api/v2/images/favicon?t=${Date.now()}`;
    const fallbackSvg = isPro ? await getColoredFavicon(faviconUrl, themeColor) : '/public/favicon.png';
    const setLink = (href: string) => {
        let link = document.querySelector("link[rel*='icon']") as HTMLLinkElement;
        if (!link) {
            link = document.createElement('link');
            link.rel = 'shortcut icon';
            link.type = 'image/x-icon';
            document.head.appendChild(link);
        }
        link.href = href;
    };

    if (favicon) {
        const testImg = new Image();
        testImg.onload = () => setLink(customFaviconUrl);
        testImg.onerror = () => setLink(fallbackSvg);
        testImg.src = customFaviconUrl;
    } else {
        setLink(fallbackSvg);
    }
}

export async function getXpackSetting() {
    const res = await searchXpackSetting();
    if (!res) {
        initFavicon();
        resetXSetting();
        return;
    }
    initFavicon();
    return res;
}

const loadDataFromDB = async () => {
    const globalStore = GlobalStore();
    const res = await getSettingBaseInfo();
    document.title = res.data.panelName;
    globalStore.entrance = res.data.securityEntrance;
    globalStore.openMenuTabs = res.data.menuTabs === 'Enable';
};

export async function loadProductProFromDB() {
    const globalStore = GlobalStore();
    if (!globalStore.isEnterprise) {
        globalStore.isEnterpriseLicenseLoaded = true;
        const res = await getLicenseStatus();
        if (!res || !res.data) {
            globalStore.isProductPro = false;
        } else {
            globalStore.isProductPro = res.data.status === 'Bound';
            if (globalStore.isProductPro) {
                globalStore.productProExpires = Number(res.data.productPro);
            }
        }
        return;
    }
    const res = await getEnterpriseLicenseStatus();
    globalStore.isEnterpriseLicenseLoaded = true;
    if (!res || !res.data) {
        globalStore.isEnterpriseLicensed = false;
        globalStore.isProductPro = false;
    } else {
        globalStore.isEnterpriseLicensed = res.data.status === 'Bound';
        globalStore.isProductPro = globalStore.isEnterpriseLicensed;
    }
}

export async function loadMasterProductProFromDB() {
    const globalStore = GlobalStore();
    if (!globalStore.isEnterprise) {
        globalStore.isEnterpriseLicenseLoaded = true;
        const res = await getMasterLicenseStatus();
        if (!res || !res.data) {
            globalStore.isMasterProductPro = false;
        } else {
            globalStore.isMasterProductPro = res.data.status === 'Bound';
        }
    } else {
        const res = await getEnterpriseLicenseStatus();
        globalStore.isEnterpriseLicenseLoaded = true;
        if (!res || !res.data) {
            globalStore.isEnterpriseLicensed = false;
            globalStore.isMasterProductPro = false;
            globalStore.isProductPro = false;
        } else {
            globalStore.isEnterpriseLicensed = res.data.status === 'Bound';
            globalStore.isMasterProductPro = res.data.status === 'Bound';
            globalStore.isProductPro = res.data.status === 'Bound';
        }
    }
    switchTheme();
    initFavicon();
    loadDataFromDB();
}

export async function getXpackSettingForTheme() {
    const globalStore = GlobalStore();
    const res2 = await searchXpackSetting();
    if (res2) {
        globalStore.themeConfig.title = res2.data?.title;
        globalStore.themeConfig.logo = res2.data?.logo;
        globalStore.themeConfig.logoWithText = res2.data?.logoWithText;
        globalStore.themeConfig.favicon = res2.data?.favicon;
        globalStore.themeConfig.loginImage = res2.data?.loginImage;
        globalStore.themeConfig.loginBgType = res2.data?.loginBgType;
        globalStore.themeConfig.loginBackground = res2.data?.loginBackground;
        globalStore.themeConfig.loginBtnLinkColor = res2.data?.loginBtnLinkColor;
        globalStore.themeConfig.themeColor = res2.data?.themeColor;
        globalStore.masterAlias = res2.data.masterAlias;
        if (res2.data?.theme) {
            globalStore.themeConfig.theme = res2.data.theme;
        }
        globalStore.watermarkShow = res2.data.watermarkShow === 'Enable';
        try {
            globalStore.watermark = JSON.parse(res2.data.watermark);
        } catch {
            globalStore.watermark = null;
        }
    } else {
        resetXSetting();
    }
    switchTheme();
    initFavicon();
}

export async function updateXpackSettingByKey(key: string, value: string) {
    return updateXpackSettingByKeyFromExtension(key, value);
}