checker.js

import { join } from 'node:path'
import * as zinniaRuntime from '../lib/zinnia.js'
import { formatActivityObject, activities } from '../lib/activity.js'
import { runPingLoop, runMachinesLoop } from '../lib/telemetry.js'
import fs from 'node:fs/promises'
import { metrics } from '../lib/metrics.js'
import { paths } from '../lib/paths.js'
import { getCheckerId } from '../lib/checker-id.js'
import pRetry from 'p-retry'
import { fetch } from 'undici'
import { ethAddressFromDelegated, isEthAddress } from '@glif/filecoin-address'
import { ethers, formatEther } from 'ethers'
import { runUpdateRewardsLoop } from '../lib/rewards.js'
import { runUpdateContractsLoop } from '../lib/contracts.js'

const {
  FIL_WALLET_ADDRESS,
  PASSPHRASE
} = process.env

const runtimeNames = [
  'zinnia'
]

/**
 * @param {string} msg
 * @param {number} [exitCode]
 */
const panic = (msg, exitCode = 1) => {
  console.error(msg)
  process.exit(exitCode)
}

export const checker = async ({ json, recreateCheckerIdOnError, experimental }) => {
  if (!FIL_WALLET_ADDRESS) panic('FIL_WALLET_ADDRESS required')
  if (FIL_WALLET_ADDRESS.startsWith('f1')) {
    panic('Invalid FIL_WALLET_ADDRESS: f1 addresses are currently not supported. Please use an f4 or 0x address.')
  }
  if (
    !FIL_WALLET_ADDRESS.startsWith('f410') &&
    !FIL_WALLET_ADDRESS.startsWith('0x')
  ) {
    panic('FIL_WALLET_ADDRESS must start with f410 or 0x')
  }
  if (FIL_WALLET_ADDRESS.startsWith('0x') && !isEthAddress(FIL_WALLET_ADDRESS)) {
    panic('Invalid FIL_WALLET_ADDRESS ethereum address', 2)
  }

  const keypair = await getCheckerId({ secretsDir: paths.secrets, passphrase: PASSPHRASE, recreateOnError: recreateCheckerIdOnError })
  const CHECKER_ID = keypair.publicKey

  const fetchRes = await pRetry(
    () => fetch(`https://station-wallet-screening.fly.dev/${FIL_WALLET_ADDRESS}`),
    {
      retries: 1000,
      onFailedAttempt: () =>
        console.error('Failed to validate FIL_WALLET_ADDRESS address. Retrying...')
    }
  )
  if (fetchRes.status === 403) panic('Invalid FIL_WALLET_ADDRESS address', 2)
  if (!fetchRes.ok) panic('Failed to validate FIL_WALLET_ADDRESS address')
  const ethAddress = FIL_WALLET_ADDRESS.startsWith('0x')
    ? FIL_WALLET_ADDRESS
    : ethAddressFromDelegated(FIL_WALLET_ADDRESS)
  for (const runtimeName of runtimeNames) {
    await fs.mkdir(join(paths.runtimeCache, runtimeName), { recursive: true })
    await fs.mkdir(join(paths.runtimeState, runtimeName), { recursive: true })
  }

  activities.onActivity(activity => {
    if (json) {
      console.log(JSON.stringify({
        type: `activity:${activity.type}`,
        subnet: activity.source,
        message: activity.message
      }))
    } else {
      process.stdout.write(formatActivityObject(activity))
    }
  })

  metrics.onUpdate(metrics => {
    if (json) {
      console.log(JSON.stringify({
        type: 'jobs-completed',
        total: metrics.totalJobsCompleted,
        rewardsScheduledForAddress: formatEther(metrics.rewardsScheduledForAddress)
      }))
    } else {
      console.log(JSON.stringify({
        totalJobsCompleted: metrics.totalJobsCompleted,
        rewardsScheduledForAddress:
          formatEther(metrics.rewardsScheduledForAddress)
      }, null, 2))
    }
  })

  const contracts = []

  const fetchRequest = new ethers.FetchRequest(
    'https://api.node.glif.io/rpc/v1'
  )
  fetchRequest.setHeader(
    'Authorization',
    'Bearer RXQ2SKH/BVuwN7wisZh3b5uXStGPj1JQIrIWD+rxF0Y='
  )
  const provider = new ethers.JsonRpcProvider(fetchRequest)

  await Promise.all([
    zinniaRuntime.run({
      provider,
      CHECKER_ID,
      FIL_WALLET_ADDRESS: ethAddress,
      ethAddress,
      STATE_ROOT: join(paths.runtimeState, 'zinnia'),
      CACHE_ROOT: join(paths.runtimeCache, 'zinnia'),
      subnetVersionsDir: paths.subnetVersionsDir,
      subnetSourcesDir: paths.subnetSourcesDir,
      onActivity: activity => {
        activities.submit({
          ...activity,
          // Zinnia will try to overwrite `source` if a subnet created the
          // activity. Using the spread syntax won't work because a
          // `source: null` would overwrite the default value.
          source: activity.source || 'Zinnia'
        })
      },
      onMetrics: m => metrics.submit('zinnia', m),
      experimental
    }),
    runPingLoop({ CHECKER_ID }),
    runMachinesLoop({ CHECKER_ID }),
    runUpdateContractsLoop({
      provider,
      contracts,
      onActivity: (activity) => activities.submit(activity)
    }),
    runUpdateRewardsLoop({
      contracts,
      ethAddress,
      onMetrics: m => metrics.submit('zinnia', m)
    })
  ])
}