568a44ef19
return values with either an explicit "return" statement or if they're the only line in the function, and the line is an expression.
64 lines
2.6 KiB
Plaintext
64 lines
2.6 KiB
Plaintext
require "lib/metaprogramming.nom"
|
|
require "lib/utils.nom"
|
|
require "lib/control_flow.nom"
|
|
require "lib/operators.nom"
|
|
require "lib/collections.nom"
|
|
|
|
|
|
compile [say %str] to:
|
|
"nomsu:writeln(\(%str as lua))" if ((%str's "type") == "String")
|
|
..else "nomsu:writeln(nomsu:stringify(\(%str as lua)))"
|
|
|
|
compile [do %action] to code:
|
|
(%action as lua statements) if ((%action's "type") == "Thunk")
|
|
..else "(\(%action as lua))(nomsu, vars);"
|
|
|
|
# With statement
|
|
compile [with %assignments %action] to code:
|
|
%data = []
|
|
for %i = %assignment in (%assignments' "value"):
|
|
%tokens = (%assignment's "value")
|
|
%var = (%tokens -> 1)
|
|
%eq = (%tokens -> 2)
|
|
assert (=lua "\%eq and \%eq.type == 'Word' and \%eq.value == '='") ".."
|
|
Invalid format for 'with' statement. List entries must have the form %var = (value)
|
|
%value = (%tokens -> 3)
|
|
add {i=%i, var=%var, value=%value} to %data
|
|
%setup = (..)
|
|
join (..)
|
|
"local old_value\(%->"i") = \((%->"var") as lua); \((%->"var") as lua) = \((%->"value") as lua);"
|
|
..for all %data
|
|
..with glue "\n "
|
|
return ".."
|
|
do
|
|
\%setup
|
|
local fell_through = false;
|
|
local ok, ret1, ret2 = pcall(function(nomsu, vars)
|
|
\(%action as lua statements);
|
|
fell_through = true;
|
|
end, nomsu, vars);
|
|
\(join ("\((%->"var") as lua) = old_value\(%->"i");" for all %data) with glue "\n ")
|
|
if not ok then nomsu:error(ret1); end
|
|
if not fell_through then
|
|
return ret1, ret2;
|
|
end
|
|
end
|
|
parse [with %thing = %value %action] as: with [%thing = %value] %action
|
|
|
|
# Any/all/none
|
|
compile [all of %items, all %items] to:
|
|
"(\(join ((% as lua) for all (%items' "value")) with glue " and "))"
|
|
..if (%items' "type") == "List" else "nomsu.utils.all(\(%items as lua))"
|
|
parse [not all of %items, not all %items] as: not (all of %items)
|
|
compile [any of %items, any %items] to:
|
|
"(\(join ((% as lua) for all (%items' "value")) with glue " or "))"
|
|
..if (%items' "type") == "List" else "nomsu.utils.any(\(%items as lua))"
|
|
parse [none of %items, none %items] as: not (any of %items)
|
|
|
|
compile [sum of %items, sum %items] to:
|
|
"(\(join ((% as lua) for all (%items' "value")) with glue " + "))"
|
|
..if (%items' "type") == "List" else "nomsu.utils.sum(\(%items as lua))"
|
|
compile [product of %items, product %items] to:
|
|
"(\(join ((% as lua) for all (%items' "value")) with glue " * "))"
|
|
..if (%items' "type") == "List" else "nomsu.utils.product(\(%items as lua))"
|