#![cfg(test)]
#![warn(clippy::dbg_macro)]
// See github.com/roc-lang/roc/issues/800 for discussion of the large_enum_variant check.
#![allow(clippy::large_enum_variant)]
// we actually want to compare against the literal float bits
#![allow(clippy::float_cmp)]

#[macro_use]
extern crate indoc;

/// Used in the with_larger_debug_stack() function, for tests that otherwise
/// run out of stack space in debug builds (but don't in --release builds)
#[allow(dead_code)]
const EXPANDED_STACK_SIZE: usize = 8 * 1024 * 1024;

use bumpalo::Bump;
use roc_collections::all::MutMap;
use roc_load::ExecutionMode;
use roc_load::LoadConfig;
use roc_load::LoadMonomorphizedError;
use roc_load::Threading;
use roc_module::symbol::Interns;
use roc_module::symbol::Symbol;
use roc_mono::ir::Proc;
use roc_mono::ir::ProcLayout;
use roc_mono::layout::STLayoutInterner;
use test_mono_macros::*;

const TARGET_INFO: roc_target::TargetInfo = roc_target::TargetInfo::default_x86_64();

/// Without this, some tests pass in `cargo test --release` but fail without
/// the --release flag because they run out of stack space. This increases
/// stack size for debug builds only, while leaving the stack space at the default
/// amount for release builds.
#[allow(dead_code)]
#[cfg(debug_assertions)]
pub fn with_larger_debug_stack<F>(run_test: F)
where
    F: FnOnce(),
    F: Send,
    F: 'static,
{
    std::thread::Builder::new()
        .stack_size(EXPANDED_STACK_SIZE)
        .spawn(run_test)
        .expect("Error while spawning expanded dev stack size thread")
        .join()
        .expect("Error while joining expanded dev stack size thread")
}

/// In --release builds, don't increase the stack size. Run the test normally.
/// This way, we find out if any of our tests are blowing the stack even after
/// optimizations in release builds.
#[allow(dead_code)]
#[cfg(not(debug_assertions))]
#[inline(always)]
pub fn with_larger_debug_stack<F>(run_test: F)
where
    F: FnOnce(),
    F: Send,
    F: 'static,
{
    run_test()
}

fn promote_expr_to_module(src: &str) -> String {
    let mut buffer = String::from("app \"test\" provides [main] to \"./platform\"\n\nmain =\n");

    for line in src.lines() {
        // indent the body!
        buffer.push_str("    ");
        buffer.push_str(line);
        buffer.push('\n');
    }

    buffer
}

fn compiles_to_ir(test_name: &str, src: &str, mode: &str, no_check: bool) {
    use roc_packaging::cache::RocCacheDir;
    use std::path::PathBuf;

    let exec_mode = match mode {
        "exec" => ExecutionMode::Executable,
        "test" => ExecutionMode::Test,
        _ => panic!("Invalid test_mono exec mode {mode}"),
    };

    let arena = &Bump::new();

    let filename = PathBuf::from("Test.roc");
    let src_dir = PathBuf::from("fake/test/path");

    let module_src;
    let temp;
    if src.starts_with("app") || src.starts_with("interface") {
        // this is already a module
        module_src = src;
    } else {
        // this is an expression, promote it to a module
        temp = promote_expr_to_module(src);
        module_src = &temp;
    }

    let load_config = LoadConfig {
        target_info: TARGET_INFO,
        threading: Threading::Single,
        render: roc_reporting::report::RenderTarget::Generic,
        palette: roc_reporting::report::DEFAULT_PALETTE,
        exec_mode,
    };
    let loaded = roc_load::load_and_monomorphize_from_str(
        arena,
        filename,
        module_src,
        src_dir,
        RocCacheDir::Disallowed,
        load_config,
    );

    let mut loaded = match loaded {
        Ok(x) => x,
        Err(LoadMonomorphizedError::LoadingProblem(roc_load::LoadingProblem::FormattedReport(
            report,
        ))) => {
            println!("{}", report);
            panic!();
        }
        Err(e) => panic!("{:?}", e),
    };

    use roc_load::MonomorphizedModule;
    let MonomorphizedModule {
        module_id: home,
        procedures,
        exposed_to_host,
        mut layout_interner,
        interns,
        ..
    } = loaded;

    let can_problems = loaded.can_problems.remove(&home).unwrap_or_default();
    let type_problems = loaded.type_problems.remove(&home).unwrap_or_default();

    if !can_problems.is_empty() {
        println!("Ignoring {} canonicalization problems", can_problems.len());
    }

    assert!(type_problems.is_empty());

    let main_fn_symbol = exposed_to_host.top_level_values.keys().copied().next();

    if !no_check {
        check_procedures(arena, &interns, &mut layout_interner, &procedures);
    }

    verify_procedures(test_name, layout_interner, procedures, main_fn_symbol);
}

fn check_procedures<'a>(
    arena: &'a Bump,
    interns: &Interns,
    interner: &mut STLayoutInterner<'a>,
    procedures: &MutMap<(Symbol, ProcLayout<'a>), Proc<'a>>,
) {
    use roc_mono::debug::{check_procs, format_problems};
    let problems = check_procs(arena, interner, procedures);
    if problems.is_empty() {
        return;
    }
    let formatted = format_problems(interns, interner, problems);
    panic!("IR problems found:\n{formatted}");
}

fn verify_procedures<'a>(
    test_name: &str,
    interner: STLayoutInterner<'a>,
    procedures: MutMap<(Symbol, ProcLayout<'a>), Proc<'a>>,
    opt_main_fn_symbol: Option<Symbol>,
) {
    let mut procs_string = procedures
        .values()
        .map(|proc| proc.to_pretty(&interner, 200, false))
        .collect::<Vec<_>>();

    let opt_main_fn = opt_main_fn_symbol.map(|main_fn_symbol| {
        let index = procedures
            .keys()
            .position(|(s, _)| *s == main_fn_symbol)
            .unwrap();
        procs_string.swap_remove(index)
    });

    procs_string.sort();

    if let Some(main_fn) = opt_main_fn {
        procs_string.push(main_fn);
    }

    let result = procs_string.join("\n");

    let path = format!("generated/{}.txt", test_name);
    std::fs::create_dir_all("generated").unwrap();
    std::fs::write(&path, result).unwrap();

    use std::process::Command;

    let is_tracked = Command::new("git")
        .args(["ls-files", "--error-unmatch", &path])
        .output()
        .unwrap();

    if !is_tracked.status.success() {
        panic!(
            "The file {:?} is not tracked by git. Try using `git add` on it",
            &path
        );
    }

    let has_changes = Command::new("git")
        .args(["diff", "--color=always", &path])
        .output()
        .unwrap();

    if !has_changes.status.success() {
        eprintln!("`git diff {:?}` failed", &path);
        unreachable!();
    }

    if !has_changes.stdout.is_empty() {
        println!("{}", std::str::from_utf8(&has_changes.stdout).unwrap());
        panic!("Output changed: resolve conflicts and `git add` the file.");
    }
}

#[mono_test]
fn ir_int_literal() {
    r#"
    5
    "#
}

#[mono_test]
fn ir_int_add() {
    r#"
    x = [1,2]
    5 + 4 + 3 + List.len x
    "#
}

#[mono_test]
fn ir_assignment() {
    r#"
    x = 5

    x
    "#
}

#[mono_test]
fn ir_when_maybe() {
    r#"
    when Just 3 is
        Just n -> n
        Nothing -> 0
    "#
}

