@helios-lang/compiler/src/parse/parseTypeParameters.js

Helios is a Domain Specific Language that compiles to Plutus-Core (i.e. Cardano on-chain validator scripts). Helios is a non-Haskell alternative to Plutus. With this library you can compile Helios scripts and build Cardano transactions, all you need to bu

import { group, symbol } from "@helios-lang/compiler-utils"
import { Expr, RefExpr } from "../expressions/index.js"
import { TypeParameter, TypeParameters } from "../statements/index.js"
import { ParseContext } from "./ParseContext.js"
import { anyName } from "./parseName.js"

/**
 * @param {ParseContext} ctx
 * @param {boolean} isForFunc
 * @returns {TypeParameters}
 */
export function parseTypeParameters(ctx, isForFunc = false) {
    const r = ctx.reader

    let m

    if ((m = r.matches(group("[", { minLength: 1 })))) {
        const params = m.fields.reduce((lst, f) => {
            if ((m = f.matches(anyName))) {
                const name = m

                /**
                 * @type {Expr | undefined}
                 */
                let typeClassExpr = undefined

                if ((m = f.matches(symbol(":")))) {
                    typeClassExpr = parseTypeClassRef(
                        ctx.atSite(m.site).withReader(f)
                    )
                } else {
                    f.endMatch(false)
                    f.end()
                }

                lst.push(new TypeParameter(name, typeClassExpr))
            } else {
                f.endMatch()
            }

            return lst
        }, /** @type {TypeParameter[]} */ ([]))

        return new TypeParameters(params, isForFunc)
    } else {
        r.endMatch(false)

        return new TypeParameters([], isForFunc)
    }
}

/**
 * @param {ParseContext} ctx
 * @returns {Expr | undefined}
 */
function parseTypeClassRef(ctx) {
    const r = ctx.reader

    let m

    if ((m = r.matches(anyName))) {
        r.end()
        return new RefExpr(m)
    } else {
        r.endMatch()
        r.end()

        return undefined
    }
}