cursornew2026/extension_clean/out/extension.js

'use strict';

// ============================================
// 蜂鸟Pro Extension
// ============================================

Object.defineProperty(exports, "__esModule", { value: true });

const vscode = require('vscode');
const provider_1 = require('./webview/provider');
const fs = require('fs');
const path = require('path');

let usageStatusBarItem;

// 创建输出通道
exports.outputChannel = vscode.window.createOutputChannel('蜂鸟Pro');

/**
 * 日志函数
 */
function log(message) {
    const timestamp = new Date().toLocaleTimeString();
    exports.outputChannel.appendLine('[' + timestamp + '] ' + message);
    console.log('[蜂鸟Pro] ' + message);
}
exports.log = log;

/**
 * 清理 Service Worker 缓存
 */
function cleanServiceWorkerCache() {
    try {
        const platform = process.platform;
        const cachePaths = [];

        if (platform === 'win32') {
            const appData = process.env.APPDATA || '';
            const localAppData = process.env.LOCALAPPDATA || '';
            cachePaths.push(
                path.join(appData, 'Cursor', 'Service Worker'),
                path.join(localAppData, 'Cursor', 'Service Worker'),
                path.join(appData, 'Cursor', 'GPUCache'),
                path.join(localAppData, 'Cursor', 'GPUCache')
            );
        } else if (platform === 'darwin') {
            const home = process.env.HOME || '';
            cachePaths.push(
                path.join(home, 'Library', 'Application Support', 'Cursor', 'Service Worker'),
                path.join(home, 'Library', 'Caches', 'Cursor', 'Service Worker')
            );
        } else {
            const home = process.env.HOME || '';
            cachePaths.push(
                path.join(home, '.config', 'Cursor', 'Service Worker'),
                path.join(home, '.cache', 'Cursor', 'Service Worker')
            );
        }

        for (const cachePath of cachePaths) {
            if (!fs.existsSync(cachePath)) continue;

            // 清理 ScriptCache
            const scriptCachePath = path.join(cachePath, 'ScriptCache');
            if (fs.existsSync(scriptCachePath)) {
                try {
                    const files = fs.readdirSync(scriptCachePath);
                    for (const file of files) {
                        try {
                            fs.unlinkSync(path.join(scriptCachePath, file));
                        } catch (e) {}
                    }
                    console.log('[蜂鸟Pro] Service Worker ScriptCache 已清理:', scriptCachePath);
                } catch (e) {}
            }

            // 清理 CacheStorage
            const cacheStoragePath = path.join(cachePath, 'CacheStorage');
            if (fs.existsSync(cacheStoragePath)) {
                try {
                    deleteFolderRecursive(cacheStoragePath);
                    console.log('[蜂鸟Pro] Service Worker CacheStorage 已清理:', cacheStoragePath);
                } catch (e) {}
            }

            // 清理 Database
            const databasePath = path.join(cachePath, 'Database');
            if (fs.existsSync(databasePath)) {
                try {
                    deleteFolderRecursive(databasePath);
                    console.log('[蜂鸟Pro] Service Worker Database 已清理:', databasePath);
                } catch (e) {}
            }
        }
    } catch (error) {
        console.log('[蜂鸟Pro] 清理 Service Worker 缓存时出错:', error);
    }
}

/**
 * 递归删除文件夹
 */
function deleteFolderRecursive(folderPath) {
    if (fs.existsSync(folderPath)) {
        fs.readdirSync(folderPath).forEach(file => {
            const curPath = path.join(folderPath, file);
            if (fs.lstatSync(curPath).isDirectory()) {
                deleteFolderRecursive(curPath);
            } else {
                try {
                    fs.unlinkSync(curPath);
                } catch (e) {}
            }
        });
        try {
            fs.rmdirSync(folderPath);
        } catch (e) {}
    }
}

/**
 * 激活扩展
 */
function activate(context) {
    // 清理 Service Worker 缓存
    cleanServiceWorkerCache();

    // 创建 WebView Provider
    const viewProvider = new provider_1.HummingbirdProViewProvider(context.extensionUri, context);

    // 注册 WebView
    context.subscriptions.push(
        vscode.window.registerWebviewViewProvider('hummingbird.mainView', viewProvider)
    );

    // 创建状态栏项
    usageStatusBarItem = vscode.window.createStatusBarItem(vscode.StatusBarAlignment.Right, 100);
    usageStatusBarItem.text = '$(dashboard) 用量: --';
    usageStatusBarItem.tooltip = '点击查看账号用量详情';
    usageStatusBarItem.command = 'hummingbird.showPanel';
    usageStatusBarItem.backgroundColor = new vscode.ThemeColor('statusBarItem.prominentBackground');

    // 如果有保存的 key，显示状态栏
    const savedKey = context.globalState.get('hummingbird.key');
    if (savedKey) {
        usageStatusBarItem.show();
    }

    context.subscriptions.push(usageStatusBarItem);

    // 设置同步的键
    context.globalState.setKeysForSync(['hummingbird.key']);

    // 注册显示面板命令
    context.subscriptions.push(
        vscode.commands.registerCommand('hummingbird.showPanel', () => {
            vscode.commands.executeCommand('hummingbird.mainView.focus');
        })
    );
}
exports.activate = activate;

/**
 * 停用扩展
 */
function deactivate() {
    console.log('蜂鸟Pro 插件已停用');
}
exports.deactivate = deactivate;

/**
 * 显示状态栏
 */
function showStatusBar() {
    if (usageStatusBarItem) {
        usageStatusBarItem.show();
    }
}
exports.showStatusBar = showStatusBar;

/**
 * 隐藏状态栏
 */
function hideStatusBar() {
    if (usageStatusBarItem) {
        usageStatusBarItem.hide();
    }
}
exports.hideStatusBar = hideStatusBar;

/**
 * 更新用量状态栏
 * @param {number} requestCount - 请求次数
 * @param {number|string} usageCost - 已用额度
 */
function updateUsageStatusBar(requestCount, usageCost) {
    if (usageStatusBarItem) {
        const count = typeof requestCount === 'number' ? requestCount : requestCount;
        const cost = typeof usageCost === 'number' ? usageCost : parseFloat(usageCost.toString().replace('$', '')) || 0;
        const costDisplay = typeof usageCost === 'number' ? '$' + usageCost.toFixed(2) : usageCost;

        usageStatusBarItem.text = '$(dashboard) ' + count + '次 | ' + costDisplay;
        usageStatusBarItem.tooltip = '请求次数: ' + count + '\n已用额度: ' + costDisplay + '\n点击查看详情';

        // 根据用量设置颜色
        if (cost >= 10) {
            // 高用量 - 红色警告
            usageStatusBarItem.backgroundColor = new vscode.ThemeColor('statusBarItem.errorBackground');
            usageStatusBarItem.color = undefined;
        } else if (cost >= 5) {
            // 中用量 - 黄色警告
            usageStatusBarItem.backgroundColor = new vscode.ThemeColor('statusBarItem.warningBackground');
            usageStatusBarItem.color = undefined;
        } else {
            // 低用量 - 绿色
            usageStatusBarItem.backgroundColor = undefined;
            usageStatusBarItem.color = '#4ade80';
        }
    }
}
exports.updateUsageStatusBar = updateUsageStatusBar;