#[mono_test]
fn ir_when_these() {
    r#"
    when These 1 2 is
        This x -> x
        That y -> y
        These x _ -> x
    "#
}

#[mono_test]
fn ir_when_record() {
    r#"
    when { x: 1, y: 3.14 } is
        { x } -> x
    "#
}

#[mono_test]
fn ir_plus() {
    r#"
    1 + 2
    "#
}

#[mono_test]
fn ir_round() {
    r#"
    Num.round 3.6
    "#
}

#[mono_test]
fn ir_when_idiv() {
    r#"
    when Num.divTruncChecked 1000 10 is
        Ok val -> val
        Err _ -> -1
    "#
}

#[mono_test]
fn ir_two_defs() {
    r#"
    x = 3
    y = 4

    x + y
    "#
}

#[mono_test]
fn ir_when_just() {
    r#"
    x : [Nothing, Just I64]
    x = Just 41

    when x is
        Just v -> v + 0x1
        Nothing -> 0x1
    "#
}

#[mono_test]
fn one_element_tag() {
    r#"
    x : [Pair I64]
    x = Pair 2

    x
    "#
}

#[mono_test]
fn guard_pattern_true() {
    r#"
    wrapper = \{} ->
        when 2 is
            2 if Bool.false -> 42
            _ -> 0

    wrapper {}
    "#
}

#[mono_test]
fn when_on_record() {
    r#"
    when { x: 0x2 } is
        { x } -> x + 3
    "#
}

#[mono_test]
fn when_nested_maybe() {
    r#"
    Maybe a : [Nothing, Just a]

    x : Maybe (Maybe I64)
    x = Just (Just 41)

    when x is
        Just (Just v) -> v + 0x1
        _ -> 0x1
    "#
}

#[mono_test]
fn when_on_two_values() {
    r#"
    when Pair 2 3 is
        Pair 4 3 -> 9
        Pair a b -> a + b
    "#
}

#[mono_test]
fn dict() {
    r#"
    Dict.len (Dict.empty {})
    "#
}

#[mono_test]
fn list_append_closure() {
    r#"
    myFunction = \l -> List.append l 42

    myFunction [1, 2]
    "#
}

#[mono_test]
fn list_append() {
    // TODO this leaks at the moment
    // ListAppend needs to decrement its arguments
    r#"
    List.append [1] 2
    "#
}

#[mono_test]
fn list_len() {
    r#"
    x = [1,2,3]
    y = [1.0]

    List.len x + List.len y
    "#
}

#[mono_test]
fn when_joinpoint() {
    r#"
    wrapper = \{} ->
        x : [Red, White, Blue]
        x = Blue

        y =
            when x is
                Red -> 1
                White -> 2
                Blue -> 3

        y

    wrapper {}
    "#
}

#[mono_test]
fn simple_if() {
    r#"
    if Bool.true then
        1
    else
        2
    "#
}

#[mono_test]
fn if_multi_branch() {
    r#"
    if Bool.true then
        1
    else if Bool.false then
        2
    else
        3
    "#
}

#[mono_test]
fn when_on_result() {
    r#"
    wrapper = \{} ->
        x : Result I64 I64
        x = Ok 2

        y =
            when x is
                Ok 3 -> 1
                Ok _ -> 2
                Err _ -> 3
        y

    wrapper {}
    "#
}

#[mono_test]
fn let_with_record_pattern() {
    r#"
    { x } = { x: 0x2, y: 3.14 }

    x
    "#
}

#[mono_test]
fn let_with_record_pattern_list() {
    r#"
    { x } = { x: [1, 3, 4], y: 3.14 }

    x
    "#
}

#[mono_test]
fn if_guard_bind_variable_false() {
    r#"
    wrapper = \{} ->
        when 10 is
            x if x == 5 -> 0
            _ -> 42

    wrapper {}
    "#
}

#[mono_test]
fn alias_variable() {
    r#"
    x = 5
    y = x

    3
    "#
}

#[mono_test]
fn alias_variable_and_return_it() {
    r#"
    x = 5
    y = x

    y
    "#
}

#[mono_test]
fn branch_store_variable() {
    r#"
    when 0 is
        1 -> 12
        a -> a
    "#
}

#[mono_test]
fn list_pass_to_function() {
    r#"
    x : List I64
    x = [1,2,3]

    id : List I64 -> List I64
    id = \y -> List.set y 0 0

    id x
    "#
}

#[mono_test]
fn record_optional_field_let_no_use_default() {
    r#"
    f = \r ->
        { x ? 10, y } = r
        x + y


    f { x: 4, y: 9 }
    "#
}

#[mono_test]
fn record_optional_field_let_use_default() {
    r#"
    f = \r ->
        { x ? 10, y } = r
        x + y


    f { y: 9 }
    "#
}

#[mono_test]
fn record_optional_field_function_no_use_default() {
    r#"
    f = \{ x ? 10, y } -> x + y


    f { x: 4, y: 9 }
    "#
}

#[mono_test]
fn record_optional_field_function_use_default() {
    r#"
    f = \{ x ? 10, y } -> x + y


    f { y: 9 }
    "#
}

#[mono_test]
fn quicksort_help() {
    // do we still need with_larger_debug_stack?
    r#"
    quicksortHelp : List (Num a), I64, I64 -> List (Num a)
    quicksortHelp = \list, low, high ->
        if low < high then
            (Pair partitionIndex partitioned) = Pair 0 []

            partitioned
            |> quicksortHelp low (partitionIndex - 1)
            |> quicksortHelp (partitionIndex + 1) high
        else
            list

    quicksortHelp [] 0 0
    "#
}

#[mono_test]
fn quicksort_swap() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        swap = \list ->
            when Pair (List.get list 0) (List.get list 0) is
                Pair (Ok atI) (Ok atJ) ->
                    list
                    |> List.set 0 atJ
                    |> List.set 0 atI

                _ ->
                    []

        main =
            swap [1, 2]
        "#
    )
}

// #[ignore]
// #[mono_test]
// fn quicksort_partition_help() {
//     indoc!(
//         r#"
//         app "test" provides [main] to "./platform"

//         partitionHelp : I64, I64, List (Num a), I64, (Num a) -> [Pair I64 (List (Num a))]
//         partitionHelp = \i, j, list, high, pivot ->
//             if j < high then
//                 when List.get list j is
//                     Ok value ->
//                         if value <= pivot then
//                             partitionHelp (i + 1) (j + 1) (swap (i + 1) j list) high pivot
//                         else
//                             partitionHelp i (j + 1) list high pivot

//                     Err _ ->
//                         Pair i list
//             else
//                 Pair i list

//         main =
//             partitionHelp 0 0 [] 0 0
//         "#
//     )
// }

// #[ignore]
// #[mono_test]
// fn quicksort_full() {
//     indoc!(
//         r#"
//         app "test" provides [main] to "./platform"

//         quicksortHelp : List (Num a), I64, I64 -> List (Num a)
//         quicksortHelp = \list, low, high ->
//             if low < high then
//                 (Pair partitionIndex partitioned) = partition low high list

//                 partitioned
//                     |> quicksortHelp low (partitionIndex - 1)
//                     |> quicksortHelp (partitionIndex + 1) high
//             else
//                 list

//         swap : I64, I64, List a -> List a
//         swap = \i, j, list ->
//             when Pair (List.get list i) (List.get list j) is
//                 Pair (Ok atI) (Ok atJ) ->
//                     list
//                         |> List.set i atJ
//                         |> List.set j atI

