From b314dfd58e718e0250a2fe9600a26054a4754de0 Mon Sep 17 00:00:00 2001
From: alex-eg <alexey.e.egorov@gmail.com>
Date: Fri, 26 Sep 2025 14:33:47 +0300
Subject: [PATCH] re-implement macro-expansion using new semantic walker

---
 semen.scm       | 12 +++++++++++-
 tests/semen.scm | 29 ++++++++++++++++++++++++-----
 2 files changed, 35 insertions(+), 6 deletions(-)

diff --git a/semen.scm b/semen.scm
index 7711518..b6a3d27 100644
--- a/semen.scm
+++ b/semen.scm
@@ -79,6 +79,15 @@
                                 (cons (car module-public-forms)
                                       acc))))))
 
+(define (semen-macro-expand form)
+  "Walk the form recursively and expand all macros, unitl none is left."
+  (semen-walk-form
+   form
+   (lambda (subform)
+     (if (sex-macro? subform)
+         (apply-macro subform)
+         subform))))
+
 ;;; TODO: for greater inspiration, see SBCL's walk.lisp
 (define (semen-walk-form form walk-fn)
   (if (atom? form) form
@@ -97,7 +106,8 @@
       (cons new-car new-cdr)))
 
 (define (process-fn sex-fn acc)
-  (cons sex-fn acc))
+  (let ((expanded (semen-macro-expand sex-fn)))
+    (cons expanded acc)))
 
 (define (process-struct sex-struct acc)
   (cons sex-struct acc))
diff --git a/tests/semen.scm b/tests/semen.scm
index 63e88a7..b29f6a5 100644
--- a/tests/semen.scm
+++ b/tests/semen.scm
@@ -23,12 +23,31 @@
 (test '(pub fn float sum ((int a) (int b))) (sex-fn-prototype sum-fn))
 (test '((return (cast float (+ a b)))) (sex-fn-body sum-fn))
 
-(define sex-code
-  '((defmacro (sum-var name a b c)
-      `(var ,name ,(+ a b c)))
+(let ((sex-code
+       '((defmacro (sum-var name a b c)
+           `(var ,name ,(+ a b c)))
 
-    (sum-var v 1 2 3)))
+         (sum-var v 1 2 3))))
 
-(test '((var v 6)) (semen-process sex-code))
+  (test '((var v 6)) (semen-process sex-code)))
+
+;;; Macro expansion
+
+(test 'a (semen-walk-form 'a identity))
+(test '(a b c) (semen-walk-form '(a b c) identity))
+
+(test 'a (semen-macro-expand 'a))
+(test '(a b c) (semen-macro-expand '(a b c)))
+
+(let ((sex-code-macro
+       '((defmacro (x10 a)
+           `(* 10 ,a))
+
+         (fn void foo ((int a) (int b))
+             (return (+ a (x10 b)))))))
+
+  (test '((fn void foo ((int a) (int b))
+              (return (+ a (* 10 b)))))
+        (semen-process sex-code-macro)))
 
 (test-end)