import { IApi } from '@umijs/max'
import { publicPath, isPro, replaceStr } from "./config/utils"

const attrReplaceString = ($: any, ele: any) => {
    const { name } = $(ele)[0]
    if (name === "link") {
        const str = $(ele).attr("href")
        $(ele).attr("href", str.substr(1, str.length - 1))
    }
    if (name === "meta") {
        const content = $(ele).attr("content")
        if (content && ~content.indexOf(replaceStr)) {
            $(ele).attr("content", content.replace(`/${replaceStr}/`, `${replaceStr}/`));
        }
    }
    if (name === "script") {
        const str = $(ele).attr("src")
        $(ele).attr("src", str.substr(1, str.length - 1))
    }
}

export default (api: IApi) => {
    api.addHTMLMetas(() => [
        {
            name: "keywords",
            content: "OpenAnolis 龙蜥操作系统开源社区 TestLib"
        },
        {
            name: "description",
            content: "Test library is a test management system"
        },
        {
            property: "og:description",
            content: "Test library is a test management system"
        },
        {
            property: "og:image",
            content: publicPath + "logo.png"
        },
    ])

    api.addHTMLLinks(() => [
        {
            rel: "icon",
            href: publicPath + "favicon.ico",
            type: "image/x-icon"
        },
        {
            type: "image/png",
            href: publicPath + "logo.png",
            rel: "shortcut icon"
        }
    ])

    api.modifyHTML(($) => {
        if (isPro) {
            $("head").children().each((i, ele) => attrReplaceString($, ele))
            $("body").children().each((i, ele) => attrReplaceString($, ele))

            const js = $(`<script>window.publicPath = "${replaceStr}/";</script>`)
            $("head").append(js)
        }

        return $;
    })
};