//                 _ ->
//                     []

//         partition : I64, I64, List (Num a) -> [Pair I64 (List (Num a))]
//         partition = \low, high, initialList ->
//             when List.get initialList high is
//                 Ok pivot ->
//                     when partitionHelp (low - 1) low initialList high pivot is
//                         Pair newI newList ->
//                             Pair (newI + 1) (swap (newI + 1) high newList)

//                 Err _ ->
//                     Pair (low - 1) initialList

//         partitionHelp : I64, I64, List (Num a), I64, (Num a) -> [Pair I64 (List (Num a))]
//         partitionHelp = \i, j, list, high, pivot ->
//             if j < high then
//                 when List.get list j is
//                     Ok value ->
//                         if value <= pivot then
//                             partitionHelp (i + 1) (j + 1) (swap (i + 1) j list) high pivot
//                         else
//                             partitionHelp i (j + 1) list high pivot

//                     Err _ ->
//                         Pair i list
//             else
//                 Pair i list

//         quicksort = \originalList ->
//             n = List.len originalList
//             quicksortHelp originalList 0 (n - 1)

//         main =
//             quicksort [1,2,3]
//         "#
//     )
// }

#[mono_test]
fn factorial() {
    r#"
    factorial = \n, accum ->
        when n is
            0 ->
                accum

            _ ->
                factorial (n - 1) (n * accum)

    factorial 10 1
    "#
}

#[mono_test]
fn is_nil() {
    r#"
    ConsList a : [Cons a (ConsList a), Nil]

    isNil : ConsList a -> Bool
    isNil = \list ->
        when list is
            Nil -> Bool.true
            Cons _ _ -> Bool.false

    isNil (Cons 0x2 Nil)
    "#
}

#[mono_test]
#[ignore]
fn has_none() {
    r#"
    Maybe a : [Just a, Nothing]
    ConsList a : [Cons a (ConsList a), Nil]

    hasNone : ConsList (Maybe a) -> Bool
    hasNone = \list ->
        when list is
            Nil -> Bool.false
            Cons Nothing _ -> Bool.true
            Cons (Just _) xs -> hasNone xs

    hasNone (Cons (Just 3) Nil)
    "#
}

#[mono_test]
fn mk_pair_of() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        mkPairOf = \x -> Pair x x

        main =
            mkPairOf [1,2,3]
        "#
    )
}

#[mono_test]
fn fst() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        fst = \x, _ -> x

        main =
            fst [1,2,3] [3,2,1]
        "#
    )
}

#[mono_test]
fn list_cannot_update_inplace() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        x : List I64
        x = [1,2,3]

        add : List I64 -> List I64
        add = \y -> List.set y 0 0

        main =
            List.len (add x) + List.len x
        "#
    )
}

#[mono_test]
fn list_get() {
    r#"
    wrapper = \{} ->
        List.get [1,2,3] 0

    wrapper {}
    "#
}

#[mono_test]
fn peano() {
    r#"
    Peano : [S Peano, Z]

    three : Peano
    three = S (S (S Z))

    three
    "#
}

#[mono_test]
fn peano1() {
    r#"
    Peano : [S Peano, Z]

    three : Peano
    three = S (S (S Z))

    when three is
        Z -> 0
        S _ -> 1
    "#
}

#[mono_test]
fn peano2() {
    r#"
    Peano : [S Peano, Z]

    three : Peano
    three = S (S (S Z))

    when three is
        S (S _) -> 1
        S (_) -> 0
        Z -> 0
    "#
}

#[mono_test]
fn optional_when() {
    r#"
    f = \r ->
        when r is
            { x: Blue, y ? 3 } -> y
            { x: Red, y ? 5 } -> y

    a = f { x: Blue, y: 7 }
    b = f { x: Blue }
    c = f { x: Red, y: 11 }
    d = f { x: Red }

    a * b * c * d
    "#
}

#[mono_test]
fn nested_pattern_match() {
    r#"
    Maybe a : [Nothing, Just a]

    x : Maybe (Maybe I64)
    x = Just (Just 41)

    when x is
        Just (Just v) -> v + 0x1
        _ -> 0x1
    "#
}

#[mono_test]
#[ignore]
fn linked_list_length_twice() {
    r#"
    LinkedList a : [Nil, Cons a (LinkedList a)]

    nil : LinkedList I64
    nil = Nil

    length : LinkedList a -> I64
    length = \list ->
        when list is
            Nil -> 0
            Cons _ rest -> 1 + length rest

    length nil + length nil
    "#
}

#[mono_test]
fn rigids() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        swap : Nat, Nat, List a -> List a
        swap = \i, j, list ->
            when Pair (List.get list i) (List.get list j) is
                Pair (Ok atI) (Ok atJ) ->
                    foo = atJ

                    list
                        |> List.set i foo
                        |> List.set j atI

                _ ->
                    []

        main =
            swap 0 0 [0x1]
        "#
    )
}

#[mono_test]
fn let_x_in_x() {
    r#"
    x = 5

    answer =
        1337

    unused =
        nested = 17
        nested

    answer
    "#
}

#[mono_test]
fn let_x_in_x_indirect() {
    r#"
    x = 5

    answer =
        1337

    unused =
        nested = 17

        i = 1

        nested

    { answer, unused }.answer
    "#
}

#[mono_test]
fn nested_closure() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        foo = \{} ->
            x = 42
            f = \{} -> x
            f

        main =
            f = foo {}
            f {}
        "#
    )
}

#[mono_test]
fn closure_in_list() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        foo = \{} ->
            x = 41

            f = \{} -> x

            [f]

        main =
            items = foo {}

            List.len items
        "#
    )
}

#[ignore]
#[mono_test]
fn somehow_drops_definitions() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        one : I64
        one = 1

        two : I64
        two = 2

        increment : I64 -> I64
        increment = \x -> x + one

        double : I64 -> I64
        double = \x -> x * two

        apply : (a -> a), a -> a
        apply = \f, x -> f x

        main =
            apply (if Bool.true then increment else double) 42
        "#
    )
}

#[mono_test]
fn specialize_closures() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"


        apply : (a -> a), a -> a
        apply = \f, x -> f x

        main =
            one : I64
            one = 1

            two : I64
            two = 2

            b : Bool
            b = Bool.true

            increment : I64 -> I64
            increment = \x -> x + one

            double : I64 -> I64
            double = \x -> if b then x * two else x

            apply (if Bool.true then increment else double) 42
        "#
    )
}

#[mono_test]
fn specialize_lowlevel() {
    indoc!(
        r#"
         app "test" provides [main] to "./platform"

         apply : (a -> a), a -> a
         apply = \f, x -> f x

         main =
             one : I64
             one = 1

             two : I64
             two = 2

             increment : I64 -> I64
             increment = \x -> x + one

             double : I64 -> I64
             double = \x -> x * two

             (if Bool.true then increment else double) 42
         "#
    )
}

#[mono_test]
fn empty_list_of_function_type() {
    // see https://github.com/roc-lang/roc/issues/1732
    indoc!(
        r#"
         app "test" provides [main] to "./platform"

         main =
            myList : List (Str -> Str)
            myList = []

            myClosure : Str -> Str
            myClosure = \_ -> "bar"

            choose =
                if Bool.false then
                    myList
                else
                    [myClosure]

            when List.get choose 0 is
                Ok f -> f "foo"
                Err _ -> "bad!"
            "#
    )
}

