trae-api

auth.js

@@ -0,0 +1,62 @@
+// ==UserScript==
+// @name         获取并保存 Trae 登录凭证
+// @namespace    a23187.cn
+// @version      0.1.0
+// @description  获取并保存 Trae 登录凭证
+// @author       A23187
+// @match        https://www.trae.ai/authorization?*
+// @match        https://www.trae.com.cn/authorization?*
+// @icon         https://www.google.com/s2/favicons?sz=64&domain=trae.ai
+// @grant        GM_setClipboard
+// @license      MIT
+// ==/UserScript==
+
+(function () {
+    'use strict';
+    async function getRefreshToken(clientId) {
+        const resp = await fetch('https://www.trae.ai/cloudide/api/v3/trae/oauth/GetRefreshToken', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({ clientID: clientId }),
+        }).then((resp) => resp.json());
+        return resp.Result.RefreshToken;
+    }
+    async function exchangeToken(clientId, refreshToken) {
+        const resp = await fetch('https://api-sg-central.trae.ai/cloudide/api/v3/trae/oauth/ExchangeToken', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                ClientID: clientId, RefreshToken: refreshToken, ClientSecret: '-', UserID: '',
+            }),
+        }).then((resp) => resp.json());
+        return resp.Result.Token;
+    }
+    const id = setInterval(() => {
+        const mainTitleDiv = document.getElementsByClassName('main-title')[0];
+        if (mainTitleDiv === undefined || mainTitleDiv?.innerText !== 'Login Successful') {
+            return;
+        }
+        const saveTokenBtn = document.createElement('button');
+        mainTitleDiv.parentElement.appendChild(saveTokenBtn);
+        saveTokenBtn.innerText = 'Save token to clipboard';
+        saveTokenBtn.onclick = async () => {
+            const params = new URLSearchParams(document.location.search);
+            const clientId = params.get('client_id');
+            const refreshToken = await getRefreshToken(clientId);
+            const token = await exchangeToken(clientId, refreshToken);
+            GM_setClipboard(`
+TRAE_APP_ID = "6eefa01c-1036-4c7e-9ca5-d891f63bfcd8"
+TRAE_DEVICE_BRAND = "${params.get('x_device_brand')}"
+TRAE_DEVICE_CPU = "Intel"
+TRAE_DEVICE_ID = "${params.get('x_device_id')}"
+TRAE_DEVICE_TYPE = "${params.get('x_device_type')}"
+TRAE_IDE_TOKEN = "${token}"
+TRAE_IDE_VERSION = ""
+TRAE_IDE_VERSION_CODE = ""
+TRAE_IDE_VERSION_TYPE = ""
+TRAE_MACHINE_ID = "${params.get('x_machine_id')}"
+TRAE_OS_VERSION = "${params.get('x_os_version')}"`, 'text', () => alert('已复制'));
+        };
+        clearInterval(id);
+    }, 1000);
+})();