hscript-improved/hscript/customclass/CustomClassDecl.hx

package hscript.customclass;

import hscript.Expr.AccessContext;
import hscript.proxy.ProxyType;
import hscript.customclass.utils.FunctionUtils;
import haxe.Constraints.Function;
import hscript.Expr.FieldDecl;
import hscript.Expr.VarDecl;
import hscript.Expr.FunctionDecl;

/**
 * The ACTUAL "StaticHandler"
 * @author Jamextreme140
 */
@:access(hscript.Interp)
@:structInit
class CustomClassDecl implements IHScriptCustomAccessBehaviour {
	public var classDecl:Expr.ClassDecl; //This holds the class instantiation info
	public var imports:Map<String, CustomClassImport>;
	public var usings:Array<String>;
	public var pkg:Null<Array<String>> = null;
	public var ogInterp:Null<Interp> = null;
	public var isInline:Null<Bool> = null;

	public var staticInterp:Interp = new Interp();

	public var superClassDecl(default, null):Dynamic = null; //This holds the super class reference. 

	var _cachedStaticFields:Map<String, FieldDecl> = [];
	var _cachedStaticFunctions:Map<String, FunctionDecl> = [];
	var _cachedStaticVariables:Map<String, VarDecl> = [];

	public var __allowSetGet:Bool = true;

	public var __allowPrivateAccess:Bool = false;

	public var accessContext:AccessContext = COuter;

	public final name:String;

	public function new(classDecl:Expr.ClassDecl, imports:Map<String, CustomClassImport>, usings:Array<String>, ?pkg:Array<String>, ?ogInterp:Interp, ?isInline:Bool) {
		this.classDecl = classDecl;
		this.imports = imports;
		this.usings = usings;
		this.pkg = pkg;
		this.ogInterp = ogInterp;
		this.isInline = isInline;

		// Cache the classname
		var pkg = pkg != null ? '${pkg.join(".")}.' : "";
		var className = classDecl.name;
		this.name = '$pkg$className';

		if(ogInterp != null) {
			staticInterp.importFailedCallback = ogInterp.importFailedCallback;
			staticInterp.errorHandler = ogInterp.errorHandler;
			staticInterp.allowStaticVariables = ogInterp.allowStaticVariables;
			staticInterp.staticVariables = ogInterp.staticVariables;

			if(isInline != null && isInline) {
				// uses public variables from the same scope as where the class was defined
				staticInterp.variables = ogInterp.variables;
				staticInterp.allowPublicVariables = ogInterp.allowPublicVariables;
				staticInterp.publicVariables = ogInterp.publicVariables;
			}
		}

		cacheImports();
		processUsings();
		cacheFields();
		if(classDecl.extend != null)
			buildSuperClass();
	}

	function cacheImports() {
		// This will make imported classes available for Static Functions
		var i:Int = 0;
		for(s => imp in imports) {
			var importedClass = imp.fullPath;
			var importAlias = imp.as;

			if(this.staticInterp.variables.exists(imp.name)) continue; // class is already imported

			if (Interp.customClassExist(importedClass) && this.staticInterp.importFailedCallback != null) {
				this.staticInterp.importFailedCallback(importedClass.split("."), importAlias);
				continue;
			}

			#if hscriptPos
			var e:Expr = {
				e: ExprDef.EImport(importedClass, importAlias),
				pmin: 0,
				pmax: 0,
				origin: this.classDecl.name,
				line: i
			};
			i++;
			#else
			var e = Expr.EImport(importedClass, importAlias);
			#end
			this.staticInterp.expr(e);
		}
	}

	function cacheFields() {
		for (f in classDecl.fields) {
			if (f.access.contains(AStatic)) {
				_cachedStaticFields.set(f.name, f);
				switch (f.kind) {
					case KFunction(fn):
						_cachedStaticFunctions.set(f.name, fn);
						#if hscriptPos
						var fexpr:Expr = {
							e: ExprDef.EFunction(fn.args, fn.body, f.name, fn.ret, false, false),
							pmin: fn.body.pmin,
							pmax: fn.body.pmax,
							line: fn.body.line,
							origin: fn.body.origin
						};
						#else
						var fexpr = Expr.EFunction(fn.args, fn.body, f.name, fn.ret, false, false);
						#end
						var func:Function = this.staticInterp.expr(fexpr);
						this.staticInterp.variables.set(f.name, func);
					case KVar(v):
						if(v.get != ADefault || v.set != ADefault)
							__allowSetGet = true;
						_cachedStaticVariables.set(f.name, v);
						if (v.expr != null) {
							var varValue = this.staticInterp.expr(v.expr);
							this.staticInterp.variables.set(f.name, varValue);
						}
				}
			}
		}
	}

