// Get the modal
var modal = document.getElementById("myModal");

// Get the <span> element that closes the modal
var span = document.getElementsByClassName("close")[0];

// When the user clicks on <span> (x), close the modal
span.onclick = function () {
    modal.style.display = "none";
}

// When the user clicks anywhere outside of the modal, close it
window.onclick = function (event) {
    if (event.target == modal) {
        modal.style.display = "none";
    }
}


let queriedApiKeys = [];
let serialNumber = 1;


async function checkBilling(apiKey, apiUrl) {
    const now = new Date();
    let startDate = new Date(now - 90 * 24 * 60 * 60 * 1000);
    const endDate = new Date(now.getTime() + 24 * 60 * 60 * 1000);
    const subDate = new Date(now);
    subDate.setDate(1);

    const headers = {
        "Authorization": "Bearer " + apiKey,
        "Content-Type": "application/json"
    };
    const modelsCheck = `${apiUrl}/v1/models`;
    const urlSubscription = `${apiUrl}/v1/dashboard/billing/subscription`;
    let urlUsage = `${apiUrl}/v1/dashboard/billing/usage?start_date=${formatDate(startDate)}&end_date=${formatDate(endDate)}`;
    const urlsetid = apiUrl + '/v1/organizations';
    const urlPaymentmethods = `${apiUrl}/v1/dashboard/billing/payment_methods`;
    const urlRatelimits = `${apiUrl}/v1/dashboard/rate_limits`;

    try {
        let totalAmount, totalUsage, remaining, GPT35CheckResult, GPT4CheckResult, GPT432kCheckResult, setid, isSubscrible;
        let SubscribleInformation = {};
        let SubInformation;
        let errors = {};

        let response = await fetch(urlSubscription, { headers });

        let currentDate = new Date();
        const subscriptionData = await response.json();
        const expiryDate = new Date(subscriptionData.access_until * 1000 + 8 * 60 * 60 * 1000);
        const formattedDate = `${expiryDate.getFullYear()}-${(expiryDate.getMonth() + 1).toString().padStart(2, '0')}-${expiryDate.getDate().toString().padStart(2, '0')}`;

        try {
            totalAmount = subscriptionData.system_hard_limit_usd;

            if (totalAmount > 20) {
                startDate = subDate;
                urlUsage = `${apiUrl}/v1/dashboard/billing/usage?start_date=${formatDate(startDate)}&end_date=${formatDate(endDate)}`;
                response = await fetch(urlUsage, { headers });
                const usageData = await response.json();
            }
            response = await fetch(urlUsage, { headers });
            const usageData = await response.json();
            totalUsage = usageData.total_usage / 100;
            remaining = currentDate > expiryDate ? "❌过期" : (totalAmount - totalUsage).toFixed(3);
        } catch (error) {
            console.error(error);
            errors['subscription'] = error.message;
        }
        //获取是否绑卡
        try {
            if (subscriptionData.plan.id.includes('payg')) {
                switch (subscriptionData.billing_mechanism) {
                    case 'advance':
                        isSubscrible = '✅预付费';
                        break;
                    case 'arrears':
                        isSubscrible = '✅已欠费';
                        break;
                    case null:
                        isSubscrible = '✅后付费';
                        break;
                    default:
                        isSubscrible = '✅';
                }
            } else {
                isSubscrible = '❌';
            }
        } catch (error) {
            console.error(error);
        }

        //获取绑卡信息
        try {
            // 从 subscriptionData 中获取 SubscribleInformation 的值...
            SubscribleInformation.account_name = subscriptionData.account_name;
            SubscribleInformation.po_number = subscriptionData.po_number;
            SubscribleInformation.billing_email = subscriptionData.billing_email;
            SubscribleInformation.tax_ids = subscriptionData.tax_ids;

            let billingAddress = subscriptionData.billing_address; // 定义并赋值 billingAddress
            let businessAddress = subscriptionData.business_address; // 定义并赋值 businessAddress

            SubInformation = "名称: " + SubscribleInformation.account_name + "\n";
            SubInformation += "PO号: " + SubscribleInformation.po_number + "\n";
            SubInformation += "帐单邮箱: " + SubscribleInformation.billing_email + "\n";
            SubInformation += "税号: " + SubscribleInformation.tax_ids + "\n";
            //使用 JavaScript 的可选链式调用来确定是否为null，避免异常控制台报错
            SubInformation += "账单地址: " + (billingAddress?.line1 ? billingAddress.line1 : '') + ", " + (billingAddress?.city ? billingAddress.city : '') + ", " + (billingAddress?.state ? billingAddress.state : '') + ", " + (billingAddress?.country ? billingAddress.country : '') + ", " + (billingAddress?.postal_code ? billingAddress.postal_code : '') + "\n";
            SubInformation += "商业地址: " + (businessAddress?.line1 ? businessAddress.line1 : '') + ", " + (businessAddress?.city ? businessAddress.city : '') + ", " + (businessAddress?.state ? businessAddress.state : '') + ", " + (businessAddress?.country ? businessAddress.country : '') + ", " + (businessAddress?.postal_code ? businessAddress.postal_code : '\n');

            // 获取付款方法信息
            response = await fetch(urlPaymentmethods, { headers });
            const paymentMethodsData = await response.json();

            if (paymentMethodsData.data && paymentMethodsData.data.length > 0) {
                paymentMethodsData.data.forEach((paymentMethod, index) => {
                    if (paymentMethod.type === 'card' && paymentMethod.card) {
                        const cardInfo = paymentMethod.card;
                        SubInformation += `\n\n付款方式 ${index + 1}: \n`;
                        SubInformation += paymentMethod.is_default ? "默认卡:\n" : "非默认卡:\n"; // Add default label
                        SubInformation += `卡种: ${cardInfo.brand} \n`; // Card Brand
                        SubInformation += `后4位: ${cardInfo.last4} \n`; // Card Last4
                        SubInformation += `卡日期: ${cardInfo.exp_month}/${cardInfo.exp_year} \n`; // Card Expiry
                        SubInformation += `卡国家: ${cardInfo.country}`; // Card Country
                    }
                });
            }

        }
        catch (error) {
            console.error(error);
        }

        //组织信息
        try {
            response = await fetch(urlsetid, { headers });
            const setiddata = await response.json();
            setid = '';
            const emailStartIndex = setiddata.data[0].description.lastIndexOf(' ') + 1;
            const email = setiddata.data[0].description.substring(emailStartIndex);
            const title = setiddata.data[0].title;
            const id1 = setiddata.data[0].id;
            const name = setiddata.data[0].name;
            const description = setiddata.data[0].description;
            const createdTimestamp = setiddata.data[0].created;
            const createdDate = new Date(createdTimestamp * 1000);

            // 格式化日期
            const formattedDate = `${createdDate.getFullYear()}.${(createdDate.getMonth() + 1).toString().padStart(2, '0')}.${createdDate.getDate().toString().padStart(2, '0')}`;

            const timeAndZone = createdDate.toString().match(/\d{2}:\d{2}:\d{2} \w+/)[0];

            if (typeof setiddata.data[1] !== 'undefined') {
                const id2 = setiddata.data[1].id;
                setid = `${title}\n${email}\n${id1}\n${id2}\n${name}\n${description}\n${formattedDate} ${timeAndZone}`;
            } else {
                setid = `${title}\n${email}\n${id1}\n${name}\n${description}\n${formattedDate} ${timeAndZone}`;
            }
            if (typeof setiddata.data[1] !== 'undefined') {
                const id2 = setiddata.data[1].id;
                setid = `人名: ${title}\n邮箱: ${email}\n组织: ${id1} ${id2}\nUserID:${name}\n\n注册时间:${formattedDate} ${timeAndZone}`;
            } else {
                setid = `人名: ${title}\n邮箱: ${email}\n组织: ${id1} \nUserID:${name}\n注册时间:${formattedDate} ${timeAndZone}`;
            }

        } catch (error) {
            console.error(error);
            errors['setid'] = error.message;
        }

        //获取速率
        let rateLimits;
        try {
            const response = await fetch(urlRatelimits, { headers });
            const rateLimitsData = await response.json();
            rateLimits = {
                'gpt-3.5-turbo': rateLimitsData['gpt-3.5-turbo'],
                'gpt-3.5-turbo-16k': rateLimitsData['gpt-3.5-turbo-16k'],
                'gpt-4': rateLimitsData['gpt-4'],
                'gpt-4-32k': rateLimitsData['gpt-4-32k']
            };
        } catch (error) {
            console.error(error);
            errors['rateLimits'] = error.message;
        }



        // 初始化模型查询结果
        GPT35CheckResult = '❌';
        GPT4CheckResult = '❌';
        GPT432kCheckResult = '❌';
        //3.5模型查询
        let GPT35CheckSuccess = false; // 初始化为 false
        try {
            const modelsCheckResponse = await fetch(modelsCheck, { headers });
            const modelsCheckData = await modelsCheckResponse.json();
            GPT35CheckSuccess = GPT35CheckResult = Array.isArray(modelsCheckData.data) && modelsCheckData.data.some(item => item.id.includes('gpt-3.5-turbo')) ? '✅' : '❌';
        } catch (error) {
            console.error(error);
            errors['modelsCheck'] = error.message;
        }
        //4模型查询
        try {
            const modelsCheckResponse = await fetch(modelsCheck, { headers });
            const modelsCheckData = await modelsCheckResponse.json();
            GPT4CheckResult = Array.isArray(modelsCheckData.data) && modelsCheckData.data.some(item => item.id.includes('gpt-4')) ? '✅' : '❌';
            GPT432kCheckResult = Array.isArray(modelsCheckData.data) && modelsCheckData.data.some(item => item.id.includes('gpt-4-32k')) ? '✅' : '❌';
        } catch (error) {
            console.error(error);
            errors['modelsCheck'] = error.message;
        }
        // 是否有效查询
        async function checkCompletion(apiKey, apiUrl) {
            const urlCompletion = `${apiUrl}/v1/chat/completions`;
            const headers = {
                "Authorization": "Bearer " + apiKey,
                "Content-Type": "application/json"
            };
            const postBody = JSON.stringify({
                "model": "gpt-3.5-turbo",
                "messages": [{
                    "role": "user",
                    "content": "Hello"
                }],
                "max_tokens": 5
            });

            let response = await fetch(urlCompletion, {
                method: 'POST',
                headers: headers,
                body: postBody
            });

            let data = await response.json();
            // 判断请求是否成功
            if (response.status === 200) {
                return ['✅', data.usage.total_tokens]; // 返回状态和 total_tokens
            } else {
                return ['❌', null];
            }
        }
        // 调用 checkCompletion 函数并获取结果
        let completionCheckResult = await checkCompletion(apiKey, apiUrl);
        //返回值
        return [totalAmount, totalUsage, remaining, formattedDate, GPT35CheckResult, GPT4CheckResult, GPT432kCheckResult, isSubscrible, SubInformation, setid, errors, GPT35CheckSuccess, completionCheckResult, rateLimits];
    } catch (error) {
        return ["Error", null, null, null, null, null, null, null, null, null];
    }
}




