Added login working

Author: nang-dev

src/extension.ts

@@ -67,6 +67,14 @@ export function activate(context: vscode.ExtensionContext) {
 			console.dir(data)
 			context.secrets.store("pearai-token", data.accessToken)
 			context.secrets.store("pearai-refresh", data.refreshToken)
+			// Update MCP server with new token
+			const provider = await ClineProvider.getInstance()
+			if (provider) {
+				const mcpHub = provider.getMcpHub()
+				if (mcpHub) {
+					await mcpHub.updatePearAiApiKey(data.accessToken)
+				}
+			}
 			vscode.commands.executeCommand("roo-cline.plusButtonClicked")
 		}),
 	)

src/services/mcp/McpHub.ts

@@ -56,11 +56,13 @@ export class McpHub {
 	private disposables: vscode.Disposable[] = []
 	private settingsWatcher?: vscode.FileSystemWatcher
 	private fileWatchers: Map<string, FSWatcher> = new Map()
+	private context: vscode.ExtensionContext
 	connections: McpConnection[] = []
 	isConnecting: boolean = false
 
-	constructor(provider: ClineProvider) {
+	constructor(provider: ClineProvider, context: vscode.ExtensionContext) {
 		this.providerRef = new WeakRef(provider)
+		this.context = context
 		this.watchMcpSettingsFile()
 		this.initializeMcpServers()
 	}
@@ -154,6 +156,16 @@ export class McpHub {
 		}
 	}
 
+	private async getPearAiApiKey(): Promise<string | null> {
+		try {
+			const token = await this.context.secrets.get("pearai-token")
+			return token || null
+		} catch (error) {
+			console.error("Failed to get PearAI token from secrets:", error)
+			return null
+		}
+	}
+
 	private async initializeMcpServers(): Promise<void> {
 		try {
 			const settingsPath = await this.getMcpSettingsFilePath()
@@ -168,6 +180,17 @@ export class McpHub {
 				if (!mergedServers[serverName]) {
 					mergedServers[serverName] = serverConfig
 				}
+
+				// If this is the pearai server, check login status and update API key
+				if (serverName === "pearai") {
+					const apiKey = await this.getPearAiApiKey()
+					if (apiKey) {
+						mergedServers[serverName] = {
+							...serverConfig,
+							args: ["pearai-mcp", apiKey],
+						}
+					}
+				}
 			}
 
 			// Update the settings file with merged settings
@@ -482,6 +505,29 @@ export class McpHub {
 		})
 	}
 
+	public async updatePearAiApiKey(apiKey: string): Promise<void> {
+		try {
+			const settingsPath = await this.getMcpSettingsFilePath()
+			const content = await fs.readFile(settingsPath, "utf-8")
+			const config = JSON.parse(content)
+
+			if (config.mcpServers?.pearai) {
+				config.mcpServers.pearai = {
+					...config.mcpServers.pearai,
+					args: ["pearai-mcp", apiKey],
+				}
+
+				await fs.writeFile(settingsPath, JSON.stringify(config, null, 2))
+				await this.updateServerConnections(config.mcpServers)
+				vscode.window.showInformationMessage("PearAI API key updated successfully")
+			}
+		} catch (error) {
+			console.error("Failed to update PearAI API key:", error)
+			vscode.window.showErrorMessage("Failed to update PearAI API key")
+			throw error
+		}
+	}
+
 	public async toggleServerDisabled(serverName: string, disabled: boolean): Promise<void> {
 		let settingsPath: string
 		try {

src/services/mcp/McpServerManager.ts

@@ -36,7 +36,7 @@ export class McpServerManager {
 			try {
 				// Double-check instance in case it was created while we were waiting
 				if (!this.instance) {
-					this.instance = new McpHub(provider)
+					this.instance = new McpHub(provider, context)
 					// Store a unique identifier in global state to track the primary instance
 					await context.globalState.update(this.GLOBAL_STATE_KEY, Date.now().toString())
 				}