@graphprotocol/toolshed/src/deployments/subgraph-service/actions.ts

A collection of tools and utilities for the Graph Protocol Typescript components

import type { SubgraphService } from '@graphprotocol/interfaces'
import type { HardhatEthersSigner } from '@nomicfoundation/hardhat-ethers/signers'
import { Interface } from 'ethers'

export function loadActions(contracts: { SubgraphService: SubgraphService }) {
  return {
    /**
     * Collects the allocated funds for a subgraph deployment
     * @param signer - The signer that will execute the collect transaction
     * @param args Parameters:
     *   - `[indexer, paymentType, data]` - The collect parameters
     * @returns The payment collected
     */
    collect: (signer: HardhatEthersSigner, args: Parameters<SubgraphService['collect']>): Promise<bigint> =>
      collect(contracts, signer, args),
  }
}

// Collects payment from the subgraph service
async function collect(
  contracts: { SubgraphService: SubgraphService },
  signer: HardhatEthersSigner,
  args: Parameters<SubgraphService['collect']>,
): Promise<bigint> {
  const { SubgraphService } = contracts
  const [indexer, paymentType, data] = args

  const tx = await SubgraphService.connect(signer).collect(indexer, paymentType, data)
  const receipt = await tx.wait()
  if (!receipt) throw new Error('Transaction failed')

  const iface = new Interface([
    'event ServicePaymentCollected(address indexed serviceProvider, uint8 indexed feeType, uint256 tokens)',
  ])
  const event = receipt.logs.find((log) => log.topics[0] === iface.getEvent('ServicePaymentCollected')?.topicHash)
  if (!event) throw new Error('ServicePaymentCollected event not found')

  return BigInt(event.data)
}