	function processUsings() {
		for(us in usings) {
			this.staticInterp.useUsing(us);
		}
	}

	function buildSuperClass() {
		var extendString = new Printer().typeToString(classDecl.extend);
		if (this.pkg != null && extendString.indexOf(".") == -1) {
			extendString = this.pkg.join(".") + "." + extendString;
		}

		var cls:Dynamic = Type.resolveClass('${extendString}_HSX');
		if(cls == null)
			cls = ProxyType.resolveClass(extendString);

		superClassDecl = cls;
		
		if(superClassDecl == null)
			staticInterp.error(ECustom("could not resolve super class: " + extendString));
	}

	public function callFunction(name:String, ?args:Array<Dynamic>):Dynamic {
		var func:Function = getFunction(name);

		return FunctionUtils.callStaticFunction(name, this, staticInterp, func, args != null ? args : []);
	}

	public function hasField(name:String):Bool {
		return _cachedStaticFields.exists(name);
	}

	private function hasFunction(name:String) {
		return _cachedStaticFunctions.exists(name);
	}

	private function getFunction(name:String):Function {
		var fn = this.staticInterp.variables.get(name);
		return Reflect.isFunction(fn) ? fn : null;
	}

	private function hasVar(name:String):Bool {
		return _cachedStaticVariables.exists(name);
	}

	private function getVar(name:String):VarDecl {
		return _cachedStaticVariables.get(name);
	}

	/**
	 * Remove a function from the cache.
	 * This is useful when a function is broken and needs to be skipped.
	 * @param name The name of the function to remove from the cache.
	 */
	private function purgeFunction(name:String):Void {
		if (_cachedStaticFunctions != null) {
			_cachedStaticFunctions.remove(name);
		}
	}

	public function hget(name:String):Dynamic {
		if(hasVar(name)) {
			var v = getVar(name);
			var getter = v.get;
			var isInner = switch (accessContext) {
				case CInner(cl): return cl == this.toString();
				default: false;
			}

			var r:Dynamic = null;

			if (getter == ANever || getter == ANull && (!isInner || !__allowPrivateAccess))
				throw 'field $name cannot be accessed for reading';

			if(__allowSetGet && getter == AGet){
				if(hasFunction('get_$name'))
					r = __callGetter(name);
				else 
					throw 'Method get_$name required by property $name is missing';
			}
			else if (this.staticInterp.variables.exists(name))
				r = this.staticInterp.variables.get(name);
			else {
				if(v.expr != null) {
					r = this.staticInterp.expr(v.expr);
					this.staticInterp.variables.set(name, r);
				}
			}
			return r;
		}
		if(hasFunction(name)) {
			// TODO: optimize this
			var fn:Function = Reflect.makeVarArgs(function(args:Array<Dynamic>) {
				return this.callFunction(name, args);
			});
			return fn;
		}

		throw "static field '" + name + "' does not exist in custom class '" + this.classDecl.name + "'";
	}

	public function hset(name:String, val:Dynamic):Dynamic {
		if (hasVar(name)) {
			var v = getVar(name);
			var setter = v.set;
			var isInner = switch (accessContext) {
				case CInner(cl): return cl == this.toString();
				default: false;
			}

			if (setter == ANever || setter == ANull && (!isInner || !__allowPrivateAccess) || v.isFinal)
				throw 'field $name cannot be accessed for writing';

			if (__allowSetGet && setter == ASet) {
				if (hasFunction('set_$name'))
					return __callSetter(name, val);
				else 
					throw 'Method set_$name required by property $name is missing';
			}
			else {
				this.staticInterp.variables.set(name, val);
				return val;
			}
		}

		throw "static field '" + name + "' does not exist in custom class '" + this.classDecl.name + "'";
	}

	public function __callGetter(name:String):Dynamic {
		__allowSetGet = false;
		var r = callFunction('get_${name}');
		__allowSetGet = true;
		return r;
	}

	public function __callSetter(name:String, val:Dynamic):Dynamic {
		__allowSetGet = false;
		var r = callFunction('set_${name}', [val]);
		__allowSetGet = true;
		return r;
	}
	
	public function toString():String {
		return name;
	}
}

typedef CustomClassImport = {
	var ?name:String;
	var ?pkg:Array<String>;
	var ?fullPath:String; // pkg.pkg.pkg.name
	var ?as:Null<String>; // import pkg.Name as OtherName
}