#[mono_test]
fn monomorphized_ints() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            x = 100

            f : U8, U32 -> Nat
            f = \_, _ -> 18

            f x x
        "#
    )
}

#[mono_test]
fn monomorphized_floats() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            x = 100.0

            f : F32, F64 -> Nat
            f = \_, _ -> 18

            f x x
        "#
    )
}

#[mono_test]
#[ignore = "TODO"]
fn monomorphized_ints_aliased() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            y = 100
            w1 = y
            w2 = y

            f = \_, _ -> 1

            f1 : U8, U32 -> Nat
            f1 = f

            f2 : U32, U8 -> Nat
            f2 = f

            f1 w1 w2 + f2 w1 w2
        "#
    )
}

#[mono_test]
fn monomorphized_tag() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            b = \{} -> Bar
            f : [Foo, Bar], [Bar, Baz] -> U8
            f = \_, _ -> 18
            f (b {}) (b {})
        "#
    )
}

#[mono_test]
fn monomorphized_tag_with_aliased_args() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            b = Bool.false
            c = Bool.false
            a = A b c
            f : [A Bool Bool] -> Nat
            f = \_ -> 1
            f a
        "#
    )
}

#[mono_test]
fn monomorphized_list() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            l = \{} -> [1, 2, 3]

            f : List U8, List U16 -> Nat
            f = \_, _ -> 18

            f (l {}) (l {})
        "#
    )
}

#[mono_test]
fn monomorphized_applied_tag() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            a = A "A"
            f = \x ->
                when x is
                    A y -> y
                    B y -> y
            f a
        "#
    )
}

#[mono_test]
#[ignore = "Cannot compile polymorphic closures yet"]
fn aliased_polymorphic_closure() {
    indoc!(
        r#"
        n : U8
        n = 1
        f = \{} -> (\a -> n)
        g = f {}
        g {}
        "#
    )
}

#[mono_test]
fn issue_2535_let_weakened_fields_referenced_in_list() {
    indoc!(
        r#"
        app "test" provides [nums] to "./platform"

        alpha = { a: 1, b: 2 }

        nums : List U8
        nums =
            [
                alpha.a,
                alpha.b,
           ]
        "#
    )
}

#[mono_test]
fn issue_2725_alias_polymorphic_lambda() {
    indoc!(
        r#"
        wrap = \value -> Tag value
        wrapIt = wrap
        wrapIt 42
        "#
    )
}

#[mono_test]
fn issue_2583_specialize_errors_behind_unified_branches() {
    indoc!(
        r#"
        if Bool.true then List.first [] else Str.toI64 ""
        "#
    )
}

#[mono_test]
fn issue_2810() {
    indoc!(
        r#"
        Command : [Command Tool]

        Job : [Job Command]

        Tool : [SystemTool, FromJob Job]

        a : Job
        a = Job (Command (FromJob (Job (Command SystemTool))))
        a
        "#
    )
}

#[mono_test]
fn issue_2811() {
    indoc!(
        r#"
        x = Command { tool: "bash" }
        Command c = x
        c.tool
        "#
    )
}

#[mono_test]
fn specialize_ability_call() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        MHash has
            hash : a -> U64 | a has MHash

        Id := U64 has [MHash {hash}]

        hash : Id -> U64
        hash = \@Id n -> n

        main = hash (@Id 1234)
        "#
    )
}

#[mono_test]
fn opaque_assign_to_symbol() {
    indoc!(
        r#"
        app "test" provides [out] to "./platform"

        Variable := U8

        fromUtf8 : U8 -> Result Variable [InvalidVariableUtf8]
        fromUtf8 = \char ->
            Ok (@Variable char)

        out = fromUtf8 98
        "#
    )
}

#[mono_test]
fn encode() {
    indoc!(
        r#"
        app "test" provides [myU8Bytes] to "./platform"

        MEncoder fmt := List U8, fmt -> List U8 | fmt has Format

        MEncoding has
          toEncoder : val -> MEncoder fmt | val has MEncoding, fmt has Format

        Format has
          u8 : U8 -> MEncoder fmt | fmt has Format


        Linear := {} has [Format {u8}]

        u8 = \n -> @MEncoder (\lst, @Linear {} -> List.append lst n)

        MyU8 := U8 has [MEncoding {toEncoder}]

        toEncoder = \@MyU8 n -> u8 n

        myU8Bytes =
            when toEncoder (@MyU8 15) is
                @MEncoder doEncode -> doEncode [] (@Linear {})
        "#
    )
}

// #[ignore]
// #[mono_test]
// fn static_str_closure() {
//     indoc!(
//         r#"
//         app "test" provides [main] to "./platform"

//         main : Str
//         main =
//             x = "long string that is malloced"

//             f : {} -> Str
//             f = (\_ -> x)

//             f {}
//         "#
//     )
// }

#[mono_test]
fn list_map_closure_borrows() {
    indoc!(
        r#"
        app "test" provides [out] to "./platform"

        list = [Str.concat "lllllllllllllllllllllooooooooooong" "g"]

        example1 = List.map list \string -> Str.repeat string 2

        out =
            when List.get example1 0 is
                Ok s -> s
                Err _ -> "Hello, World!\n"
        "#
    )
}

#[mono_test]
fn list_map_closure_owns() {
    indoc!(
        r#"
        app "test" provides [out] to "./platform"

        list = [Str.concat "lllllllllllllllllllllooooooooooong" "g"]

        example2 = List.map list \string -> Str.concat string "!"

        out =
            when List.get example2 0 is
                Ok s -> s
                Err _ -> "Hello, World!\n"
        "#
    )
}

#[mono_test]
fn list_sort_asc() {
    indoc!(
        r#"
        app "test" provides [out] to "./platform"

        out = List.sortAsc [4, 3, 2, 1]
        "#
    )
}

#[mono_test]
#[ignore]
fn encode_custom_type() {
    indoc!(
        r#"
        app "test"
            imports [Encode.{ toEncoder }, Json]
            provides [main] to "./platform"

        HelloWorld := {}
        toEncoder = \@HelloWorld {} ->
            Encode.custom \bytes, fmt ->
                bytes
                    |> Encode.appendWith (Encode.string "Hello, World!\n") fmt

        main =
            result = Str.fromUtf8 (Encode.toBytes (@HelloWorld {}) Json.toUtf8)
            when result is
                Ok s -> s
                _ -> "<bad>"
        "#
    )
}

#[mono_test]
fn encode_derived_string() {
    indoc!(
        r#"
        app "test"
            imports [Encode.{ toEncoder }, Json]
            provides [main] to "./platform"

        main =
            result = Str.fromUtf8 (Encode.toBytes "abc" Json.toUtf8)
            when result is
                Ok s -> s
                _ -> "<bad>"
        "#
    )
}

#[mono_test]
#[ignore = "TODO"]
fn encode_derived_record() {
    indoc!(
        r#"
        app "test"
            imports [Encode.{ toEncoder }, Json]
            provides [main] to "./platform"

        main =
            result = Str.fromUtf8 (Encode.toBytes {a: "a"} Json.toUtf8)
            when result is
                Ok s -> s
                _ -> "<bad>"
        "#
    )
}

