revanced-cli/src/main/kotlin/app/revanced/cli/utils/SignatureParser.kt

package app.revanced.cli.utils

import app.revanced.patcher.signature.MethodSignature
import com.google.gson.JsonParser
import org.jf.dexlib2.AccessFlags
import org.jf.dexlib2.Opcodes
import java.io.File

class SignatureParser {
    companion object {
        fun parse(signatureJsonPath: String): List<MethodSignature> {
            val json = File(signatureJsonPath).readText()
            val signatures = JsonParser.parseString(json).asJsonObject.get("signatures").asJsonArray.map { sig ->
                val signature = sig.asJsonObject

                val returnType = signature.get("returns").asString

                var accessFlags = 0

                signature
                    .get("accessors").asJsonArray
                    .forEach { accessFlags = accessFlags or AccessFlags.getAccessFlag(it.asString).value }

                val parameters = signature.get("parameters").asJsonArray
                    .map { it.asString }
                    .toTypedArray()

                val opcodes = signature.get("opcodes").asJsonArray
                    .map { Opcodes.getDefault().getOpcodeByName(it.asString)!! }
                    .toTypedArray()

                MethodSignature(
                    signature.get("name").asString,
                    returnType,
                    accessFlags,
                    parameters,
                    opcodes
                )
            }

            return signatures
        }
    }
}