candle 0.0.1

data/lib/candle/generators/lua/wax/lib/wax.m

@@ -0,0 +1,260 @@
+//
+//  ObjLua.m
+//  Lua
+//
+//  Created by ProbablyInteractive on 5/27/09.
+//  Copyright 2009 Probably Interactive. All rights reserved.
+//
+
+#import "ProtocolLoader.h"
+
+#import "wax.h"
+#import "wax_class.h"
+#import "wax_instance.h"
+#import "wax_struct.h"
+#import "wax_helpers.h"
+#import "wax_gc.h"
+#import "wax_server.h"
+#import "wax_stdlib.h"
+
+#import "lauxlib.h"
+#import "lobject.h"
+#import "lualib.h"
+
+static void addGlobals(lua_State *L);
+static int waxRoot(lua_State *L);
+static int waxPrint(lua_State *L);
+static int tolua(lua_State *L);
+static int toobjc(lua_State *L);
+static int exitApp(lua_State *L);
+static int objcDebug(lua_State *L);
+
+static lua_State *currentL;
+lua_State *wax_currentLuaState() {
+    
+    if (!currentL) 
+        currentL = lua_open();
+    
+    return currentL;
+}
+
+void uncaughtExceptionHandler(NSException *e) {
+    NSLog(@"ERROR: Uncaught exception %@", [e description]);
+    lua_State *L = wax_currentLuaState();
+    
+    if (L) {
+        wax_getStackTrace(L);
+        const char *stackTrace = luaL_checkstring(L, -1);
+        NSLog(@"%s", stackTrace);
+        lua_pop(L, -1); // remove the stackTrace
+    }
+}
+
+int wax_panic(lua_State *L) {
+	printf("Lua panicked and quit: %s\n", luaL_checkstring(L, -1));
+	exit(EXIT_FAILURE);
+}
+
+lua_CFunction lua_atpanic (lua_State *L, lua_CFunction panicf);
+
+void wax_setup() {
+	NSSetUncaughtExceptionHandler(&uncaughtExceptionHandler); 
+	
+    NSFileManager *fileManager = [NSFileManager defaultManager];
+    [fileManager changeCurrentDirectoryPath:[[NSBundle mainBundle] bundlePath]];
+    
+    lua_State *L = wax_currentLuaState();
+	lua_atpanic(L, &wax_panic);
+    
+    luaL_openlibs(L); 
+
+	luaopen_wax_class(L);
+    luaopen_wax_instance(L);
+    luaopen_wax_struct(L);
+	
+    addGlobals(L);
+	
+	[wax_gc start];
+}
+
+void wax_start(char* initScript, lua_CFunction extensionFunction, ...) {
+	wax_setup();
+	
+	lua_State *L = wax_currentLuaState();
+	
+	// Load extentions
+	// ---------------
+	if (extensionFunction) {
+        extensionFunction(L);
+		
+        va_list ap;
+        va_start(ap, extensionFunction);
+        while((extensionFunction = va_arg(ap, lua_CFunction))) extensionFunction(L);
+		
+        va_end(ap);
+    }
+
+	// Load stdlib
+	// ---------------
+	#ifdef WAX_STDLIB 
+		// If the stdlib was autogenerated and included in the source, load
+		char stdlib[] = WAX_STDLIB;
+		size_t stdlibSize = sizeof(stdlib);
+	#else
+		char stdlib[] = "require 'wax'";
+		size_t stdlibSize = strlen(stdlib);
+	#endif
+
+	if (luaL_loadbuffer(L, stdlib, stdlibSize, "loading wax stdlib") || lua_pcall(L, 0, LUA_MULTRET, 0)) {
+		fprintf(stderr,"Error opening wax scripts: %s\n", lua_tostring(L,-1));
+	}
+
+	// Run Tests or the REPL?
+	// ----------------------
+	NSDictionary *env = [[NSProcessInfo processInfo] environment];
+    if ([[env objectForKey:@"WAX_TEST"] isEqual:@"YES"]) {
+		printf("Running Tests\n");
+		if (luaL_dostring(L, "require 'tests'") != 0) {
+			fprintf(stderr,"Fatal error running tests: %s\n", lua_tostring(L,-1));
+        }
+        exit(1);
+    }
+	else if ([[env objectForKey:@"WAX_REPL"] isEqual:@"YES"]) {
+		printf("Starting REPL\n");
+		if (luaL_dostring(L, "require 'wax.repl'") != 0) {
+            fprintf(stderr,"Fatal error starting the REPL: %s\n", lua_tostring(L,-1));
+        }		
+		exit(1);
+	}
+	else {
+		// Load app
+		char appLoadString[512];
+		snprintf(appLoadString, sizeof(appLoadString), "local f = '%s' require(f:gsub('%%.[^.]*$', ''))", initScript); // Strip the extension off the file.
+		if (luaL_dostring(L, appLoadString) != 0) {
+			fprintf(stderr,"Error opening wax scripts: %s\n", lua_tostring(L,-1));
+		}
+	}
+
+}
+
+void wax_startWithServer() {		
+	wax_setup();
+	[wax_server class]; // You need to load the class somehow via the wax.framework
+	lua_State *L = wax_currentLuaState();
+	
+	// Load all the wax lua scripts
+    if (luaL_dofile(L, WAX_SCRIPTS_DIR "/scripts/wax/init.lua") != 0) {
+        fprintf(stderr,"Fatal error opening wax scripts: %s\n", lua_tostring(L,-1));
+    }
+	
+	Class WaxServer = objc_getClass("WaxServer");
+	if (!WaxServer) [NSException raise:@"Wax Server Error" format:@"Could load Wax Server"];
+	
+	[WaxServer start];
+}
+
+void wax_end() {
+    [wax_gc stop];
+    lua_close(wax_currentLuaState());
+    currentL = 0;
+}
+
+static void addGlobals(lua_State *L) {
+    lua_getglobal(L, "wax");
+    if (lua_isnil(L, -1)) {
+        lua_pop(L, 1); // Get rid of the nil
+        lua_newtable(L);
+        lua_pushvalue(L, -1);
+        lua_setglobal(L, "wax");
+    }
+    
+    lua_pushnumber(L, WAX_VERSION);
+    lua_setfield(L, -2, "version");
+    
+    lua_pushstring(L, [[[NSBundle mainBundle] objectForInfoDictionaryKey:@"CFBundleVersion"] UTF8String]);
+    lua_setfield(L, -2, "appVersion");
+    
+    lua_pushcfunction(L, waxRoot);
+    lua_setfield(L, -2, "root");
+
+    lua_pushcfunction(L, waxPrint);
+    lua_setfield(L, -2, "print");    
+    
+#ifdef DEBUG
+    lua_pushboolean(L, YES);
+    lua_setfield(L, -2, "isDebug");
+#endif
+    
+    lua_pop(L, 1); // pop the wax global off
+    
+
+    lua_pushcfunction(L, tolua);
+    lua_setglobal(L, "tolua");
+    
+    lua_pushcfunction(L, toobjc);
+    lua_setglobal(L, "toobjc");
+    
+    lua_pushcfunction(L, exitApp);
+    lua_setglobal(L, "exitApp");
+    
+    lua_pushstring(L, [[NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) objectAtIndex:0] UTF8String]);
+    lua_setglobal(L, "NSDocumentDirectory");
+    
+    lua_pushstring(L, [[NSSearchPathForDirectoriesInDomains(NSLibraryDirectory, NSUserDomainMask, YES) objectAtIndex:0] UTF8String]);
+    lua_setglobal(L, "NSLibraryDirectory");
+    
+    NSString *cachePath = [NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES) objectAtIndex:0];
+    lua_pushstring(L, [cachePath UTF8String]);
+    lua_setglobal(L, "NSCacheDirectory");
+
+    NSError *error = nil;
+    [[NSFileManager defaultManager] createDirectoryAtPath:cachePath withIntermediateDirectories:YES attributes: nil error:&error];
+    if (error) {
+        wax_log(LOG_DEBUG, @"Error creating cache path. %@", [error localizedDescription]);
+    }
+}
+
+static int waxPrint(lua_State *L) {
+    NSLog(@"%s", luaL_checkstring(L, 1));
+    return 0;
+}
+
+static int waxRoot(lua_State *L) {
+    luaL_Buffer b;
+    luaL_buffinit(L, &b);
+    luaL_addstring(&b, WAX_SCRIPTS_DIR);
+    
+    for (int i = 1; i <= lua_gettop(L); i++) {
+        luaL_addstring(&b, "/");
+        luaL_addstring(&b, luaL_checkstring(L, i));
+    }
+
+    luaL_pushresult(&b);
+                       
+    return 1;
+}
+
+static int tolua(lua_State *L) {
+    if (lua_isuserdata(L, 1)) { // If it's not userdata... it's already lua!
+        wax_instance_userdata *instanceUserdata = (wax_instance_userdata *)luaL_checkudata(L, 1, WAX_INSTANCE_METATABLE_NAME);
+        wax_fromInstance(L, instanceUserdata->instance);
+    }
+    
+    return 1;
+}
+
+static int toobjc(lua_State *L) {
+    id *instancePointer = wax_copyToObjc(L, "@", 1, nil);
+    id instance = *(id *)instancePointer;
+    
+    wax_instance_create(L, instance, NO);
+    
+    if (instancePointer) free(instancePointer);
+    
+    return 1;
+}
+
+static int exitApp(lua_State *L) {
+    exit(0);
+    return 0;
+}