#[mono_test]
fn choose_correct_recursion_var_under_record() {
    indoc!(
        r#"
        Parser : [
            Specialize Parser,
            Record (List {parser: Parser}),
        ]

        printCombinatorParser : Parser -> Str
        printCombinatorParser = \parser ->
            when parser is
                Specialize p ->
                    printed = printCombinatorParser p
                    if Bool.false then printed else "foo"
                Record fields ->
                    fields
                        |> List.map \f ->
                            printed = printCombinatorParser f.parser
                            if Bool.false then printed else "foo"
                        |> List.first
                        |> Result.withDefault ("foo")

        printCombinatorParser (Record [])
        "#
    )
}

#[mono_test]
fn tail_call_elimination() {
    indoc!(
        r#"
        sum = \n, accum ->
            when n is
                0 -> accum
                _ -> sum (n - 1) (n + accum)

        sum 1_000_000 0
        "#
    )
}

#[mono_test]
fn tail_call_with_same_layout_different_lambda_sets() {
    indoc!(
        r#"
        chain = \in, buildLazy ->
            \{} ->
                thunk = buildLazy in
                thunk {}

        chain 1u8 \_ -> chain 1u8 \_ -> (\{} -> "")
        "#
    )
}

#[mono_test]
fn tail_call_with_different_layout() {
    indoc!(
        r#"
        chain = \in, buildLazy ->
            \{} ->
                thunk = buildLazy in
                thunk {}

        chain 1u8 \_ -> chain 1u16 \_ -> (\{} -> "")
        "#
    )
}

#[mono_test]
fn lambda_capture_niche_u8_vs_u64() {
    indoc!(
        r#"
        capture : _ -> ({} -> Str)
        capture = \val ->
            \{} ->
                Num.toStr val

        x : [True, False]
        x = True

        fun =
            when x is
                True -> capture 123u64
                False -> capture 18u8

        fun {}
        "#
    )
}

#[mono_test]
fn lambda_capture_niches_with_other_lambda_capture() {
    indoc!(
        r#"
        capture : a -> ({} -> Str)
        capture = \val ->
            \{} ->
                when val is
                    _ -> ""

        capture2 = \val -> \{} -> "\(val)"

        x : [A, B, C]
        x = A

        fun =
            when x is
                A -> capture {}
                B -> capture2 "foo"
                C -> capture 1u64

        fun {}
        "#
    )
}

#[mono_test]
fn lambda_capture_niches_with_non_capturing_function() {
    indoc!(
        r#"
        capture : a -> ({} -> Str)
        capture = \val ->
            \{} ->
                when val is
                    _ -> ""

        triv = \{} -> ""

        x : [A, B, C]
        x = A

        fun =
            when x is
                A -> capture {}
                B -> triv
                C -> capture 1u64

        fun {}
        "#
    )
}

#[mono_test]
fn lambda_capture_niches_have_captured_function_in_closure() {
    indoc!(
        r#"
        Lazy a : {} -> a

        after : Lazy a, (a -> Lazy b) -> Lazy b
        after = \effect, map ->
            thunk = \{} ->
                when map (effect {}) is
                    b -> b {}
            thunk

        f = \_ -> \_ -> ""
        g = \{ s1 } -> \_ -> s1

        x : [True, False]
        x = True

        fun =
            when x is
                True -> after (\{} -> "") f
                False -> after (\{} -> {s1: "s1"}) g

        fun {}
        "#
    )
}

#[mono_test]
fn lambda_set_niche_same_layout_different_constructor() {
    indoc!(
        r#"
        capture : a -> ({} -> Str)
        capture = \val ->
            thunk =
                \{} ->
                    when val is
                        _ -> ""
            thunk

        x : [True, False]
        x = True

        fun =
            when x is
                True -> capture {a: ""}
                False -> capture (A "")
        fun
        "#
    )
}

#[mono_test]
fn choose_u64_layout() {
    indoc!(
        r#"
        9999999999999999999 + 1
        "#
    )
}

#[mono_test]
fn choose_i128_layout() {
    indoc!(
        r#"
        {
            a: 18446744073709551616 + 1,
            b: -9223372036854775809 + 1,
        }
        "#
    )
}

#[mono_test]
fn choose_u128_layout() {
    indoc!(
        r#"
        170141183460469231731687303715884105728 + 1
        "#
    )
}

#[mono_test]
fn recursive_call_capturing_function() {
    indoc!(
        r#"
        a = \b ->
            c : U32 -> U32
            c = \d ->
                if Bool.true then d else c (d+b)
            c 0

        a 6
        "#
    )
}

#[mono_test]
fn call_function_in_empty_list() {
    indoc!(
        r#"
        lst : List ({} -> {})
        lst = []
        List.map lst \f -> f {}
        "#
    )
}

#[mono_test]
fn call_function_in_empty_list_unbound() {
    indoc!(
        r#"
        lst = []
        List.map lst \f -> f {}
        "#
    )
}

#[mono_test]
fn instantiate_annotated_as_recursive_alias_toplevel() {
    indoc!(
        r#"
        app "test" provides [it] to "./platform"

        Value : [Nil, Array (List Value)]

        foo : [Nil]_
        foo = Nil

        it : Value
        it = foo
        "#
    )
}

#[mono_test]
fn instantiate_annotated_as_recursive_alias_polymorphic_expr() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            Value : [Nil, Array (List Value)]

            foo : [Nil]_
            foo = Nil

            it : Value
            it = foo

            it
        "#
    )
}

#[mono_test]
fn instantiate_annotated_as_recursive_alias_multiple_polymorphic_expr() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            Value : [Nil, Array (List Value)]

            foo : {} -> [Nil]_
            foo = \{} -> Nil

            v1 : Value
            v1 = foo {}

            Value2 : [Nil, B U16, Array (List Value)]

            v2 : Value2
            v2 = foo {}

            {v1, v2}
        "#
    )
}

#[mono_test]
fn encode_derived_record_one_field_string() {
    indoc!(
        r#"
        app "test"
            imports [Encode.{ toEncoder }, Json]
            provides [main] to "./platform"

        main =
            result = Str.fromUtf8 (Encode.toBytes {a: "foo"} Json.toUtf8)
            when result is
                Ok s -> s
                _ -> "<bad>"
        "#
    )
}

#[mono_test]
fn encode_derived_record_two_field_strings() {
    indoc!(
        r#"
        app "test"
            imports [Encode.{ toEncoder }, Json]
            provides [main] to "./platform"

        main =
            result = Str.fromUtf8 (Encode.toBytes {a: "foo", b: "bar"} Json.toUtf8)
            when result is
                Ok s -> s
                _ -> "<bad>"
        "#
    )
}

#[mono_test]
fn encode_derived_nested_record_string() {
    indoc!(
        r#"
        app "test"
            imports [Encode.{ toEncoder }, Json]
            provides [main] to "./platform"

        main =
            result = Str.fromUtf8 (Encode.toBytes {a: {b: "bar"}} Json.toUtf8)
            when result is
                Ok s -> s
                _ -> "<bad>"
        "#
    )
}

#[mono_test]
fn encode_derived_tag_one_field_string() {
    indoc!(
        r#"
        app "test"
            imports [Encode.{ toEncoder }, Json]
            provides [main] to "./platform"

        main =
            x : [A Str]
            x = A "foo"
            result = Str.fromUtf8 (Encode.toBytes x Json.toUtf8)
            when result is
                Ok s -> s
                _ -> "<bad>"
        "#
    )
}

#[mono_test]
fn polymorphic_expression_unification() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        RenderTree : [
            Text Str,
            Indent (List RenderTree),
        ]
        parseFunction : Str -> RenderTree
        parseFunction = \name ->
            last = Indent [Text ".trace(\"\(name)\")" ]
            Indent [last]

        values = parseFunction "interface_header"

        main = values == Text ""
        "#
    )
}

