poke-gate 0.0.5 → 0.0.7

package/bin/poke-gate.js

@@ -21,10 +21,20 @@ function saveConfig(config) {
   writeFileSync(CONFIG_PATH, JSON.stringify(config, null, 2));
 }
 
+function loadPokeCredentials() {
+  try {
+    const creds = JSON.parse(readFileSync(join(CONFIG_DIR, "poke", "credentials.json"), "utf-8"));
+    if (creds.token) return creds.token;
+  } catch {}
+  return null;
+}
+
 function resolveToken() {
   if (process.env.POKE_API_KEY) return process.env.POKE_API_KEY;
   const config = loadConfig();
   if (config.apiKey) return config.apiKey;
+  const pokeCreds = loadPokeCredentials();
+  if (pokeCreds) return pokeCreds;
   return null;
 }
 
@@ -47,19 +57,24 @@ async function onboarding() {
   console.log();
   console.log("  ⚠  This grants full shell access. Only run on trusted networks.");
   console.log();
-  console.log("  To get started, you need a Poke API key:");
+  console.log("  To get started, either:");
   console.log();
-  console.log("  1. Go to https://poke.com/kitchen/api-keys");
-  console.log("  2. Generate a new key");
-  console.log("  3. Paste it below");
+  console.log("  Option 1: Run 'npx poke login' (recommended)");
+  console.log("  Option 2: Paste an API key from https://poke.com/kitchen/api-keys");
   console.log();
 
-  const key = await ask("  API key: ");
+  const key = await ask("  API key (or press Enter if you ran poke login): ");
 
   if (!key) {
+    const pokeCreds = loadPokeCredentials();
+    if (pokeCreds) {
+      console.log();
+      console.log("  Found poke login credentials! Starting...");
+      console.log();
+      return pokeCreds;
+    }
     console.log();
-    console.log("  No key provided. Set it later:");
-    console.log("    export POKE_API_KEY=your_key_here");
+    console.log("  No credentials found. Run: npx poke login");
     console.log();
     process.exit(1);
   }

package/clients/Poke macOS Gate/Poke macOS Gate/GateService.swift

@@ -1,6 +1,5 @@
 import Foundation
 import Combine
-import ScreenCaptureKit
 
 @MainActor
 class GateService: ObservableObject {
@@ -28,29 +27,17 @@ class GateService: ObservableObject {
     }
 
     var hasAPIKey: Bool {
-        let key = loadAPIKey()
-        return key != nil && !key!.isEmpty
+        resolveToken() != nil
     }
 
     func autoStartIfNeeded() {
         guard !hasAutoStarted else { return }
         hasAutoStarted = true
-        requestScreenCapturePermission()
         if hasAPIKey {
             start()
         }
     }
 
-    private func requestScreenCapturePermission() {
-        Task {
-            do {
-                _ = try await SCShareableContent.current
-            } catch {
-                appendLog("Screen capture permission not granted yet.")
-            }
-        }
-    }
-
     func start() {
         guard hasAPIKey else {
             status = .error
@@ -123,7 +110,7 @@ class GateService: ObservableObject {
         proc.arguments = ["-c", "npx -y poke-gate --verbose"]
         proc.environment = ProcessInfo.processInfo.environment.merging(
             [
-                "POKE_API_KEY": loadAPIKey() ?? "",
+                "POKE_API_KEY": resolveToken() ?? "",
                 "PATH": fullPath,
             ],
             uniquingKeysWith: { _, new in new }
@@ -238,4 +225,80 @@ class GateService: ObservableObject {
         }
         objectWillChange.send()
     }
+
+    // MARK: - Poke Login Credentials
+
+    private var pokeCredentialsURL: URL {
+        let configDir: URL
+        if let xdg = ProcessInfo.processInfo.environment["XDG_CONFIG_HOME"] {
+            configDir = URL(fileURLWithPath: xdg)
+        } else {
+            configDir = FileManager.default.homeDirectoryForCurrentUser
+                .appendingPathComponent(".config")
+        }
+        return configDir
+            .appendingPathComponent("poke")
+            .appendingPathComponent("credentials.json")
+    }
+
+    var hasPokeLoginCredentials: Bool {
+        loadPokeLoginToken() != nil
+    }
+
+    func loadPokeLoginToken() -> String? {
+        guard let data = try? Data(contentsOf: pokeCredentialsURL),
+              let json = try? JSONSerialization.jsonObject(with: data) as? [String: Any],
+              let token = json["token"] as? String else {
+            return nil
+        }
+        return token
+    }
+
+    var authSource: AuthSource {
+        get {
+            let config = (try? Data(contentsOf: configURL))
+                .flatMap { try? JSONSerialization.jsonObject(with: $0) as? [String: Any] }
+            if let source = config?["authSource"] as? String, source == "pokeLogin" {
+                return .pokeLogin
+            }
+            if loadAPIKey() != nil {
+                return .apiKey
+            }
+            if hasPokeLoginCredentials {
+                return .pokeLogin
+            }
+            return .none
+        }
+        set {
+            let dir = configURL.deletingLastPathComponent()
+            try? FileManager.default.createDirectory(at: dir, withIntermediateDirectories: true)
+            var json: [String: Any] = [:]
+            if let data = try? Data(contentsOf: configURL),
+               let existing = try? JSONSerialization.jsonObject(with: data) as? [String: Any] {
+                json = existing
+            }
+            json["authSource"] = newValue == .pokeLogin ? "pokeLogin" : "apiKey"
+            if let data = try? JSONSerialization.data(withJSONObject: json, options: [.prettyPrinted]) {
+                try? data.write(to: configURL)
+            }
+            objectWillChange.send()
+        }
+    }
+
+    func resolveToken() -> String? {
+        switch authSource {
+        case .pokeLogin:
+            return loadPokeLoginToken()
+        case .apiKey:
+            return loadAPIKey()
+        case .none:
+            return loadAPIKey() ?? loadPokeLoginToken()
+        }
+    }
+
+    enum AuthSource {
+        case pokeLogin
+        case apiKey
+        case none
+    }
 }

package/clients/Poke macOS Gate/Poke macOS Gate/SettingsView.swift

@@ -3,6 +3,7 @@ import SwiftUI
 struct SettingsView: View {
     @ObservedObject var service: GateService
     @State private var apiKeyInput: String = ""
+    @State private var usePokeLogin: Bool = true
     @Environment(\.dismiss) private var dismiss
 
     var body: some View {
@@ -10,18 +11,64 @@ struct SettingsView: View {
             Text("Poke Gate Settings")
                 .font(.headline)
 
-            VStack(alignment: .leading, spacing: 8) {
-                Text("API Key")
-                    .font(.subheadline)
-                    .foregroundStyle(.secondary)
+            Picker("Authentication", selection: $usePokeLogin) {
+                Text("Use poke login").tag(true)
+                Text("Use API key").tag(false)
+            }
+            .pickerStyle(.segmented)
+
+            if usePokeLogin {
+                VStack(alignment: .leading, spacing: 8) {
+                    if service.hasPokeLoginCredentials {
+                        Label("poke login credentials found", systemImage: "checkmark.circle.fill")
+                            .foregroundStyle(.green)
+                            .font(.subheadline)
+                    } else {
+                        Label("No credentials found", systemImage: "xmark.circle")
+                            .foregroundStyle(.red)
+                            .font(.subheadline)
+
+                        Text("Run this in your terminal:")
+                            .font(.caption)
+                            .foregroundStyle(.secondary)
+
+                        HStack {
+                            Text("npx poke login")
+                                .font(.system(.caption, design: .monospaced))
+                                .padding(.horizontal, 8)
+                                .padding(.vertical, 4)
+                                .background(.quaternary)
+                                .cornerRadius(4)
 
-                SecureField("Paste your API key", text: $apiKeyInput)
-                    .textFieldStyle(.roundedBorder)
+                            Button {
+                                NSPasteboard.general.clearContents()
+                                NSPasteboard.general.setString("npx poke login", forType: .string)
+                            } label: {
+                                Image(systemName: "doc.on.doc")
+                            }
+                            .buttonStyle(.plain)
+                        }
 
-                Link("Get your key at poke.com/kitchen/api-keys",
-                     destination: URL(string: "https://poke.com/kitchen/api-keys")!)
-                    .font(.caption)
-                    .foregroundStyle(.blue)
+                        Text("Then come back here and click Save.")
+                            .font(.caption)
+                            .foregroundStyle(.secondary)
+                    }
+                }
+                .frame(maxWidth: .infinity, alignment: .leading)
+            } else {
+                VStack(alignment: .leading, spacing: 8) {
+                    Text("API Key")
+                        .font(.subheadline)
+                        .foregroundStyle(.secondary)
+
+                    SecureField("Paste your API key", text: $apiKeyInput)
+                        .textFieldStyle(.roundedBorder)
+
+                    Link("Get your key at poke.com/kitchen/api-keys",
+                         destination: URL(string: "https://poke.com/kitchen/api-keys")!)
+                        .font(.caption)
+                        .foregroundStyle(.blue)
+                }
             }
 
             HStack {
@@ -33,17 +80,24 @@ struct SettingsView: View {
                 Spacer()
 
                 Button("Save") {
-                    service.apiKey = apiKeyInput
+                    if usePokeLogin {
+                        service.authSource = .pokeLogin
+                    } else {
+                        service.apiKey = apiKeyInput
+                        service.authSource = .apiKey
+                    }
                     dismiss()
                     service.restart()
                 }
                 .keyboardShortcut(.defaultAction)
-                .disabled(apiKeyInput.isEmpty)
+                .disabled(!usePokeLogin && apiKeyInput.isEmpty)
+                .disabled(usePokeLogin && !service.hasPokeLoginCredentials)
             }
         }
         .padding(20)
-        .frame(width: 360)
+        .frame(width: 380)
         .onAppear {
+            usePokeLogin = service.authSource != .apiKey
             apiKeyInput = service.apiKey
         }
     }

package/clients/Poke macOS Gate/Poke macOS Gate.xcodeproj/project.pbxproj

@@ -264,7 +264,7 @@
 					"$(inherited)",
 					"@executable_path/../Frameworks",
 				);
-				MARKETING_VERSION = 0.0.4;
+				MARKETING_VERSION = 0.0.5;
 				PRODUCT_BUNDLE_IDENTIFIER = "dev.fka.Poke-macOS-Gate";
 				PRODUCT_NAME = "$(TARGET_NAME)";
 				REGISTER_APP_GROUPS = YES;
@@ -296,7 +296,7 @@
 					"$(inherited)",
 					"@executable_path/../Frameworks",
 				);
-				MARKETING_VERSION = 0.0.4;
+				MARKETING_VERSION = 0.0.5;
 				PRODUCT_BUNDLE_IDENTIFIER = "dev.fka.Poke-macOS-Gate";
 				PRODUCT_NAME = "$(TARGET_NAME)";
 				REGISTER_APP_GROUPS = YES;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "poke-gate",
-  "version": "0.0.5",
+  "version": "0.0.7",
   "description": "Expose your machine to your Poke AI assistant via MCP tunnel",
   "type": "module",
   "bin": {

package/src/app.js

@@ -1,5 +1,6 @@
 import { startMcpServer, enableLogging } from "./mcp-server.js";
 import { startTunnel } from "./tunnel.js";
+import { Poke } from "poke";
 import { readFileSync } from "node:fs";
 import { join } from "node:path";
 import { homedir } from "node:os";
@@ -17,6 +18,11 @@ function resolveToken() {
     if (cfg.apiKey) return cfg.apiKey;
   } catch {}
 
+  try {
+    const creds = JSON.parse(readFileSync(join(configDir, "poke", "credentials.json"), "utf-8"));
+    if (creds.token) return creds.token;
+  } catch {}
+
   return null;
 }
 
@@ -50,6 +56,7 @@ async function main() {
           case "connected":
             log(`Tunnel connected (${data.connectionId})`);
             log("Ready — your Poke agent can now access this machine.");
+            notifyPoke(data.connectionId);
             break;
           case "disconnected":
             log("Tunnel disconnected. Reconnecting...");
@@ -72,6 +79,20 @@ async function main() {
   }
 }
 
+async function notifyPoke(connectionId) {
+  try {
+    const poke = new Poke({ apiKey: API_KEY });
+    await poke.sendMessage(
+      `Poke macOS Gate is connected. Tunnel ID: ${connectionId}. ` +
+      `You now have access to this machine's terminal, files, and screen. ` +
+      `Use the available tools (run_command, read_file, write_file, list_directory, system_info, read_image, take_screenshot) to help the user.`
+    );
+    log("Notified Poke agent about connection.");
+  } catch (err) {
+    log(`Failed to notify Poke: ${err.message}`);
+  }
+}
+
 process.on("SIGINT", () => {
   log("Shutting down...");
   process.exit(0);