// Copyright (c) 2026 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
module ssa

import v2.ast
import v2.types

fn int_contract_num(value string) ast.Expr {
    return ast.Expr(ast.BasicLiteral{
        kind:  .number
        value: value
    })
}

fn int_contract_cast(type_name string, value string) ast.Expr {
    return ast.Expr(ast.CallOrCastExpr{
        lhs:  ast.Expr(ast.Ident{
            name: type_name
        })
        expr: int_contract_num(value)
    })
}

fn test_const_untyped_integer_array_serializes_as_dense_i32() {
    mut mod := Module.new('array_int_contract')
    env := types.Environment.new()
    mut b := Builder.new_with_env(mod, env)
    data := b.try_serialize_const_array(ast.ArrayInitExpr{
        exprs: [int_contract_num('1'), int_contract_num('2'),
            int_contract_num('3')]
    })
    assert data == [u8(1), 0, 0, 0, 2, 0, 0, 0, 3, 0, 0, 0]
}

fn test_const_explicit_i64_array_serializes_as_dense_i64() {
    mut mod := Module.new('array_int_contract_i64')
    env := types.Environment.new()
    mut b := Builder.new_with_env(mod, env)
    data := b.try_serialize_const_array(ast.ArrayInitExpr{
        exprs: [int_contract_cast('i64', '1'), int_contract_cast('i64', '2')]
    })
    assert data == [u8(1), 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 0]
}

fn test_const_explicit_u8_array_serializes_as_dense_u8() {
    mut mod := Module.new('array_int_contract_u8')
    env := types.Environment.new()
    mut b := Builder.new_with_env(mod, env)
    data := b.try_serialize_const_array(ast.ArrayInitExpr{
        exprs: [int_contract_cast('u8', '1'), int_contract_cast('u8', '2')]
    })
    assert data == [u8(1), 2]
}

fn test_const_explicit_rune_array_serializes_as_dense_i32() {
    mut mod := Module.new('array_int_contract_rune')
    env := types.Environment.new()
    mut b := Builder.new_with_env(mod, env)
    data := b.try_serialize_const_array(ast.ArrayInitExpr{
        exprs: [int_contract_cast('rune', '65'), int_contract_cast('rune', '8364')]
    })
    assert data == [u8(65), 0, 0, 0, 172, 32, 0, 0]
}