function formatDate(date) {
    const year = date.getFullYear();
    const month = (date.getMonth() + 1).toString().padStart(2, '0');
    const day = date.getDate().toString().padStart(2, '0');

    return `${year}-${month}-${day}`;
}

//查询函数
function sendRequest() {
    let button = document.querySelector("button");
    button.textContent = "加载中...";
    button.disabled = true;
    button.classList.add("loading")

    let apiKeyInput = document.getElementById("api-key-input");
    let apiUrlSelect = document.getElementById("api-url-select");
    let customUrlInput = document.getElementById("custom-url-input");
    let table = document.getElementById("result-table");
    let h2 = document.getElementById("result-head");
    h2.style.visibility = "visible";
    table.style.visibility = "visible";

    if (apiKeyInput.value.trim() === "") {
        alert("请填写API KEY");
        return;
    }

    document.getElementById("result-table").getElementsByTagName('tbody')[0].innerHTML = "";

    let apiUrl = "";
    if (apiUrlSelect.value === "custom") {
        if (customUrlInput.value.trim() === "") {
            alert("请设置API链接");
            return;
        } else {
            apiUrl = customUrlInput.value.trim();
            if (!apiUrl.startsWith("http://") && !apiUrl.startsWith("https://")) {
                apiUrl = "https://" + apiUrl;
            }
        }
    } else {
        apiUrl = apiUrlSelect.value;
    }

    let apiKeys = apiKeyInput.value.split(/[,\s，\n]+/);

    if (apiKeys.length === 0) {
        alert("未匹配到 API-KEY，请检查输入内容");
        return;
    }

    alert("成功匹配到 API Key，确认后开始查询：" + apiKeys);

    let tableBody = document.querySelector("#result-table tbody");
    for (let i = 0; i < apiKeys.length; i++) {
        let apiKey = apiKeys[i].trim();

        if (queriedApiKeys.includes(apiKey)) {
            console.log(`API KEY ${apiKey} 已查询过，跳过此次查询`);
            continue;
        }
        queriedApiKeys.push(apiKey);

        checkBilling(apiKey, apiUrl).then((data) => {
            data = data.map(item => {
                if (item === undefined) {
                    return 'Not Found.'
                } else {
                    return item
                }
            }
            )

            let row = document.createElement("tr");

            let serialNumberCell = document.createElement("td"); // 创建序列号单元格
            serialNumberCell.textContent = serialNumber; // 设置序列号文本
            row.appendChild(serialNumberCell); // 将序列号单元格添加到行中

            let apiKeyCell = document.createElement("td");
            apiKeyCell.textContent = apiKey.replace(/^(.{5}).*(.{4})$/, "$1***$2");
            row.appendChild(apiKeyCell);

            console.log('查看查询结果', data); // 添加 console.log 以查看 data 的值

            if (data[0] === undefined) {
                let errorMessageCell = document.createElement("td");
                errorMessageCell.colSpan = "8";
                errorMessageCell.classList.add("status-error");
                errorMessageCell.textContent = "不正确或已失效的API-KEY";
                row.appendChild(errorMessageCell);
            } else {
                let totalAmount = document.createElement("td");
                totalAmount.textContent = data[0];
                row.appendChild(totalAmount);

                let totalUsedCell = document.createElement("td");
                if (!isNaN(data[1])) {
                    totalUsedCell.textContent = data[1].toFixed(3);
                } else {
                    totalUsedCell.textContent = '❌'
                }
                row.appendChild(totalUsedCell);

                let totalAvailableCell = document.createElement("td");
                totalAvailableCell.textContent = typeof data[2] === 'number' ? data[2] : data[2];
                row.appendChild(totalAvailableCell);
                let progressCell = document.createElement("td");
                let progressContainer = document.createElement("div"); // 添加一个新的容器元素
                progressContainer.style.width = "100%";
                progressContainer.style.height = "20px";
                progressContainer.style.backgroundColor = "#f3f3f3"; // 设置容器的背景色为灰白色
                let progressBar = document.createElement("div");
                progressBar.style.width = (data[1] / data[0] * 100).toFixed(2) + "%";
                progressBar.style.height = "20px";
                progressBar.style.backgroundColor = "#4CAF50";
                progressBar.style.position = "relative"; // 设置进度条的 position 为 relative
                progressBar.textContent = (data[1] / data[0] * 100).toFixed(2) + "%"; // 显示百分比
                progressBar.style.textAlign = "right"; // 设置文本对齐方式为右对齐
                progressBar.style.paddingRight = "5px"; // 设置右边距以确保文本不超出边界
                progressBar.style.color = "black"; // 设置文本颜色为白色
                progressContainer.appendChild(progressBar); // 将进度条添加到容器中
                progressCell.appendChild(progressContainer); // 将容器添加到单元格中
                row.appendChild(progressCell);

                function createSeeMoreLink(text) {
                    let seeMoreLink = document.createElement("a");
                    seeMoreLink.href = "#";
                    seeMoreLink.textContent = '全部';
                    seeMoreLink.style.cursor = "pointer";
                    seeMoreLink.style.textDecoration = "underline";
                    seeMoreLink.onclick = (event) => {
                        event.preventDefault();
                        document.getElementById('modalText').textContent = text;
                        document.getElementById('myModal').style.display = "block";
                    };
                    return seeMoreLink;
                }



                let expireTime = document.createElement("td");
                expireTime.textContent = data[3];
                row.appendChild(expireTime);

                let GPT35CheckResult = document.createElement("td");
                GPT35CheckResult.textContent = data[4];
                row.appendChild(GPT35CheckResult);

                let GPT4CheckResult = document.createElement("td");
                GPT4CheckResult.textContent = data[5];
                row.appendChild(GPT4CheckResult);

                let GPT432kCheckResult = document.createElement("td");
                GPT432kCheckResult.textContent = data[6];
                row.appendChild(GPT432kCheckResult);

                let isSubscribe = document.createElement("td");
                isSubscribe.style.whiteSpace = "pre"; // 添加这一行来保留换行
                isSubscribe.textContent = data[7];
                row.appendChild(isSubscribe);

                let SubInformation = document.createElement("td");
                let SubInformationContainer = document.createElement("div");
                SubInformationContainer.style.whiteSpace = "pre-wrap";
                if (data[8].length > 50) {
                    SubInformationContainer.textContent = data[8].slice(0, 25) + '... ';
                    SubInformationContainer.appendChild(createSeeMoreLink(data[8]));
                } else {
                    SubInformationContainer.textContent = data[8];
                }
                SubInformation.appendChild(SubInformationContainer);
                row.appendChild(SubInformation);


                let setidCell = document.createElement("td");
                let setidCellContainer = document.createElement("div");
                setidCellContainer.style.whiteSpace = "pre-wrap";
                if (data[9].length > 50) {
                    setidCellContainer.textContent = data[9].slice(0, 25) + '... ';
                    setidCellContainer.appendChild(createSeeMoreLink(data[9]));

                } else {
                    setidCellContainer.textContent = data[9];
                }
                setidCell.appendChild(setidCellContainer);
                row.appendChild(setidCell);


                let rateLimitsDataCell = document.createElement("td");
                let rateLimitsDataContainer = document.createElement("div");
                rateLimitsDataContainer.style.whiteSpace = "pre-wrap";
                if (data[13]) {
                    let rateLimitsData = data[13];
                    let models = ['gpt-3.5-turbo', 'gpt-3.5-turbo-16k', 'gpt-4', 'gpt-4-32k'];
                    let rateLimitsText = '';
                    for (let model of models) {
                        if (rateLimitsData[model]) {
                            rateLimitsText += `${model}:\n\tRPM: ${rateLimitsData[model].max_requests_per_1_minute}\n\tTPM: ${rateLimitsData[model].max_tokens_per_1_minute}\n\n`;
                        } else {
                            rateLimitsText += model + ": ❌\n";
                        }
                    }
                    if (rateLimitsText.length > 50) {
                        rateLimitsDataContainer.textContent = rateLimitsText.slice(0, 25) + '... ';
                        rateLimitsDataContainer.appendChild(createSeeMoreLink(rateLimitsText));
                    } else {
                        rateLimitsDataContainer.textContent = rateLimitsText;
                    }
                }

                rateLimitsDataCell.appendChild(rateLimitsDataContainer);
                row.appendChild(rateLimitsDataCell);


                // 是否有效列
                let completionCheckResultCell = document.createElement("td");
                completionCheckResultCell.innerHTML = `<span style="font-size:24px">${data[12][0]}</span><br>消耗${data[12][1]} tokens`; // 使用 innerHTML 添加两行内容
                row.appendChild(completionCheckResultCell);


            }
            tableBody.appendChild(row);

            if (i === apiKeys.length - 1) {
                queriedApiKeys = [];
            }
            serialNumber++; // 增加序列号
            h2.style.display = 'block';
            table.style.display = 'table';

            button.textContent = "查询";
            button.disabled = false;
            button.classList.remove("loading")
        })

    }
}

let apiUrlSelect = document.getElementById("api-url-select");
let customUrlInput = document.getElementById("custom-url-input");

apiUrlSelect.addEventListener("change", function () {
    if (apiUrlSelect.value === "custom") {
        customUrlInput.style.display = "inline-block";
        customUrlInput.style.marginTop = "5px";
    } else {
        customUrlInput.style.display = "none";
    }
});