@sentio/sdk 2.60.0-rc.4 → 2.60.0-rc.6

package/src/iota/builtin/index.ts

@@ -0,0 +1,6 @@
+/* Autogenerated file. Do not edit manually. */
+/* tslint:disable */
+/* eslint-disable */
+export * as _0x1 from './0x1.js'
+export * as _0x2 from './0x2.js'
+export * as _0x3 from './0x3.js'

package/src/iota/codegen/codegen.ts

@@ -0,0 +1,118 @@
+import { IotaMoveNormalizedModule, IotaEvent, IotaMoveObject } from '@iota/iota-sdk/client'
+
+import * as fs from 'fs'
+import chalk from 'chalk'
+import { InternalMoveModule, InternalMoveStruct, structQname } from '@typemove/move'
+import { IotaCodegen as BaseIotaCodegen } from '@typemove/iota/codegen'
+import path, { join } from 'path'
+import { SharedNetworkCodegen } from '../../move/shared-network-codegen.js'
+import { getRpcEndpoint, IotaNetwork } from '../network.js'
+import { recursiveCodegen } from '../../core/codegen.js'
+
+export async function codegen(
+  abisDir: string,
+  outDir = join('src', 'types', 'iota'),
+  genExample = false,
+  builtin = false
+) {
+  if (!fs.existsSync(abisDir)) {
+    return
+  }
+  const gen = new IotaCodegen()
+  const numFiles = await gen.generate(abisDir, outDir, builtin)
+  console.log(chalk.green(`Generated ${numFiles} for Iota`))
+}
+
+class IotaNetworkCodegen extends BaseIotaCodegen {
+  moduleGenerator: SharedNetworkCodegen<IotaNetwork, IotaMoveNormalizedModule, IotaEvent | IotaMoveObject>
+  SYSTEM_PACKAGE = '@sentio/sdk/iota'
+
+  constructor(network: IotaNetwork) {
+    const endpoint = getRpcEndpoint(network)
+    super(endpoint)
+    const generator = this
+    this.moduleGenerator = new (class extends SharedNetworkCodegen<
+      IotaNetwork,
+      IotaMoveNormalizedModule,
+      IotaEvent | IotaMoveObject
+    > {
+      ADDRESS_TYPE = 'string'
+      PREFIX = 'Iota'
+      SYSTEM_PACKAGE = '@sentio/sdk/iota'
+      NETWORK = IotaNetwork
+
+      generateStructs(module: InternalMoveModule, struct: InternalMoveStruct, events: Set<string>): string {
+        let content = ''
+        switch (structQname(module, struct)) {
+          // TODO they should still have module code generated
+          case '0x1::ascii::Char':
+          case '0x1::ascii::String':
+          case '0x2::object::ID':
+            content += `export type ${struct.name} = string`
+            break
+          case '0x2::coin::Coin':
+            content += `export type ${struct.name}<T> = string`
+            break
+          case '0x2::balance::Balance':
+            content += `export type ${struct.name}<T> = bigint`
+            break
+          case '0x1::option::Option':
+            content += `export type Option<T> = T | undefined`
+            break
+        }
+        return content + super.generateStructs(module, struct, events, content !== '')
+      }
+
+      generateForOnEvents(module: InternalMoveModule, struct: InternalMoveStruct): string {
+        switch (structQname(module, struct)) {
+          case '0x1::ascii::Char':
+          case '0x1::ascii::String':
+          case '0x2::object::ID':
+          case '0x2::coin::Coin':
+          case '0x1::option::Option':
+          case '0x2::balance::Balance':
+            return ''
+        }
+        return super.generateForOnEvents(module, struct)
+      }
+
+      protected generateExtra(address: string | undefined, module: InternalMoveModule): string {
+        return generator.generateExtra(address, module)
+      }
+    })(network, this.chainAdapter)
+  }
+
+  generateModule(module: InternalMoveModule, allEventStructs: Map<string, InternalMoveStruct>) {
+    return this.moduleGenerator.generateModule(module, allEventStructs)
+  }
+  generateImports() {
+    return super.generateImports() + this.moduleGenerator.generateImports()
+  }
+  generateLoadAll(isSystem: boolean): string {
+    return this.moduleGenerator.generateLoadAll(isSystem)
+  }
+}
+//
+const MAINNET_CODEGEN = new IotaNetworkCodegen(IotaNetwork.MAIN_NET)
+const TESTNET_CODEGEN = new IotaNetworkCodegen(IotaNetwork.TEST_NET)
+
+class IotaCodegen {
+  async generate(srcDir: string, outputDir: string, builtin = false): Promise<number> {
+    let numFiles = 0
+    const generators: [string, IotaNetworkCodegen][] = [
+      ['', MAINNET_CODEGEN],
+      ['testnet', TESTNET_CODEGEN]
+    ]
+
+    for (const [network, gen] of generators) {
+      const exclude = network ? [] : ['testnet']
+      numFiles += await recursiveCodegen(
+        path.join(srcDir, network),
+        path.join(outputDir, network),
+        (src, dst) => gen.generate(src, dst, builtin),
+        exclude
+      )
+    }
+    return numFiles
+  }
+}

package/src/iota/codegen/index.ts

@@ -0,0 +1 @@
+export * from './codegen.js'

package/src/iota/codegen/run.ts

@@ -0,0 +1,13 @@
+// #!/usr/bin/env node
+
+import { codegen } from './codegen.js'
+
+if (process.argv.length > 3) {
+  const abisDir = process.argv[2]
+  const targetDir = process.argv[3]
+
+  await codegen(abisDir, targetDir, false, true)
+} else {
+  console.error('Not enough argument')
+  process.exit(1)
+}

