sway-ast 0.71.2

use serde::{Deserialize, Serialize};
use std::fmt;

#[derive(Eq, PartialEq, Debug, Clone, Hash, Serialize, Deserialize)]
pub enum Intrinsic {
    IsReferenceType,
    SizeOfType,
    SizeOfVal,
    SizeOfStr,
    IsStrArray,
    AssertIsStrArray,
    ToStrArray,
    Eq,
    Gt,
    Lt,
    Gtf,
    AddrOf,
    StateClear,
    StateClearSlots,
    StateLoadWord,
    StateStoreWord,
    StateLoadQuad,
    StateStoreQuad,
    StateLoadSlot,
    StateStoreSlot,
    StateUpdateSlot,
    StatePreload,
    Log,
    Add,
    Sub,
    Mul,
    Div,
    And,
    Or,
    Xor,
    Lsh,
    Rsh,
    Mod,
    Revert,
    PtrAdd,
    PtrSub,
    Smo,
    Not,
    JmpMem,
    ContractCall,           // __contract_call(params, coins, asset_id, gas)
    ContractRet,            // __contract_ret(ptr, len)
    EncodeBufferEmpty,      // let buffer: (raw_ptr, u64, u64) = __encode_buffer_empty()
    EncodeBufferAppend, // let buffer: (raw_ptr, u64, u64) = __encode_buffer_append(buffer, primitive data type)
    EncodeBufferAsRawSlice, // let slice: raw_slice = __encode_buffer_as_raw_slice(buffer)
    Slice, // let ref_to_slice = __slice::<T: array or ref_to_slice>(item: T, inclusive_start_index, exclusive_end_index)
    ElemAt, // let elem: &T = __elem_at::<T: array or ref_to_slice>(item: T, index)
    Transmute, // let dst: B = __transmute::<A, B>(src)
    Dbg,   // __dbg(value)
    Alloc, // __alloc<T>(size: u64) -> raw_ptr
    RuntimeMemoryId, // __runtime_mem_id::<T>() -> u64
    EncodingMemoryId, // __encoding_mem_id::<T>() -> u64
}

impl fmt::Display for Intrinsic {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        let s = match self {
            Intrinsic::IsReferenceType => "is_reference_type",
            Intrinsic::IsStrArray => "is_str_type",
            Intrinsic::SizeOfType => "size_of",
            Intrinsic::SizeOfVal => "size_of_val",
            Intrinsic::SizeOfStr => "size_of_str_array",
            Intrinsic::AssertIsStrArray => "assert_is_str_array",
            Intrinsic::ToStrArray => "to_str_array",
            Intrinsic::Eq => "eq",
            Intrinsic::Gt => "gt",
            Intrinsic::Lt => "lt",
            Intrinsic::Gtf => "gtf",
            Intrinsic::AddrOf => "addr_of",
            Intrinsic::StateClear => "state_clear",
            Intrinsic::StateClearSlots => "state_clear_slots",
            Intrinsic::StateLoadWord => "state_load_word",
            Intrinsic::StateStoreWord => "state_store_word",
            Intrinsic::StateLoadQuad => "state_load_quad",
            Intrinsic::StateStoreQuad => "state_store_quad",
            Intrinsic::StateLoadSlot => "state_load_slot",
            Intrinsic::StateStoreSlot => "state_store_slot",
            Intrinsic::StateUpdateSlot => "state_update_slot",
            Intrinsic::StatePreload => "state_preload",
            Intrinsic::Log => "log",
            Intrinsic::Add => "add",
            Intrinsic::Sub => "sub",
            Intrinsic::Mul => "mul",
            Intrinsic::Div => "div",
            Intrinsic::And => "and",
            Intrinsic::Or => "or",
            Intrinsic::Xor => "xor",
            Intrinsic::Lsh => "lsh",
            Intrinsic::Rsh => "rsh",
            Intrinsic::Mod => "mod",
            Intrinsic::Revert => "revert",
            Intrinsic::PtrAdd => "ptr_add",
            Intrinsic::PtrSub => "ptr_sub",
            Intrinsic::Smo => "smo",
            Intrinsic::Not => "not",
            Intrinsic::JmpMem => "jmp_mem",
            Intrinsic::ContractCall => "contract_call",
            Intrinsic::ContractRet => "contract_ret",
            Intrinsic::EncodeBufferEmpty => "encode_buffer_empty",
            Intrinsic::EncodeBufferAppend => "encode_buffer_append",
            Intrinsic::EncodeBufferAsRawSlice => "encode_buffer_as_raw_slice",
            Intrinsic::Slice => "slice",
            Intrinsic::ElemAt => "elem_at",
            Intrinsic::Transmute => "transmute",
            Intrinsic::Dbg => "dbg",
            Intrinsic::Alloc => "alloc",
            Intrinsic::RuntimeMemoryId => "runtime_mem_id",
            Intrinsic::EncodingMemoryId => "encoding_mem_id",
        };
        write!(f, "{s}")
    }
}

impl Intrinsic {
    pub fn try_from_str(raw: &str) -> Option<Intrinsic> {
        use Intrinsic::*;
        Some(match raw {
            "__is_reference_type" => IsReferenceType,
            "__is_str_array" => IsStrArray,
            "__size_of" => SizeOfType,
            "__size_of_val" => SizeOfVal,
            "__size_of_str_array" => SizeOfStr,
            "__assert_is_str_array" => AssertIsStrArray,
            "__to_str_array" => ToStrArray,
            "__eq" => Eq,
            "__gt" => Gt,
            "__lt" => Lt,
            "__gtf" => Gtf,
            "__addr_of" => AddrOf,
            "__state_clear" => StateClear,
            "__state_clear_slots" => StateClearSlots,
            "__state_load_word" => StateLoadWord,
            "__state_store_word" => StateStoreWord,
            "__state_load_quad" => StateLoadQuad,
            "__state_store_quad" => StateStoreQuad,
            "__state_load_slot" => StateLoadSlot,
            "__state_store_slot" => StateStoreSlot,
            "__state_update_slot" => StateUpdateSlot,
            "__state_preload" => StatePreload,
            "__log" => Log,
            "__add" => Add,
            "__sub" => Sub,
            "__mul" => Mul,
            "__div" => Div,
            "__and" => And,
            "__or" => Or,
            "__xor" => Xor,
            "__lsh" => Lsh,
            "__rsh" => Rsh,
            "__mod" => Mod,
            "__revert" => Revert,
            "__ptr_add" => PtrAdd,
            "__ptr_sub" => PtrSub,
            "__smo" => Smo,
            "__not" => Not,
            "__jmp_mem" => JmpMem,
            "__contract_call" => ContractCall,
            "__contract_ret" => ContractRet,
            "__encode_buffer_empty" => EncodeBufferEmpty,
            "__encode_buffer_append" => EncodeBufferAppend,
            "__encode_buffer_as_raw_slice" => EncodeBufferAsRawSlice,
            "__slice" => Slice,
            "__elem_at" => ElemAt,
            "__transmute" => Transmute,
            "__dbg" => Dbg,
            "__alloc" => Alloc,
            "__runtime_mem_id" => RuntimeMemoryId,
            "__encoding_mem_id" => EncodingMemoryId,
            _ => return None,
        })
    }
}