@sentio/sdk 1.7.14 → 1.7.15

package/src/target-ethers-sentio/codegen.ts

@@ -0,0 +1,289 @@
+import {
+  Contract,
+  createImportsForUsedIdentifiers,
+  EventDeclaration,
+  FunctionDeclaration,
+  getFullSignatureAsSymbolForEvent,
+} from 'typechain'
+
+import { reservedKeywords } from '@typechain/ethers-v5/dist/codegen/reserved-keywords'
+import { generateInputTypes } from '@typechain/ethers-v5/dist/codegen/types'
+import { getFullSignatureForEvent } from 'typechain/dist/utils/signatures'
+
+export function codeGenIndex(contract: Contract): string {
+  return ` 
+  export * from '../internal/${contract.name.toLowerCase()}_processor'
+  export * from '../internal/${contract.name}'
+  `
+}
+
+export function codeGenSentioFile(contract: Contract): string {
+  const source = `
+  const templateContract = ${contract.name}__factory.connect("", DummyProvider)
+
+  export class ${contract.name}ContractView extends ContractView<${contract.name}> {
+    constructor (contract: ${contract.name}) {
+      super(contract);
+    }
+
+    ${Object.values(contract.functions)
+      .filter((f) => !reservedKeywords.has(f[0].name))
+      .flatMap((v) => v.filter((f) => f.stateMutability == 'view').map(generateViewFunction))
+      .join('\n')}
+  }
+  
+  export class ${contract.name}BoundContractView extends BoundContractView<${contract.name}, ${
+    contract.name
+  }ContractView> {
+    // constructor (contract: ${contract.name}) {
+    //   super(contract);
+    // }
+
+    ${Object.values(contract.functions)
+      .filter((f) => !reservedKeywords.has(f[0].name))
+      .flatMap((v) => v.filter((f) => f.stateMutability == 'view').map(generateBoundViewFunction))
+      .join('\n')}
+  }
+
+  export type ${contract.name}Context = Context<${contract.name}, ${contract.name}BoundContractView>
+
+  export class ${contract.name}ProcessorTemplate extends BaseProcessorTemplate<${contract.name}, ${
+    contract.name
+  }BoundContractView> {
+    bindInternal(options: BindOptions) {
+      let processor = getProcessor("${contract.name}", options) as ${contract.name}Processor
+      if (!processor) {
+        const finalOptions = Object.assign({}, options)
+        finalOptions.name = getContractName("${contract.name}", options.name, options.address, options.network)
+        processor = new ${contract.name}Processor(finalOptions)
+        addProcessor("${contract.name}", options, processor)
+      }
+      return processor
+    }
+
+    ${Object.values(contract.events)
+      .map((events) => {
+        if (events.length === 1) {
+          return generateOnEventFunction(events[0], contract.name, false)
+        } else {
+          return events.map((e) => generateOnEventFunction(e, contract.name, true)).join('\n')
+        }
+      })
+      .join('\n')}
+    }
+
+    export class ${contract.name}Processor extends BaseProcessor<${contract.name}, ${contract.name}BoundContractView> {
+      ${Object.values(contract.events)
+        .map((events) => {
+          if (events.length === 1) {
+            return generateOnEventFunction(events[0], contract.name, false)
+          } else {
+            return events.map((e) => generateOnEventFunction(e, contract.name, true)).join('\n')
+          }
+        })
+        .join('\n')}
+
+    public static filters = templateContract.filters
+    
+    protected CreateBoundContractView(): ${contract.name}BoundContractView {
+      const view = get${contract.name}Contract(this.config.address, this.config.network)
+      return new ${contract.name}BoundContractView(view)
+    }
+
+    public static bind(options: BindOptions): ${contract.name}Processor {
+      let processor = getProcessor("${contract.name}", options) as ${contract.name}Processor
+      if (!processor) {
+        // const wrapper = get${contract.name}Contract(options.address, options.network)
+
+        const finalOptions = Object.assign({}, options)
+        finalOptions.name = getContractName("${contract.name}", options.name, options.address, options.network)
+        processor = new ${contract.name}Processor(finalOptions)
+        addProcessor("${contract.name}", options, processor)
+      }
+      return processor
+    }
+  }
+
+  export function get${contract.name}Contract(address: string, network: Networkish = 1): ${contract.name}ContractView {
+    let contract = getContractByABI("${contract.name}", address, network) as ${contract.name}ContractView
+    if (!contract) {
+      const rawContract = ${contract.name}__factory.connect(address, getProvider(network))
+      contract = new ${contract.name}ContractView(rawContract)
+      addContractByABI("${contract.name}", address, network, contract)
+    }
+    return contract
+  }
+  `
+
+  const imports = createImportsForUsedIdentifiers(
+    {
+      ethers: ['BigNumber', 'BigNumberish', 'CallOverrides', 'BytesLike'],
+      '@ethersproject/providers': ['Networkish'],
+      '@sentio/sdk': [
+        'addContractByABI',
+        'getContractByABI',
+        'addProcessor',
+        'getProcessor',
+        'getProvider',
+        'transformEtherError',
+        'BindOptions',
+        'BaseProcessor',
+        'BaseProcessorTemplate',
+        'BoundContractView',
+        'Context',
+        'ContractView',
+        'DummyProvider',
+        'getContractName',
+      ],
+      './common': ['PromiseOrValue'],
+      './index': [`${contract.name}`, `${contract.name}__factory`],
+      [`./${contract.name}`]: (<string[]>[]).concat(
+        ...Object.values(contract.events).map((events) => {
+          if (events.length === 1) {
+            return [`${events[0].name}Event`, `${events[0].name}EventFilter`]
+          } else {
+            return events.flatMap((e) => [
+              `${getFullSignatureAsSymbolForEvent(e)}_Event`,
+              `${getFullSignatureAsSymbolForEvent(e)}_EventFilter`,
+            ])
+          }
+        })
+      ),
+    },
+    source
+  )
+
+  return imports + source
+}
+
+export function codeGenTestUtilsFile(contract: Contract): string {
+  const source = `
+  const mockField = {
+    blockHash:
+      "0x0000000000000000000000000000000000000000000000000000000000000000",
+    blockNumber: 0,
+    logIndex: 0,
+    removed: false,
+    transactionHash:
+      "0x0000000000000000000000000000000000000000000000000000000000000000",
+    transactionIndex: 0,
+  }
+  ${Object.values(contract.events)
+    .map((events) => {
+      if (events.length === 1) {
+        return generateMockEventLogFunction(events[0], contract.name, false)
+      } else {
+        return events.map((e) => generateMockEventLogFunction(e, contract.name, true)).join('\n')
+      }
+    })
+    .join('\n')}
+  `
+
+  const imports = createImportsForUsedIdentifiers(
+    {
+      '@ethersproject/providers': ['Log'],
+      '.': [
+        `get${contract.name}Contract`,
+        ...Object.values(contract.events).flatMap((events) => {
+          if (events.length === 1) {
+            return `${events[0].name}EventObject`
+          } else {
+            return events.flatMap((e) => `${getFullSignatureAsSymbolForEvent(e)}_EventObject`)
+          }
+        }),
+      ],
+    },
+    source
+  )
+
+  return imports + source
+}
+
+function generateOnEventFunction(event: EventDeclaration, contractName: string, includeArgTypes: boolean): string {
+  let eventName = event.name
+  if (includeArgTypes) {
+    eventName = getFullSignatureAsSymbolForEvent(event) + '_'
+  }
+
+  const filterName = getFullSignatureForEvent(event)
+
+  return `
+  on${eventName}(
+    handler: (event: ${eventName}Event, ctx: ${contractName}Context) => void,
+    filter?: ${eventName}EventFilter | ${eventName}EventFilter[]
+  ) {
+    if (!filter) {
+      filter = ${contractName}Processor.filters['${filterName}'](${event.inputs.map(() => 'null').join(',')})
+    }
+    return super.onEvent(handler, filter)
+  }
+  `
+}
+
+function generateViewFunction(func: FunctionDeclaration): string {
+  return `
+  async ${func.name}(${generateInputTypes(func.inputs, { useStructs: true })}overrides?: CallOverrides) {
+    try {
+      if (overrides) {
+        return await this.contract.${func.name}(${
+    func.inputs.length > 0 ? func.inputs.map((input, index) => input.name || `arg${index}`).join(',') + ',' : ''
+  } overrides)
+      } else {
+        return await this.contract.${func.name}(${func.inputs
+    .map((input, index) => input.name || `arg${index}`)
+    .join(',')})
+      }
+    } catch (e) {
+      throw transformEtherError(e, undefined)
+    }
+  }
+  `
+}
+
+function generateBoundViewFunction(func: FunctionDeclaration): string {
+  return `
+  async ${func.name}(${generateInputTypes(func.inputs, { useStructs: true })}overrides?: CallOverrides) {
+    try {
+      if (!overrides && this.context) {
+        overrides = {
+          blockTag: this.context.blockNumber.toNumber(),
+        }
+      }
+      if (overrides) {
+        return await this.view.${func.name}(${
+    func.inputs.length > 0 ? func.inputs.map((input, index) => input.name || `arg${index}`).join(',') + ',' : ''
+  } overrides)
+      } else {
+        return await this.view.${func.name}(${func.inputs.map((input, index) => input.name || `arg${index}`).join(',')})
+      }
+    } catch (e) {
+      throw transformEtherError(e, this.context)
+    }
+  }
+  `
+}
+
+function generateMockEventLogFunction(event: EventDeclaration, contractName: string, includeArgTypes: boolean): string {
+  let eventName = event.name
+  let eventNameWithSignature = event.name
+  if (includeArgTypes) {
+    eventName = getFullSignatureAsSymbolForEvent(event) + '_'
+    eventNameWithSignature = getFullSignatureForEvent(event)
+  }
+
+  return `
+    export function mock${eventName}Log(contractAddress: string, event: ${eventName}EventObject): Log {
+      const contract = get${contractName}Contract(contractAddress)
+      const encodedLog = contract.rawContract.interface.encodeEventLog(
+        contract.rawContract.interface.getEvent('${eventNameWithSignature}'),
+        Object.values(event)
+      )
+      return {
+        ...mockField,
+        address: contractAddress,
+        data: encodedLog.data,
+        topics: encodedLog.topics,
+      }
+    }
+  `
+}

