wao 0.9.0 → 0.9.2

package/esm/armem-base.js

@@ -0,0 +1,172 @@
+import _Arweave from "arweave"
+const Arweave = _Arweave.default ?? _Arweave
+
+import { last } from "ramda"
+import { buildTags, tags } from "./utils.js"
+
+export default class ArMemBase {
+  constructor({ MU_URL, CU_URL, SU_URL, GATEWAY_URL, scheduler, cache } = {}) {
+    this.__type__ = "mem"
+    this.isInit = false
+    this.keyInit = false
+    this.keys = {}
+    this.arweave = Arweave.init()
+    this.arweave.transactions.getTransactionAnchor = () => this.getAnchor()
+    this.arweave.transactions.getPrice = () => 0
+    this.scheduler = scheduler
+    this.SU_URL = SU_URL
+  }
+  getAnchor() {
+    return this.blocks.length === 0
+      ? "Do_Uc2Sju_ffp6Ev0AnLVdPtot15rvMjP-a9VVaA5fM"
+      : last(this.blockmap[last(this.blocks)].txs)
+  }
+  async get(key, field) {
+    await this.init()
+    if (!field) return this[key]
+    return this[key]?.[field]
+  }
+  async set(val, key, field) {
+    await this.init()
+    if (!field) {
+      this[key] = val
+      if (this.db) await this.db.put(`${key}`, this[key])
+    } else {
+      this[key] ??= {}
+      this[key][field] = val
+      if (this.db) await this.db.put(`${key}.${field}`, this[key][field])
+    }
+  }
+  initSync() {
+    this.addrmap = {}
+    this.txs = {}
+    this.jwks = {}
+    this.blocks = []
+    this.blockmap = {}
+    this.env = {}
+    this.modules = {
+      aos2_0_1: "Do_Uc2Sju_ffp6Ev0AnLVdPtot15rvMjP-a9VVaA5fM",
+      aos1: "cNlipBptaF9JeFAf4wUmpi43EojNanIBos3EfNrEOWo",
+      sqlite: "ghSkge2sIUD_F00ym5sEimC63BDBuBrq4b5OcwxOjiw",
+    }
+    this.modmap = {}
+    this.msgs = {}
+    this.wasms = {
+      "Do_Uc2Sju_ffp6Ev0AnLVdPtot15rvMjP-a9VVaA5fM": {
+        file: "aos2_0_1",
+        format: "wasm64-unknown-emscripten-draft_2024_02_15",
+      },
+      cNlipBptaF9JeFAf4wUmpi43EojNanIBos3EfNrEOWo: {
+        file: "aos_1",
+        format: "wasm64-unknown-emscripten-draft_2024_02_15",
+      },
+      ghSkge2sIUD_F00ym5sEimC63BDBuBrq4b5OcwxOjiw: {
+        file: "sqlite",
+        format: "wasm64-unknown-emscripten-draft_2024_02_15",
+      },
+    }
+    let txs = []
+    for (const k in this.modules) {
+      const key = this.modules[k]
+      txs.push(key)
+      this.txs[key] = {
+        id: key,
+        block: 0,
+        tags: buildTags(null, {
+          "Data-Protocol": "ao",
+          Variant: "ao.TN.1",
+          Type: "Module",
+          "Module-Format": "wasm64-unknown-emscripten-draft_2024_02_15",
+          "Input-Encoding": "JSON-V1",
+          "Output-Encoding": "JSON-V1",
+          "Memory-Limit": "1-gb",
+          "Compute-Limit": "9000000000000",
+          Extension: "WeaveDrive",
+        }),
+      }
+    }
+    if (this.scheduler && this.SU_URL) {
+      const key = this.scheduler
+      txs.push(key)
+      this.addrmap[this.scheduler] = { address: this.scheduler }
+      this.txs[key] = {
+        id: key,
+        block: 0,
+        owner: this.scheduler,
+        tags: buildTags(null, {
+          "Data-Protocol": "ao",
+          Variant: "ao.TN.1",
+          Type: "Scheduler-Location",
+          Url: this.SU_URL,
+          "Time-To-Live": 1000 * 60 * 60 * 24 * 365 * 10,
+        }),
+      }
+    }
+    this.blockmap["0"] = {
+      txs,
+      timestamp: Date.now(),
+      height: 0,
+      previous: "",
+      id: "0",
+    }
+    this.blocks.push("0")
+    this.height = 1
+  }
+  async putAll(key) {
+    for (let k in this[key]) await this.set(this[key][k], key, k)
+  }
+  async init() {
+    if (this.isInit) return
+    this.isInit = true
+    if (this.db) {
+      for (const v of ["height", "blocks"]) this[v] = await this.get(v)
+      for (const v of [
+        "txs",
+        "jwks",
+        "env",
+        "modules",
+        "wasms",
+        "addrmap",
+        "blockmap",
+        "modmap",
+        "msgs",
+      ]) {
+        this[v] ??= {}
+        const items = await this.db.getKeys({ start: v, end: v + "a" })
+        for (const v2 of items || []) {
+          const key = v2.split(".")[0]
+          const field = v2.split(".")[1]
+          if (key === v) this[v][field] = await this.db.get(v2)
+        }
+      }
+    } else {
+      for (const v of ["height", "blocks"]) await this.set(this[v], v)
+      for (const v of ["modules", "wasms", "addrmap", "blockmap"]) {
+        await this.putAll(v)
+      }
+    }
+  }
+  async getWasm(module) {
+    let mod = module ?? this.modules.aos2_0_1
+    if (!mod) throw Error("module not found")
+    let format = null
+    let _wasm = await this.wasms[mod]
+    let wasm = _wasm?.data
+    if (!wasm) {
+      if (_wasm?.file) {
+        wasm = await this._getWasm(this.wasms[mod].file)
+        format = _wasm.format
+      } else {
+        const tx = await this.get("txs", mod)
+        if (tx) {
+          wasm = Buffer.from(tx.data, "base64")
+          format = tags(tx.tags)["Module-Format"]
+        }
+      }
+    } else {
+      format = _wasm.format
+    }
+    format ??= "wasm64-unknown-emscripten-draft_2024_02_15"
+    return { format, mod, wasm }
+  }
+}

