@positronic/cloudflare 0.0.3 → 0.0.5

package/dist/src/schedule-do.js

@@ -0,0 +1,705 @@
+function _array_like_to_array(arr, len) {
+    if (len == null || len > arr.length) len = arr.length;
+    for(var i = 0, arr2 = new Array(len); i < len; i++)arr2[i] = arr[i];
+    return arr2;
+}
+function _array_without_holes(arr) {
+    if (Array.isArray(arr)) return _array_like_to_array(arr);
+}
+function _assert_this_initialized(self) {
+    if (self === void 0) {
+        throw new ReferenceError("this hasn't been initialised - super() hasn't been called");
+    }
+    return self;
+}
+function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) {
+    try {
+        var info = gen[key](arg);
+        var value = info.value;
+    } catch (error) {
+        reject(error);
+        return;
+    }
+    if (info.done) {
+        resolve(value);
+    } else {
+        Promise.resolve(value).then(_next, _throw);
+    }
+}
+function _async_to_generator(fn) {
+    return function() {
+        var self = this, args = arguments;
+        return new Promise(function(resolve, reject) {
+            var gen = fn.apply(self, args);
+            function _next(value) {
+                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value);
+            }
+            function _throw(err) {
+                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err);
+            }
+            _next(undefined);
+        });
+    };
+}
+function _call_super(_this, derived, args) {
+    derived = _get_prototype_of(derived);
+    return _possible_constructor_return(_this, _is_native_reflect_construct() ? Reflect.construct(derived, args || [], _get_prototype_of(_this).constructor) : derived.apply(_this, args));
+}
+function _class_call_check(instance, Constructor) {
+    if (!(instance instanceof Constructor)) {
+        throw new TypeError("Cannot call a class as a function");
+    }
+}
+function _defineProperties(target, props) {
+    for(var i = 0; i < props.length; i++){
+        var descriptor = props[i];
+        descriptor.enumerable = descriptor.enumerable || false;
+        descriptor.configurable = true;
+        if ("value" in descriptor) descriptor.writable = true;
+        Object.defineProperty(target, descriptor.key, descriptor);
+    }
+}
+function _create_class(Constructor, protoProps, staticProps) {
+    if (protoProps) _defineProperties(Constructor.prototype, protoProps);
+    if (staticProps) _defineProperties(Constructor, staticProps);
+    return Constructor;
+}
+function _define_property(obj, key, value) {
+    if (key in obj) {
+        Object.defineProperty(obj, key, {
+            value: value,
+            enumerable: true,
+            configurable: true,
+            writable: true
+        });
+    } else {
+        obj[key] = value;
+    }
+    return obj;
+}
+function _get_prototype_of(o) {
+    _get_prototype_of = Object.setPrototypeOf ? Object.getPrototypeOf : function getPrototypeOf(o) {
+        return o.__proto__ || Object.getPrototypeOf(o);
+    };
+    return _get_prototype_of(o);
+}
+function _inherits(subClass, superClass) {
+    if (typeof superClass !== "function" && superClass !== null) {
+        throw new TypeError("Super expression must either be null or a function");
+    }
+    subClass.prototype = Object.create(superClass && superClass.prototype, {
+        constructor: {
+            value: subClass,
+            writable: true,
+            configurable: true
+        }
+    });
+    if (superClass) _set_prototype_of(subClass, superClass);
+}
+function _instanceof(left, right) {
+    if (right != null && typeof Symbol !== "undefined" && right[Symbol.hasInstance]) {
+        return !!right[Symbol.hasInstance](left);
+    } else {
+        return left instanceof right;
+    }
+}
+function _iterable_to_array(iter) {
+    if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter);
+}
+function _non_iterable_spread() {
+    throw new TypeError("Invalid attempt to spread non-iterable instance.\\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method.");
+}
+function _possible_constructor_return(self, call) {
+    if (call && (_type_of(call) === "object" || typeof call === "function")) {
+        return call;
+    }
+    return _assert_this_initialized(self);
+}
+function _set_prototype_of(o, p) {
+    _set_prototype_of = Object.setPrototypeOf || function setPrototypeOf(o, p) {
+        o.__proto__ = p;
+        return o;
+    };
+    return _set_prototype_of(o, p);
+}
+function _to_consumable_array(arr) {
+    return _array_without_holes(arr) || _iterable_to_array(arr) || _unsupported_iterable_to_array(arr) || _non_iterable_spread();
+}
+function _type_of(obj) {
+    "@swc/helpers - typeof";
+    return obj && typeof Symbol !== "undefined" && obj.constructor === Symbol ? "symbol" : typeof obj;
+}
+function _unsupported_iterable_to_array(o, minLen) {
+    if (!o) return;
+    if (typeof o === "string") return _array_like_to_array(o, minLen);
+    var n = Object.prototype.toString.call(o).slice(8, -1);
+    if (n === "Object" && o.constructor) n = o.constructor.name;
+    if (n === "Map" || n === "Set") return Array.from(n);
+    if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _array_like_to_array(o, minLen);
+}
+function _is_native_reflect_construct() {
+    try {
+        var result = !Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function() {}));
+    } catch (_) {}
+    return (_is_native_reflect_construct = function() {
+        return !!result;
+    })();
+}
+function _ts_generator(thisArg, body) {
+    var f, y, t, _ = {
+        label: 0,
+        sent: function() {
+            if (t[0] & 1) throw t[1];
+            return t[1];
+        },
+        trys: [],
+        ops: []
+    }, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() {
+        return this;
+    }), g;
+    function verb(n) {
+        return function(v) {
+            return step([
+                n,
+                v
+            ]);
+        };
+    }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while(g && (g = 0, op[0] && (_ = 0)), _)try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [
+                op[0] & 2,
+                t.value
+            ];
+            switch(op[0]){
+                case 0:
+                case 1:
+                    t = op;
+                    break;
+                case 4:
+                    _.label++;
+                    return {
+                        value: op[1],
+                        done: false
+                    };
+                case 5:
+                    _.label++;
+                    y = op[1];
+                    op = [
+                        0
+                    ];
+                    continue;
+                case 7:
+                    op = _.ops.pop();
+                    _.trys.pop();
+                    continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) {
+                        _ = 0;
+                        continue;
+                    }
+                    if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) {
+                        _.label = op[1];
+                        break;
+                    }
+                    if (op[0] === 6 && _.label < t[1]) {
+                        _.label = t[1];
+                        t = op;
+                        break;
+                    }
+                    if (t && _.label < t[2]) {
+                        _.label = t[2];
+                        _.ops.push(op);
+                        break;
+                    }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop();
+                    continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) {
+            op = [
+                6,
+                e
+            ];
+            y = 0;
+        } finally{
+            f = t = 0;
+        }
+        if (op[0] & 5) throw op[1];
+        return {
+            value: op[0] ? op[1] : void 0,
+            done: true
+        };
+    }
+}
+import { DurableObject } from 'cloudflare:workers';
+import { v4 as uuidv4 } from 'uuid';
+import { parseCronExpression } from 'cron-schedule';
+import { BRAIN_EVENTS } from '@positronic/core';
+var ALARM_INTERVAL = 60 * 1000;
+export var ScheduleDO = /*#__PURE__*/ function(DurableObject) {
+    "use strict";
+    _inherits(ScheduleDO, DurableObject);
+    function ScheduleDO(state, env) {
+        _class_call_check(this, ScheduleDO);
+        var _this;
+        _this = _call_super(this, ScheduleDO, [
+            state,
+            env
+        ]), _define_property(_this, "storage", void 0);
+        _this.storage = state.storage.sql;
+        // Initialize database schema
+        _this.storage.exec("\n      CREATE TABLE IF NOT EXISTS schedules (\n        id TEXT PRIMARY KEY,\n        brain_name TEXT NOT NULL,\n        cron_expression TEXT NOT NULL,\n        enabled INTEGER NOT NULL DEFAULT 1,\n        created_at INTEGER NOT NULL,\n        next_run_at INTEGER\n      );\n\n      CREATE INDEX IF NOT EXISTS idx_schedules_brain\n      ON schedules(brain_name);\n\n      CREATE INDEX IF NOT EXISTS idx_schedules_enabled\n      ON schedules(enabled);\n\n      CREATE TABLE IF NOT EXISTS scheduled_runs (\n        id INTEGER PRIMARY KEY AUTOINCREMENT,\n        schedule_id TEXT NOT NULL,\n        brain_run_id TEXT UNIQUE,\n        status TEXT NOT NULL CHECK(status IN ('triggered', 'failed', 'complete')),\n        ran_at INTEGER NOT NULL,\n        completed_at INTEGER,\n        error TEXT,\n        FOREIGN KEY (schedule_id) REFERENCES schedules(id) ON DELETE CASCADE\n      );\n\n      CREATE INDEX IF NOT EXISTS idx_runs_schedule\n      ON scheduled_runs(schedule_id, ran_at DESC);\n    ");
+        return _this;
+    }
+    _create_class(ScheduleDO, [
+        {
+            key: "createSchedule",
+            value: function createSchedule(brainName, cronExpression) {
+                return _async_to_generator(function() {
+                    var id, createdAt, alarm, cron, nextRunAt;
+                    return _ts_generator(this, function(_state) {
+                        switch(_state.label){
+                            case 0:
+                                id = uuidv4();
+                                createdAt = Date.now();
+                                if (!(this.env.IS_TEST !== 'true')) return [
+                                    3,
+                                    3
+                                ];
+                                return [
+                                    4,
+                                    this.ctx.storage.getAlarm()
+                                ];
+                            case 1:
+                                alarm = _state.sent();
+                                if (!!alarm) return [
+                                    3,
+                                    3
+                                ];
+                                return [
+                                    4,
+                                    this.ctx.storage.setAlarm(Date.now() + ALARM_INTERVAL)
+                                ];
+                            case 2:
+                                _state.sent();
+                                _state.label = 3;
+                            case 3:
+                                // Note: Cron expression is validated at the API level before calling this method
+                                // Calculate next run time
+                                cron = parseCronExpression(cronExpression);
+                                nextRunAt = this.calculateNextRunTime(cron, createdAt);
+                                this.storage.exec("INSERT INTO schedules (id, brain_name, cron_expression, enabled, created_at, next_run_at)\n       VALUES (?, ?, ?, 1, ?, ?)", id, brainName, cronExpression, createdAt, nextRunAt);
+                                return [
+                                    2,
+                                    {
+                                        id: id,
+                                        brainName: brainName,
+                                        cronExpression: cronExpression,
+                                        enabled: true,
+                                        createdAt: createdAt,
+                                        nextRunAt: nextRunAt
+                                    }
+                                ];
+                        }
+                    });
+                }).call(this);
+            }
+        },
+        {
+            key: "getSchedule",
+            value: function getSchedule(scheduleId) {
+                return _async_to_generator(function() {
+                    var results, result;
+                    return _ts_generator(this, function(_state) {
+                        results = this.storage.exec("SELECT id, brain_name, cron_expression, enabled, created_at, next_run_at\n         FROM schedules WHERE id = ?", scheduleId).toArray();
+                        if (results.length === 0) {
+                            return [
+                                2,
+                                null
+                            ];
+                        }
+                        result = results[0];
+                        return [
+                            2,
+                            {
+                                id: result.id,
+                                brainName: result.brain_name,
+                                cronExpression: result.cron_expression,
+                                enabled: result.enabled === 1,
+                                createdAt: result.created_at,
+                                nextRunAt: result.next_run_at
+                            }
+                        ];
+                    });
+                }).call(this);
+            }
+        },
+        {
+            key: "deleteSchedule",
+            value: function deleteSchedule(scheduleId) {
+                return _async_to_generator(function() {
+                    var existing;
+                    return _ts_generator(this, function(_state) {
+                        switch(_state.label){
+                            case 0:
+                                return [
+                                    4,
+                                    this.getSchedule(scheduleId)
+                                ];
+                            case 1:
+                                existing = _state.sent();
+                                if (!existing) {
+                                    return [
+                                        2,
+                                        false
+                                    ];
+                                }
+                                this.storage.exec("DELETE FROM schedules WHERE id = ?", scheduleId);
+                                return [
+                                    2,
+                                    true
+                                ];
+                        }
+                    });
+                }).call(this);
+            }
+        },
+        {
+            key: "listSchedules",
+            value: function listSchedules() {
+                return _async_to_generator(function() {
+                    var alarm, schedules;
+                    return _ts_generator(this, function(_state) {
+                        switch(_state.label){
+                            case 0:
+                                if (!(this.env.NODE_ENV === 'development')) return [
+                                    3,
+                                    4
+                                ];
+                                console.log('[ScheduleDO] Checking alarm');
+                                return [
+                                    4,
+                                    this.ctx.storage.getAlarm()
+                                ];
+                            case 1:
+                                alarm = _state.sent();
+                                if (!alarm) return [
+                                    3,
+                                    4
+                                ];
+                                console.log('[ScheduleDO] Deleting alarm');
+                                return [
+                                    4,
+                                    this.ctx.storage.deleteAlarm()
+                                ];
+                            case 2:
+                                _state.sent();
+                                console.log('[ScheduleDO] Running alarm handler');
+                                return [
+                                    4,
+                                    this.alarm()
+                                ];
+                            case 3:
+                                _state.sent();
+                                _state.label = 4;
+                            case 4:
+                                schedules = this.storage.exec("SELECT id, brain_name, cron_expression, enabled, created_at, next_run_at\n         FROM schedules\n         ORDER BY created_at DESC").toArray().map(function(row) {
+                                    return {
+                                        id: row.id,
+                                        brainName: row.brain_name,
+                                        cronExpression: row.cron_expression,
+                                        enabled: row.enabled === 1,
+                                        createdAt: row.created_at,
+                                        nextRunAt: row.next_run_at
+                                    };
+                                });
+                                return [
+                                    2,
+                                    {
+                                        schedules: schedules,
+                                        count: schedules.length
+                                    }
+                                ];
+                        }
+                    });
+                }).call(this);
+            }
+        },
+        {
+            key: "getAllRuns",
+            value: function getAllRuns(scheduleId) {
+                var limit = arguments.length > 1 && arguments[1] !== void 0 ? arguments[1] : 100;
+                return _async_to_generator(function() {
+                    var _this_storage, _this_storage1, query, params, runs, countQuery, countParams, countResult, count;
+                    return _ts_generator(this, function(_state) {
+                        query = "\n      SELECT id, schedule_id, brain_run_id, status, ran_at, completed_at, error\n      FROM scheduled_runs\n    ";
+                        params = [];
+                        if (scheduleId) {
+                            query += " WHERE schedule_id = ?";
+                            params.push(scheduleId);
+                        }
+                        query += " ORDER BY ran_at DESC LIMIT ?";
+                        params.push(limit);
+                        runs = (_this_storage = this.storage).exec.apply(_this_storage, [
+                            query
+                        ].concat(_to_consumable_array(params))).toArray().map(function(row) {
+                            return {
+                                id: row.id,
+                                scheduleId: row.schedule_id,
+                                brainRunId: row.brain_run_id,
+                                status: row.status,
+                                ranAt: row.ran_at,
+                                completedAt: row.completed_at,
+                                error: row.error
+                            };
+                        });
+                        // Get total count
+                        countQuery = "SELECT COUNT(*) as count FROM scheduled_runs";
+                        countParams = [];
+                        if (scheduleId) {
+                            countQuery += " WHERE schedule_id = ?";
+                            countParams.push(scheduleId);
+                        }
+                        countResult = (_this_storage1 = this.storage).exec.apply(_this_storage1, [
+                            countQuery
+                        ].concat(_to_consumable_array(countParams))).one();
+                        count = (countResult === null || countResult === void 0 ? void 0 : countResult.count) || 0;
+                        return [
+                            2,
+                            {
+                                runs: runs,
+                                count: count
+                            }
+                        ];
+                    });
+                }).call(this);
+            }
+        },
+        {
+            key: "alarm",
+            value: // Handle the alarm trigger - runs every minute in a perpetual cycle
+            function alarm() {
+                return _async_to_generator(function() {
+                    var now, dueSchedules, _iteratorNormalCompletion, _didIteratorError, _iteratorError, _iterator, _step, schedule, scheduleId, brainName, cronExpression, brainRunId, error, errorMessage, cron, nextRunAt, err;
+                    return _ts_generator(this, function(_state) {
+                        switch(_state.label){
+                            case 0:
+                                _state.trys.push([
+                                    0,
+                                    ,
+                                    12,
+                                    15
+                                ]);
+                                // This alarm runs every minute to check for schedules that need to be executed.
+                                // Since cron expressions have minute-level granularity at most (e.g., * * * * *),
+                                // checking every minute ensures we never miss a scheduled run.
+                                // Get all enabled schedules that are due
+                                now = Date.now();
+                                dueSchedules = this.storage.exec("SELECT id, brain_name, cron_expression\n           FROM schedules\n           WHERE enabled = 1 AND next_run_at <= ?", now).toArray();
+                                _iteratorNormalCompletion = true, _didIteratorError = false, _iteratorError = undefined;
+                                _state.label = 1;
+                            case 1:
+                                _state.trys.push([
+                                    1,
+                                    9,
+                                    10,
+                                    11
+                                ]);
+                                _iterator = dueSchedules[Symbol.iterator]();
+                                _state.label = 2;
+                            case 2:
+                                if (!!(_iteratorNormalCompletion = (_step = _iterator.next()).done)) return [
+                                    3,
+                                    8
+                                ];
+                                schedule = _step.value;
+                                scheduleId = schedule.id;
+                                brainName = schedule.brain_name;
+                                cronExpression = schedule.cron_expression;
+                                _state.label = 3;
+                            case 3:
+                                _state.trys.push([
+                                    3,
+                                    5,
+                                    ,
+                                    6
+                                ]);
+                                return [
+                                    4,
+                                    this.triggerBrainRun(brainName)
+                                ];
+                            case 4:
+                                brainRunId = _state.sent();
+                                // Record successful run
+                                this.storage.exec("INSERT INTO scheduled_runs (schedule_id, brain_run_id, status, ran_at)\n             VALUES (?, ?, 'triggered', ?)", scheduleId, brainRunId, now);
+                                return [
+                                    3,
+                                    6
+                                ];
+                            case 5:
+                                error = _state.sent();
+                                // Record failed run
+                                errorMessage = _instanceof(error, Error) ? error.message : 'Unknown error';
+                                this.storage.exec("INSERT INTO scheduled_runs (schedule_id, status, ran_at, error)\n               VALUES (?, 'failed', ?, ?)", scheduleId, now, errorMessage);
+                                console.error("[ScheduleDO] Failed to trigger brain ".concat(brainName, ":"), error);
+                                return [
+                                    3,
+                                    6
+                                ];
+                            case 6:
+                                // Calculate and update next run time
+                                cron = parseCronExpression(cronExpression);
+                                nextRunAt = this.calculateNextRunTime(cron, now);
+                                this.storage.exec("UPDATE schedules SET next_run_at = ? WHERE id = ?", nextRunAt, scheduleId);
+                                _state.label = 7;
+                            case 7:
+                                _iteratorNormalCompletion = true;
+                                return [
+                                    3,
+                                    2
+                                ];
+                            case 8:
+                                return [
+                                    3,
+                                    11
+                                ];
+                            case 9:
+                                err = _state.sent();
+                                _didIteratorError = true;
+                                _iteratorError = err;
+                                return [
+                                    3,
+                                    11
+                                ];
+                            case 10:
+                                try {
+                                    if (!_iteratorNormalCompletion && _iterator.return != null) {
+                                        _iterator.return();
+                                    }
+                                } finally{
+                                    if (_didIteratorError) {
+                                        throw _iteratorError;
+                                    }
+                                }
+                                return [
+                                    7
+                                ];
+                            case 11:
+                                return [
+                                    3,
+                                    15
+                                ];
+                            case 12:
+                                if (!(this.env.IS_TEST !== 'true')) return [
+                                    3,
+                                    14
+                                ];
+                                return [
+                                    4,
+                                    this.ctx.storage.setAlarm(Date.now() + ALARM_INTERVAL)
+                                ];
+                            case 13:
+                                _state.sent();
+                                _state.label = 14;
+                            case 14:
+                                return [
+                                    7
+                                ];
+                            case 15:
+                                return [
+                                    2
+                                ];
+                        }
+                    });
+                }).call(this);
+            }
+        },
+        {
+            key: "triggerBrainRun",
+            value: function triggerBrainRun(brainName) {
+                return _async_to_generator(function() {
+                    var brainRunId, namespace, doId, stub;
+                    return _ts_generator(this, function(_state) {
+                        switch(_state.label){
+                            case 0:
+                                brainRunId = uuidv4();
+                                namespace = this.env.BRAIN_RUNNER_DO;
+                                doId = namespace.idFromName(brainRunId);
+                                stub = namespace.get(doId);
+                                console.log("[ScheduleDO] Triggering brain run ".concat(brainName, " with id ").concat(brainRunId));
+                                return [
+                                    4,
+                                    stub.start(brainName, brainRunId)
+                                ];
+                            case 1:
+                                _state.sent();
+                                return [
+                                    2,
+                                    brainRunId
+                                ];
+                        }
+                    });
+                }).call(this);
+            }
+        },
+        {
+            key: "handleBrainEvent",
+            value: // Called by ScheduleAdapter when brain events occur
+            function handleBrainEvent(event) {
+                return _async_to_generator(function() {
+                    var result, scheduledRun, completedAt, status, error;
+                    return _ts_generator(this, function(_state) {
+                        // We only care about completion events for scheduled runs
+                        if (event.type !== BRAIN_EVENTS.COMPLETE && event.type !== BRAIN_EVENTS.ERROR) {
+                            return [
+                                2
+                            ];
+                        }
+                        // Check if this brain run was triggered by a schedule
+                        result = this.storage.exec("SELECT id FROM scheduled_runs WHERE brain_run_id = ?", event.brainRunId).toArray();
+                        if (result.length === 0) {
+                            // This brain run wasn't triggered by a schedule, ignore it
+                            return [
+                                2
+                            ];
+                        }
+                        scheduledRun = result[0];
+                        completedAt = Date.now();
+                        status = event.type === BRAIN_EVENTS.COMPLETE ? 'complete' : 'failed';
+                        error = event.type === BRAIN_EVENTS.ERROR ? event.error ? JSON.stringify(event.error) : 'Unknown error' : null;
+                        // Update the scheduled run record
+                        this.storage.exec("UPDATE scheduled_runs\n       SET status = ?, completed_at = ?, error = ?\n       WHERE brain_run_id = ?", status, completedAt, error, event.brainRunId);
+                        return [
+                            2
+                        ];
+                    });
+                }).call(this);
+            }
+        },
+        {
+            key: "isValidCronExpression",
+            value: function isValidCronExpression(expression) {
+                try {
+                    // Try to parse the expression - if it throws, it's invalid
+                    parseCronExpression(expression);
+                    return true;
+                } catch (error) {
+                    return false;
+                }
+            }
+        },
+        {
+            key: "calculateNextRunTime",
+            value: function calculateNextRunTime(cron, afterTime) {
+                var nextDate = cron.getNextDate(new Date(afterTime));
+                return nextDate.getTime();
+            }
+        }
+    ]);
+    return ScheduleDO;
+}(DurableObject);