【接力奥特曼大佬】Share token使用更更优雅!

原贴:兄弟们,Share token使用更优雅了!

部署方式不变

一、新增功能
1.增加访问权限,必须输入密码,如https://域名地址/?password=密码
2.增加页面 access_token旁边增加“获取?”链接自动跳转到始皇的获取页面

二、改良的代码如下

addEventListener('fetch', event => {
  event.respondWith(handleRequest(event.request))
})
function parseJwt(token) {
  const base64Url = token.split('.')[1]; // 获取载荷部分
  const base64 = base64Url.replace(/-/g, '+').replace(/_/g, '/'); // 将 Base64Url 转为 Base64
  const jsonPayload = decodeURIComponent(atob(base64).split('').map(function(c) {
    return '%' + ('00' + c.charCodeAt(0).toString(16)).slice(-2);
  }).join(''));

  return JSON.parse(jsonPayload); // 返回载荷解析后的 JSON 对象
}

function isTokenExpired(token) {
  const payload = parseJwt(token);
  const currentTime = Math.floor(Date.now() / 1000); // 获取当前时间戳(秒)
  return payload.exp < currentTime; // 检查 token 是否过期
}
async function handleRequest(request) {
  const PASSWORD = 'password'; // 设置你的密码
  const url = new URL(request.url);
  const params = new URLSearchParams(url.search);
  const password = params.get('password');

  if (password !== PASSWORD) {
    return new Response('Unauthorized access', { status: 401 });
  }

  // @ts-ignore
  const token = await oai_global_variables.get('at'); // 这里填入你的 JWT

  if (isTokenExpired(token)) {
    // 如果 Token 过期,执行获取新 Token 的逻辑
    const url = 'https://token.oaifree.com/api/auth/refresh';
    // @ts-ignore
    const refreshToken = await oai_global_variables.get('rt');  // 实际情况下你可能会从某处获取这个值
  
    // 发送 POST 请求
    const response = await fetch(url, {
        method: 'POST',
        headers: {
            'Content-Type': 'application/x-www-form-urlencoded;charset=UTF-8'
        },
        body: `refresh_token=${refreshToken}`
    });
  
    // 检查响应状态
    if (response.ok) {
        const data = await response.json();  // 假设服务器返回的是 JSON 格式数据
        const token = data.access_token; // 直接从 JSON 中获取 access_token

        // @ts-ignore
        await oai_global_variables.put('at', token);
    } else {
        return new Response('Error fetching access token', { status: response.status });
    }
  }

  // 如果 Token 未过期,继续执行原来的逻辑
  if (request.method === "POST") {
    const formData = await request.formData();    
    // @ts-ignore
    const at = await oai_global_variables.get('at')
    const access_token = formData.get('access_token') || at; //删除行尾at变量以停止使用内置的access_token(两个竖线也需要删除)
    const unique_name = formData.get('unique_name');
    const site_limit = formData.get('site_limit') || '';
    const expires_in = formData.get('expires_in') || '';
    const gpt35_limit = formData.get('gpt35_limit') || '';
    const gpt4_limit = formData.get('gpt4_limit') || '';
    const show_conversations = formData.get('show_conversations') || 'false';
    const show_userinfo = formData.get('show_userinfo') || 'false';

    const url = 'https://chat.oaifree.com/token/register';
    const body = new URLSearchParams({
      unique_name,
      access_token, // 使用来自表单或KV变量的 access_token
      site_limit,
      expires_in,
      gpt35_limit,
      gpt4_limit,
      show_conversations,
      show_userinfo
    }).toString();

    const apiResponse = await fetch(url, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/x-www-form-urlencoded'
      },
      body: body
    });

    const responseText = await apiResponse.text();
    const tokenKeyMatch = /"token_key":"([^"]+)"/.exec(responseText);
    const tokenKey = tokenKeyMatch ? tokenKeyMatch[1] : '未找到 Share_token';

    const resultHtml = `
    <!DOCTYPE html>
    <html lang="en">
    <head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <style>
      body {
        font-family: Arial, sans-serif;
        background-color: #f4f4f4;
        padding: 20px;
      }
      form, .response-container {
        background: white;
        padding: 20px;
        border-radius: 8px;
        box-shadow: 0 2px 5px rgba(0,0,0,0.1);
        max-width: 600px;
        margin: 20px auto;
      }
      label, h1, p {
        display: block;
        margin-bottom: 10px;
        font-size: 16px;
      }
      input[type="text"], input[type="checkbox"], textarea {
        width: calc(100% - 22px);
        padding: 10px;
        margin-top: 5px;
        margin-bottom: 20px;
        border-radius: 5px;
        border: 1px solid #ccc;
      }
      textarea {
        font-family: 'Courier New', monospace;
        background-color: #f8f8f8;
        height: 150px; /* Adjust height based on your needs */
      }
      button {
        background-color: #007BFF;
        color: white;
        padding: 10px 20px;
        border: none;
        border-radius: 5px;
        cursor: pointer;
        font-size: 16px;
      }
      button:hover {
        background-color: #0056b3;
      }
      @media (max-width: 768px) {
        body, form, .response-container {
          padding: 10px;
        }
      }
    </style>
    </head>
    <body>
    <div class="response-container">
      <h1>Share Token</h1>
      <textarea id="tokenKey" readonly>${tokenKey}</textarea>
      <button onclick="copyTokenKey()">一键复制</button>
      <br></br>
      <button onclick="goToOtherPage1()">去聊天(旧页面)</button>
      <br></br>
      <button onclick="goToOtherPage2()">去聊天(新页面)</button>
      <script>
        function copyTokenKey() {
          const copyText = document.getElementById("tokenKey");
          copyText.select();
          document.execCommand("copy");
          alert("已复制: " + copyText.value);
        }
      
        function goToOtherPage1() {
          window.location.href = 'https://demo.oaifree.com/auth/login_share?token=${tokenKey}'; 
        }
        function goToOtherPage2() {
          window.location.href = 'https://new.oaifree.com'; 
        }
      </script>
      <a href="/?password=123">返回</a>
      <h1>响应原文</h1>
      <textarea id="response" readonly>${responseText}</textarea>
      
      
    </div>
    </body>
    </html>
    `;

    return new Response(resultHtml, {
      headers: {
        'Content-Type': 'text/html; charset=utf-8'
      },
    });
  } else {
    const formHtml = `
    <!DOCTYPE html>
    <html lang="en">
    <head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <style>
      body {
        font-family: Arial, sans-serif;
        background-color: #f4f4f4;
        padding: 20px;
      }
      form {
        background: white;
        padding: 20px;
        border-radius: 8px;
        box-shadow: 0 2px 5px rgba(0,0,0,0.1);
        max-width: 600px;
        margin: 20px auto;
      }
      label {
        display: block;
        margin-bottom: 10px;
        font-size: 16px;
      }
      input[type="text"], input[type="checkbox"] {
        width: calc(100% - 22px);
        padding: 10px;
        margin-top: 5px;
        margin-bottom: 20px;
        border-radius: 5px;
        border: 1px solid #ccc;
      }
      button {
        background-color: #007BFF;
        color: white;
        padding: 10px 20px;
        border: none;
        border-radius: 5px;
        cursor: pointer;
        font-size: 16px;
      }
      button:hover {
        background-color: #0056b3;
      }
      @media (max-width: 768px) {
        body {
          padding: 10px;
        }
        form {
          padding: 15px;
        }
      }
    </style>
    </head>
    <body>
    <h1>欢迎使用ChatGPT</h1>
    <form method="POST">
      <label for="unique_name">Unique Name(必须输入,以区分身份、会话隔离、聊天记录保存。):</label>
      <input type="text" id="unique_name" name="unique_name" placeholder="只能由数字和字母构成" required value="">
      
      <details>
        <summary>更多选项(不懂则无需更改)</summary>
        <br></br>
        <div>
            <label for="access_token">Access Token:<a target="_blank" href="https://token.oaifree.com/auth">获取?</a></label>
            <input type="text" id="access_token" name="access_token" placeholder="your access token" value="">

            <label for="site_limit">Site Limit:</label>
            <input type="text" id="site_limit" name="site_limit" placeholder="All" value="">

            <label for="expires_in">Expires In:</label>
            <input type="text" id="expires_in" name="expires_in" placeholder="以秒为单位,0表示永不过期" value="0">

            <label for="gpt35_limit">GPT-3.5 Limit:</label>
            <input type="text" id="gpt35_limit" name="gpt35_limit" placeholder="Usage counts, with negative values indicating no limitation" value="-1">

            <label for="gpt4_limit">GPT-4 Limit:</label>
            <input type="text" id="gpt4_limit" name="gpt4_limit" placeholder="Usage counts, with negative values indicating no limitation" value="-1">
        </div>
    </details>
    <br></br>
      <button type="submit">获取你的访问门票 share token</button>
    </form>
    </body>
    </html>
    
    `;

    return new Response(formHtml, {
      headers: {
        'Content-Type': 'text/html; charset=utf-8'
      },
    });
  }
}

谢谢社区!

15 Likes

感谢分享,顶一个

1 Like

给个小建议,在html渲染头部添加,用来防止烦人的浏览器询问是否要开启翻译。
PASSWORD更适合不用const定义,而是存储在环境变量中。

1 Like

优雅,你们都优雅

1 Like

我记得原来奥特曼的代码刷新fk的机制是点击进入后刷新,能不能整一个定时自动刷新,例如5天,这样就不用手动触发刷新了。

1 Like

设置下workers的触发器
支持Cron 触发器

1 Like

牛逼

1 Like

优雅!!!实在优雅!!