From 3363ab2da764825558c859f4419ff99528ed2274 Mon Sep 17 00:00:00 2001
From: Arnaud Bailly <arnaud.bailly@iohk.io>
Date: Fri, 11 Oct 2024 15:09:29 +0200
Subject: Add elixir code

---
 elixir/ast.ex | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 elixir/ast.ex

(limited to 'elixir/ast.ex')

diff --git a/elixir/ast.ex b/elixir/ast.ex
new file mode 100644
index 0000000..4305b7b
--- /dev/null
+++ b/elixir/ast.ex
@@ -0,0 +1,34 @@
+defmodule Ast do
+  def var(name), do: {:var, name}
+  def abs(param, body), do: {:abs, param, body}
+  def app(func, param), do: {:app, func, param}
+
+  def pprint(expr) do
+    case expr do
+      {:var, name} -> name
+      {:abs, param, body} -> "(λ #{param} . #{pprint(body)})"
+      {:app, func, param} -> "#{pprint(func)} #{pprint(param)}"
+    end
+  end
+
+  def free_vars(expr) do
+    case expr do
+      {:var, name} -> MapSet.new([name])
+      {:abs, param, body} -> MapSet.delete(free_vars(body), param)
+      {:app, func, param} -> MapSet.union(free_vars(func), free_vars(param))
+    end
+  end
+
+  def subst(expr, old_var, new_expr) do
+    case expr do
+      {:var, name} ->
+        if name == old_var, do: new_expr, else: expr
+
+      {:abs, param, body} ->
+        if param == old_var, do: expr, else: {:abs, param, subst(body, old_var, new_expr)}
+
+      {:app, func, param} ->
+        {:app, subst(func, old_var, new_expr), subst(param, old_var, new_expr)}
+    end
+  end
+end
-- 
cgit v1.2.3