main.ts

import { readdirSync, readFileSync, existsSync, writeFileSync, mkdirSync } from 'fs'
import { resolve, extname } from 'path'
import { abi } from 'thor-devkit';


const baseDir = resolve(__dirname, 'ABIs')
const distDir = resolve(__dirname, 'dist')
const qDir = resolve(distDir, 'q')
const cDir = resolve(distDir, 'c')

const entries = readdirSync(baseDir, { withFileTypes: true })

const fsOpt = { encoding: 'utf8' }

mkdirSync(qDir, { recursive: true })
mkdirSync(cDir, { recursive: true })

writeFileSync(
    resolve(distDir, 'contracts.json'),
    JSON.stringify(entries.map(v => {
        if (!v.isFile() || v.name.startsWith('.')) {
            return
        }
        const str = readFileSync(resolve(baseDir, v.name), fsOpt)
        const abiJSONArray = JSON.parse(str)
        if (!Array.isArray(abiJSONArray)) {
            throw new Error('ABI expected array')
        }

        const contractName = v.name.slice(0, -extname(v.name).length)
        abiJSONArray.forEach(abiJSON => save(abiJSON, contractName))
        return v.name
    })),
    fsOpt)

function save(jsonABI: any, contractName: string) {
    jsonABI = { ...jsonABI, $contractName: contractName }
    let sig = ''
    if (jsonABI.type === 'event') {
        const ev = new abi.Event(jsonABI)
        sig = ev.signature
    } else {
        if (!jsonABI.inputs) {
            // fallback / constructor
            return
        }
        const fn = new abi.Function(jsonABI)
        sig = fn.signature
    }

    const path = resolve(qDir, sig + '.json')
    if (existsSync(path)) {
        const exist = JSON.parse(readFileSync(path, fsOpt)) as any[]
        exist.push(jsonABI)
        writeFileSync(path, JSON.stringify(exist), fsOpt)
    } else {
        writeFileSync(path, JSON.stringify([jsonABI]), fsOpt)
    }
}