328 lines
10 KiB
TypeScript
328 lines
10 KiB
TypeScript
import * as vscode from "vscode";
|
||
import { ICViewProvider } from "./views/ICViewProvider";
|
||
import { showICHelperPanel } from "./panels/ICHelperPanel";
|
||
import { VCDViewerPanel, VCDViewerEditorProvider } from "./panels/VCDViewerPanel";
|
||
import { ChatHistoryManager } from "./utils/chatHistoryManager";
|
||
import { ICCoderAuthenticationProvider } from "./services/icCoderAuthProvider";
|
||
import { VCDFileServer } from "./services/vcdFileServer";
|
||
import { initUserService } from "./services/userService";
|
||
import { initCreditsService } from "./services/creditsService";
|
||
import { isTokenExpired } from "./utils/jwtUtils";
|
||
import { NotificationService } from "./services/notificationService";
|
||
import { InvitationService } from "./services/invitationService";
|
||
|
||
export async function activate(context: vscode.ExtensionContext) {
|
||
console.log("🎉 IC Coder 插件已激活！");
|
||
|
||
// 初始化通知服务
|
||
const notificationService = NotificationService.getInstance(context);
|
||
console.log('[Extension] 通知服务已初始化');
|
||
|
||
// 【关键】在创建 AuthProvider 之前，先检查并清除过期的 session
|
||
const storedSessions = context.globalState.get<any[]>('icCoderSessions', []);
|
||
console.log('[Extension] 检查 sessions 数量:', storedSessions.length);
|
||
|
||
if (storedSessions.length > 0) {
|
||
const session = storedSessions[0];
|
||
const token = session.accessToken;
|
||
console.log('[Extension] 检查 token 是否过期...');
|
||
|
||
if (token) {
|
||
const expired = isTokenExpired(token);
|
||
console.log('[Extension] token 过期检查结果:', expired);
|
||
|
||
if (expired) {
|
||
// 必须等待清除完成后再创建 AuthProvider
|
||
await context.globalState.update('icCoderSessions', []);
|
||
await context.globalState.update('icCoderUserInfo', undefined);
|
||
console.log('[Extension] Token 已过期，已清除所有登录状态');
|
||
}
|
||
}
|
||
}
|
||
|
||
// 初始化用户服务
|
||
initUserService(context);
|
||
|
||
// 初始化 Credits 服务
|
||
initCreditsService(context);
|
||
|
||
// 初始化 VCD 文件服务器
|
||
const vcdFileServer = new VCDFileServer(context.extensionUri);
|
||
vcdFileServer.start().then((port) => {
|
||
console.log(`VCD 文件服务器已启动，端口: ${port}`);
|
||
}).catch((error) => {
|
||
console.error("启动 VCD 文件服务器失败:", error);
|
||
});
|
||
|
||
// 在插件停用时关闭服务器
|
||
context.subscriptions.push({
|
||
dispose: () => vcdFileServer.stop()
|
||
});
|
||
|
||
// 注册 Authentication Provider（此时 icCoderSessions 已经被清除）
|
||
const authProvider = new ICCoderAuthenticationProvider(context);
|
||
context.subscriptions.push(
|
||
vscode.authentication.registerAuthenticationProvider(
|
||
"iccoder",
|
||
"IC Coder",
|
||
authProvider
|
||
)
|
||
);
|
||
|
||
// 检查登录状态，如果已登录则自动打开聊天面板
|
||
vscode.authentication.getSession("iccoder", [], { createIfNone: false })
|
||
.then((session) => {
|
||
if (session) {
|
||
vscode.commands.executeCommand("ic-coder.openChat");
|
||
}
|
||
}, () => {
|
||
// 未登录，不做任何操作
|
||
});
|
||
|
||
// 注册命令：打开助手面板
|
||
const openPanelCommand = vscode.commands.registerCommand(
|
||
"ic-coder.openPanel",
|
||
async () => {
|
||
await showICHelperPanel(context);
|
||
}
|
||
);
|
||
|
||
// 注册命令：打开聊天（用于侧边栏）
|
||
const openChatCommand = vscode.commands.registerCommand(
|
||
"ic-coder.openChat",
|
||
async () => {
|
||
await showICHelperPanel(context);
|
||
}
|
||
);
|
||
|
||
// 注册命令：打开 VCD 波形查看器
|
||
const openVCDViewerCommand = vscode.commands.registerCommand(
|
||
"ic-coder.openVCDViewer",
|
||
async (vcdFilePath?: string) => {
|
||
if (!vcdFilePath) {
|
||
// 如果没有提供文件路径，让用户选择 VCD 文件
|
||
const fileUri = await vscode.window.showOpenDialog({
|
||
canSelectFiles: true,
|
||
canSelectFolders: false,
|
||
canSelectMany: false,
|
||
filters: {
|
||
"VCD 文件": ["vcd"],
|
||
"所有文件": ["*"],
|
||
},
|
||
title: "选择 VCD 文件",
|
||
});
|
||
|
||
if (fileUri && fileUri[0]) {
|
||
vcdFilePath = fileUri[0].fsPath;
|
||
} else {
|
||
return;
|
||
}
|
||
}
|
||
|
||
VCDViewerPanel.createOrShow(context.extensionUri, vcdFilePath, vcdFileServer);
|
||
}
|
||
);
|
||
|
||
// 注册命令：在浏览器中打开 VCD 波形查看器
|
||
const openVCDViewerInBrowserCommand = vscode.commands.registerCommand(
|
||
"ic-coder.openVCDViewerInBrowser",
|
||
async (vcdFilePath?: string) => {
|
||
if (!vcdFilePath) {
|
||
const fileUri = await vscode.window.showOpenDialog({
|
||
canSelectFiles: true,
|
||
canSelectFolders: false,
|
||
canSelectMany: false,
|
||
filters: {
|
||
"VCD 文件": ["vcd"],
|
||
"所有文件": ["*"],
|
||
},
|
||
title: "选择 VCD 文件",
|
||
});
|
||
|
||
if (fileUri && fileUri[0]) {
|
||
vcdFilePath = fileUri[0].fsPath;
|
||
} else {
|
||
return;
|
||
}
|
||
}
|
||
|
||
// 注册文件到服务器
|
||
const fileId = vcdFileServer.registerFile(vcdFilePath);
|
||
const viewerUrl = vcdFileServer.getViewerUrl(fileId);
|
||
|
||
// 在默认浏览器中打开
|
||
vscode.env.openExternal(vscode.Uri.parse(viewerUrl));
|
||
vscode.window.showInformationMessage(`波形查看器已在浏览器中打开`);
|
||
}
|
||
);
|
||
|
||
// 注册命令：用户登录
|
||
const loginCommand = vscode.commands.registerCommand(
|
||
"ic-coder.login",
|
||
async () => {
|
||
try {
|
||
// 先清除 session 偏好，避免 VSCode 弹出"账户不一致"确认框
|
||
try {
|
||
await vscode.authentication.getSession("iccoder", [], {
|
||
clearSessionPreference: true,
|
||
createIfNone: false
|
||
});
|
||
} catch {
|
||
// 忽略错误
|
||
}
|
||
|
||
// 创建新 session
|
||
await vscode.authentication.getSession("iccoder", [], { createIfNone: true });
|
||
} catch (error) {
|
||
vscode.window.showErrorMessage(`登录失败: ${error}`);
|
||
}
|
||
}
|
||
);
|
||
|
||
// 注册命令：用户登出
|
||
const logoutCommand = vscode.commands.registerCommand(
|
||
"ic-coder.logout",
|
||
async () => {
|
||
try {
|
||
const session = await vscode.authentication.getSession("iccoder", [], { createIfNone: false });
|
||
if (session) {
|
||
// 调用 authProvider 的 removeSession 方法
|
||
await authProvider.removeSession(session.id);
|
||
// 清除邀请码验证状态
|
||
await InvitationService.clearVerificationStatus(context);
|
||
} else {
|
||
vscode.window.showInformationMessage("当前未登录");
|
||
}
|
||
} catch (error) {
|
||
vscode.window.showInformationMessage("当前未登录");
|
||
}
|
||
}
|
||
);
|
||
|
||
// 注册命令：更换邀请码
|
||
const changeInvitationCodeCommand = vscode.commands.registerCommand(
|
||
"ic-coder.changeInvitationCode",
|
||
async () => {
|
||
const confirm = await vscode.window.showWarningMessage(
|
||
'确定要更换邀请码吗？',
|
||
'确定',
|
||
'取消'
|
||
);
|
||
|
||
if (confirm === '确定') {
|
||
await InvitationService.clearVerificationStatus(context);
|
||
vscode.window.showInformationMessage('已清除邀请码，请重新验证');
|
||
}
|
||
}
|
||
);
|
||
|
||
// 注册命令：测试系统通知
|
||
const testNotificationCommand = vscode.commands.registerCommand(
|
||
"ic-coder.testNotification",
|
||
() => {
|
||
console.log('[Extension] ========== 测试通知命令被调用 ==========');
|
||
|
||
// 先显示 VS Code 通知确认命令执行
|
||
vscode.window.showInformationMessage('正在测试系统通知...');
|
||
|
||
// 发送系统通知
|
||
notificationService.success(
|
||
'IC Coder - 测试通知',
|
||
'系统通知功能正常工作！',
|
||
() => {
|
||
vscode.window.showInformationMessage('您点击了系统通知！');
|
||
}
|
||
);
|
||
|
||
console.log('[Extension] 测试通知命令执行完成');
|
||
}
|
||
);
|
||
|
||
// 注册命令：查看会话历史
|
||
// TODO: 这些命令需要根据新的任务架构重新实现
|
||
// 暂时注释掉，等待重新实现
|
||
/*
|
||
const viewHistoryCommand = vscode.commands.registerCommand(
|
||
"ic-coder.viewHistory",
|
||
() => {
|
||
vscode.window.showInformationMessage("此功能正在重构中，敬请期待");
|
||
}
|
||
);
|
||
|
||
const newSessionCommand = vscode.commands.registerCommand(
|
||
"ic-coder.newSession",
|
||
() => {
|
||
vscode.window.showInformationMessage("此功能正在重构中，敬请期待");
|
||
}
|
||
);
|
||
|
||
const exportSessionCommand = vscode.commands.registerCommand(
|
||
"ic-coder.exportSession",
|
||
() => {
|
||
vscode.window.showInformationMessage("此功能正在重构中，敬请期待");
|
||
}
|
||
);
|
||
|
||
const deleteSessionCommand = vscode.commands.registerCommand(
|
||
"ic-coder.deleteSession",
|
||
() => {
|
||
vscode.window.showInformationMessage("此功能正在重构中，敬请期待");
|
||
}
|
||
);
|
||
|
||
const clearHistoryCommand = vscode.commands.registerCommand(
|
||
"ic-coder.clearHistory",
|
||
() => {
|
||
vscode.window.showInformationMessage("此功能正在重构中，敬请期待");
|
||
}
|
||
);
|
||
|
||
const searchSessionCommand = vscode.commands.registerCommand(
|
||
"ic-coder.searchSession",
|
||
() => {
|
||
vscode.window.showInformationMessage("此功能正在重构中，敬请期待");
|
||
}
|
||
);
|
||
*/
|
||
|
||
// 注册侧边栏视图
|
||
const viewProvider = new ICViewProvider(context.extensionUri, context);
|
||
const viewRegistration = vscode.window.registerWebviewViewProvider(
|
||
"ic-coder.mainView",
|
||
viewProvider,
|
||
{
|
||
webviewOptions: {
|
||
retainContextWhenHidden: true
|
||
}
|
||
}
|
||
);
|
||
|
||
// 注册 VCD 自定义编辑器
|
||
const vcdEditorProvider = VCDViewerEditorProvider.register(context, vcdFileServer);
|
||
|
||
// 添加到订阅
|
||
context.subscriptions.push(
|
||
openPanelCommand,
|
||
openChatCommand,
|
||
openVCDViewerCommand,
|
||
openVCDViewerInBrowserCommand,
|
||
loginCommand,
|
||
logoutCommand,
|
||
changeInvitationCodeCommand,
|
||
testNotificationCommand,
|
||
// testTrialUserCommand,
|
||
// testExpiredUserCommand,
|
||
// TODO: 等待重新实现这些命令
|
||
// viewHistoryCommand,
|
||
// newSessionCommand,
|
||
// exportSessionCommand,
|
||
// deleteSessionCommand,
|
||
// clearHistoryCommand,
|
||
// searchSessionCommand,
|
||
viewRegistration,
|
||
vcdEditorProvider
|
||
);
|
||
}
|
||
|
||
export function deactivate() {}
|