package/esm/armem-web.js

@@ -1,174 +1,14 @@
-import _Arweave from "arweave"
-import { last } from "ramda"
-import { tags, buildTags, dirname } from "./utils.js"
-const Arweave = _Arweave.default ?? _Arweave
-//import { readFileSync } from "fs"
-//import { resolve } from "path"
 import wdb from "./lua/weavedb-lite.js"
-export default class ArMem {
-  constructor({ MU_URL, CU_URL, SU_URL, GATEWAY_URL, scheduler, cache } = {}) {
-    this.__type__ = "mem"
-    this.isInit = false
-    //if (cache) this.db = open({ path: cache, compression: true })
-    this.arweave = Arweave.init()
-    this.arweave.transactions.getTransactionAnchor = () => this.getAnchor()
-    this.arweave.transactions.getPrice = () => 0
-    this.scheduler = scheduler
-    this.SU_URL = SU_URL
+import Base from "./armem-base.js"
+import db from "./lfdb.js"
+
+export default class ArMem extends Base {
+  constructor(args = {}) {
+    super(args)
+    this.db = db(this)
     this.initSync()
   }
-  async putAll(key) {
-    for (let k in this[key]) await this.set(this[key][k], key, k)
-  }
-  async init() {
-    if (this.isInit) return
-    this.isInit = true
-    if (this.db) {
-      for (const v of ["height", "blocks"]) this[v] = await this.get(v)
-      for (const v of [
-        "txs",
-        "jwks",
-        "env",
-        "modules",
-        "wasms",
-        "addrmap",
-        "blockmap",
-        "modmap",
-        "msgs",
-      ]) {
-        this[v] ??= {}
-        const items = await this.db.getKeys({ start: v, end: v + "a" })
-        for (const v2 of items) {
-          const key = v2.split(".")[0]
-          const field = v2.split(".")[1]
-          if (key === v) this[v][field] = await this.db.get(v2)
-        }
-      }
-    } else {
-      for (const v of ["height", "blocks"]) await this.set(this[v], v)
-      for (const v of ["modules", "wasms", "addrmap", "blockmap"]) {
-        await this.putAll(v)
-      }
-    }
-  }
-  async getWasm(module) {
-    let mod = module ?? this.modules.aos2_0_1
-    if (!mod) throw Error("module not found")
-    const __dirname = await dirname()
-    let format = null
-    let _wasm = await this.wasms[mod]
-    let wasm = _wasm?.data
-    if (!wasm) {
-      if (_wasm?.file) {
-        wasm = Buffer.from(wdb, "base64")
-        format = _wasm.format
-      } else {
-        const tx = await this.get("txs", mod)
-        if (tx) {
-          wasm = Buffer.from(tx.data, "base64")
-          format = tags(tx.tags)["Module-Format"]
-        }
-      }
-    } else {
-      format = _wasm.format
-    }
-    format ??= "wasm64-unknown-emscripten-draft_2024_02_15"
-    return { format, mod, wasm }
-  }
-  initSync() {
-    this.addrmap = {}
-    this.txs = {}
-    this.jwks = {}
-    this.blocks = []
-    this.blockmap = {}
-    this.env = {}
-    this.modules = {
-      aos2_0_1: "Do_Uc2Sju_ffp6Ev0AnLVdPtot15rvMjP-a9VVaA5fM",
-      aos1: "cNlipBptaF9JeFAf4wUmpi43EojNanIBos3EfNrEOWo",
-      sqlite: "ghSkge2sIUD_F00ym5sEimC63BDBuBrq4b5OcwxOjiw",
-    }
-    this.modmap = {}
-    this.msgs = {}
-    this.wasms = {
-      "Do_Uc2Sju_ffp6Ev0AnLVdPtot15rvMjP-a9VVaA5fM": {
-        file: "aos2_0_1",
-        format: "wasm64-unknown-emscripten-draft_2024_02_15",
-      },
-      cNlipBptaF9JeFAf4wUmpi43EojNanIBos3EfNrEOWo: {
-        file: "aos_1",
-        format: "wasm64-unknown-emscripten-draft_2024_02_15",
-      },
-      ghSkge2sIUD_F00ym5sEimC63BDBuBrq4b5OcwxOjiw: {
-        file: "sqlite",
-        format: "wasm64-unknown-emscripten-draft_2024_02_15",
-      },
-    }
-    let txs = []
-    for (const k in this.modules) {
-      const key = this.modules[k]
-      txs.push(key)
-      this.txs[key] = {
-        id: key,
-        block: 0,
-        tags: buildTags(null, {
-          "Data-Protocol": "ao",
-          Variant: "ao.TN.1",
-          Type: "Module",
-          "Module-Format": "wasm64-unknown-emscripten-draft_2024_02_15",
-          "Input-Encoding": "JSON-V1",
-          "Output-Encoding": "JSON-V1",
-          "Memory-Limit": "1-gb",
-          "Compute-Limit": "9000000000000",
-          Extension: "WeaveDrive",
-        }),
-      }
-    }
-    if (this.scheduler && this.SU_URL) {
-      const key = this.scheduler
-      txs.push(key)
-      this.addrmap[this.scheduler] = { address: this.scheduler }
-      this.txs[key] = {
-        id: key,
-        block: 0,
-        owner: this.scheduler,
-        tags: buildTags(null, {
-          "Data-Protocol": "ao",
-          Variant: "ao.TN.1",
-          Type: "Scheduler-Location",
-          Url: this.SU_URL,
-          "Time-To-Live": 1000 * 60 * 60 * 24 * 365 * 10,
-        }),
-      }
-    }
-    this.blockmap["0"] = {
-      txs,
-      timestamp: Date.now(),
-      height: 0,
-      previous: "",
-      id: "0",
-    }
-    this.blocks.push("0")
-    this.height = 1
-  }
-  getAnchor() {
-    return this.blocks.length === 0
-      ? "Do_Uc2Sju_ffp6Ev0AnLVdPtot15rvMjP-a9VVaA5fM"
-      : last(this.blockmap[last(this.blocks)].txs)
-  }
-  async get(key, field) {
-    await this.init()
-    if (!field) return this[key]
-    return this[key]?.[field]
-  }
-  async set(val, key, field) {
-    await this.init()
-    if (!field) {
-      this[key] = val
-      if (this.db) await this.db.put(`${key}`, this[key])
-    } else {
-      this[key] ??= {}
-      this[key][field] = val
-      if (this.db) await this.db.put(`${key}.${field}`, this[key][field])
-    }
+  async _getWasm(file) {
+    return Buffer.from(wdb, "base64")
   }
 }

package/esm/armem.js

@@ -4,173 +4,17 @@ import { tags, buildTags, dirname } from "./utils.js"
 import { open } from "lmdb"
 import { readFileSync } from "fs"
 import { resolve } from "path"
+import Base from "./armem-base.js"
 
-export default class ArMem {
-  constructor({ MU_URL, CU_URL, SU_URL, GATEWAY_URL, scheduler, cache } = {}) {
-    this.__type__ = "mem"
-    this.isInit = false
+export default class ArMem extends Base {
+  constructor(args = {}) {
+    const { cache } = args
+    super(args)
     if (cache) this.db = open({ path: cache, compression: true })
-    this.arweave = Arweave.init()
-    this.arweave.transactions.getTransactionAnchor = () => this.getAnchor()
-    this.arweave.transactions.getPrice = () => 0
-    this.scheduler = scheduler
-    this.SU_URL = SU_URL
     this.initSync()
   }
-  async putAll(key) {
-    for (let k in this[key]) await this.set(this[key][k], key, k)
-  }
-  async init() {
-    if (this.isInit) return
-    this.isInit = true
-    if (this.db) {
-      for (const v of ["height", "blocks"]) this[v] = await this.get(v)
-      for (const v of [
-        "txs",
-        "jwks",
-        "env",
-        "modules",
-        "wasms",
-        "addrmap",
-        "blockmap",
-        "modmap",
-        "msgs",
-      ]) {
-        this[v] ??= {}
-        const items = await this.db.getKeys({ start: v, end: v + "a" })
-        for (const v2 of items) {
-          const key = v2.split(".")[0]
-          const field = v2.split(".")[1]
-          if (key === v) this[v][field] = await this.db.get(v2)
-        }
-      }
-    } else {
-      for (const v of ["height", "blocks"]) await this.set(this[v], v)
-      for (const v of ["modules", "wasms", "addrmap", "blockmap"]) {
-        await this.putAll(v)
-      }
-    }
-  }
-  async getWasm(module) {
-    let mod = module ?? this.modules.aos2_0_1
-    if (!mod) throw Error("module not found")
+  async _getWasm(file) {
     const __dirname = await dirname()
-    let format = null
-    let _wasm = await this.wasms[mod]
-    let wasm = _wasm?.data
-    if (!wasm) {
-      if (_wasm?.file) {
-        wasm = readFileSync(
-          resolve(__dirname, `lua/${this.wasms[mod].file}.wasm`),
-        )
-        format = _wasm.format
-      } else {
-        const tx = await this.get("txs", mod)
-        if (tx) {
-          wasm = Buffer.from(tx.data, "base64")
-          format = tags(tx.tags)["Module-Format"]
-        }
-      }
-    } else {
-      format = _wasm.format
-    }
-    format ??= "wasm64-unknown-emscripten-draft_2024_02_15"
-    return { format, mod, wasm }
-  }
-  initSync() {
-    this.addrmap = {}
-    this.txs = {}
-    this.jwks = {}
-    this.blocks = []
-    this.blockmap = {}
-    this.env = {}
-    this.modules = {
-      aos2_0_1: "Do_Uc2Sju_ffp6Ev0AnLVdPtot15rvMjP-a9VVaA5fM",
-      aos1: "cNlipBptaF9JeFAf4wUmpi43EojNanIBos3EfNrEOWo",
-      sqlite: "ghSkge2sIUD_F00ym5sEimC63BDBuBrq4b5OcwxOjiw",
-    }
-    this.modmap = {}
-    this.msgs = {}
-    this.wasms = {
-      "Do_Uc2Sju_ffp6Ev0AnLVdPtot15rvMjP-a9VVaA5fM": {
-        file: "aos2_0_1",
-        format: "wasm64-unknown-emscripten-draft_2024_02_15",
-      },
-      cNlipBptaF9JeFAf4wUmpi43EojNanIBos3EfNrEOWo: {
-        file: "aos_1",
-        format: "wasm64-unknown-emscripten-draft_2024_02_15",
-      },
-      ghSkge2sIUD_F00ym5sEimC63BDBuBrq4b5OcwxOjiw: {
-        file: "sqlite",
-        format: "wasm64-unknown-emscripten-draft_2024_02_15",
-      },
-    }
-    let txs = []
-    for (const k in this.modules) {
-      const key = this.modules[k]
-      txs.push(key)
-      this.txs[key] = {
-        id: key,
-        block: 0,
-        tags: buildTags(null, {
-          "Data-Protocol": "ao",
-          Variant: "ao.TN.1",
-          Type: "Module",
-          "Module-Format": "wasm64-unknown-emscripten-draft_2024_02_15",
-          "Input-Encoding": "JSON-V1",
-          "Output-Encoding": "JSON-V1",
-          "Memory-Limit": "1-gb",
-          "Compute-Limit": "9000000000000",
-          Extension: "WeaveDrive",
-        }),
-      }
-    }
-    if (this.scheduler && this.SU_URL) {
-      const key = this.scheduler
-      txs.push(key)
-      this.addrmap[this.scheduler] = { address: this.scheduler }
-      this.txs[key] = {
-        id: key,
-        block: 0,
-        owner: this.scheduler,
-        tags: buildTags(null, {
-          "Data-Protocol": "ao",
-          Variant: "ao.TN.1",
-          Type: "Scheduler-Location",
-          Url: this.SU_URL,
-          "Time-To-Live": 1000 * 60 * 60 * 24 * 365 * 10,
-        }),
-      }
-    }
-    this.blockmap["0"] = {
-      txs,
-      timestamp: Date.now(),
-      height: 0,
-      previous: "",
-      id: "0",
-    }
-    this.blocks.push("0")
-    this.height = 1
-  }
-  getAnchor() {
-    return this.blocks.length === 0
-      ? "Do_Uc2Sju_ffp6Ev0AnLVdPtot15rvMjP-a9VVaA5fM"
-      : last(this.blockmap[last(this.blocks)].txs)
-  }
-  async get(key, field) {
-    await this.init()
-    if (!field) return this[key]
-    return this[key]?.[field]
-  }
-  async set(val, key, field) {
-    await this.init()
-    if (!field) {
-      this[key] = val
-      if (this.db) await this.db.put(`${key}`, this[key])
-    } else {
-      this[key] ??= {}
-      this[key][field] = val
-      if (this.db) await this.db.put(`${key}.${field}`, this[key][field])
-    }
+    return readFileSync(resolve(__dirname, `lua/${file}.wasm`))
   }
 }

package/esm/bao.js

@@ -0,0 +1,125 @@
+import MAO from "./ao.js"
+import { createDataItemSigner } from "@permaweb/aoconnect"
+import { srcs, buildTags } from "./utils.js"
+import { mergeLeft, is, map } from "ramda"
+
+let log = `
+local json = require("json")
+ao = ao or {}
+
+function ao.log(...)
+    local args = {...}
+    local txt = json.encode(args)
+    if type(ao.outbox.Output) == 'string' then
+      ao.outbox.Output = {ao.outbox.Output}
+    end
+    table.insert(ao.outbox.Output, txt)
+    ao.outbox.Logs = ao.outbox.Logs or {}
+    table.insert(ao.outbox.Logs, txt)
+end
+
+function ao.result(result)
+    if ao.outbox.Error or result.Error then
+      return { Error = result.Error or ao.outbox.Error, Output = ao.outbox.Logs }
+    end
+    return {
+        Output = ao.outbox.Logs or result.Output or ao.outbox.Output,
+        Messages = ao.outbox.Messages,
+        Spawns = ao.outbox.Spawns,
+        Assignments = ao.outbox.Assignments,
+    }
+end
+`
+
+const renderLogs = logs => {
+  if (is(Array, logs) && logs.length > 0) {
+    for (const v of logs) {
+      let l = v
+      try {
+        const p = JSON.parse(v)
+        if (JSON.stringify(p) === v) l = p
+        console.log(...l)
+      } catch (e) {
+        console.log(l)
+      }
+    }
+  }
+}
+
+class AO extends MAO {
+  constructor(opt = {}) {
+    super({ ...opt, in_memory: true })
+    this.in_memory = true
+    const {
+      modules,
+      results,
+      assign,
+      result,
+      message,
+      spawn,
+      dryrun,
+      monitor,
+      unmonitor,
+      mem,
+    } = opt.connect(opt.mem, {
+      extensions: opt.extensions,
+      cache: opt.cache,
+      reset: opt.reset,
+    })
+    this.assign = assign
+    this.result = async (...opt) => {
+      const res = await result(...opt)
+      renderLogs(res.Output)
+      return res
+    }
+
+    this.results = results
+    this.message = message
+    this.spawn = async (...opt) => {
+      const res = await spawn(...opt)
+      await this.load({ data: log, pid: res })
+      return res
+    }
+    this.dryrun = async (...opt) => {
+      const res = await dryrun(...opt)
+      renderLogs(res.Output)
+      return res
+    }
+    this.monitor = monitor
+    this.unmonitor = unmonitor
+    this.mem = mem
+    this.ar = new opt.AR({ mem })
+  }
+
+  async init(jwk) {
+    if (!jwk && this.acc[0]) jwk = this.acc[0].jwk
+    await this.ar.init(jwk)
+    return this
+  }
+
+  async postModule({ data, tags = {}, jwk }) {
+    let err = null
+    ;({ jwk, err } = await this.ar.checkWallet({ jwk }))
+    if (err) return { err }
+    const t = mergeLeft(tags, {
+      "Data-Protocol": "ao",
+      Variant: "ao.TN.1",
+      Type: "Module",
+      "Module-Format": "wasm64-unknown-emscripten-draft_2024_02_15",
+      "Input-Encoding": "JSON-V1",
+      "Output-Encoding": "JSON-V1",
+      "Memory-Limit": "1-gb",
+      "Compute-Limit": "9000000000000",
+    })
+    const signer = createDataItemSigner(jwk)
+    const { id, owner, item } = await this.ar.dataitem({
+      tags: t,
+      data,
+      signer,
+    })
+    await this.ar.postItems(item, jwk)
+    return { id }
+  }
+}
+
+export default AO