From 4b6cb0727fe5ed1ed70c4199c98f7323d3daae43 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Mon, 19 May 2025 13:33:06 +0900
Subject: [PATCH] Add MCP related functions

---
 src/ts/process/mcp.ts             | 135 ++++++++++++++++++++++++++++++
 src/ts/process/request.ts         |   1 +
 src/ts/storage/database.svelte.ts |   1 +
 3 files changed, 137 insertions(+)
 create mode 100644 src/ts/process/mcp.ts

diff --git a/src/ts/process/mcp.ts b/src/ts/process/mcp.ts
new file mode 100644
index 00000000..e3a67687
--- /dev/null
+++ b/src/ts/process/mcp.ts
@@ -0,0 +1,135 @@
+import { v4 } from "uuid"
+import { fetchNative } from "../globalApi.svelte"
+import { DBState } from "../stores.svelte";
+import { message } from "@tauri-apps/plugin-dialog";
+
+type MCPPrompt = {
+    name: string;              // Unique identifier for the prompt
+    description?: string;      // Human-readable description
+    arguments?:{               // Optional list of arguments
+        name: string;          // Argument identifier
+        description?: string;  // Argument description
+        required?: boolean;    // Whether argument is required
+    }[]
+    url?:string
+}
+
+type RPCRequestResult = {
+    jsonrpc: "2.0",
+    id: number | string,
+    result?: any,
+    error?: {
+      code: number,
+      message: string,
+      data?: any
+    }
+}
+
+let requestId = ''
+
+async function MPCClientRequest(url:string|null, method:string, params?:any):Promise<RPCRequestResult>{
+
+    const body = {
+        jsonrpc: "2.0",
+        id: requestId,
+        method: method,
+        params: params
+    }
+
+    if(!params){
+        delete body.params
+    }
+
+    try {
+
+        const response = await fetchNative(url, {
+            body: JSON.stringify(body),
+            method: "POST"
+        })
+        const d:RPCRequestResult = await response.json() 
+        return d  
+    } catch (error) {
+        return {
+            jsonrpc: "2.0",
+            id: requestId,
+            error: {
+                code: 400,
+                message: "None"
+            }
+        }
+    }
+}
+
+async function getMCPPromptList(){
+    let prompts:MCPPrompt[] = []
+    for(let i=0;i<DBState.db.mcpURLs.length;i++){
+        const url = DBState.db.mcpURLs[i]
+        await MCPHandshake(url)
+        const res = await MPCClientRequest(url, 'prompts/list');
+        if(res.error){
+            console.error(res.error)
+        }
+        if(res?.result?.prompts){
+            const rPrompts:MCPPrompt[] = res?.result?.prompts
+            for(let i =0;i<rPrompts.length;i++){
+                const part = rPrompts[i]
+                part.url = url
+                prompts.push(part)
+            }
+        }
+    }
+
+    return prompts
+
+}
+
+async function loadMCPPrompt(mcpPrompt:MCPPrompt){
+    await MCPHandshake(mcpPrompt.url)
+    const d = await MPCClientRequest(mcpPrompt.url, "prompts/get", {
+        name: mcpPrompt.name
+    })
+
+    return d
+}
+
+async function MCPHandshake(url:string, arg:{
+    forceReload?:boolean,
+} = {}){
+
+    if(!requestId || arg.forceReload){
+        requestId = v4()
+    }
+
+    const d = await MPCClientRequest(url, 'initialize', {
+        "protocolVersion": "2025-03-26",
+        "capabilities": {},
+        "clientInfo": {
+            name: "RS-MCP-CLIENT",
+            version: "1.0.0"
+        }
+    })
+    if(d?.result?.serverInfo){
+        const serverInfo:{
+            name:string,
+            version:string,
+            warnings:string[]
+        } = d.result.serverInfo
+
+        await fetchNative(url, {
+            body: JSON.stringify({
+                "jsonrpc": "2.0",
+                "method": "notifications/initialized"
+            }),
+            method: "POST"
+        })
+        
+        if(d?.result?.protocolVersion !== "2025-03-26"){
+            serverInfo.warnings.push(`Protocol version is ${d?.result?.protocolVersion}, expected "2025-03-26"`)
+        }
+
+        return serverInfo
+    }
+    else{
+        throw "MCP Handshake Failed"
+    }
+}
\ No newline at end of file
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index b466e298..b82202c1 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -23,6 +23,7 @@ import { getModelInfo, LLMFlags, LLMFormat, type LLMModel } from "../model/model
 import { runTrigger } from "./triggers";
 import { registerClaudeObserver } from "../observer.svelte";
 import { v4 } from "uuid";
+import { DBState } from "../stores.svelte";
 
 
 
diff --git a/src/ts/storage/database.svelte.ts b/src/ts/storage/database.svelte.ts
index 7a5fb478..fd67acd0 100644
--- a/src/ts/storage/database.svelte.ts
+++ b/src/ts/storage/database.svelte.ts
@@ -1026,6 +1026,7 @@ export interface Database{
     igpPrompt:string
     useTokenizerCaching:boolean
     showMenuHypaMemoryModal:boolean
+    mcpURLs:string[]
 }
 
 interface SeparateParameters{