ai-worktool 1.0.0

package/dist/user.js

@@ -0,0 +1,505 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setEnvironment = setEnvironment;
+exports.getEnvironment = getEnvironment;
+exports.setTokenStorage = setTokenStorage;
+exports.getTokenStorage = getTokenStorage;
+exports.openExternalUrl = openExternalUrl;
+exports.login = login;
+exports.cancelLogin = cancelLogin;
+exports.logout = logout;
+exports.isLoggedIn = isLoggedIn;
+exports.getUserData = getUserData;
+exports.decodeJwt = decodeJwt;
+exports.isJwtExpired = isJwtExpired;
+const http = __importStar(require("http"));
+const url_1 = require("url");
+const crypto_1 = __importDefault(require("crypto"));
+const jsonwebtoken_1 = __importDefault(require("jsonwebtoken"));
+const util_1 = __importDefault(require("util"));
+const child_process_1 = require("child_process");
+// 转换exec为Promise形式，便于async/await使用
+const execAsync = util_1.default.promisify(child_process_1.exec);
+let tokenStorage = {
+    get: async () => {
+        return localStorage.getItem('token');
+    },
+    set: async (token) => {
+        localStorage.setItem('token', token);
+    },
+    clear: async () => {
+        localStorage.removeItem('token');
+    }
+};
+let environment = {
+    showInfoMessage: async (message) => {
+        console.log(message);
+    },
+    showErrorMessage: async (message) => {
+        console.error(message);
+    }
+};
+function setEnvironment(env) {
+    environment = env;
+}
+function getEnvironment() {
+    return environment;
+}
+function setTokenStorage(storage) {
+    tokenStorage = storage;
+}
+function getTokenStorage() {
+    return tokenStorage;
+}
+// 生成随机字符串用于CSRF保护
+function generateState() {
+    return crypto_1.default.randomBytes(16).toString('hex');
+}
+let lastLoginServer = null;
+let lastReject = null;
+/**
+ * 启动临时HTTP服务器接收登录回调
+ * @param state 用于验证的state值
+ * @returns 包含token的Promise
+ */
+function startCallbackServer(state, port) {
+    return new Promise((resolve, reject) => {
+        lastReject = reject;
+        const server = (lastLoginServer = http.createServer((req, res) => {
+            if (!req.url) {
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                res.writeHead(400, { 'Content-Type': 'text/html; charset=utf-8' });
+                res.end('Bad Request');
+                return;
+            }
+            const parsedUrl = new url_1.URL(req.url, `http://${req.headers.host}`);
+            const callbackState = parsedUrl.searchParams.get('state');
+            const token = parsedUrl.searchParams.get('token');
+            // 验证state防止CSRF攻击
+            if (callbackState !== state) {
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                res.writeHead(403, { 'Content-Type': 'text/html; charset=utf-8' });
+                res.end(`
+                    <html>
+                        <head>
+                            <style>
+                                body {
+                                    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                                    display: flex;
+                                    flex-direction: column;
+                                    align-items: center;
+                                    justify-content: center;
+                                    height: 100vh;
+                                    margin: 0;
+                                    background-color: #f8f9fa;
+                                    color: #333;
+                                }
+                                .card {
+                                    background-color: white;
+                                    border-radius: 12px;
+                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+                                    padding: 2rem;
+                                    text-align: center;
+                                    max-width: 400px;
+                                    width: 90%;
+                                    border-left: 4px solid #dc3545;
+                                }
+                                h1 {
+                                    color: #dc3545;
+                                    margin-bottom: 1rem;
+                                    font-size: 1.8rem;
+                                }
+                                p {
+                                    margin: 0.5rem 0;
+                                    color: #6c757d;
+                                }
+                                .countdown {
+                                    font-weight: bold;
+                                    color: #dc3545;
+                                    font-size: 1.2rem;
+                                }
+                            </style>
+                        </head>
+                        <body>
+                            <div class="card">
+                                <h1>登录验证失败</h1>
+                                <p>无效的状态参数，可能存在安全风险</p>
+                            </div>
+                        </body>
+                    </html>
+                `);
+                reject(new Error('登录验证失败：无效的状态参数'));
+                server.close();
+                return;
+            }
+            // 处理token
+            if (token) {
+                // 构建清理后的URL（移除token和state参数）
+                parsedUrl.searchParams.delete('token');
+                parsedUrl.searchParams.delete('state');
+                const cleanUrl = parsedUrl.toString();
+                // 返回包含自动刷新和关闭逻辑的HTML
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                res.writeHead(200, { 'Content-Type': 'text/html; charset=utf-8' });
+                res.end(`
+                    <html>
+                        <head>
+                            <title>登录成功</title>
+                            <style>
+                                body {
+                                    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                                    display: flex;
+                                    flex-direction: column;
+                                    align-items: center;
+                                    justify-content: center;
+                                    height: 100vh;
+                                    margin: 0;
+                                    background-color: #f8f9fa;
+                                    color: #333;
+                                }
+                                .card {
+                                    background-color: white;
+                                    border-radius: 12px;
+                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+                                    padding: 2rem;
+                                    text-align: center;
+                                    max-width: 400px;
+                                    width: 90%;
+                                    border-left: 4px solid #28a745;
+                                }
+                                h1 {
+                                    color: #28a745;
+                                    margin-bottom: 1rem;
+                                    font-size: 1.8rem;
+                                }
+                                p {
+                                    margin: 0.5rem 0;
+                                    color: #6c757d;
+                                }
+                                .countdown {
+                                    font-weight: bold;
+                                    color: #28a745;
+                                    font-size: 1.2rem;
+                                }
+                                .checkmark {
+                                    font-size: 3rem;
+                                    color: #28a745;
+                                    margin-bottom: 1rem;
+                                }
+                            </style>
+                        </head>
+                        <body>
+                            <div class="card">
+                                <div class="checkmark">✓</div>
+                                <h1>登录成功！</h1>
+                                <p>您可以关闭此窗口返回应用。</p>
+                            </div>
+                        </body>
+                    </html>
+                `);
+                resolve(token);
+            }
+            else {
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                res.writeHead(400, { 'Content-Type': 'text/html; charset=utf-8' });
+                res.end(`
+                    <html>
+                        <head>
+                            <style>
+                                body {
+                                    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                                    display: flex;
+                                    flex-direction: column;
+                                    align-items: center;
+                                    justify-content: center;
+                                    height: 100vh;
+                                    margin: 0;
+                                    background-color: #f8f9fa;
+                                    color: #333;
+                                }
+                                .card {
+                                    background-color: white;
+                                    border-radius: 12px;
+                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+                                    padding: 2rem;
+                                    text-align: center;
+                                    max-width: 400px;
+                                    width: 90%;
+                                    border-left: 4px solid #dc3545;
+                                }
+                                h1 {
+                                    color: #dc3545;
+                                    margin-bottom: 1rem;
+                                    font-size: 1.8rem;
+                                }
+                                p {
+                                    margin: 0.5rem 0;
+                                    color: #6c757d;
+                                }
+                                .countdown {
+                                    font-weight: bold;
+                                    color: #dc3545;
+                                    font-size: 1.2rem;
+                                }
+                                .error-icon {
+                                    font-size: 3rem;
+                                    color: #dc3545;
+                                    margin-bottom: 1rem;
+                                }
+                            </style>
+                        </head>
+                        <body>
+                            <div class="card">
+                                <div class="error-icon">✗</div>
+                                <h1>登录失败</h1>
+                                <p>未收到有效的认证信息</p>
+                            </div>
+                        </body>
+                    </html>
+                `);
+                reject(new Error('登录失败：未收到token'));
+            }
+            server.close();
+        }));
+        // 监听随机可用端口
+        server.listen(port, 'localhost', () => {
+            const address = server.address();
+            if (typeof address === 'object' && address) {
+                console.log(`回调服务器启动在端口 ${address.port}`);
+            }
+        });
+        // 设置超时（5分钟）
+        setTimeout(() => {
+            server.close();
+            reject(new Error('登录超时，请重试'));
+        }, 5 * 60 * 1000);
+    });
+}
+/**
+ * 获取可用端口
+ * @returns 可用端口号
+ */
+function getAvailablePort() {
+    return new Promise((resolve, reject) => {
+        const tempServer = http.createServer();
+        tempServer.listen(0, 'localhost', () => {
+            const address = tempServer.address();
+            if (typeof address === 'object' && address && address.port) {
+                const port = address.port;
+                tempServer.close(() => resolve(port));
+            }
+            else {
+                tempServer.close();
+                reject(new Error('无法获取可用端口'));
+            }
+        });
+    });
+}
+async function openExternalUrl(url) {
+    try {
+        // 根据不同操作系统，使用相应的命令打开浏览器
+        switch (process.platform) {
+            case 'darwin': // macOS
+                await execAsync(`open "${url}"`);
+                break;
+            case 'win32': // Windows
+                await execAsync(`start "" "${url}"`);
+                break;
+            default: // Linux及其他类Unix系统
+                await execAsync(`xdg-open "${url}"`);
+                break;
+        }
+        return true;
+    }
+    catch (error) {
+        console.error('无法打开外部浏览器：', error);
+        return false;
+    }
+}
+const AUTH_SERVER_URL = 'https://login.jianguoke.cn/login';
+/**
+ * 执行登录流程
+ * @param env 环境接口
+ * @param tokenStorage token存储接口
+ * @param authServerUrl 认证服务器URL
+ * @param clientId 客户端ID
+ * @returns 登录成功后的token
+ */
+async function login(clientId) {
+    const authServerUrl = AUTH_SERVER_URL;
+    const env = getEnvironment();
+    const tokenStorage = getTokenStorage();
+    const state = generateState();
+    const port = await getAvailablePort();
+    const callbackUrl = `http://localhost:${port}/callback`;
+    await cancelLogin();
+    // 启动回调服务器
+    const serverPromise = startCallbackServer(state, port);
+    // 构建认证服务器URL
+    const authUrl = new url_1.URL(authServerUrl);
+    authUrl.searchParams.set('clientId', clientId);
+    authUrl.searchParams.set('redirectUrl', callbackUrl);
+    authUrl.searchParams.set('state', state);
+    authUrl.searchParams.set('responseType', 'token');
+    // 打开外部登录页面
+    const opened = await openExternalUrl(authUrl.toString());
+    if (!opened) {
+        throw new Error('无法打开外部浏览器进行登录');
+    }
+    // 等待回调返回token
+    const token = await serverPromise;
+    // 存储token
+    await tokenStorage.set(token);
+    env.showInfoMessage('登录成功！');
+    return token;
+}
+async function cancelLogin() {
+    if (lastLoginServer) {
+        lastLoginServer.close();
+        lastLoginServer = null;
+    }
+    if (lastReject) {
+        lastReject(new Error('登录已取消'));
+        lastReject = null;
+    }
+}
+/**
+ * 执行登出操作
+ * @param tokenStorage token存储接口
+ * @param env 环境接口
+ */
+async function logout() {
+    const env = getEnvironment();
+    const tokenStorage = getTokenStorage();
+    await tokenStorage.clear();
+    env.showInfoMessage('已成功登出');
+}
+/**
+ * 检查用户是否已登录
+ * @param tokenStorage token存储接口
+ * @returns 是否已登录
+ */
+function isLoggedIn() {
+    const tokenStorage = getTokenStorage();
+    const token = tokenStorage.get();
+    if (!token) {
+        return false;
+    }
+    if (isJwtExpired(token)) {
+        return false;
+    }
+    return true;
+}
+async function getUserData() {
+    const token = await getTokenStorage().get();
+    if (!token) {
+        return null;
+    }
+    try {
+        const decoded = decodeJwt(token);
+        return decoded;
+    }
+    catch (error) {
+        console.error('解析用户数据失败:', error);
+    }
+    return null;
+}
+function isTokenExpired(payload) {
+    if (!payload || !payload.exp) {
+        // 没有过期时间字段，视为过期
+        return true;
+    }
+    // JWT的exp是秒级时间戳，需要转换为毫秒
+    const expirationTime = payload.exp * 1000;
+    const currentTime = Date.now();
+    // 检查是否已过期（提前30秒过期，避免网络延迟问题）
+    return currentTime >= expirationTime - 30000;
+}
+/**
+ * 解析JWT获取其中的信息
+ * @param token JWT字符串
+ * @returns 解析后的payload对象
+ */
+function decodeJwt(token) {
+    try {
+        // 不验证签名，仅解码获取信息
+        const decoded = jsonwebtoken_1.default.decode(token);
+        if (isTokenExpired(decoded)) {
+            return null;
+        }
+        return decoded;
+    }
+    catch (error) {
+        console.error('JWT解码失败:', error);
+        return null;
+    }
+}
+/**
+ * 验证JWT是否过期
+ * @param token JWT字符串 或 已解析的JWT payload
+ * @returns 如果过期返回true，否则返回false
+ */
+function isJwtExpired(token) {
+    try {
+        let payload;
+        // 如果传入的是字符串，则先解析
+        if (typeof token === 'string') {
+            payload = decodeJwt(token);
+            if (!payload) {
+                return true;
+            } // 解析失败视为过期
+        }
+        else {
+            payload = token;
+        }
+        // 检查是否包含过期时间
+        if (!payload.exp) {
+            return false;
+        } // 没有过期时间，视为不过期
+        // 计算当前时间（秒级时间戳）
+        const currentTime = Math.floor(Date.now() / 1000);
+        // 比较当前时间和过期时间
+        return currentTime > payload.exp;
+    }
+    catch (error) {
+        console.error('JWT过期检查失败:', error);
+        return true; // 发生错误时视为过期
+    }
+}
+//# sourceMappingURL=user.js.map