@rainfall-devkit/sdk 0.1.8 → 0.2.1

package/README.md

@@ -339,6 +339,57 @@ try {
 }
 ```
 
+## Daemon
+
+Run a local daemon with WebSocket (MCP) and OpenAI-compatible API endpoints:
+
+```bash
+# Start the daemon
+rainfall daemon start
+
+# Start with custom ports
+rainfall daemon start --port 8765 --openai-port 8787
+
+# Check status
+rainfall daemon status
+
+# Stop the daemon
+rainfall daemon stop
+```
+
+### Using the Daemon
+
+**WebSocket (MCP) Endpoint:** `ws://localhost:8765`
+
+**OpenAI-compatible API:** `http://localhost:8787/v1/chat/completions`
+
+```bash
+# Test the OpenAI-compatible endpoint
+curl http://localhost:8787/v1/chat/completions \
+  -H "Content-Type: application/json" \
+  -d '{
+    "model": "rainfall",
+    "messages": [{"role": "user", "content": "hello"}]
+  }'
+```
+
+### Programmatic Usage
+
+```typescript
+import { RainfallDaemon } from '@rainfall-devkit/sdk/daemon';
+
+const daemon = new RainfallDaemon({
+  port: 8765,        // WebSocket/MCP port
+  openaiPort: 8787,  // OpenAI-compatible API port
+});
+
+await daemon.start();
+// Daemon is running with all Rainfall tools loaded
+
+// Later...
+await daemon.stop();
+```
+
 ## MCP Server
 
 Use Rainfall with Claude, Cursor, and other MCP-compatible assistants:

package/dist/chunk-7MRE4ZVI.mjs

