harsta/node/features/deployer/storage.ts

Harsta is a contract development tool based on Hardhat, designed to streamline the development, testing, and referencing of contracts, addresses, ABIs, and contract instances.

import path from 'pathe'
import fs from 'fs-extra'
import { loadFile, writeFile } from 'magicast'
import { absolutePaths } from '../../constants'
import { network } from '../environment'

export async function getAddresses() {
  const addrFile = `${absolutePaths.userAddresses}.ts`
  const jsonFile = `${absolutePaths.userAddresses}.json`

  if (fs.existsSync(addrFile))
    return loadFile(addrFile).then(mod => mod.exports.default)

  if (fs.existsSync(jsonFile))
    return fs.readJSON(jsonFile).then(mod => mod)
}

export async function getAddress(nameOrAddress: string): Promise<string | undefined> {
  if (nameOrAddress.startsWith('0x'))
    return nameOrAddress
  return getAddresses().then(mod => mod?.[network.id]?.[nameOrAddress])
}

export async function setAddress(name: string, address: string) {
  const addrFile = `${absolutePaths.userAddresses}.ts`
  const jsonFile = `${absolutePaths.userAddresses}.json`

  if (fs.existsSync(addrFile)) {
    const mod = await loadFile(addrFile)
    mod.exports.default ??= {}
    mod.exports.default[network.id] ??= {}
    mod.exports.default[network.id][name] = address
    await writeFile(mod, addrFile)
  }

  if (fs.existsSync(jsonFile)) {
    const mod = await fs.readJSON(jsonFile)
    mod[network.id] ??= {}
    mod[network.id][name] = address
    await fs.writeJson(jsonFile, mod, { spaces: 2 })
  }
}

export async function getDeployed(name: string) {
  const dirPath = path.resolve(absolutePaths.harstaDeployments, network.alias)
  const filePath = path.resolve(dirPath, `${name}.json`)
  if (!fs.existsSync(filePath)) {
    return
  }

  return fs.readJSON(filePath)
}

export async function setDeployed(name: string, deployed: any) {
  const dirPath = path.resolve(absolutePaths.harstaDeployments, network.alias)
  const filePath = path.resolve(dirPath, `${name}.json`)
  await fs.ensureDir(dirPath)
  await fs.writeJSON(filePath, deployed, { spaces: 2 })
}