assert macro: Allow destructuring calls.

Neat-Lang · May 26, 2023 · 15adb44 · 15adb44
1 parent ec443dd
commit 15adb44
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/src/neat/base.nt b/src/neat/base.nt
@@ -839,6 +839,8 @@ abstract class CompilerBase
 
  abstract (string | :none) destructAstStringLiteral(ASTSymbol sym);
 
+ abstract (ASTSymbol target, ASTArgument[] args | :none) destructAstCall(ASTSymbol sym);
+
  abstract ASTSymbol astFormatString(ASTSymbol[] parts, LocRange locRange);
 
  ASTSymbol astBoolLiteral(bool value, LocRange locRange) {

diff --git a/src/neat/compiler.nt b/src/neat/compiler.nt
@@ -189,6 +189,13 @@ class CompilerImpl : CompilerBase
  return :none;
  }
 
+ override (ASTSymbol target, ASTArgument[] args | :none) destructAstCall(ASTSymbol sym) {
+ if (auto call = sym.instanceOf(ASTCall)) {
+ return (call.target, call.args);
+ }
+ return :none;
+ }
+
  override ASTSymbol astFormatString(ASTSymbol[] parts, LocRange locRange) {
  return new ASTFormatString(parts, locRange);
  }

diff --git a/src/std/macro/assert.nt b/src/std/macro/assert.nt
@@ -256,9 +256,24 @@ string repr(ASTSymbol sym, Context context) {
  string text: return "\"$text\"";
  (:none): {}
  }
+ context.compiler.destructAstCall(sym).case {
+ (ASTSymbol target, ASTArgument[] args): return repr(target, context) ~ "(" ~ reprArgs(args, context) ~ ")";
+ (:none): {}
+ }
+
  return "TODO";
 }
 
+string reprArgs(ASTArgument[] args, Context context) {
+ mut string res;
+ for (arg in args) {
+ if (!res.empty) res ~= ", ";
+ if (arg.name) res ~= "$(arg.name)=";
+ res ~= repr(arg.sym, context);
+ }
+ return res;
+}
+
 class ParseAssert : Macro
 {
  this() { }