nomsu/core/coroutines.nom

#!/usr/bin/env nomsu -V6.14
#
    This file defines the code that creates and manipulates coroutines
    
use "core/metaprogramming.nom"
use "core/operators.nom"
use "core/control_flow.nom"

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

test:
    $co =
        ->:
            yield 4
            yield 5
            repeat 3 times:
                yield 6
    $nums = []
    for $ in coroutine $co:
        $nums, add $
    
    unless ($nums == [4, 5, 6, 6, 6]):
        fail "Coroutine iteration failed"
    
    $d = {.x = 0}
    $co2 =
        coroutine:
            $d.x += 1
            yield 1
            $d.x += 1
            yield
            $d.x += 1
    repeat while ((coroutine status of $co2) != "dead"): resume $co2
    assume $d.x == 3
(coroutine $body) parses as (coroutine from (-> $body))
(for $ in coroutine $co $body) compiles to ("
    for \($ as lua expr) in coroutine_wrap(\($co as lua expr)) do
        \($body as lua)
    end
")