#[mono_test]
fn encode_derived_tag_two_payloads_string() {
    indoc!(
        r#"
        app "test"
            imports [Encode.{ toEncoder }, Json]
            provides [main] to "./platform"

        main =
            x : [A Str Str]
            x = A "foo" "foo"
            result = Str.fromUtf8 (Encode.toBytes x Json.toUtf8)
            when result is
                Ok s -> s
                _ -> "<bad>"
        "#
    )
}

#[mono_test]
fn issue_3560_nested_tag_constructor_is_newtype() {
    indoc!(
        r#"
        when Wrapper (Payload "err") is
            Wrapper (Payload str) -> str
            Wrapper (AlternatePayload str) -> str
        "#
    )
}

#[mono_test]
fn issue_3669() {
    indoc!(
        r#"
        Peano a := [
            Zero,
            Successor (Peano a)
        ]

        unwrap : Peano a -> {}
        unwrap = \@Peano p ->
            when p is
                Zero -> {}
                Successor inner -> unwrap inner

        when unwrap (@Peano Zero) == {} is
            _ -> ""
        "#
    )
}

#[mono_test]
fn num_width_gt_u8_layout_as_float() {
    indoc!(
        r#"
        1 / 200
        "#
    )
}

#[mono_test]
fn match_on_result_with_uninhabited_error_branch() {
    indoc!(
        r#"
        x : Result Str []
        x = Ok "abc"

        when x is
            Ok s -> s
        "#
    )
}

#[mono_test]
fn unreachable_void_constructor() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        x : []

        main = if Bool.true then Ok x else Err "abc"
        "#
    )
}

#[mono_test]
fn unreachable_branch_is_eliminated_but_produces_lambda_specializations() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        provideThunk = \x ->
            when x is
                Ok _ ->
                    t1 = \{} -> "t1"
                    t1
                # During specialization of `main` we specialize this function,
                # which leads to elimination of this branch, because it is unreachable
                # (it can only match the uninhabited type `Err []`).
                #
                # However, naive elimination of this branch would mean we don't traverse
                # the branch body. If we don't do so, we will fail to see and specialize `t2`,
                # which is problematic - while `t2` won't ever be reached in this specialization,
                # it is still part of the lambda set, and `thunk {}` (in main) will match over
                # it before calling.
                #
                # So, this test verifies that we eliminate this branch, but still specialize
                # everything we need.
                Err _ ->
                    t2 = \{} -> "t2"
                    t2

        main =
            x : Result Str []
            x = Ok "abc"

            thunk = provideThunk x

            thunk {}
        "#
    )
}

#[mono_test]
fn match_list() {
    indoc!(
        r#"
        l = [A, B]

        when l is
            [] -> "A"
            [A] -> "B"
            [A, A, ..] -> "C"
            [A, B, ..] -> "D"
            [B, ..] -> "E"
        "#
    )
}

#[mono_test]
fn recursive_function_and_union_with_inference_hole() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        Html state : [
            Element (List (Html state)),
        ]

        translateStatic : Html _ -> Html _
        translateStatic = \node ->
            when node is
                Element children ->
                    newChildren = List.map children translateStatic

                    Element newChildren

        main = when translateStatic (Element []) is
            _ -> ""
        "#
    )
}

#[mono_test]
fn crash() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        getInfallible = \result -> when result is
            Ok x -> x
            _ -> crash "turns out this was fallible"

        main =
            x : [Ok U64, Err Str]
            x = Ok 78
            getInfallible x
        "#
    )
}

#[mono_test]
fn function_pointer_lambda_set() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        number = \{} -> 1u64

        parse = \parser -> parser {}

        main =
            parser = number
            parse parser
        "#
    )
}

#[mono_test]
fn anonymous_closure_lifted_to_named_issue_2403() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            f =
                n = 1
                \{} -> n
            g = f {}
            g
        "#
    )
}

#[mono_test]
fn toplevel_accessor_fn_thunk() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        ra = .field

        main =
            ra { field : 15u8 }
        "#
    )
}

#[mono_test]
fn list_one_vs_one_spread_issue_4685() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main = when [""] is
            [] -> "A"
            [_] -> "B"
            [_, ..] -> "C"
        "#
    )
}

#[mono_test]
fn tuple_pattern_match() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main = when (1, 2) is
            (1, _) -> "A"
            (_, 2) -> "B"
            (_, _) -> "C"
        "#
    )
}

#[mono_test(mode = "test")]
fn issue_4705() {
    indoc!(
        r###"
        interface Test exposes [] imports []

        go : {} -> Bool
        go = \{} -> Bool.true

        expect
            input = {}
            x = go input
            x
        "###
    )
}

#[mono_test(mode = "test")]
fn issue_4749() {
    indoc!(
        r###"
        interface Test exposes [] imports [Json]

        expect
            input = [82, 111, 99]
            got = Decode.fromBytes input Json.fromUtf8
            got == Ok "Roc"
        "###
    )
}

#[mono_test(mode = "test")]
fn lambda_set_with_imported_toplevels_issue_4733() {
    indoc!(
        r###"
        interface Test exposes [] imports []

        fn = \{} ->
            instr : [ Op (U64, U64 -> U64) ]
            instr = if Bool.true then (Op Num.mul) else (Op Num.add)

            Op op = instr

            \a -> op a a

        expect ((fn {}) 3) == 9
        "###
    )
}

#[mono_test]
fn order_list_size_tests_issue_4732() {
    indoc!(
        r###"
        when [] is
            [1, ..]          -> "B1"
            [2, 1, ..]       -> "B2"
            [3, 2, 1, ..]    -> "B3"
            [4, 3, 2, 1, ..] -> "B4"
            _                -> "Catchall"
        "###
    )
}

#[mono_test]
fn anonymous_closure_in_polymorphic_expression_issue_4717() {
    indoc!(
        r###"
        app "test" provides [main] to "platform"

        chompWhile : (List U8) -> (List U8)
        chompWhile = \input ->
                index = List.walkUntil input 0 \i, _ -> Break i

                if index == 0 then
                    input
                else
                    List.drop input index

        main = chompWhile [1u8, 2u8, 3u8]
        "###
    )
}

#[mono_test]
fn list_map_take_capturing_or_noncapturing() {
    indoc!(
        r###"
        app "test" provides [main] to "platform"

        main =
            x = 1u8
            y = 2u8
            f = when "" is
                "A" ->
                    g = \n -> n + x
                    g
                "B" ->
                    h = \n -> n + y
                    h
                _   ->
                    k = \n -> n + n
                    k
            List.map [1u8, 2u8, 3u8] f
        "###
    )
}

#[mono_test]
fn issue_4557() {
    indoc!(
        r###"
        app "test" provides [main] to "./platform"

        isEqQ = \q1, q2 -> when T q1 q2 is
            T (U f1) (U f2) -> Bool.or (isEqQ (U f2) (U f1)) (f1 {} == f2 {})

        main = isEqQ (U \{} -> "a") (U \{} -> "a")
        "###
    )
}

#[mono_test]
fn nullable_wrapped_with_non_nullable_singleton_tags() {
    indoc!(
        r###"
        app "test" provides [main] to "./platform"

        F : [
            A F,
            B,
            C,
        ]

        g : F -> Str
        g = \f -> when f is
                A _ -> "A"
                B -> "B"
                C -> "C"

        main =
            g (A (B))
            |> Str.concat (g B)
            |> Str.concat (g C)
        "###
    )
}

