@rainfall-devkit/sdk 0.2.1 → 0.2.3

package/dist/chunk-6FXRLPLR.mjs

@@ -0,0 +1,436 @@
+// src/security/edge-node.ts
+import sodium from "libsodium-wrappers-sumo";
+var EdgeNodeSecurity = class {
+  sodiumReady;
+  backendSecret;
+  keyPair;
+  constructor(options = {}) {
+    this.sodiumReady = sodium.ready;
+    this.backendSecret = options.backendSecret;
+    this.keyPair = options.keyPair;
+  }
+  /**
+   * Initialize libsodium
+   */
+  async initialize() {
+    await this.sodiumReady;
+  }
+  // ============================================================================
+  // JWT Token Management
+  // ============================================================================
+  /**
+   * Generate a JWT token for an edge node
+   * Note: In production, this is done by the backend. This is for testing.
+   */
+  generateJWT(edgeNodeId, subscriberId, expiresInDays = 30) {
+    if (!this.backendSecret) {
+      throw new Error("Backend secret not configured");
+    }
+    const now = Math.floor(Date.now() / 1e3);
+    const exp = now + expiresInDays * 24 * 60 * 60;
+    const jti = this.generateTokenId();
+    const payload = {
+      sub: edgeNodeId,
+      iss: "rainfall-backend",
+      iat: now,
+      exp,
+      jti,
+      scope: ["edge:heartbeat", "edge:claim", "edge:submit", "edge:queue"]
+    };
+    const header = { alg: "HS256", typ: "JWT" };
+    const encodedHeader = this.base64UrlEncode(JSON.stringify(header));
+    const encodedPayload = this.base64UrlEncode(JSON.stringify(payload));
+    const signature = this.hmacSha256(
+      `${encodedHeader}.${encodedPayload}`,
+      this.backendSecret
+    );
+    const encodedSignature = this.base64UrlEncode(signature);
+    return `${encodedHeader}.${encodedPayload}.${encodedSignature}`;
+  }
+  /**
+   * Validate a JWT token
+   */
+  validateJWT(token) {
+    const parts = token.split(".");
+    if (parts.length !== 3) {
+      throw new Error("Invalid JWT format");
+    }
+    const [encodedHeader, encodedPayload, encodedSignature] = parts;
+    if (this.backendSecret) {
+      const expectedSignature = this.hmacSha256(
+        `${encodedHeader}.${encodedPayload}`,
+        this.backendSecret
+      );
+      const expectedEncoded = this.base64UrlEncode(expectedSignature);
+      if (!this.timingSafeEqual(encodedSignature, expectedEncoded)) {
+        throw new Error("Invalid JWT signature");
+      }
+    }
+    const payload = JSON.parse(this.base64UrlDecode(encodedPayload));
+    const now = Math.floor(Date.now() / 1e3);
+    if (payload.exp < now) {
+      throw new Error("JWT token expired");
+    }
+    if (payload.iss !== "rainfall-backend") {
+      throw new Error("Invalid JWT issuer");
+    }
+    return {
+      edgeNodeId: payload.sub,
+      subscriberId: payload.sub,
+      // Same as edge node ID for now
+      scopes: payload.scope,
+      expiresAt: payload.exp
+    };
+  }
+  /**
+   * Extract bearer token from Authorization header
+   */
+  extractBearerToken(authHeader) {
+    if (!authHeader) return null;
+    const match = authHeader.match(/^Bearer\s+(.+)$/i);
+    return match ? match[1] : null;
+  }
+  // ============================================================================
+  // ACL Enforcement
+  // ============================================================================
+  /**
+   * Check if an edge node is allowed to perform an action on a job
+   * Rule: Edge nodes can only access jobs for their own subscriber
+   */
+  checkACL(check) {
+    if (check.subscriberId !== check.jobSubscriberId) {
+      return {
+        allowed: false,
+        reason: `Edge node ${check.edgeNodeId} cannot access jobs from subscriber ${check.jobSubscriberId}`
+      };
+    }
+    const allowedActions = ["heartbeat", "claim", "submit", "queue"];
+    if (!allowedActions.includes(check.action)) {
+      return {
+        allowed: false,
+        reason: `Unknown action: ${check.action}`
+      };
+    }
+    return { allowed: true };
+  }
+  /**
+   * Middleware-style ACL check for job operations
+   */
+  requireSameSubscriber(edgeNodeSubscriberId, jobSubscriberId, operation) {
+    const result = this.checkACL({
+      edgeNodeId: edgeNodeSubscriberId,
+      subscriberId: edgeNodeSubscriberId,
+      jobSubscriberId,
+      action: operation
+    });
+    if (!result.allowed) {
+      throw new Error(result.reason);
+    }
+  }
+  // ============================================================================
+  // Encryption (Libsodium)
+  // ============================================================================
+  /**
+   * Generate a new Ed25519 key pair for an edge node
+   */
+  async generateKeyPair() {
+    await this.sodiumReady;
+    const keyPair = sodium.crypto_box_keypair();
+    return {
+      publicKey: this.bytesToBase64(keyPair.publicKey),
+      privateKey: this.bytesToBase64(keyPair.privateKey)
+    };
+  }
+  /**
+   * Encrypt job parameters for a target edge node using its public key
+   */
+  async encryptForEdgeNode(plaintext, targetPublicKeyBase64) {
+    await this.sodiumReady;
+    if (!this.keyPair) {
+      throw new Error("Local key pair not configured");
+    }
+    const targetPublicKey = this.base64ToBytes(targetPublicKeyBase64);
+    const ephemeralKeyPair = sodium.crypto_box_keypair();
+    const nonce = sodium.randombytes_buf(sodium.crypto_box_NONCEBYTES);
+    const message = new TextEncoder().encode(plaintext);
+    const ciphertext = sodium.crypto_box_easy(
+      message,
+      nonce,
+      targetPublicKey,
+      ephemeralKeyPair.privateKey
+    );
+    return {
+      ciphertext: this.bytesToBase64(ciphertext),
+      nonce: this.bytesToBase64(nonce),
+      ephemeralPublicKey: this.bytesToBase64(ephemeralKeyPair.publicKey)
+    };
+  }
+  /**
+   * Decrypt job parameters received from the backend
+   */
+  async decryptFromBackend(encrypted) {
+    await this.sodiumReady;
+    if (!this.keyPair) {
+      throw new Error("Local key pair not configured");
+    }
+    const privateKey = this.base64ToBytes(this.keyPair.privateKey);
+    const ephemeralPublicKey = this.base64ToBytes(encrypted.ephemeralPublicKey);
+    const nonce = this.base64ToBytes(encrypted.nonce);
+    const ciphertext = this.base64ToBytes(encrypted.ciphertext);
+    const decrypted = sodium.crypto_box_open_easy(
+      ciphertext,
+      nonce,
+      ephemeralPublicKey,
+      privateKey
+    );
+    if (!decrypted) {
+      throw new Error("Decryption failed - invalid ciphertext or keys");
+    }
+    return new TextDecoder().decode(decrypted);
+  }
+  /**
+   * Encrypt job parameters for local storage (using secretbox)
+   */
+  async encryptLocal(plaintext, key) {
+    await this.sodiumReady;
+    const keyBytes = this.deriveKey(key);
+    const nonce = sodium.randombytes_buf(sodium.crypto_secretbox_NONCEBYTES);
+    const message = new TextEncoder().encode(plaintext);
+    const ciphertext = sodium.crypto_secretbox_easy(message, nonce, keyBytes);
+    return {
+      ciphertext: this.bytesToBase64(ciphertext),
+      nonce: this.bytesToBase64(nonce)
+    };
+  }
+  /**
+   * Decrypt locally stored job parameters
+   */
+  async decryptLocal(encrypted, key) {
+    await this.sodiumReady;
+    const keyBytes = this.deriveKey(key);
+    const nonce = this.base64ToBytes(encrypted.nonce);
+    const ciphertext = this.base64ToBytes(encrypted.ciphertext);
+    const decrypted = sodium.crypto_secretbox_open_easy(ciphertext, nonce, keyBytes);
+    if (!decrypted) {
+      throw new Error("Local decryption failed");
+    }
+    return new TextDecoder().decode(decrypted);
+  }
+  // ============================================================================
+  // Utility Methods
+  // ============================================================================
+  generateTokenId() {
+    return `${Date.now()}-${Math.random().toString(36).substring(2, 15)}`;
+  }
+  base64UrlEncode(str) {
+    return btoa(str).replace(/\+/g, "-").replace(/\//g, "_").replace(/=/g, "");
+  }
+  base64UrlDecode(str) {
+    const padding = "=".repeat((4 - str.length % 4) % 4);
+    const base64 = str.replace(/-/g, "+").replace(/_/g, "/") + padding;
+    return atob(base64);
+  }
+  hmacSha256(message, secret) {
+    const key = new TextEncoder().encode(secret);
+    const msg = new TextEncoder().encode(message);
+    const hash = sodium.crypto_auth(msg, key);
+    return this.bytesToBase64(hash);
+  }
+  timingSafeEqual(a, b) {
+    if (a.length !== b.length) return false;
+    let result = 0;
+    for (let i = 0; i < a.length; i++) {
+      result |= a.charCodeAt(i) ^ b.charCodeAt(i);
+    }
+    return result === 0;
+  }
+  bytesToBase64(bytes) {
+    const binString = Array.from(bytes, (b) => String.fromCharCode(b)).join("");
+    return btoa(binString);
+  }
+  base64ToBytes(base64) {
+    const binString = atob(base64);
+    return Uint8Array.from(binString, (m) => m.charCodeAt(0));
+  }
+  deriveKey(password) {
+    const passwordBytes = new TextEncoder().encode(password);
+    return sodium.crypto_generichash(32, passwordBytes, null);
+  }
+};
+async function createEdgeNodeSecurity(options = {}) {
+  const security = new EdgeNodeSecurity(options);
+  await security.initialize();
+  return security;
+}
+
+// src/cli/core/types.ts
+var ToolHandlerRegistry = class {
+  handlers = [];
+  register(handler) {
+    this.handlers.push(handler);
+  }
+  findHandler(toolId) {
+    return this.handlers.find((h) => {
+      if (typeof h.toolId === "string") {
+        return h.toolId === toolId;
+      }
+      return h.toolId.test(toolId);
+    });
+  }
+  getAllHandlers() {
+    return [...this.handlers];
+  }
+};
+var globalHandlerRegistry = new ToolHandlerRegistry();
+
+// src/cli/handlers/_registry.ts
+var imageGenerationHandler = {
+  toolId: /image-generation|generate-image/,
+  async display(context) {
+    const { detectImageData, displayImage } = await import("./display-KKJPO6UA.mjs");
+    const { result, flags } = context;
+    const imageInfo = detectImageData(result);
+    if (imageInfo.hasImage && !flags.raw) {
+      try {
+        if (imageInfo.imageData) {
+          await displayImage(imageInfo.imageData);
+          return true;
+        }
+      } catch (error) {
+        console.warn("Failed to display image:", error instanceof Error ? error.message : error);
+      }
+    }
+    return false;
+  }
+};
+var finvizQuotesHandler = {
+  toolId: "finviz-quotes",
+  async preflight(context) {
+    const { parseValue } = await import("./param-parser-PAKCNDBX.mjs");
+    const params = { ...context.params };
+    if (params.tickers && typeof params.tickers === "string") {
+      params.tickers = parseValue(params.tickers, { type: "array", items: { type: "string" } });
+    }
+    return { params };
+  },
+  async display(context) {
+    const { result, flags } = context;
+    if (flags.raw) {
+      return false;
+    }
+    const obj = result;
+    const quotes = obj?.quotes;
+    if (Array.isArray(quotes) && quotes.length > 0) {
+      const { formatAsTable } = await import("./display-KKJPO6UA.mjs");
+      const tableData = quotes.map((q) => {
+        const quote = q;
+        const data = quote.data || {};
+        return {
+          Ticker: quote.ticker || data.Ticker || "-",
+          Price: data.Price || data.Close || "-",
+          Change: data.Change || "-",
+          Volume: data.Volume || "-",
+          "Market Cap": data.MarketCap || "-"
+        };
+      });
+      console.log(formatAsTable(tableData));
+      const summary = obj?.summary;
+      if (summary && typeof summary === "string") {
+        console.log(`
+${summary}`);
+      }
+      return true;
+    }
+    return false;
+  }
+};
+var csvQueryHandler = {
+  toolId: /query-csv|csv-query/,
+  async display(context) {
+    const { result, flags } = context;
+    if (flags.raw) {
+      return false;
+    }
+    if (Array.isArray(result) && result.length > 0) {
+      const { formatAsTable } = await import("./display-KKJPO6UA.mjs");
+      console.log(formatAsTable(result));
+      return true;
+    }
+    return false;
+  }
+};
+var webSearchHandler = {
+  toolId: /web-search|exa-web-search|perplexity/,
+  async display(context) {
+    const { result, flags } = context;
+    if (flags.raw) {
+      return false;
+    }
+    const obj = result;
+    if (obj.results && typeof obj.results === "string") {
+      console.log(obj.results);
+      return true;
+    }
+    if (obj.answer || obj.summary) {
+      console.log(obj.answer || obj.summary);
+      if (obj.sources && Array.isArray(obj.sources)) {
+        console.log("\n--- Sources ---");
+        obj.sources.forEach((source, i) => {
+          if (typeof source === "string") {
+            console.log(`  ${i + 1}. ${source}`);
+          } else if (source && typeof source === "object") {
+            const s = source;
+            console.log(`  ${i + 1}. ${s.title || s.url || JSON.stringify(source)}`);
+          }
+        });
+      }
+      return true;
+    }
+    return false;
+  }
+};
+var memoryRecallHandler = {
+  toolId: /memory-recall|recall/,
+  async display(context) {
+    const { result, flags } = context;
+    if (flags.raw) {
+      return false;
+    }
+    if (Array.isArray(result)) {
+      if (result.length === 0) {
+        console.log("No memories found.");
+        return true;
+      }
+      console.log(`Found ${result.length} memory(s):
+`);
+      result.forEach((mem, i) => {
+        const memory = mem;
+        console.log(`\u2500`.repeat(60));
+        console.log(`  ${i + 1}. ${memory.content || memory.text || JSON.stringify(memory).slice(0, 100)}`);
+        if (memory.similarity) {
+          console.log(`     Similarity: ${(Number(memory.similarity) * 100).toFixed(1)}%`);
+        }
+        if (memory.keywords && Array.isArray(memory.keywords)) {
+          console.log(`     Keywords: ${memory.keywords.join(", ")}`);
+        }
+        console.log();
+      });
+      return true;
+    }
+    return false;
+  }
+};
+function registerBuiltInHandlers(registry = globalHandlerRegistry) {
+  registry.register(imageGenerationHandler);
+  registry.register(finvizQuotesHandler);
+  registry.register(csvQueryHandler);
+  registry.register(webSearchHandler);
+  registry.register(memoryRecallHandler);
+}
+registerBuiltInHandlers();
+
+export {
+  EdgeNodeSecurity,
+  createEdgeNodeSecurity,
+  globalHandlerRegistry
+};