mirror of
https://github.com/kanaka/mal.git
synced 2024-09-11 21:57:38 +03:00
- Reorder README to have implementation list after "learning tool" bullet. - This also moves tests/ and libs/ into impls. It would be preferrable to have these directories at the top level. However, this causes difficulties with the wasm implementations which need pre-open directories and have trouble with paths starting with "../../". So in lieu of that, symlink those directories to the top-level. - Move the run_argv_test.sh script into the tests directory for general hygiene.
150 lines
4.5 KiB
Objective-C
150 lines
4.5 KiB
Objective-C
#import <Foundation/Foundation.h>
|
|
|
|
#import "mal_readline.h"
|
|
#import "types.h"
|
|
#import "reader.h"
|
|
#import "printer.h"
|
|
#import "env.h"
|
|
#import "malfunc.h"
|
|
#import "core.h"
|
|
|
|
// read
|
|
NSObject *READ(NSString *str) {
|
|
return read_str(str);
|
|
}
|
|
|
|
// eval
|
|
NSObject *eval_ast(NSObject *ast, Env *env) {
|
|
if ([ast isMemberOfClass:[MalSymbol class]]) {
|
|
return [env get:(MalSymbol *)ast];
|
|
} else if ([ast isKindOfClass:[NSArray class]]) {
|
|
NSMutableArray *newLst = [NSMutableArray array];
|
|
for (NSObject * x in (NSArray *)ast) {
|
|
[newLst addObject:EVAL(x, env)];
|
|
}
|
|
if ([ast isKindOfClass:[MalVector class]]) {
|
|
return [MalVector fromArray:newLst];
|
|
} else {
|
|
return newLst;
|
|
}
|
|
} else if ([ast isKindOfClass:[NSDictionary class]]) {
|
|
NSMutableDictionary *newDict = [NSMutableDictionary dictionary];
|
|
for (NSString * k in (NSDictionary *)ast) {
|
|
newDict[k] = EVAL(((NSDictionary *)ast)[k], env);
|
|
}
|
|
return newDict;
|
|
} else {
|
|
return ast;
|
|
}
|
|
}
|
|
|
|
NSObject *EVAL(NSObject *ast, Env *env) {
|
|
//NSLog(@"EVAL: %@ (%@)", _pr_str(ast, true), env);
|
|
if (!list_Q(ast)) {
|
|
return eval_ast(ast, env);
|
|
}
|
|
|
|
// apply list
|
|
if ([(NSArray *)ast count] == 0) {
|
|
return ast;
|
|
}
|
|
NSArray * alst = (NSArray *)ast;
|
|
id a0 = alst[0];
|
|
NSString * a0sym = [a0 isKindOfClass:[MalSymbol class]] ? (NSString *)a0
|
|
: @"__<*fn*>__";
|
|
|
|
if ([a0sym isEqualTo:@"def!"]) {
|
|
return [env set:((MalSymbol *)alst[1]) val:EVAL(alst[2], env)];
|
|
} else if ([(NSString *)a0 isEqualTo:@"let*"]) {
|
|
Env *let_env = [Env fromOuter:env];
|
|
NSArray * binds = (NSArray *)alst[1];
|
|
for (int i=0; i < [binds count]; i+=2) {
|
|
[let_env set:binds[i] val:EVAL(binds[i+1], let_env)];
|
|
}
|
|
return EVAL(alst[2], let_env);
|
|
} else if ([a0sym isEqualTo:@"do"]) {
|
|
NSArray * el = (NSArray *)eval_ast(_rest(alst), env);
|
|
return [el lastObject];
|
|
} else if ([a0sym isEqualTo:@"if"]) {
|
|
NSObject * cond = EVAL(alst[1], env);
|
|
if ([cond isKindOfClass:[NSNull class]] ||
|
|
[cond isKindOfClass:[MalFalse class]]) {
|
|
if ([alst count] > 3) {
|
|
return EVAL(alst[3], env);
|
|
} else {
|
|
return [NSNull alloc];
|
|
}
|
|
} else {
|
|
return EVAL(alst[2], env);
|
|
}
|
|
} else if ([a0sym isEqualTo:@"fn*"]) {
|
|
return [[MalFunc alloc] init:alst[2] env:env params:alst[1]];
|
|
} else {
|
|
NSArray * el = (NSArray *) eval_ast(ast, env);
|
|
NSArray * args = @[];
|
|
if ([el count] > 1) {
|
|
args = _rest(el);
|
|
}
|
|
return apply(el[0], args);
|
|
/*
|
|
if ([el[0] isKindOfClass:[MalFunc class]]) {
|
|
MalFunc * mf = el[0];
|
|
return [mf apply:args];
|
|
} else {
|
|
NSObject * (^ f)(NSArray *) = el[0];
|
|
return f(args);
|
|
}
|
|
*/
|
|
}
|
|
}
|
|
|
|
// print
|
|
NSString *PRINT(NSObject *exp) {
|
|
return _pr_str(exp, true);
|
|
}
|
|
|
|
// REPL
|
|
NSString *REP(NSString *line, Env *env) {
|
|
return PRINT(EVAL(READ(line), env));
|
|
}
|
|
|
|
int main () {
|
|
Env * repl_env = [[Env alloc] init];
|
|
|
|
// Create an autorelease pool to manage the memory into the program
|
|
NSAutoreleasePool * pool = [[NSAutoreleasePool alloc] init];
|
|
// If using automatic reference counting (ARC), use @autoreleasepool instead:
|
|
// @autoreleasepool {
|
|
|
|
// core.m: defined using Objective-C
|
|
NSDictionary * core_ns = [Core ns];
|
|
for (NSString* key in core_ns) {
|
|
[repl_env set:(MalSymbol *)key val:[core_ns objectForKey:key]];
|
|
}
|
|
|
|
// core.mal: defined using the language itself
|
|
REP(@"(def! not (fn* (a) (if a false true)))", repl_env);
|
|
|
|
while (true) {
|
|
char *rawline = _readline("user> ");
|
|
if (!rawline) { break; }
|
|
NSString *line = [NSString stringWithUTF8String:rawline];
|
|
if ([line length] == 0) { continue; }
|
|
@try {
|
|
printf("%s\n", [[REP(line, repl_env) description] UTF8String]);
|
|
} @catch(NSString *e) {
|
|
printf("Error: %s\n", [e UTF8String]);
|
|
} @catch(NSObject *e) {
|
|
NSObject * exc = e;
|
|
printf("Exception: %s\n", [_pr_str(exc, true) UTF8String]);
|
|
} @catch(NSException *e) {
|
|
if ([[e name] isEqualTo:@"ReaderContinue"]) { continue; }
|
|
printf("Exception: %s\n", [[e reason] UTF8String]);
|
|
}
|
|
}
|
|
|
|
[pool drain];
|
|
|
|
// }
|
|
}
|