@rainfall-devkit/sdk 0.2.0 → 0.2.1

package/dist/index.js

@@ -1,7 +1,9 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -15,12 +17,21 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
   AuthenticationError: () => AuthenticationError,
+  EdgeNodeSecurity: () => EdgeNodeSecurity,
   NetworkError: () => NetworkError,
   NotFoundError: () => NotFoundError,
   Rainfall: () => Rainfall,
@@ -30,13 +41,16 @@ __export(index_exports, {
   RainfallListenerRegistry: () => RainfallListenerRegistry,
   RainfallNetworkedExecutor: () => RainfallNetworkedExecutor,
   RateLimitError: () => RateLimitError,
+  SecureEdgeClient: () => SecureEdgeClient,
   ServerError: () => ServerError,
   TimeoutError: () => TimeoutError,
   ToolNotFoundError: () => ToolNotFoundError,
   VERSION: () => VERSION,
   ValidationError: () => ValidationError,
   createCronWorkflow: () => createCronWorkflow,
-  createFileWatcherWorkflow: () => createFileWatcherWorkflow
+  createEdgeNodeSecurity: () => createEdgeNodeSecurity,
+  createFileWatcherWorkflow: () => createFileWatcherWorkflow,
+  createSecureEdgeClient: () => createSecureEdgeClient
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -1550,11 +1564,482 @@ function createCronWorkflow(name, cron, workflow) {
   };
 }
 
+// src/security/edge-node.ts
+var sodium = __toESM(require("libsodium-wrappers"));
+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/security/edge-client.ts
+var import_fs = require("fs");
+var import_path = require("path");
+var SecureEdgeClient = class {
+  client;
+  security;
+  edgeNodeId;
+  edgeNodeSecret;
+  keysPath;
+  jwtPayload;
+  keyPair;
+  constructor(config) {
+    this.client = config.client;
+    this.edgeNodeId = config.edgeNodeId;
+    this.edgeNodeSecret = config.edgeNodeSecret;
+    this.keysPath = config.keysPath;
+    this.security = new EdgeNodeSecurity({
+      backendSecret: config.backendSecret
+    });
+  }
+  /**
+   * Initialize the secure client
+   */
+  async initialize() {
+    await this.security.initialize();
+    await this.loadKeyPair();
+    this.jwtPayload = this.security.validateJWT(this.edgeNodeSecret);
+    if (this.jwtPayload.edgeNodeId !== this.edgeNodeId) {
+      throw new Error("JWT edge node ID mismatch");
+    }
+  }
+  /**
+   * Load key pair from disk
+   */
+  async loadKeyPair() {
+    const publicKeyPath = (0, import_path.join)(this.keysPath, "edge-node.pub");
+    const privateKeyPath = (0, import_path.join)(this.keysPath, "edge-node.key");
+    if (!(0, import_fs.existsSync)(publicKeyPath) || !(0, import_fs.existsSync)(privateKeyPath)) {
+      throw new Error("Key pair not found. Run: rainfall edge generate-keys");
+    }
+    this.keyPair = {
+      publicKey: (0, import_fs.readFileSync)(publicKeyPath, "utf-8"),
+      privateKey: (0, import_fs.readFileSync)(privateKeyPath, "utf-8")
+    };
+  }
+  /**
+   * Get public key for sharing with backend
+   */
+  getPublicKey() {
+    if (!this.keyPair) {
+      throw new Error("Key pair not loaded");
+    }
+    return this.keyPair.publicKey;
+  }
+  /**
+   * Send heartbeat with authentication
+   */
+  async heartbeat() {
+    this.requireAuth();
+    return this.client.request("/edge/heartbeat", {
+      method: "POST",
+      body: {
+        edgeNodeId: this.edgeNodeId,
+        timestamp: Date.now()
+      },
+      headers: {
+        "Authorization": `Bearer ${this.edgeNodeSecret}`
+      }
+    });
+  }
+  /**
+   * Claim a job from the queue
+   */
+  async claimJob() {
+    this.requireAuth();
+    const job = await this.client.request("/edge/claim-job", {
+      method: "POST",
+      body: {
+        edgeNodeId: this.edgeNodeId,
+        subscriberId: this.jwtPayload.subscriberId
+      },
+      headers: {
+        "Authorization": `Bearer ${this.edgeNodeSecret}`
+      }
+    });
+    if (job && job.encrypted && job.params) {
+      const decrypted = await this.decryptJobParams(job.params);
+      return { ...job, params: decrypted };
+    }
+    return job;
+  }
+  /**
+   * Submit job result
+   */
+  async submitJobResult(result) {
+    this.requireAuth();
+    let encryptedOutput;
+    if (result.output) {
+      encryptedOutput = await this.encryptJobResult(result.output);
+    }
+    await this.client.request("/edge/submit-job-result", {
+      method: "POST",
+      body: {
+        edgeNodeId: this.edgeNodeId,
+        subscriberId: this.jwtPayload.subscriberId,
+        result: {
+          ...result,
+          output: encryptedOutput,
+          encrypted: !!encryptedOutput
+        }
+      },
+      headers: {
+        "Authorization": `Bearer ${this.edgeNodeSecret}`
+      }
+    });
+  }
+  /**
+   * Queue a job for processing
+   */
+  async queueJob(type, params, targetPublicKey) {
+    this.requireAuth();
+    let encryptedParams;
+    let encrypted = false;
+    if (targetPublicKey) {
+      encryptedParams = await this.encryptJobParamsForTarget(
+        JSON.stringify(params),
+        targetPublicKey
+      );
+      encrypted = true;
+    }
+    return this.client.request("/edge/queue-job", {
+      method: "POST",
+      body: {
+        edgeNodeId: this.edgeNodeId,
+        subscriberId: this.jwtPayload.subscriberId,
+        job: {
+          type,
+          params: encryptedParams || JSON.stringify(params),
+          encrypted
+        }
+      },
+      headers: {
+        "Authorization": `Bearer ${this.edgeNodeSecret}`
+      }
+    });
+  }
+  /**
+   * Decrypt job params received from backend
+   */
+  async decryptJobParams(encryptedParams) {
+    const encrypted = JSON.parse(encryptedParams);
+    return this.security.decryptFromBackend(encrypted);
+  }
+  /**
+   * Encrypt job result for sending to backend
+   */
+  async encryptJobResult(output) {
+    const encrypted = await this.security.encryptLocal(output, this.keyPair.privateKey);
+    return JSON.stringify(encrypted);
+  }
+  /**
+   * Encrypt job params for a specific target edge node
+   */
+  async encryptJobParamsForTarget(params, targetPublicKey) {
+    const encrypted = await this.security.encryptForEdgeNode(params, targetPublicKey);
+    return JSON.stringify(encrypted);
+  }
+  /**
+   * Check if client is authenticated
+   */
+  requireAuth() {
+    if (!this.jwtPayload) {
+      throw new Error("Client not authenticated. Call initialize() first.");
+    }
+    const now = Math.floor(Date.now() / 1e3);
+    if (this.jwtPayload.expiresAt < now) {
+      throw new Error("JWT token expired. Re-register edge node.");
+    }
+  }
+  /**
+   * Get current authentication status
+   */
+  getAuthStatus() {
+    if (!this.jwtPayload) {
+      return { authenticated: false };
+    }
+    const now = Math.floor(Date.now() / 1e3);
+    return {
+      authenticated: this.jwtPayload.expiresAt > now,
+      edgeNodeId: this.jwtPayload.edgeNodeId,
+      subscriberId: this.jwtPayload.subscriberId,
+      expiresAt: this.jwtPayload.expiresAt,
+      scopes: this.jwtPayload.scopes
+    };
+  }
+};
+async function createSecureEdgeClient(client, options) {
+  const secureClient = new SecureEdgeClient({
+    client,
+    ...options
+  });
+  await secureClient.initialize();
+  return secureClient;
+}
+
 // src/index.ts
 var VERSION = "0.1.0";
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AuthenticationError,
+  EdgeNodeSecurity,
   NetworkError,
   NotFoundError,
   Rainfall,
@@ -1564,11 +2049,14 @@ var VERSION = "0.1.0";
   RainfallListenerRegistry,
   RainfallNetworkedExecutor,
   RateLimitError,
+  SecureEdgeClient,
   ServerError,
   TimeoutError,
   ToolNotFoundError,
   VERSION,
   ValidationError,
   createCronWorkflow,
-  createFileWatcherWorkflow
+  createEdgeNodeSecurity,
+  createFileWatcherWorkflow,
+  createSecureEdgeClient
 });