sweetspot-remote-agent 1.8.5 → 1.8.7

package/mcp-server.js

@@ -273,20 +273,34 @@ if (!isStdio && !noRegister && !regName) {
   process.exit(1);
 }
 
-// ── 토큰 자동 생성/저장 ──
+// ── RSA 키페어 자동 생성/저장 ──
 import fs from "fs";
 
-const tokenFile = `${os.homedir()}/.sweetspot-token`;
+const keypairFile = `${os.homedir()}/.sweetspot-keypair.json`;
 
-function loadOrCreateToken() {
-  if (fs.existsSync(tokenFile)) return fs.readFileSync(tokenFile, "utf-8").trim();
-  const token = crypto.randomUUID();
-  fs.writeFileSync(tokenFile, token, "utf-8");
-  return token;
+function loadOrCreateKeypair() {
+  if (fs.existsSync(keypairFile)) {
+    const { privateKey, publicKey } = JSON.parse(fs.readFileSync(keypairFile, "utf-8"));
+    return { privateKey, publicKey };
+  }
+  const { privateKey, publicKey } = crypto.generateKeyPairSync("rsa", {
+    modulusLength: 2048,
+    publicKeyEncoding:  { type: "spki",  format: "pem" },
+    privateKeyEncoding: { type: "pkcs8", format: "pem" },
+  });
+  fs.writeFileSync(keypairFile, JSON.stringify({ privateKey, publicKey }), { mode: 0o600 });
+  console.error(`[remote-agent] RSA 키페어 생성 완료: ${keypairFile}`);
+  return { privateKey, publicKey };
+}
+
+function signChallenge(privateKey, name, timestamp) {
+  const payload = `${name}:${timestamp}`;
+  return crypto.sign("sha256", Buffer.from(payload), { key: privateKey, padding: crypto.constants.RSA_PKCS1_PSS_PADDING })
+    .toString("base64");
 }
 
 // ── WebSocket 클라이언트 (GCP 역방향 연결) ──
-async function connectWs(name, token) {
+async function connectWs(name, keypair) {
   const { default: WebSocket } = await import("ws");
   const wsUrl = BOT_SERVER.replace(/^http/, "ws") + "/agent";
   let alive = true;
@@ -296,7 +310,15 @@ async function connectWs(name, token) {
     const ws = new WebSocket(wsUrl);
 
     ws.on("open", () => {
-      ws.send(JSON.stringify({ type: "register", name, token }));
+      const timestamp = Date.now();
+      const signature = signChallenge(keypair.privateKey, name, timestamp);
+      ws.send(JSON.stringify({
+        type: "register",
+        name,
+        publicKey: keypair.publicKey,
+        timestamp,
+        signature,
+      }));
       console.error(`[remote-agent] GCP 서버에 연결 중... (${name})`);
     });
 
@@ -377,6 +399,24 @@ async function executeAction(msg) {
           try { unlinkSync(tmpFile); } catch {}
         }
       }
+      case "write_note": {
+        // 메모 앱 노트 작성 — body를 temp 파일로 전달해 인코딩 문제 방지
+        const { execSync } = await import("child_process");
+        const { writeFileSync, unlinkSync } = await import("fs");
+        const ts = Date.now();
+        const contentFile = `/tmp/sweetspot_note_${ts}.txt`;
+        const scriptFile = `/tmp/sweetspot_script_${ts}.applescript`;
+        const title = (msg.title || "Sophy 노트").replace(/\\/g, "\\\\").replace(/"/g, '\\"');
+        writeFileSync(contentFile, msg.body || "", "utf-8");
+        const script = `tell application "Notes"\nset noteBody to read POSIX file "${contentFile}" as «class utf8»\nmake new note with properties {name:"${title}", body:noteBody}\nend tell`;
+        writeFileSync(scriptFile, script, "utf-8");
+        try {
+          const output = execSync(`osascript '${scriptFile}'`, { timeout: 15000, encoding: "utf-8" });
+          return { action, success: true, output: output || "(노트 작성 완료)" };
+        } finally {
+          try { unlinkSync(scriptFile); unlinkSync(contentFile); } catch {}
+        }
+      }
       case "list_files": {
         const { execSync } = await import("child_process");
         const resolved = (msg.path || "~").replace(/^~/, process.env.HOME || "~");
@@ -414,9 +454,9 @@ async function main() {
     console.error("[remote-agent] MCP 서버 시작 (stdio)");
   } else {
     // WebSocket 모드: GCP 역방향 연결 (ngrok 불필요)
-    const token = loadOrCreateToken();
-    console.error(`[remote-agent] 토큰 파일: ${tokenFile}`);
-    await connectWs(regName, token);
+    const keypair = loadOrCreateKeypair();
+    console.error(`[remote-agent] RSA 키페어 파일: ${keypairFile}`);
+    await connectWs(regName, keypair);
   }
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sweetspot-remote-agent",
-  "version": "1.8.5",
+  "version": "1.8.7",
   "description": "Sweetspot 원격 제어 MCP 서버 — 스크린샷, 마우스/키보드, 앱 제어, 파일 탐색, 셸 실행",
   "type": "module",
   "main": "mcp-server.js",