#[mono_test]
fn nullable_wrapped_with_nullable_not_last_index() {
    indoc!(
        r###"
        app "test" provides [main] to "./platform"

        Parser : [
            OneOrMore Parser,
            Keyword Str,
            CharLiteral,
        ]

        toIdParser : Parser -> Str
        toIdParser = \parser ->
            when parser is
                OneOrMore _ -> "a"
                Keyword _ -> "b"
                CharLiteral -> "c"

        main = toIdParser CharLiteral == "c"
        "###
    )
}

#[mono_test]
fn pattern_as_toplevel() {
    indoc!(
        r###"
        app "test" provides [main] to "./platform"

        record = { a: 42i64, b: "foo" }

        main =
            when record is
                { a: 42i64 } as r -> record == r
                _ -> Bool.false
        "###
    )
}

#[mono_test]
fn pattern_as_nested() {
    indoc!(
        r###"
        app "test" provides [main] to "./platform"

        record = { a: 42i64, b: "foo" }

        main =
            when Pair {} record is
                Pair {} ({ a: 42i64 } as r) -> record == r
                _ -> Bool.false
        "###
    )
}

#[mono_test]
fn pattern_as_of_symbol() {
    indoc!(
        r###"
        app "test" provides [main] to "./platform"

        main =
            when "foo" is
                a as b -> a == b
        "###
    )
}

#[mono_test]
fn function_specialization_information_in_lambda_set_thunk() {
    // https://github.com/roc-lang/roc/issues/4734
    // https://rwx.notion.site/Let-generalization-Let-s-not-742a3ab23ff742619129dcc848a271cf#6b08b0a203fb443db2d7238a0eb154eb
    indoc!(
        r###"
        app "test" provides [main] to "./platform"

        andThen = \{} ->
            x = 10
            \newFn -> Num.add (newFn {}) x

        between = andThen {}

        main = between \{} -> between \{} -> 10
        "###
    )
}

#[mono_test]
fn function_specialization_information_in_lambda_set_thunk_independent_defs() {
    // https://github.com/roc-lang/roc/issues/4734
    // https://rwx.notion.site/Let-generalization-Let-s-not-742a3ab23ff742619129dcc848a271cf#6b08b0a203fb443db2d7238a0eb154eb
    indoc!(
        r###"
        app "test" provides [main] to "./platform"

        andThen = \{} ->
            x = 10u8
            \newFn -> Num.add (newFn {}) x

        between1 = andThen {}

        between2 = andThen {}

        main = between1 \{} -> between2 \{} -> 10u8
        "###
    )
}

#[mono_test(mode = "test")]
fn issue_4772_weakened_monomorphic_destructure() {
    indoc!(
        r###"
        interface Test exposes [] imports [Json]

        getNumber =
            { result, rest } = Decode.fromBytesPartial (Str.toUtf8 "-1234") Json.fromUtf8

            when result is
                Ok val ->
                    when Str.toI64 val is
                        Ok number ->
                            Ok {val : number, input : rest}
                        Err InvalidNumStr ->
                            Err (ParsingFailure "not a number")

                Err _ ->
                    Err (ParsingFailure "not a number")

        expect
            result = getNumber
            result == Ok {val : -1234i64, input : []}
        "###
    )
}

#[mono_test]
fn weakening_avoids_overspecialization() {
    // Without weakening of let-bindings, this program would force two specializations of
    // `index` - to `Nat` and the default integer type, `I64`. The test is to ensure only one
    // specialization, that of `Nat`, exists.
    indoc!(
        r###"
        app "test" provides [main] to "./platform"

        main : (List U8) -> (List U8)
        main = \input ->
            index = List.walkUntil input 0 \i, _ -> Break i

            if index == 0 then
                input
            else
                List.drop input index
        "###
    )
}

#[mono_test]
fn recursively_build_effect() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        greeting =
            hi = "Hello"
            name = "World"

            "\(hi), \(name)!"

        main =
            when nestHelp 4 is
                _ -> greeting

        nestHelp : I64 -> XEffect {}
        nestHelp = \m ->
            when m is
                0 ->
                    always {}

                _ ->
                    always {} |> after \_ -> nestHelp (m - 1)


        XEffect a := {} -> a

        always : a -> XEffect a
        always = \x -> @XEffect (\{} -> x)

        after : XEffect a, (a -> XEffect b) -> XEffect b
        after = \(@XEffect e), toB ->
            @XEffect \{} ->
                when toB (e {}) is
                    @XEffect e2 ->
                        e2 {}
        "#
    )
}

#[mono_test]
#[ignore = "roc glue code generation cannot handle a type that this test generates"]
fn recursive_lambda_set_has_nested_non_recursive_lambda_sets_issue_5026() {
    indoc!(
        r#"
        app "test" provides [looper] to "./platform"

        Effect : {} -> Str

        after = \buildNext ->
            afterInner = \{} -> (buildNext "foobar") {}
            afterInner

        await : (Str -> Effect) -> Effect
        await = \cont -> after (\result -> cont result)

        looper = await \_ -> if Bool.true then looper else \{} -> "done"
        "#
    )
}

#[mono_test]
fn unspecialized_lambda_set_unification_keeps_all_concrete_types_without_unification() {
    // This is a regression test for the ambient lambda set specialization algorithm.
    //
    // In the program below, monomorphization of `toEncoderQ` with the `Q` in `main` induces the
    // resolution of `t.a` and `t.b`, and the unification of their pending unspecialization lambda
    // sets, when `t.a` and `t.b` have been resolved to concrete types, but before the
    // specialization procedure steps in to resolve the lambda sets concretely. That's because
    // monomorphization unifies the general type of `toEncoderQ` with the concrete type, forcing
    // concretization of `t`, but the specialization procedure runs only after the unification is
    // complete.
    //
    // In this case, it's imperative that the unspecialized lambda sets of `toEncoder t.a` and
    // `toEncoder t.b` wind up in the same lambda set, that is in
    //
    // tag : @MEncoder (Bytes, Linear -[[] + @MU8:toEncoder:1 + @MStr:toEncoder+1] -> Bytes)
    //       -[lTag]->
    //       @MEncoder (Bytes, Linear -[[Linear:lTag:3 { @MEncoder (Bytes, Linear -[[] + @MU8:toEncoder:1 + @MStr:toEncoder:1] -> Bytes) }]] -> Bytes)
    //
    // rather than forcing the lambda set inside to `tag` to become disjoint, as e.g.
    //
    // tag : @MEncoder (Bytes, Linear -[[] + @MU8:toEncoder:1 + @MStr:toEncoder+1] -> Bytes)
    //       -[lTag]->
    //       @MEncoder (Bytes, Linear -[[
    //                      Linear:lTag:3 { @MEncoder (Bytes, Linear -[[] + @MU8:toEncoder:1] -> Bytes) },
    //                      Linear:lTag:3 { @MEncoder (Bytes, Linear -[[] + @MStr:toEncoder:1] -> Bytes) },
    //                  ]] -> Bytes)
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        MEncoder fmt := List U8, fmt -> List U8 | fmt has Format

        MEncoding has
          toEncoder : val -> MEncoder fmt | val has MEncoding, fmt has Format

        Format has
          u8 : {} -> MEncoder fmt | fmt has Format
          str : {} -> MEncoder fmt | fmt has Format
          tag : MEncoder fmt -> MEncoder fmt | fmt has Format

        Linear := {} has [Format {u8: lU8, str: lStr, tag: lTag}]

        MU8 := U8 has [MEncoding {toEncoder: toEncoderU8}]
        MStr := Str has [MEncoding {toEncoder: toEncoderStr}]

        Q a b := { a: a, b: b }

        lU8 = \{} -> @MEncoder (\lst, @Linear {} -> lst)
        lStr = \{} -> @MEncoder (\lst, @Linear {} -> lst)

        lTag = \@MEncoder doFormat -> @MEncoder (\lst, @Linear {} ->
            doFormat lst (@Linear {})
        )

        toEncoderU8 = \@MU8 _ -> u8 {}

        toEncoderStr = \@MStr _ -> str {}

        toEncoderQ =
            \@Q t -> \fmt ->
                @MEncoder doit = if Bool.true
                    then tag (toEncoder t.a)
                    else tag (toEncoder t.b)

                doit [] fmt

        main =
            fmt = toEncoderQ (@Q {a : @MStr "", b: @MU8 7})
            fmt (@Linear {})
        "#
    )
}

