fix: refresh mcp server functionality

Author: chungjac

server/aws-lsp-codewhisperer/src/language-server/agenticChat/agenticChatController.ts

@@ -414,6 +414,7 @@ export class AgenticChatController implements ChatHandlers {
             list: groups,
         }
     }
+
     async onMcpServerClick(params: McpServerClickParams) {
         return {
             id: params.id,

server/aws-lsp-codewhisperer/src/language-server/agenticChat/tools/mcp/mcpManager.test.ts

@@ -556,3 +556,49 @@ describe('close()', () => {
         expect(() => McpManager.instance).to.throw()
     })
 })
+
+// reinitializeMcpServers()
+describe('reinitializeMcpServers()', () => {
+    let loadStub: sinon.SinonStub
+
+    beforeEach(() => {
+        loadStub = sinon.stub(mcpUtils, 'loadMcpServerConfigs')
+    })
+
+    afterEach(async () => {
+        sinon.restore()
+        try {
+            await McpManager.instance.close()
+        } catch {}
+    })
+
+    it('calls necessary methods during reinitialization', async () => {
+        const cfg: MCPServerConfig = {
+            command: 'cmd',
+            args: [],
+            env: {},
+            disabled: false,
+            autoApprove: false,
+            toolOverrides: {},
+            __configPath__: 'cfg.json',
+        }
+        loadStub.resolves(new Map([['srv', cfg]]))
+
+        const mgr = await McpManager.init(['cfg.json'], features)
+
+        const closeStub = sinon.stub(mgr, 'close').resolves()
+        const initStub = sinon.stub(McpManager, 'init').resolves(mgr)
+
+        loadStub.resetHistory()
+
+        await mgr.reinitializeMcpServers()
+
+        expect(closeStub.calledOnce).to.be.true
+        expect(initStub.calledOnce).to.be.true
+        expect(initStub.firstCall.args[0]).to.deep.equal(['cfg.json'])
+        expect(initStub.firstCall.args[1]).to.equal(features)
+
+        closeStub.restore()
+        initStub.restore()
+    })
+})

server/aws-lsp-codewhisperer/src/language-server/agenticChat/tools/mcp/mcpManager.ts

@@ -412,6 +412,24 @@ export class McpManager {
         McpManager.#instance = undefined
     }
 
+    /**
+     * Reinitialize all MCP servers by closing existing connections and rediscovering servers
+     */
+    public async reinitializeMcpServers(): Promise<void> {
+        this.features.logging.info('Reinitializing MCP servers')
+
+        try {
+            await this.close()
+            await McpManager.init(this.configPaths, this.features)
+
+            const reinitializedServerCount = McpManager.#instance?.mcpServers.size
+            this.features.logging.info(`MCP servers reinitialized completed with length of ${reinitializedServerCount}`)
+        } catch (err: any) {
+            this.features.logging.error(`Error reinitializing MCP servers: ${err.message}`)
+            throw err
+        }
+    }
+
     /**
      * Check if a tool requires approval.
      */