package/src/iota/context.ts

@@ -0,0 +1,200 @@
+import { RecordMetaData } from '@sentio/protos'
+import { type Labels, normalizeLabels } from '../index.js'
+import { getClient, IotaNetwork } from './network.js'
+import {
+  IotaTransactionBlockResponse,
+  IotaEvent,
+  IotaMoveNormalizedModule,
+  IotaMoveObject,
+  IotaClient
+} from '@iota/iota-sdk/client'
+import { MoveCoder } from './index.js'
+import { defaultMoveCoder } from './move-coder.js'
+import { MoveAccountContext, MoveContext } from '../move/index.js'
+
+export class IotaContext extends MoveContext<IotaNetwork, IotaMoveNormalizedModule, IotaEvent | IotaMoveObject> {
+  moduleName: string
+  timestamp: Date
+  checkpoint: bigint
+  transaction: IotaTransactionBlockResponse
+  eventIndex: number
+  coder: MoveCoder
+
+  constructor(
+    moduleName: string,
+    network: IotaNetwork,
+    address: string,
+    timestamp: Date,
+    checkpoint: bigint,
+    transaction: IotaTransactionBlockResponse,
+    eventIndex: number,
+    baseLabels: Labels | undefined
+  ) {
+    super(baseLabels)
+    this.address = address.toLowerCase()
+    this.network = network
+    this.moduleName = moduleName
+    this.timestamp = timestamp
+    this.checkpoint = checkpoint
+    this.eventIndex = eventIndex
+    this.coder = defaultMoveCoder(network)
+    if (transaction) {
+      this.transaction = transaction
+    }
+  }
+
+  getChainId() {
+    return this.network
+  }
+
+  getTimestamp(): number {
+    return this.timestamp.getDate()
+  }
+
+  getMetaDataInternal(name: string, labels: Labels): RecordMetaData {
+    return {
+      address: this.address,
+      contractName: this.moduleName,
+      blockNumber: this.checkpoint,
+      transactionIndex: 0,
+      transactionHash: this.transaction?.digest || '', // TODO
+      logIndex: this.eventIndex,
+      chainId: this.getChainId(),
+      name: name,
+      labels: normalizeLabels(labels)
+    }
+  }
+
+  get client(): IotaClient {
+    return getClient(this.network)
+  }
+}
+
+export class IotaObjectChangeContext extends MoveContext<
+  IotaNetwork,
+  IotaMoveNormalizedModule,
+  IotaEvent | IotaMoveObject
+> {
+  timestamp: Date
+  checkpoint: bigint
+  coder: MoveCoder
+  txDigest: string
+
+  constructor(
+    network: IotaNetwork,
+    address: string,
+    timestamp: Date,
+    checkpoint: bigint,
+    txDigest: string,
+    baseLabels: Labels | undefined
+  ) {
+    super(baseLabels)
+    this.address = address
+    this.network = network
+    this.timestamp = timestamp
+    this.checkpoint = checkpoint
+    this.txDigest = txDigest
+    this.coder = defaultMoveCoder(network)
+  }
+
+  getChainId() {
+    return this.network
+  }
+
+  getTimestamp(): number {
+    return this.timestamp.getDate()
+  }
+
+  getMetaDataInternal(name: string, labels: Labels): RecordMetaData {
+    return {
+      address: this.address,
+      contractName: '*',
+      blockNumber: this.checkpoint,
+      transactionIndex: -1,
+      transactionHash: this.txDigest,
+      logIndex: 0,
+      chainId: this.getChainId(),
+      name: name,
+      labels: normalizeLabels(labels)
+    }
+  }
+
+  get client(): IotaClient {
+    return getClient(this.network)
+  }
+}
+
+// TODO address handler should use this context
+export class IotaAddressContext extends MoveAccountContext<
+  IotaNetwork,
+  IotaMoveNormalizedModule,
+  IotaEvent | IotaMoveObject
+> {
+  address: string
+  network: IotaNetwork
+  checkpoint: bigint
+  timestamp: Date
+  coder: MoveCoder
+
+  protected contractName = 'address'
+
+  constructor(
+    network: IotaNetwork,
+    address: string,
+    checkpoint: bigint,
+    timestamp: Date,
+    baseLabels: Labels | undefined
+  ) {
+    super(baseLabels)
+    this.address = address
+    this.network = network
+    this.checkpoint = checkpoint
+    this.timestamp = timestamp
+    this.coder = defaultMoveCoder(network)
+  }
+
+  getChainId() {
+    return this.network
+  }
+
+  getMetaDataInternal(name: string, labels: Labels): RecordMetaData {
+    return {
+      address: this.address,
+      contractName: this.contractName,
+      blockNumber: this.checkpoint,
+      transactionIndex: 0,
+      transactionHash: '',
+      logIndex: 0,
+      chainId: this.getChainId(),
+      name: name,
+      labels: normalizeLabels(labels)
+    }
+  }
+
+  get client(): IotaClient {
+    return getClient(this.network)
+  }
+
+  getTimestamp(): number {
+    return this.timestamp.getDate()
+  }
+}
+
+export class IotaObjectContext extends IotaAddressContext {
+  contractName = 'object'
+  objectId: string
+  objectVersion: bigint
+
+  constructor(
+    network: IotaNetwork,
+    objectId: string,
+    objectVersion: bigint,
+    checkpoint: bigint,
+    timestamp: Date,
+    baseLabels: Labels | undefined
+  ) {
+    super(network, objectId, checkpoint, timestamp, baseLabels)
+    this.objectId = objectId
+    this.objectVersion = objectVersion
+  }
+}