defmodule Orb do
@moduledoc """
Write WebAssembly modules with Elixir.
WebAssembly is a low-level language. You work with integers and floats, can perform operations on them like adding or multiplication, and then read and write those values to a block of memory. There’s no concept of a “string” or an “array”, let alone a “hash map” or “HTTP request”.
That’s where a library like Orb can help out. It takes full advantage of Elixir’s language features by becoming a compiler for WebAssembly. You can define WebAssembly modules in Elixir for “string” or “hash map”, and compose them together into a final module.
That WebAssembly module can then run in every major application environment: browsers, servers, the edge, and mobile devices like phones, tablets & laptops. This story is still being developed, but I believe like other web standards like JSON, HTML, and HTTP, that WebAssembly will become a first-class citizen on any platform. It’s Turing-complete, designed to be backwards compatible, fast, and works almost everywhere.
## Example
Let’s create a module that calculates the average of a set of numbers.
WebAssembly modules can have state. Here will have two pieces of state: a total `count` and a running `tally`. These are stored as **globals**. (If you are familiar with object-oriented programming, you can think of them as instance variables).
Our module will export two functions: `insert` and `calculate_mean`. These two functions will work with the `count` and `tally` globals.
```elixir
defmodule CalculateMean do
use Orb
I32.global(
count: 0,
tally: 0
)
wasm do
func insert(element: I32) do
@count = @count + 1
@tally = @tally + element
end
func calculate_mean(), I32 do
@tally / @count
end
end
end
```
One thing you’ll notice is that we must specify the type of function parameters and return values. Our `insert` function accepts a 32-bit integer, denoted using `I32`. It returns no value, while `calculate_mean` is annotated to return a 32-bit integer.
We get to write math with the intuitive `+` and `/` operators. Let’s see the same module without the magic: no math operators and without `@` conveniences for working with globals:
```elixir
defmodule CalculateMean do
use Orb
I32.global(
count: 0,
tally: 0
)
wasm do
func insert(element: I32) do
I32.add(global_get(:count), 1)
global_set(:count)
I32.add(global_get(:tally), element)
global_set(:tally)
end
func calculate_mean(), I32 do
I32.div_u(global_get(:tally), global_get(:count))
end
end
end
```
This is the exact same logic as before. In fact, this is what the first version expands to. Orb adds “sugar syntax” to make authoring WebAssembly nicer, to make it feel like writing Elixir or Ruby.
## Functions
In Elixir you define functions publicly available outside the module with `def/1`, and functions private to the module with `defp/1`. Orb follows the same suffix convention with `func/2` and `funcp/2`.
Consumers of your WebAssembly module will only be able to call exported functions defined using `func/2`. Making a function public in WebAssembly is known as “exporting”.
## Stack based
While it looks like Elixir, there are some key differences between it and programs written in Orb. The first is that state is mutable. While immutability is one of the best features of Elixir, in WebAssembly variables are mutable because raw computer memory is mutable.
The second key difference is that WebAssembly is stack based. Every function has an implicit stack of values that you can push and pop from. This paradigm allows WebAssembly runtimes to efficiently optimize for raw CPU registers whilst not being platform specific.
In Elixir when you write:
```elixir
def example() do
1
2
3
end
```
The first two lines with `1` and `2` are inert — they have no effect — and the result from the function is the last line `3`.
In WebAssembly / Orb when you write the same sort of thing:
```elixir
wasm do
func example() do
1
2
3
end
end
```
Then what’s happening is that we are pushing `1` onto the stack, then `2`, and then `3`. Now the stack has three items on it. Which will become our return value: a tuple of 3 integers. (Our function has no return type specified, so this will be an error if you attempted to compile the resulting module).
So the correct return type from this function would be a tuple of three integers:
```elixir
wasm do
func example(), {I32, I32, I32} do
1
2
3
end
end
```
If you prefer, Orb allows you to be explicit with your stack pushes with `Orb.DSL.push/1`:
```elixir
wasm do
func example(), {I32, I32, I32} do
push(1)
push(2)
push(3)
end
end
```
You can use the stack to unlock novel patterns, but for the most part Orb avoids the need to interact with it. It’s just something to keep in mind if you are used to lines of code with simple values not having any side effects.
## Locals
Locals are variables that live for the lifetime of a function. They must be specified upfront with their type alongside the function’s definition, and are initialized to zero.
Here we have two locals: `under?` and `over?`, both 32-bit integers. We can set their value and then read them again at the bottom of the function.
```elixir
defmodule WithinRange do
use Orb
wasm do
func validate(num: I32), I32, under?: I32, over?: I32 do
under? = num < 1
over? = num > 255
not (under? or over?)
end
end
end
```
## Globals
Globals are like locals, but live for the duration of the entire running module’s life. Their initial type and value are specified upfront.
Globals by default are internal: nothing outside the module can see them. They can be exported to expose them to the outside world.
When exporting a global you decide if it is `:readonly` or `:mutable`. Internal globals are mutable by default.
```elixir
I32.global(some_internal_global: 99)
I32.global(:readonly, some_internal_global: 99)
I32.export_global(:readonly, some_public_constant: 1001)
I32.export_global(:mutable, some_public_variable: 42)
# You can define multiple globals at once:
I32.global(magic_number_a: 99, magic_number_b: 12, magic_number_c: -5)
```
You can read or write to a global using the `@` prefix:
```elixir
defmodule Counter do
use Orb
I32.global(counter: 0)
wasm do
func increment() do
@counter = @counter + 1
end
end
end
```
## Memory
WebAssembly provides a buffer of memory when you need more than a handful global integers or floats. This is a contiguous array of random-access memory which you can freely read and write to.
### Pages
WebAssembly Memory comes in 64 KiB segments called pages. You use some multiple of these 64 KiB (64 * 1024 = 65,536 bytes) pages.
By default your module will have **no** memory, so you must specify how much memory you want upfront.
Here’s an example with 16 pages (1 MiB) of memory:
```elixir
defmodule Example do
use Orb
Memory.pages(16)
end
```
### Reading & writing memory
To read from memory, you can use the `Memory.load/2` function. This loads a value at the given memory address. Addresses are themselves 32-bit integers. This mean you can perform pointer arithmetic to calculate whatever address you need to access.
However, this can prove unsafe as it’s easy to calculate the wrong address and corrupt your memory. For this reason, Orb provides higher level constructs for making working with memory pointers more pleasant, which are detailed later on.
```elixir
defmodule Example do
use Orb
Memory.pages(1)
wasm do
func get_int32(), I32 do
Memory.load!(I32, 0x100)
end
func set_int32(value: I32) do
Memory.store!(I32, 0x100, value)
end
end
end
```
### Initializing memory with data
You can populate the initial memory of your module using `Orb.Memory.initial_data/1`. This accepts an memory offset and the string to write there.
```elixir
defmodule MimeTypeDataExample do
use Orb
Memory.pages(1)
wasm do
Memory.initial_data(offset: 0x100, string: "text/html")
Memory.initial_data(offset: 0x200, string: \"""
<!doctype html>
<meta charset=utf-8>
<h1>Hello world</h1>
\""")
func get_mime_type(), I32 do
0x100
end
func get_body(), I32 do
0x200
end
end
end
```
Having to manually allocate and remember each memory offset is a pain, so Orb provides conveniences which are detailed in the next section.
## Strings constants
You can use constant strings with the `~S` sigil. These will be extracted as initial data definitions at the start of the WebAssembly module, and their memory offsets substituted in their place.
Each string is packed together for maximum efficiency of memory space. Strings are deduplicated, so you can use the same string constant multiple times and a single allocation will be made.
String constants in Orb are nul-terminated.
```elixir
defmodule MimeTypeStringExample do
use Orb
Memory.pages(1)
wasm do
func get_mime_type(), I32 do
~S"text/html"
end
func get_body(), I32 do
~S\"""
<!doctype html>
<meta charset=utf-8>
<h1>Hello world</h1>
\"""
end
end
end
```
## Control flow
Orb supports control flow with `if`, `block`, and `loop` statements.
### If statements
If you want to run logic conditionally, use an `if` statement.
```elixir
if @party_mode? do
music_volume = 100
end
```
You can add an `else` clause:
```elixir
if @party_mode? do
music_volume = 100
else
music_volume = 30
end
```
If you want a ternary operator (e.g. to map from one value to another), you can use `Orb.I32.when?/2` instead:
```elixir
music_volume = I32.when? @party_mode? do
100
else
30
end
```
These can be written on single line too:
```elixir
music_volume = I32.when?(@party_mode?, do: 100, else: 30)
```
### Loops
Loops look like the familiar construct in other languages like JavaScript, with two key differences: each loop has a name, and loops by default stop unless you explicitly tell them to continue.
```elixir
i = 0
loop CountUp do
i = i + 1
CountUp.continue(if: i < 10)
end
```
Each loop is named, so if you nest them you can specify which particular one to continue.
```elixir
total_weeks = 10
weekday_count = 7
week = 0
weekday = 0
loop Weeks do
loop Weekdays do
# Do something here with week and weekday
weekday = weekday + 1
Weekdays.continue(if: weekday < weekday_count)
end
week = week + 1
Weeks.continue(if: week < total_weeks)
end
```
#### Iterators
Iterators are an upcoming feature, currently part of SilverOrb that will hopefully become part of Orb itself.
### Blocks
Blocks provide a structured way to skip code.
```elixir
defblock Validate do
break(Validate, if: i < 0)
# Do something with i
end
```
Blocks can have a type.
```elixir
defblock Double, I32 do
if i < 0 do
push(0)
break(Double)
end
push(i * 2)
end
```
## Calling other functions
You can `Orb.DSL.call/1` other functions defined within your module. Currently, the parameters and return type are not checked, so you must ensure you are calling with the correct arity and types.
```elixir
char = call(:encode_html_char, char)
```
## Composing modules together
Any functions from one module can be included into another to allow code reuse.
When you `use Orb`, `funcp` and `func` functions are defined on your Elixir module for you. Calling these from another module will copy any functions across.
```elixir
defmodule A do
use Orb
wasm do
func square(n: I32), I32 do
n * n
end
end
end
```
```elixir
defmodule B do
use Orb
# Copies all functions defined in A as private functions into this module.
A.funcp()
wasm do
func example(n: I32), I32 do
call(:square, 42)
end
end
end
```
You can pass a name to `YourSourceModule.funcp(name)` to only copy that particular function across.
```elixir
defmodule A do
use Orb
wasm do
func square(n: I32), I32 do
n * n
end
func double(n: I32), I32 do
2 * n
end
end
end
```
```elixir
defmodule B do
use Orb
A.funcp(:square)
wasm do
func example(n: I32), I32 do
call(:square, 42)
end
end
end
```
## Importing
Your running WebAssembly module can interact with the outside world by importing globals and functions.
## Use Elixir features
- Piping
- Module attributes
- Inline for
### Custom types with `Access`
TODO: extract this into its own section.
## Define your own functions and macros
## Hex packages
- GoldenOrb
- String builder
- SilverOrb
## Running your module
"""
alias Orb.Ops
alias Orb.Memory
require Ops
defmacro __using__(opts) do
inline? = Keyword.get(opts, :inline, false)
attrs =
case inline? do
true ->
nil
false ->
quote do
@before_compile unquote(__MODULE__).BeforeCompile
end
end
quote do
import Orb
alias Orb.{I32, I64, S32, U32, F32, Memory}
require Orb.{I32, Memory}
# @wasm_name __MODULE__ |> Module.split() |> List.last()
# @before_compile {unquote(__MODULE__), :register_attributes}
# unless unquote(inline?) do
# @before_compile unquote(__MODULE__).BeforeCompile
# end
unquote(attrs)
if Module.open?(__MODULE__) do
# @before_compile unquote(__MODULE__).BeforeCompile
# Module.put_attribute(__MODULE__, :before_compile, unquote(__MODULE__).BeforeCompile)
Module.put_attribute(__MODULE__, :wasm_name, __MODULE__ |> Module.split() |> List.last())
Module.register_attribute(__MODULE__, :wasm_memory, accumulate: true)
Module.register_attribute(__MODULE__, :wasm_globals, accumulate: true)
Module.register_attribute(__MODULE__, :wasm_imports, accumulate: true)
Module.register_attribute(__MODULE__, :wasm_body, accumulate: true)
Module.register_attribute(__MODULE__, :wasm_constants, accumulate: true)
end
end
end
# TODO: break up into multiple modules. Perhaps add/2 etc can be put on Orb.I32.DSL?
defmodule I32 do
@moduledoc """
Type for 32-bit integer.
"""
import Kernel, except: [and: 2, or: 2]
require Ops
def wasm_type(), do: :i32
def add(a, b)
def sub(a, b)
def mul(a, b)
def div_u(a, divisor)
def div_s(a, divisor)
def rem_u(a, divisor)
def rem_s(a, divisor)
def unquote(:or)(a, b)
def xor(a, b)
def shl(a, b)
def shr_u(a, b)
def shr_s(a, b)
def rotl(a, b)
def rotr(a, b)
# def store(offset, i32)
# def store8(offset, i8)
for op <- Ops.i32(1) do
def unquote(op)(a) do
{:i32, unquote(op), a}
end
end
for op <- Ops.i32(2) do
case op do
:eq ->
def eq(0, n), do: {:i32, :eqz, n}
def eq(n, 0), do: {:i32, :eqz, n}
def eq(a, b), do: {:i32, :eq, {a, b}}
:and ->
def band(a, b) do
{:i32, :and, {a, b}}
end
_ ->
def unquote(op)(a, b) do
{:i32, unquote(op), {a, b}}
end
end
end
# for op <- Ops.i32(:load) do
# def unquote(op)(offset) do
# {:i32, unquote(op), offset}
# end
# end
# for op <- Ops.i32(:store) do
# def unquote(op)(offset, value) do
# {:i32, unquote(op), offset, value}
# end
# end
def memory8!(offset) do
%{
unsigned: {:i32, :load8_u, offset},
signed: {:i32, :load8_s, offset}
}
end
# Replaced by |||
# defp _or(a, b), do: {:i32, :or, {a, b}}
def sum!(items) when is_list(items) do
Enum.reduce(items, &add/2)
end
def in_inclusive_range?(value, lower, upper) do
{:i32, :and, {I32.ge_u(value, lower), I32.le_u(value, upper)}}
end
def in?(value, list) when is_list(list) do
for {item, index} <- Enum.with_index(list) do
case index do
0 ->
eq(value, item)
_ ->
[eq(value, item), {:i32, :or}]
end
end
end
defmacro when?(condition, do: when_true, else: when_false) do
quote do
Orb.IfElse.new(
:i32,
unquote(condition),
unquote(__get_block_items(when_true)),
unquote(__get_block_items(when_false))
)
end
end
# This only works with WebAssembly 1.1
# Sadly wat2wasm doesn’t like it
def select(condition, do: when_true, else: when_false) do
[
when_true,
when_false,
condition,
:select
]
end
# TODO: remove?
def eqz?(value, do: when_true, else: when_false) do
Orb.IfElse.new(:i32, eqz(value), when_true, when_false)
end
def calculate_enum(cases) do
Map.new(Enum.with_index(cases), fn {key, index} -> {key, {:i32_const, index}} end)
end
def from_4_byte_ascii(<<int::little-size(32)>>), do: int
defmacro match(value, do: transform) do
statements =
for {:->, _, [input, result]} <- transform do
case input do
# _ ->
# like an else clause
[{:_, _, _}] ->
__get_block_items(result)
[match] ->
quote do
%Orb.IfElse{
condition: I32.eq(unquote(value), unquote(match)),
when_true: [unquote(__get_block_items(result)), break(:i32_match)]
}
end
matches ->
quote do
%Orb.IfElse{
condition: I32.in?(unquote(value), unquote(matches)),
when_true: [unquote(__get_block_items(result)), break(:i32_match)]
}
end
end
end
# catchall = for {:->, _, [[{:_, _, _}], _]} <- transform, do: true
has_catchall? = Enum.any?(transform, &match?({:->, _, [[{:_, _, _}], _]}, &1))
final_instruction =
case has_catchall? do
false -> :unreachable
true -> []
end
quote do
defblock :i32_match, result: I32 do
unquote(statements)
unquote(final_instruction)
end
end
end
defmacro cond(do: transform) do
statements =
for {:->, _, [input, target]} <- transform do
case input do
# true ->
# like an else clause
[true] ->
target
[match] ->
quote do
%Orb.IfElse{
condition: unquote(match),
when_true: [unquote(__get_block_items(target)), break(:i32_map)]
}
end
end
end
catchall = for {:->, _, [[true], _]} <- transform, do: true
final_instruction =
case catchall do
[] -> :unreachable
[true] -> []
end
quote do
defblock :i32_map, result: I32 do
unquote(statements)
unquote(final_instruction)
end
end
end
defp __get_block_items(block) do
case block do
nil -> nil
{:__block__, _meta, block_items} -> block_items
single -> [single]
end
end
def __global_value(value) when is_integer(value), do: Orb.DSL.i32(value)
def __global_value(false), do: Orb.DSL.i32(false)
def __global_value(true), do: Orb.DSL.i32(true)
# TODO: stash away which module so we can do smart stuff like with local types
def __global_value(mod) when is_atom(mod), do: mod.initial_i32() |> Orb.DSL.i32()
defmacro global(mutability \\ :mutable, list)
when mutability in ~w{readonly mutable}a do
quote do
@wasm_globals (for {key, value} <- unquote(list) do
Orb.Global.new(
:i32,
key,
unquote(mutability),
:internal,
Orb.I32.__global_value(value)
)
end)
end
end
defmacro export_global(mutability, list)
when mutability in ~w{readonly mutable}a do
quote do
@wasm_globals (for {key, value} <- unquote(list) do
Orb.Global.new(
:i32,
key,
unquote(mutability),
:exported,
Orb.I32.__global_value(value)
)
end)
end
end
defmacro export_enum(keys, offset \\ 0) do
quote do
unquote(__MODULE__).export_global(
:readonly,
Enum.with_index(unquote(keys), unquote(offset))
)
end
end
defmacro enum(keys, offset \\ 0) do
quote do
unquote(__MODULE__).global(:readonly, Enum.with_index(unquote(keys), unquote(offset)))
end
end
defmacro attr_writer(global_name) when is_atom(global_name) do
quote do
func unquote(String.to_atom("#{global_name}="))(new_value: I32) do
local_get(:new_value)
global_set(unquote(global_name))
end
end
end
defmacro attr_writer(global_name, as: func_name)
when is_atom(global_name) |> Kernel.and(is_atom(func_name)) do
quote do
func unquote(func_name)(new_value: I32) do
local_get(:new_value)
global_set(unquote(global_name))
end
end
end
end
# TODO: extract?
defmodule VariableReference do
@moduledoc false
defstruct [:global_or_local, :identifier, :type]
def global(identifier, type) do
%__MODULE__{global_or_local: :global, identifier: identifier, type: type}
end
def local(identifier, type) do
%__MODULE__{global_or_local: :local, identifier: identifier, type: type}
end
def as_set(%__MODULE__{global_or_local: :local, identifier: identifier}) do
{:local_set, identifier}
end
@behaviour Access
@impl Access
def fetch(%__MODULE__{global_or_local: :local, identifier: identifier, type: :i32} = ref,
at: offset
) do
ast = {:i32, :load, {:i32, :add, {ref, offset}}}
{:ok, ast}
end
def fetch(
%__MODULE__{global_or_local: :local, identifier: identifier, type: mod} = ref,
key
) do
mod.fetch(ref, key)
end
defimpl Orb.ToWat do
def to_wat(%VariableReference{global_or_local: :global, identifier: identifier}, indent) do
[indent, "(global.get $", to_string(identifier), ?)]
end
def to_wat(%VariableReference{global_or_local: :local, identifier: identifier}, indent) do
[indent, "(local.get $", to_string(identifier), ?)]
end
end
end
defp do_module_body(block, options, env, env_module) do
# TODO split into readonly_globals and mutable_globals?
internal_global_types = Keyword.get(options, :globals, [])
# TODO rename to export_readonly_globals?
exported_global_types = Keyword.get(options, :exported_globals, [])
exported_mutable_global_types = Keyword.get(options, :exported_mutable_globals, [])
internal_global_types =
internal_global_types ++
List.flatten(List.wrap(Module.get_attribute(env_module, :wasm_global)))
# dbg(env_module)
# dbg(Module.get_attribute(env_module, :wasm_global))
globals =
(internal_global_types ++ exported_global_types ++ exported_mutable_global_types)
|> Keyword.new(fn {key, _} -> {key, nil} end)
|> Map.new()
block = interpolate_external_values(block, env)
block_items =
case block do
{:__block__, _meta, block_items} -> block_items
single -> List.wrap(single)
end
# block_items = Macro.expand(block_items, env)
# block_items = block_items
{block_items, constants} =
Macro.prewalk(block_items, [], fn
# TODO: remove, replaced by @global_name =
{:=, _meta1, [{global, _meta2, nil}, input]}, constants
when is_atom(global) and is_map_key(globals, global) ->
{[input, Orb.DSL.global_set(global)], constants}
{atom, _meta, nil}, constants when is_atom(atom) and is_map_key(globals, atom) ->
{quote(do: Orb.VariableReference.global(unquote(atom), unquote(globals[atom]))),
constants}
{:const, _, [str]}, constants when is_binary(str) ->
{quote(do: Orb.__data_for_constant(unquote(str))), [str | constants]}
{:sigil_S, _, [{:<<>>, _, [str]}, _]}, constants ->
{
quote(do: Orb.__data_for_constant(unquote(str))),
[str | constants]
}
# FIXME: have to decide whether supporting ~s and interpolation is too hard.
{:sigil_s, _, [{:<<>>, _, [str]}, _]}, constants ->
{
quote(do: Orb.__data_for_constant(unquote(str))),
[str | constants]
}
# {quote(do: Orb.__data_for_constant(unquote(str))), [str | constants]}
other, constants ->
{other, constants}
end)
constants = Enum.reverse(constants)
%{
body: block_items,
constants: constants
}
end
defmodule BeforeCompile do
@moduledoc false
defmacro __before_compile__(_env) do
quote do
def __wasm_module__() do
Orb.ModuleDefinition.new(
name: @wasm_name,
imports: @wasm_imports |> Enum.reverse() |> List.flatten(),
globals: @wasm_globals |> Enum.reverse() |> List.flatten(),
memory: Memory.from(@wasm_memory),
constants: Orb.Constants.from_attribute(@wasm_constants),
body: @wasm_body |> Enum.reverse() |> List.flatten()
)
end
# def func(),
# do: Orb.ModuleDefinition.func_ref_all!(__MODULE__)
def _func(name),
do: Orb.ModuleDefinition.func_ref!(__MODULE__, name)
@doc "Import all WebAssembly functions from this module’s Orb definition."
def funcp(),
do: Orb.ModuleDefinition.funcp_ref_all!(__MODULE__)
@doc "Import a specific WebAssembly function from this module’s Orb definition."
def funcp(name),
do: Orb.ModuleDefinition.funcp_ref!(__MODULE__, name)
@doc "Convert this module’s Orb definition to WebAssembly text (Wat) format."
def to_wat(), do: Orb.to_wat(__wasm_module__())
end
end
end
defmacro __data_for_constant(value) do
quote do
@wasm_constants
|> Orb.Constants.from_attribute()
|> Orb.Constants.to_map()
|> Orb.Constants.resolve(unquote(value))
end
end
defp mode_pre(mode) do
dsl =
case mode do
Orb.S32 ->
quote do
import Orb.I32.DSL
import Orb.S32.DSL
import Orb.Global.DSL
end
Orb.U32 ->
quote do
import Orb.I32.DSL
import Orb.U32.DSL
import Orb.Global.DSL
end
Orb.F32 ->
quote do
import Orb.F32.DSL
import Orb.Global.DSL
end
:no_magic ->
[]
end
quote do
import Kernel,
except: [
if: 2,
@: 1,
+: 2,
-: 2,
*: 2,
/: 2,
<: 2,
>: 2,
<=: 2,
>=: 2,
===: 2,
!==: 2,
not: 1,
or: 2
]
# TODO: should this be omitted if :no_magic is passed?
import Orb.IfElse.DSL
unquote(dsl)
end
end
defp mode_post(mode) do
dsl =
case mode do
Orb.S32 ->
quote do
import Orb.I32.DSL, only: []
import Orb.S32.DSL, only: []
import Orb.Global.DSL, only: []
end
Orb.U32 ->
quote do
import Orb.I32.DSL, only: []
import Orb.U32.DSL, only: []
import Orb.Global.DSL, only: []
end
Orb.F32 ->
quote do
import Orb.F32.DSL, only: []
import Orb.Global.DSL, only: []
end
:no_magic ->
[]
end
quote do
import Kernel
import Orb.IfElse.DSL, only: []
unquote(dsl)
end
end
@doc """
Enter WebAssembly.
"""
defmacro wasm(mode \\ Orb.S32, do: block) do
# block = interpolate_external_values(block, __ENV__)
mode = Macro.expand_literals(mode, __CALLER__)
pre = mode_pre(mode)
post = mode_post(mode)
%{body: body, constants: constants} = do_module_body(block, [], __CALLER__, __CALLER__.module)
Module.register_attribute(__CALLER__.module, :wasm_constants, accumulate: true)
Module.put_attribute(__CALLER__.module, :wasm_constants, constants)
quote do
unquote(pre)
import Orb.DSL
@wasm_body unquote(body)
import Orb.DSL, only: []
unquote(post)
end
end
@doc """
Declare a snippet of Orb AST for reuse. Enables DSL, with additions from `mode`.
"""
defmacro snippet(mode \\ Orb.S32, locals \\ [], do: block) do
block = interpolate_external_values(block, __CALLER__)
mode = Macro.expand_literals(mode, __CALLER__)
pre = mode_pre(mode)
post = mode_post(mode)
block_items =
case block do
{:__block__, _meta, items} -> items
single -> [single]
end
locals =
for {key, type} <- locals, into: %{} do
{key, Orb.ToWat.Instructions.expand_type(type, __CALLER__)}
end
quote do
# We want to import and un-import. so we use `with` as a finite scope.
with do
unquote(pre)
import Orb.DSL
dsl_items = unquote(Orb.DSL.do_snippet(locals, block_items))
import Orb.DSL, only: []
unquote(post)
dsl_items
end
end
end
defp interpolate_external_values(ast, env) do
Macro.postwalk(ast, fn
{:^, _, [term]} ->
Macro.postwalk(term, &Macro.expand_once(&1, env))
{:^, _, _other} ->
raise "Invalid ^. Expected single argument."
other ->
other
end)
end
@doc """
Declare a WebAssembly import for a function or global.
"""
defmacro wasm_import(mod, entries) when is_atom(mod) and is_list(entries) do
quote do
@wasm_imports (for {name, type} <-
unquote(entries) do
%Orb.Import{module: unquote(mod), name: name, type: type}
end)
end
end
@doc """
Convert Orb AST into WebAssembly text format.
"""
def to_wat(term) when is_atom(term) do
term.__wasm_module__() |> Orb.ToWat.to_wat("") |> IO.chardata_to_string()
end
def to_wat(term) do
term |> Orb.ToWat.to_wat("") |> IO.chardata_to_string()
end
def __get_block_items(block) do
case block do
nil -> nil
{:__block__, _meta, block_items} -> block_items
single -> [single]
end
end
end
