transcendental-lisp/src/main/kotlin/function/builtin/SYMBOL_FUNCTION.java

64 lines
2.0 KiB
Java

package function.builtin;
import error.LispException;
import function.ArgumentValidator;
import function.FunctionNames;
import function.LispFunction;
import function.LispSpecialFunction;
import function.UserDefinedFunction;
import sexpression.Cons;
import sexpression.SExpression;
import sexpression.Symbol;
import table.FunctionTable;
import static java.text.MessageFormat.format;
@FunctionNames({ "SYMBOL-FUNCTION" })
public class SYMBOL_FUNCTION extends LispFunction {
private ArgumentValidator argumentValidator;
public SYMBOL_FUNCTION(String name) {
this.argumentValidator = new ArgumentValidator(name);
this.argumentValidator.setExactNumberOfArguments(1);
this.argumentValidator.setEveryArgumentExpectedType(Symbol.class);
}
@Override
public SExpression call(Cons argumentList) {
argumentValidator.validate(argumentList);
SExpression symbol = argumentList.getFirst();
LispFunction function = FunctionTable.INSTANCE.lookupFunction(symbol.toString());
if (function != null)
return createRepresentation(symbol, function);
throw new UndefinedSymbolFunctionException(symbol);
}
private SExpression createRepresentation(SExpression symbol, LispFunction function) {
if (function instanceof UserDefinedFunction)
return ((UserDefinedFunction) function).getLambdaExpression();
String typeIndicator = function instanceof LispSpecialFunction ? "SPECIAL-FUNCTION" : "FUNCTION";
return new Symbol(format("#<{0} {1}>", typeIndicator, symbol));
}
public static class UndefinedSymbolFunctionException extends LispException {
private static final long serialVersionUID = 1L;
private SExpression function;
public UndefinedSymbolFunctionException(SExpression function) {
this.function = function;
}
@Override
public String getMessage() {
return format("SYMBOL-FUNCTION: undefined function: {0}", function);
}
}
}