added support for quasiquote

This commit is contained in:
Pekka Laiho 2020-12-05 16:47:49 +07:00
parent 52b7bc6934
commit f4baaca01c

View File

@ -277,6 +277,19 @@ class Evaller
} }
return $astData[1]; return $astData[1];
} elseif ($symbolName == 'quasiquote') {
if ($astLength != 2) {
throw new MadLispException("quasiquote requires exactly 1 argument");
}
$ast = $this->quasiquote($astData[1]);
continue; // tco
} elseif ($symbolName == 'quasiquote-expand') {
if ($astLength != 2) {
throw new MadLispException("quasiquote-expand requires exactly 1 argument");
}
return $this->quasiquote($astData[1]);
} }
} }
@ -330,4 +343,51 @@ class Evaller
return $ast; return $ast;
} }
private function quasiquote($ast)
{
if ($ast instanceof MList) {
$data = $ast->getData();
// Check for unquote
if (count($data) > 0 && $data[0] instanceof Symbol && $data[0]->getName() == 'unquote') {
if (count($data) == 2) {
return $data[1];
} else {
throw new MadLispException("unquote requires exactly 1 argument");
}
}
$result = new MList();
for ($i = count($data) - 1; $i >= 0; $i--) {
$elt = $data[$i];
if ($elt instanceof MList && count($elt->getData()) > 0 && $elt->get(0) instanceof Symbol && $elt->get(0)->getName() == 'splice-unquote') {
if (count($elt->getData()) == 2) {
$result = new MList([
new Symbol('concat'),
$elt->get(1),
$result
]);
} else {
throw new MadLispException("splice-unquote requires exactly 1 argument");
}
} else {
$result = new MList([
new Symbol('pull'),
$this->quasiquote($elt),
$result
]);
}
}
return $result;
} elseif ($ast instanceof Symbol || $ast instanceof Collection) {
// Quote other forms which are affected by evaluation
return new MList([new Symbol('quote'), $ast]);
} else {
return $ast;
}
}
} }