package/src/target-ethers-sentio/index.ts

@@ -0,0 +1,50 @@
+import Ethers from '@typechain/ethers-v5'
+import { Config, extractAbi, extractDocumentation, FileDescription, parse, shortenFullJsonFilePath } from 'typechain'
+import { dirname, join, relative } from 'path'
+import { codeGenIndex, codeGenSentioFile, codeGenTestUtilsFile } from './codegen'
+
+export default class EthersSentio extends Ethers {
+  constructor(config: Config) {
+    if (!config.outDir) {
+      throw new Error('Out put path not specificed')
+    }
+    super(config)
+  }
+
+  // TODO(pc): also have to override transformBinFile, transformFile
+  override transformAbiOrFullJsonFile(file: FileDescription): FileDescription[] | void {
+    const abi = extractAbi(file.contents)
+    if (abi.length === 0) {
+      return
+    }
+
+    const documentation = extractDocumentation(file.contents)
+
+    const jsonPath = relative(this.cfg.inputDir, shortenFullJsonFilePath(file.path, this.cfg.allFiles))
+    const contract = parse(abi, jsonPath, documentation)
+    const files = super.transformAbiOrFullJsonFile(file)
+
+    if (files !== undefined) {
+      // files.forEach(this.transformFilePath)
+      // for (const file of files) {
+      //   this.transformFilePath(file)
+      // }
+
+      return [
+        ...files,
+        {
+          path: join(dirname(files[0].path), `${contract.name.toLowerCase()}_processor.ts`),
+          contents: codeGenSentioFile(contract),
+        },
+        {
+          path: join(dirname(files[0].path), '..', contract.name.toLowerCase(), 'index.ts'),
+          contents: codeGenIndex(contract),
+        },
+        {
+          path: join(dirname(files[0].path), '..', contract.name.toLowerCase(), 'test-utils.ts'),
+          contents: codeGenTestUtilsFile(contract),
+        },
+      ]
+    }
+  }
+}

package/src/target-ethers-sentio/tsconfig.json

@@ -0,0 +1,9 @@
+{
+  "extends": "../../../tsconfig.json",
+  "compilerOptions": {
+    "rootDir": "..",
+    "outDir": "../../lib"
+  },
+  "exclude": ["lib"],
+  "include": ["."]
+}