data/lib/candle/generators/lua/wax/lib/wax_class.h

@@ -0,0 +1,18 @@
+//
+//  wax_class.h
+//  Lua
+//
+//  Created by ProbablyInteractive on 5/20/09.
+//  Copyright 2009 Probably Interactive. All rights reserved.
+//
+
+#import <Foundation/Foundation.h>
+#import <objc/runtime.h>
+#import <objc/message.h>
+
+#import "lua.h"
+
+#define WAX_CLASS_METATABLE_NAME "wax.class"
+#define WAX_CLASS_INSTANCE_USERDATA_IVAR_NAME "wax_instance_userdata"
+
+int luaopen_wax_class(lua_State *L);

data/lib/candle/generators/lua/wax/lib/wax_class.m

@@ -0,0 +1,190 @@
+//
+//  wax_class.m
+//  Lua
+//
+//  Created by ProbablyInteractive on 5/20/09.
+//  Copyright 2009 Probably Interactive. All rights reserved.
+//
+
+#import "wax_class.h"
+
+#import "wax.h"
+#import "wax_instance.h"
+#import "wax_helpers.h"
+
+#import "lua.h"
+#import "lauxlib.h"
+
+static int __index(lua_State *L);
+static int __call(lua_State *L);
+
+static int addProtocols(lua_State *L);
+static int name(lua_State *L);
+static id alloc(id self, SEL _cmd);
+static id allocWithZone(id self, SEL _cmd, NSZone *);
+static id valueForUndefinedKey(id self, SEL cmd, NSString *key);
+static void setValueForUndefinedKey(id self, SEL cmd, id value, NSString *key);
+
+static const struct luaL_Reg MetaMethods[] = {
+    {"__index", __index},
+    {"__call", __call},
+    {NULL, NULL}
+};
+
+static const struct luaL_Reg Methods[] = {
+    {"addProtocols", addProtocols},
+    {"name", name},
+    {NULL, NULL}
+};
+
+int luaopen_wax_class(lua_State *L) {
+    BEGIN_STACK_MODIFY(L);
+    
+    luaL_newmetatable(L, WAX_CLASS_METATABLE_NAME);
+    luaL_register(L, NULL, MetaMethods);
+    luaL_register(L, WAX_CLASS_METATABLE_NAME, Methods);    
+
+    // Set the metatable for the module
+    luaL_getmetatable(L, WAX_CLASS_METATABLE_NAME);
+    lua_setmetatable(L, -2);
+    
+    END_STACK_MODIFY(L, 0)
+    
+    return 1;
+}
+
+
+// Finds an ObjC class
+static int __index(lua_State *L) {
+    const char *className = luaL_checkstring(L, 2);
+    Class klass = objc_getClass(className);
+    if (klass) {
+        wax_instance_create(L, klass, YES);
+    }
+    else {
+        lua_pushnil(L);
+    }
+    
+    return 1;
+}
+
+// Creates a new ObjC class
+static int __call(lua_State *L) {   
+    const char *className = luaL_checkstring(L, 2);
+    Class klass = objc_getClass(className);
+    
+    if (!klass) {
+        Class superClass;    
+        if (lua_isuserdata(L, 3)) {
+            wax_instance_userdata *instanceUserdata = (wax_instance_userdata *)luaL_checkudata(L, 3, WAX_INSTANCE_METATABLE_NAME);
+            superClass = instanceUserdata->instance;
+        }
+        else if (lua_isnoneornil(L, 3)) {
+            superClass = [NSObject class];
+        }
+        else {
+            const char *superClassName = luaL_checkstring(L, 3);
+            superClass = objc_getClass(superClassName);
+        }
+        
+        if (!superClass) {
+            luaL_error(L, "Failed to create '%s'. Unknown superclass \"%s\" received.", className, luaL_checkstring(L, 3));
+        }
+        
+        klass = objc_allocateClassPair(superClass, className, 0);
+        NSUInteger size;
+        NSUInteger alignment;
+        NSGetSizeAndAlignment("*", &size, &alignment);
+        class_addIvar(klass, WAX_CLASS_INSTANCE_USERDATA_IVAR_NAME, size, alignment, "*"); // Holds a reference to the lua userdata
+        objc_registerClassPair(klass);        
+
+        // Make Key-Value complient
+        class_addMethod(klass, @selector(setValue:forUndefinedKey:), (IMP)setValueForUndefinedKey, "v@:@@");
+        class_addMethod(klass, @selector(valueForUndefinedKey:), (IMP)valueForUndefinedKey, "@@:@");        
+
+        id metaclass = object_getClass(klass);
+        
+        // So objects created in ObjC will get an associated lua object
+        // Store the original allocWithZone implementation in case something secret goes on in there. 
+        // Calls to `alloc` always are end up calling `allocWithZone:` so we don't bother handling alloc here.
+        Method m = class_getInstanceMethod(metaclass, @selector(allocWithZone:));
+        
+        // If we the method has already been swizzled (by the class's super, then
+        // just leave it up to the super!
+        if (method_getImplementation(m) != (IMP)allocWithZone) {
+            class_addMethod(metaclass, @selector(wax_originalAllocWithZone:), method_getImplementation(m), method_getTypeEncoding(m));
+            class_addMethod(metaclass, @selector(allocWithZone:), (IMP)allocWithZone, "@@:^{_NSZone=}");
+        }
+    }
+        
+    wax_instance_create(L, klass, YES);
+    
+    return 1;
+}
+
+static int addProtocols(lua_State *L) {
+    wax_instance_userdata *instanceUserdata = (wax_instance_userdata *)luaL_checkudata(L, 1, WAX_INSTANCE_METATABLE_NAME);
+    
+    if (!instanceUserdata->isClass) {
+        luaL_error(L, "ERROR: Can only set a protocol on a class (You are trying to set one on an instance)");
+        return 0;
+    }
+    
+    for (int i = 2; i <= lua_gettop(L); i++) {
+        const char *protocolName = luaL_checkstring(L, i);
+        Protocol *protocol = objc_getProtocol(protocolName);
+        if (!protocol) luaL_error(L, "Could not find protocol named '%s'\nHint: Sometimes the runtime cannot automatically find a protocol. Try adding it (via xCode) to the file ProtocolLoader.h", protocolName);
+        class_addProtocol(instanceUserdata->instance, protocol);
+    }
+    
+    return 0;
+}
+
+static int name(lua_State *L) {
+    wax_instance_userdata *instanceUserdata = (wax_instance_userdata *)luaL_checkudata(L, 1, WAX_INSTANCE_METATABLE_NAME);
+    lua_pushstring(L, [NSStringFromClass([instanceUserdata->instance class]) UTF8String]);
+    return 1;
+}
+
+static id allocWithZone(id self, SEL _cmd, NSZone *zone) {
+    lua_State *L = wax_currentLuaState(); 
+    BEGIN_STACK_MODIFY(L);
+
+    id instance = [self wax_originalAllocWithZone:zone];
+    object_setInstanceVariable(instance, WAX_CLASS_INSTANCE_USERDATA_IVAR_NAME, @"YEAP");
+    
+    END_STACK_MODIFY(L, 0);
+    
+    return instance;
+}
+
+
+static void setValueForUndefinedKey(id self, SEL cmd, id value, NSString *key) {
+    lua_State *L = wax_currentLuaState();
+    
+    BEGIN_STACK_MODIFY(L);
+    
+    wax_instance_pushUserdata(L, self);
+    wax_fromObjc(L, "@", &value);
+    lua_setfield(L, -2, [key UTF8String]);
+    
+    END_STACK_MODIFY(L, 0);
+}
+
+static id valueForUndefinedKey(id self, SEL cmd, NSString *key) {
+    lua_State *L = wax_currentLuaState();    
+    id result = nil;
+    
+    BEGIN_STACK_MODIFY(L);
+    
+    wax_instance_pushUserdata(L, self);
+    lua_getfield(L, -1, [key UTF8String]);
+    
+    id *keyValue = wax_copyToObjc(L, "@", -1, nil);
+    result = *keyValue;
+    free(keyValue);
+    
+    END_STACK_MODIFY(L, 0);
+    
+    return result;
+}