#[mono_test]
fn unspecialized_lambda_set_unification_keeps_all_concrete_types_without_unification_of_unifiable()
{
    // This is a regression test for the ambient lambda set specialization algorithm.
    //
    // The principle of the test is equivalent to that of `unspecialized_lambda_set_unification_keeps_all_concrete_types_without_unification`.
    //
    // However, this test requires a larger reproduction because it is negative behavior is only
    // visible in the presence of builtin ability usage (in this case, `Encoding` and
    // `EncoderFormatting`).
    //
    // In this test, the payload types `[A]*` and `[B]*` of the encoded type `Q` are unifiable in
    // their unspecialized lambda set representations under `toEncoderQ`; however, they must not
    // be, because they in fact represent to different specializations of needed encoders. In
    // particular, the lambda set `[[] + [A]:toEncoder:1 + [B]:toEncoder:1]` must be preserved,
    // rather than collapsing to `[[] + [A, B]:toEncoder:1]`.
    indoc!(
        r#"
        app "test" imports [Json] provides [main] to "./platform"

        Q a b := { a: a, b: b } has [Encoding {toEncoder: toEncoderQ}]

        toEncoderQ =
            \@Q t -> Encode.custom \bytes, fmt ->
                f = if Bool.true
                    then Encode.tag "A" [Encode.toEncoder t.a]
                    else Encode.tag "B" [Encode.toEncoder t.b]

                Encode.appendWith bytes f fmt

        accessor = @Q {a : A, b: B}

        main =
            Encode.toBytes accessor Json.toUtf8
        "#
    )
}

#[mono_test]
fn unspecialized_lambda_set_unification_does_not_duplicate_identical_concrete_types() {
    // This is a regression test for the ambient lambda set specialization algorithm.
    //
    // The principle of the test is equivalent to that of `unspecialized_lambda_set_unification_keeps_all_concrete_types_without_unification`.
    //
    // However, this test requires a larger reproduction because it is negative behavior is only
    // visible in the presence of builtin ability usage (in this case, `Encoding` and
    // `EncoderFormatting`).
    //
    // In this test, the payload types `Str` and `Str` of the encoded type `Q` are unifiable in
    // their unspecialized lambda set representations under `toEncoderQ`, and moreoever they are
    // equivalent specializations, since they both come from the same root variable `x`. In as
    // such, the lambda set `[[] + Str:toEncoder:1]` should be produced during compaction, rather
    // than staying as the expanded `[[] + Str:toEncoder:1 + Str:toEncoder:1]` after the types of
    // `t.a` and `t.b` are filled in.
    indoc!(
        r#"
        app "test" imports [Json] provides [main] to "./platform"

        Q a b := { a: a, b: b } has [Encoding {toEncoder: toEncoderQ}]

        toEncoderQ =
            \@Q t -> Encode.custom \bytes, fmt ->
                f = if Bool.true
                    then Encode.tag "A" [Encode.toEncoder t.a]
                    else Encode.tag "B" [Encode.toEncoder t.b]

                Encode.appendWith bytes f fmt

        accessor =
            x = ""
            @Q {a : x, b: x}

        main =
            Encode.toBytes accessor Json.toUtf8
        "#
    )
}

#[mono_test]
fn inline_return_joinpoints_in_bool_lambda_set() {
    indoc!(
        r#"
        app "test" provides [f] to "./platform"

        f = \x ->
            caller = if Bool.false then f else \n -> n
            caller (x + 1)
        "#
    )
}

#[mono_test]
fn inline_return_joinpoints_in_enum_lambda_set() {
    indoc!(
        r#"
        app "test" provides [f] to "./platform"

        f = \x ->
            caller = \t -> when t is
                A -> f
                B -> \n -> n
                C -> \n -> n + 1
                D -> \n -> n + 2
            (caller A) (x + 1)
        "#
    )
}

#[mono_test]
fn inline_return_joinpoints_in_union_lambda_set() {
    indoc!(
        r#"
        app "test" provides [f] to "./platform"

        f = \x ->
            caller = \t -> when t is
                A -> f
                B -> \n -> n + x
            (caller A) (x + 1)
        "#
    )
}

#[mono_test]
fn recursive_closure_with_transiently_used_capture() {
    indoc!(
        r#"
        app "test" provides [f] to "./platform"

        thenDo = \x, callback ->
            callback x

        f = \{} ->
            code = 10u16

            bf = \{} ->
                thenDo code \_ -> bf {}

            bf {}
        "#
    )
}

#[mono_test]
fn when_guard_appears_multiple_times_in_compiled_decision_tree_issue_5176() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        go : U8 -> U8
        go = \byte ->
            when byte is
                15 if Bool.true -> 1
                b if Bool.true -> b + 2
                _ -> 3

        main = go '.'
        "#
    )
}

#[mono_test]
fn recursive_lambda_set_resolved_only_upon_specialization() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        factCPS = \n, cont ->
            if n == 0u8 then
                cont 1u8
            else
                factCPS (n - 1) \value -> cont (n * value)

        main =
            factCPS 5 \x -> x
        "#
    )
}

#[mono_test]
fn compose_recursive_lambda_set_productive_nullable_wrapped() {
    indoc!(
        r#"
         app "test" provides [main] to "./platform"

         compose = \forward -> \f, g ->
            if forward
            then \x -> g (f x)
            else \x -> f (g x)

         identity = \x -> x
         exclame = \s -> "\(s)!"
         whisper = \s -> "(\(s))"

         main =
             res: Str -> Str
             res = List.walk [ exclame, whisper ] identity (compose Bool.true)
             res "hello"
         "#
    )
}

#[mono_test]
fn issue_4759() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        main =
            update { a : { x : "x", y: "y" } }

        update = \state -> { state & a : { x : "ux", y: "uy" } }
        "#
    )
}

#[mono_test]
fn layout_cache_structure_with_multiple_recursive_structures() {
    indoc!(
        r#"
        app "test" provides [main] to "./platform"

        Chain : [
            End,
            Link Chain,
        ]

        LinkedList : [Nil, Cons { first : Chain, rest : LinkedList }]

        main =
            base : LinkedList 
            base = Nil

            walker : LinkedList, Chain -> LinkedList
            walker = \rest, first -> Cons { first, rest } 

            list : List Chain
            list = []

            r = List.walk list base walker

            r
        "#
    )
}