@@ -0,0 +1,662 @@
+// src/services/networked.ts
+var RainfallNetworkedExecutor = class {
+  rainfall;
+  options;
+  edgeNodeId;
+  jobCallbacks = /* @__PURE__ */ new Map();
+  resultPollingInterval;
+  constructor(rainfall, options = {}) {
+    this.rainfall = rainfall;
+    this.options = {
+      wsPort: 8765,
+      httpPort: 8787,
+      hostname: process.env.HOSTNAME || "local-daemon",
+      capabilities: {
+        localExec: true,
+        fileWatch: true,
+        passiveListen: true
+      },
+      ...options
+    };
+  }
+  /**
+   * Register this edge node with the Rainfall backend
+   */
+  async registerEdgeNode() {
+    const capabilities = this.buildCapabilitiesList();
+    try {
+      const result = await this.rainfall.executeTool("register-edge-node", {
+        hostname: this.options.hostname,
+        capabilities,
+        wsPort: this.options.wsPort,
+        httpPort: this.options.httpPort,
+        version: "0.1.0"
+      });
+      this.edgeNodeId = result.edgeNodeId;
+      console.log(`\u{1F310} Edge node registered with Rainfall as ${this.edgeNodeId}`);
+      return this.edgeNodeId;
+    } catch (error) {
+      this.edgeNodeId = `edge-${this.options.hostname}-${Date.now()}`;
+      console.log(`\u{1F310} Edge node running in local mode (ID: ${this.edgeNodeId})`);
+      return this.edgeNodeId;
+    }
+  }
+  /**
+   * Unregister this edge node on shutdown
+   */
+  async unregisterEdgeNode() {
+    if (!this.edgeNodeId) return;
+    try {
+      await this.rainfall.executeTool("unregister-edge-node", {
+        edgeNodeId: this.edgeNodeId
+      });
+      console.log(`\u{1F310} Edge node ${this.edgeNodeId} unregistered`);
+    } catch {
+    }
+    if (this.resultPollingInterval) {
+      clearInterval(this.resultPollingInterval);
+    }
+  }
+  /**
+   * Queue a tool execution for distributed processing
+   * Non-blocking - returns immediately with a job ID
+   */
+  async queueToolExecution(toolId, params, options = {}) {
+    const executionMode = options.executionMode || "any";
+    try {
+      const result = await this.rainfall.executeTool("queue-job", {
+        toolId,
+        params,
+        executionMode,
+        requesterEdgeNodeId: this.edgeNodeId
+      });
+      if (options.callback) {
+        this.jobCallbacks.set(result.jobId, options.callback);
+        this.startResultPolling();
+      }
+      return result.jobId;
+    } catch (error) {
+      if (executionMode === "local-only" || executionMode === "any") {
+        try {
+          const result = await this.rainfall.executeTool(toolId, params);
+          if (options.callback) {
+            options.callback(result);
+          }
+          return `local-${Date.now()}`;
+        } catch (execError) {
+          if (options.callback) {
+            options.callback(null, String(execError));
+          }
+          throw execError;
+        }
+      }
+      throw error;
+    }
+  }
+  /**
+   * Get status of a queued job
+   */
+  async getJobStatus(jobId) {
+    try {
+      const result = await this.rainfall.executeTool("get-job-status", {
+        jobId
+      });
+      return result.job;
+    } catch {
+      return null;
+    }
+  }
+  /**
+   * Subscribe to job results via polling (WebSocket fallback)
+   * In the future, this will use WebSocket push from ApresMoi
+   */
+  async subscribeToResults(callback) {
+    console.log("\u{1F4E1} Subscribed to job results via Rainfall (polling mode)");
+    this.onResultReceived = callback;
+  }
+  onResultReceived;
+  /**
+   * Start polling for job results (fallback until WebSocket push is ready)
+   */
+  startResultPolling() {
+    if (this.resultPollingInterval) return;
+    this.resultPollingInterval = setInterval(async () => {
+      for (const [jobId, callback] of this.jobCallbacks) {
+        try {
+          const job = await this.getJobStatus(jobId);
+          if (job?.status === "completed" || job?.status === "failed") {
+            callback(job.result, job.error);
+            this.jobCallbacks.delete(jobId);
+            if (this.onResultReceived) {
+              this.onResultReceived(jobId, job.result, job.error);
+            }
+          }
+        } catch {
+        }
+      }
+      if (this.jobCallbacks.size === 0 && this.resultPollingInterval) {
+        clearInterval(this.resultPollingInterval);
+        this.resultPollingInterval = void 0;
+      }
+    }, 2e3);
+  }
+  /**
+   * Claim a job for execution on this edge node
+   */
+  async claimJob() {
+    try {
+      const result = await this.rainfall.executeTool("claim-job", {
+        edgeNodeId: this.edgeNodeId,
+        capabilities: this.buildCapabilitiesList()
+      });
+      return result.job;
+    } catch {
+      return null;
+    }
+  }
+  /**
+   * Submit job result after execution
+   */
+  async submitJobResult(jobId, result, error) {
+    try {
+      await this.rainfall.executeTool("submit-job-result", {
+        jobId,
+        edgeNodeId: this.edgeNodeId,
+        result,
+        error
+      });
+    } catch {
+    }
+  }
+  /**
+   * Get this edge node's ID
+   */
+  getEdgeNodeId() {
+    return this.edgeNodeId;
+  }
+  /**
+   * Build capabilities list from options
+   */
+  buildCapabilitiesList() {
+    const caps = this.options.capabilities || {};
+    const list = [];
+    if (caps.localExec) list.push("local-exec");
+    if (caps.fileWatch) list.push("file-watch");
+    if (caps.passiveListen) list.push("passive-listen");
+    if (caps.browser) list.push("browser");
+    if (caps.custom) list.push(...caps.custom);
+    return list;
+  }
+};
+
+// src/services/context.ts
+var RainfallDaemonContext = class {
+  rainfall;
+  options;
+  localMemories = /* @__PURE__ */ new Map();
+  sessions = /* @__PURE__ */ new Map();
+  executionHistory = [];
+  currentSessionId;
+  constructor(rainfall, options = {}) {
+    this.rainfall = rainfall;
+    this.options = {
+      maxLocalMemories: 1e3,
+      maxMessageHistory: 100,
+      maxExecutionHistory: 500,
+      sessionTtl: 24 * 60 * 60 * 1e3,
+      // 24 hours
+      ...options
+    };
+  }
+  /**
+   * Initialize the context - load recent memories from cloud
+   */
+  async initialize() {
+    try {
+      const recentMemories = await this.rainfall.memory.recall({
+        query: "daemon:context",
+        topK: this.options.maxLocalMemories
+      });
+      for (const memory of recentMemories) {
+        this.localMemories.set(memory.id, {
+          id: memory.id,
+          content: memory.content,
+          keywords: memory.keywords || [],
+          timestamp: memory.timestamp,
+          source: memory.source,
+          metadata: memory.metadata
+        });
+      }
+      console.log(`\u{1F9E0} Loaded ${this.localMemories.size} memories into context`);
+    } catch (error) {
+      console.warn("\u26A0\uFE0F  Could not sync memories:", error instanceof Error ? error.message : error);
+    }
+  }
+  /**
+   * Create or get a session
+   */
+  getSession(sessionId) {
+    if (sessionId && this.sessions.has(sessionId)) {
+      const session = this.sessions.get(sessionId);
+      session.lastActivity = (/* @__PURE__ */ new Date()).toISOString();
+      return session;
+    }
+    const newSession = {
+      id: sessionId || `session-${Date.now()}`,
+      startedAt: (/* @__PURE__ */ new Date()).toISOString(),
+      lastActivity: (/* @__PURE__ */ new Date()).toISOString(),
+      variables: {},
+      messageHistory: []
+    };
+    this.sessions.set(newSession.id, newSession);
+    this.currentSessionId = newSession.id;
+    return newSession;
+  }
+  /**
+   * Get the current active session
+   */
+  getCurrentSession() {
+    if (this.currentSessionId) {
+      return this.sessions.get(this.currentSessionId);
+    }
+    return void 0;
+  }
+  /**
+   * Set the current active session
+   */
+  setCurrentSession(sessionId) {
+    if (this.sessions.has(sessionId)) {
+      this.currentSessionId = sessionId;
+    }
+  }
+  /**
+   * Add a message to the current session history
+   */
+  addMessage(role, content) {
+    const session = this.getCurrentSession();
+    if (!session) return;
+    session.messageHistory.push({
+      role,
+      content,
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    });
+    if (session.messageHistory.length > this.options.maxMessageHistory) {
+      session.messageHistory = session.messageHistory.slice(-this.options.maxMessageHistory);
+    }
+  }
+  /**
+   * Store a memory (local + cloud sync)
+   */
+  async storeMemory(content, options = {}) {
+    const id = `mem-${Date.now()}-${Math.random().toString(36).slice(2)}`;
+    const entry = {
+      id,
+      content,
+      keywords: options.keywords || [],
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      source: options.source || "daemon",
+      metadata: options.metadata
+    };
+    this.localMemories.set(id, entry);
+    try {
+      await this.rainfall.memory.create({
+        content,
+        keywords: [...options.keywords || [], "daemon:context"],
+        metadata: {
+          ...options.metadata,
+          daemonMemoryId: id,
+          source: options.source || "daemon"
+        }
+      });
+    } catch (error) {
+      console.warn("\u26A0\uFE0F  Could not sync memory to cloud:", error instanceof Error ? error.message : error);
+    }
+    this.trimLocalMemories();
+    return id;
+  }
+  /**
+   * Recall memories by query
+   */
+  async recallMemories(query, topK = 5) {
+    const localResults = Array.from(this.localMemories.values()).filter(
+      (m) => m.content.toLowerCase().includes(query.toLowerCase()) || m.keywords.some((k) => k.toLowerCase().includes(query.toLowerCase()))
+    ).slice(0, topK);
+    try {
+      const cloudResults = await this.rainfall.memory.recall({ query, topK });
+      const seen = new Set(localResults.map((r) => r.id));
+      for (const mem of cloudResults) {
+        if (!seen.has(mem.id)) {
+          localResults.push({
+            id: mem.id,
+            content: mem.content,
+            keywords: mem.keywords || [],
+            timestamp: mem.timestamp,
+            source: mem.source,
+            metadata: mem.metadata
+          });
+        }
+      }
+    } catch {
+    }
+    return localResults.slice(0, topK);
+  }
+  /**
+   * Set a session variable
+   */
+  setVariable(key, value) {
+    const session = this.getCurrentSession();
+    if (session) {
+      session.variables[key] = value;
+    }
+  }
+  /**
+   * Get a session variable
+   */
+  getVariable(key) {
+    const session = this.getCurrentSession();
+    return session?.variables[key];
+  }
+  /**
+   * Record a tool execution
+   */
+  recordExecution(toolId, params, result, options = { duration: 0 }) {
+    const record = {
+      id: `exec-${Date.now()}-${Math.random().toString(36).slice(2)}`,
+      toolId,
+      params,
+      result,
+      error: options.error,
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      duration: options.duration,
+      edgeNodeId: options.edgeNodeId
+    };
+    this.executionHistory.push(record);
+    if (this.executionHistory.length > this.options.maxExecutionHistory) {
+      this.executionHistory = this.executionHistory.slice(-this.options.maxExecutionHistory);
+    }
+  }
+  /**
+   * Get recent execution history
+   */
+  getExecutionHistory(limit = 10) {
+    return this.executionHistory.slice(-limit).reverse();
+  }
+  /**
+   * Get execution statistics
+   */
+  getExecutionStats() {
+    const stats = {
+      total: this.executionHistory.length,
+      successful: 0,
+      failed: 0,
+      averageDuration: 0,
+      byTool: {}
+    };
+    let totalDuration = 0;
+    for (const exec of this.executionHistory) {
+      if (exec.error) {
+        stats.failed++;
+      } else {
+        stats.successful++;
+      }
+      totalDuration += exec.duration;
+      stats.byTool[exec.toolId] = (stats.byTool[exec.toolId] || 0) + 1;
+    }
+    stats.averageDuration = stats.total > 0 ? totalDuration / stats.total : 0;
+    return stats;
+  }
+  /**
+   * Clear old sessions based on TTL
+   */
+  cleanupSessions() {
+    const now = Date.now();
+    const ttl = this.options.sessionTtl;
+    for (const [id, session] of this.sessions) {
+      const lastActivity = new Date(session.lastActivity).getTime();
+      if (now - lastActivity > ttl) {
+        this.sessions.delete(id);
+        if (this.currentSessionId === id) {
+          this.currentSessionId = void 0;
+        }
+      }
+    }
+  }
+  /**
+   * Get context summary for debugging
+   */
+  getStatus() {
+    return {
+      memoriesCached: this.localMemories.size,
+      activeSessions: this.sessions.size,
+      currentSession: this.currentSessionId,
+      executionHistorySize: this.executionHistory.length
+    };
+  }
+  trimLocalMemories() {
+    if (this.localMemories.size <= this.options.maxLocalMemories) return;
+    const entries = Array.from(this.localMemories.entries()).sort((a, b) => new Date(a[1].timestamp).getTime() - new Date(b[1].timestamp).getTime());
+    const toRemove = entries.slice(0, entries.length - this.options.maxLocalMemories);
+    for (const [id] of toRemove) {
+      this.localMemories.delete(id);
+    }
+  }
+};
+
+// src/services/listeners.ts
+var RainfallListenerRegistry = class {
+  rainfall;
+  context;
+  executor;
+  watchers = /* @__PURE__ */ new Map();
+  cronIntervals = /* @__PURE__ */ new Map();
+  eventHistory = [];
+  maxEventHistory = 100;
+  constructor(rainfall, context, executor) {
+    this.rainfall = rainfall;
+    this.context = context;
+    this.executor = executor;
+  }
+  /**
+   * Register a file watcher
+   * Note: Actual file watching requires fs.watch or chokidar
+   * This is the registry - actual watching is done by the daemon
+   */
+  async registerFileWatcher(config) {
+    console.log(`\u{1F441}\uFE0F  Registering file watcher: ${config.name} (${config.watchPath})`);
+    const existing = Array.from(this.watchers.keys());
+    if (existing.includes(config.id)) {
+      await this.unregisterFileWatcher(config.id);
+    }
+    this.watchers.set(config.id, {
+      stop: () => {
+        console.log(`\u{1F441}\uFE0F  Stopped file watcher: ${config.name}`);
+      }
+    });
+    await this.context.storeMemory(`File watcher registered: ${config.name}`, {
+      keywords: ["listener", "file-watcher", config.name],
+      metadata: { config }
+    });
+  }
+  /**
+   * Unregister a file watcher
+   */
+  async unregisterFileWatcher(id) {
+    const watcher = this.watchers.get(id);
+    if (watcher) {
+      watcher.stop();
+      this.watchers.delete(id);
+    }
+  }
+  /**
+   * Register a cron trigger
+   */
+  async registerCronTrigger(config) {
+    console.log(`\u23F0 Registering cron trigger: ${config.name} (${config.cron})`);
+    if (this.cronIntervals.has(config.id)) {
+      clearInterval(this.cronIntervals.get(config.id));
+      this.cronIntervals.delete(config.id);
+    }
+    const interval = this.parseCronToMs(config.cron);
+    if (interval) {
+      const intervalId = setInterval(async () => {
+        await this.handleCronTick(config);
+      }, interval);
+      this.cronIntervals.set(config.id, intervalId);
+    }
+    await this.context.storeMemory(`Cron trigger registered: ${config.name}`, {
+      keywords: ["listener", "cron", config.name],
+      metadata: { config }
+    });
+  }
+  /**
+   * Unregister a cron trigger
+   */
+  unregisterCronTrigger(id) {
+    const interval = this.cronIntervals.get(id);
+    if (interval) {
+      clearInterval(interval);
+      this.cronIntervals.delete(id);
+    }
+  }
+  /**
+   * Handle a file event
+   */
+  async handleFileEvent(watcherId, eventType, filePath) {
+    const event = {
+      id: `evt-${Date.now()}`,
+      type: "file",
+      source: watcherId,
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      data: { eventType, filePath }
+    };
+    this.recordEvent(event);
+    console.log(`\u{1F4C1} File event: ${eventType} ${filePath}`);
+  }
+  /**
+   * Handle a cron tick
+   */
+  async handleCronTick(config) {
+    const event = {
+      id: `evt-${Date.now()}`,
+      type: "cron",
+      source: config.id,
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      data: { cron: config.cron }
+    };
+    this.recordEvent(event);
+    console.log(`\u23F0 Cron tick: ${config.name}`);
+    for (const step of config.workflow) {
+      try {
+        await this.executor.queueToolExecution(step.toolId, {
+          ...step.params,
+          _event: event
+        });
+      } catch (error) {
+        console.error(`\u274C Workflow step failed: ${step.toolId}`, error);
+      }
+    }
+  }
+  /**
+   * Trigger a manual event (for testing or programmatic triggers)
+   */
+  async triggerManual(name, data = {}) {
+    const event = {
+      id: `evt-${Date.now()}`,
+      type: "manual",
+      source: name,
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      data
+    };
+    this.recordEvent(event);
+    console.log(`\u{1F446} Manual trigger: ${name}`);
+    await this.context.storeMemory(`Manual trigger fired: ${name}`, {
+      keywords: ["trigger", "manual", name],
+      metadata: { event }
+    });
+  }
+  /**
+   * Get recent events
+   */
+  getRecentEvents(limit = 10) {
+    return this.eventHistory.slice(-limit).reverse();
+  }
+  /**
+   * Get active listeners status
+   */
+  getStatus() {
+    return {
+      fileWatchers: this.watchers.size,
+      cronTriggers: this.cronIntervals.size,
+      recentEvents: this.eventHistory.length
+    };
+  }
+  /**
+   * Stop all listeners
+   */
+  async stopAll() {
+    for (const [id] of this.watchers) {
+      await this.unregisterFileWatcher(id);
+    }
+    for (const [id] of this.cronIntervals) {
+      this.unregisterCronTrigger(id);
+    }
+    console.log("\u{1F6D1} All listeners stopped");
+  }
+  recordEvent(event) {
+    this.eventHistory.push(event);
+    if (this.eventHistory.length > this.maxEventHistory) {
+      this.eventHistory = this.eventHistory.slice(-this.maxEventHistory);
+    }
+  }
+  /**
+   * Simple cron parser - converts basic cron expressions to milliseconds
+   * Supports: @hourly, @daily, @weekly, and simple intervals like every N minutes
+   */
+  parseCronToMs(cron) {
+    switch (cron) {
+      case "@hourly":
+        return 60 * 60 * 1e3;
+      case "@daily":
+        return 24 * 60 * 60 * 1e3;
+      case "@weekly":
+        return 7 * 24 * 60 * 60 * 1e3;
+      case "@minutely":
+        return 60 * 1e3;
+    }
+    const match = cron.match(/^\*\/(\d+)\s/);
+    if (match) {
+      const minutes = parseInt(match[1], 10);
+      if (minutes > 0 && minutes <= 60) {
+        return minutes * 60 * 1e3;
+      }
+    }
+    console.warn(`\u26A0\uFE0F  Unrecognized cron pattern "${cron}", using 1 minute interval`);
+    return 60 * 1e3;
+  }
+};
+function createFileWatcherWorkflow(name, watchPath, options) {
+  return {
+    id: `fw-${Date.now()}-${Math.random().toString(36).slice(2)}`,
+    name,
+    watchPath,
+    pattern: options.pattern,
+    events: options.events || ["create"],
+    workflow: options.workflow
+  };
+}
+function createCronWorkflow(name, cron, workflow) {
+  return {
+    id: `cron-${Date.now()}-${Math.random().toString(36).slice(2)}`,
+    name,
+    cron,
+    workflow
+  };
+}
+
+export {
+  RainfallNetworkedExecutor,
+  RainfallDaemonContext,
+  RainfallListenerRegistry,
+  createFileWatcherWorkflow,
+  createCronWorkflow
+};