☰

OAIFree RefreshTokens登录

支持多个refresh token，用户可随时选择token并自动登录，方便切换账号

// ==UserScript==
// @name         OAIFree RefreshTokens登录
// @namespace    http://tampermonkey.net/
// @version      0.3
// @description  支持多个refresh token，用户可随时选择token并自动登录，方便切换账号
// @author       You
// @license      Liunx.do
// @match        https://new.oaifree.com/*
// @grant        GM_xmlhttpRequest
// @grant        GM_setValue
// @grant        GM_getValue
// @grant        GM_registerMenuCommand
// @connect      token.oaifree.com
// @connect      chat.oaifree.com
// ==/UserScript==

(function() {
    'use strict';
    // 预设的多个refresh token
    const refreshTokens = [
        'You RefreshTokens',
        // 添加更多的refresh token
    ];

    // 获取当前时间戳
    function getCurrentTimestamp() {
        return Math.floor(Date.now() / 1000);
    }

    // 检查 st 是否过期
    function isSTExpired() {
        const expireAt = GM_getValue('expire_at', 0);
        console.log("st expire at: "+ expireAt);
        return isNaN(expireAt) || getCurrentTimestamp() >= expireAt;
    }

    // 使用 rt 换取 at
    function getAccessToken(refreshToken) {
        return new Promise((resolve, reject) => {
            GM_xmlhttpRequest({
                method: 'POST',
                url: 'https://token.oaifree.com/api/auth/refresh',
                headers: {
                    'Content-Type': 'application/x-www-form-urlencoded'
                },
                data: `refresh_token=${refreshToken}`,
                onload: function(response) {
                    if (response.status === 200) {
                        const data = JSON.parse(response.responseText);
                        if(data.access_token){
                            resolve(data.access_token);
                        }else{
                            reject('Failed to generate access token, response: ' + JSON.stringify(data));
                        }
                    } else {
                        reject('Failed to refresh access token');
                    }
                },
                onerror: function(e) {
                    console.error(e);
                    reject('Failed to refresh access token');
                }
            });
        });
    }

    // 使用 at 生成 st
    function getShareToken(accessToken) {
        return new Promise((resolve, reject) => {
            GM_xmlhttpRequest({
                method: 'POST',
                url: 'https://chat.oaifree.com/token/register',
                headers: {
                    'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8'
                },
                data: `unique_name=${generateRandomHex(8)}&access_token=${accessToken}&expires_in=0&site_limit=&gpt35_limit=-1&gpt4_limit=-1&show_conversations=true`,
                onload: function(response) {
                    if (response.status === 200) {
                        const data = JSON.parse(response.responseText);
                        GM_setValue('expire_at', data.expire_at);
                        if(data.token_key){
                            resolve(data.token_key);
                        }else{
                            reject('Failed to generate share token, response: ' + JSON.stringify(data));
                        }
                    } else {
                        reject('Failed to generate share token');
                    }
                },
                onerror: function(e) {
                    console.error(e);
                    reject('Failed to generate share token');
                }
            });
        });
    }

    // 生成随机字符串
    function generateRandomHex(length) {
        let result = '';
        const characters = '0123456789abcdef';
        const charactersLength = characters.length;
        for (let i = 0; i < length; i++) {
            result += characters.charAt(Math.floor(Math.random() * charactersLength));
        }
        return result;
    }

    // 自动登录
    function autoLogin(shareToken) {
        const loginUrl = `https://new.oaifree.com/auth/login_share?token=${shareToken}`;
        console.log('Logging in with URL: ' + loginUrl);
        window.location.href = loginUrl;
    }

    // 选择token并登录
    async function selectTokenAndLogin(index) {
        try {
            const refreshToken = refreshTokens[index];
            console.log('Using refresh token: ' + refreshToken);
            const accessToken = await getAccessToken(refreshToken);
            console.log('Access token obtained: ' + accessToken);
            const shareToken = await getShareToken(accessToken);
            console.log('Share token obtained: ' + shareToken);
            GM_setValue("share_token", shareToken);
            autoLogin(shareToken);
        } catch (error) {
            console.error('Login failed:', error);
            alert('登录失败，请查看控制台获取详细信息。');
        }
    }

    // 为每个token注册菜单命令
    refreshTokens.forEach((token, index) => {
        GM_registerMenuCommand(`切换到 Token ${index + 1}`, () => selectTokenAndLogin(index));
    });

    // 检查是否需要刷新token并自动登录（仅在登录页面执行）
    if (window.location.pathname === '/auth/login_auth0') {
        (async function() {
            try {
                let shareToken = GM_getValue("share_token");
                if (isSTExpired() || !shareToken) {
                    console.log('ST token is expired or not found. Please select a token from the userscript menu to login.');
                } else {
                    console.log('ST token is still valid. Auto-logging in...');
                    autoLogin(shareToken);
                }
            } catch (error) {
                console.error(error);
            }
        })();
    }
})();