@graphprotocol/graph-cli/src/protocols/ethereum/codegen/template.js

CLI for building for and deploying to The Graph

const tsCodegen = require('../../../codegen/typescript')

module.exports = class EthereumTemplateCodeGen {
  constructor(template) {
    this.template = template
  }

  generateModuleImports() {
    return [
      'Address',
    ]
  }

  generateCreateMethod() {
    const name = this.template.get('name')

    return tsCodegen.staticMethod(
      'create',
      [tsCodegen.param('address', tsCodegen.namedType('Address'))],
      tsCodegen.namedType('void'),
      `
      DataSourceTemplate.create('${name}', [address.toHex()])
      `,
    )
  }

  generateCreateWithContextMethod() {
    const name = this.template.get('name')

    return tsCodegen.staticMethod(
      'createWithContext',
      [
        tsCodegen.param('address', tsCodegen.namedType('Address')),
        tsCodegen.param('context', tsCodegen.namedType('DataSourceContext')),
      ],
      tsCodegen.namedType('void'),
      `
      DataSourceTemplate.createWithContext('${name}', [address.toHex()], context)
      `,
    )
  }
}