HaxeFoundation.haxe/std/neko/vm/Loader.hx

/*
 * Copyright (C)2005-2019 Haxe Foundation
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */

package neko.vm;

/**
	The Neko object that implements the loader.
**/
@:callable
@:coreType
abstract LoaderHandle {}

/**
	Loaders can be used to dynamically load Neko primitives stored in NDLL libraries.


	Loaders can be used to dynamically load other Neko modules (.n bytecode files).
	Modules are referenced by names. To lookup the corresponding bytecode file, the
	default loader first look in its cache, then eventually adds the .n extension
	to the name and lookup the bytecode in its path.


	Loaders can be used for sandbox security. When a Module is loaded with a given
	Loader, this loader can manager the module security by filtering which
	primitives can be loaded by this module or by rewrapping them at loading-time
	with custom secured versions. Loaders are inherited in loaded submodules.
**/
@:haxe.warning("-WUnboundMonomorph")
class Loader {
	/**
		The abstract handle.
	**/
	public var l:LoaderHandle;

	public function new(l) {
		this.l = l;
	}

	/**
		The default loader contains a search path in its `path` field. It's a
		linked list of Neko strings that is a parsed version of the `NEKOPATH`.
		This path is used to lookup for modules and libraries.
	**/
	public function getPath() {
		var p = untyped l.path;
		var path = new Array<String>();
		while (p != null) {
			path.push(new String(p[0]));
			p = cast p[1];
		}
		return path;
	}

	/**
		Adds a directory to the search path. See `getPath`.
	**/
	public function addPath(s:String) {
		untyped l.path = __dollar__array(s.__s, l.path);
	}

	/**
		The default loader contains a cache of already loaded modules. It's
		ensuring that the same module does not get loaded twice when circular
		references are occurring. The same module can eventually be loaded twice
		but with different names, for example with two relative paths representing
		the same file, since the cache is done on a by-name basic.
	**/
	public function getCache():Map<String, Module> {
		var h = new haxe.ds.StringMap<Module>();
		var cache = untyped l.cache;
		for (f in Reflect.fields(cache))
			h.set(f, new Module(Reflect.field(cache, f)));
		return h;
	}

	/**
		Set a module in the loader cache.
	**/
	public function setCache(name:String, m:Module) {
		if (m == null)
			Reflect.deleteField(untyped l.cache, name);
		else
			Reflect.setField(untyped l.cache, name, m.m);
	}

	/**
		Change the cache value and returns the old value. This can be used
		to backup the loader cache and restore it later.
	**/
	public function backupCache(c:Dynamic):Dynamic {
		var old = untyped l.cache;
		untyped l.cache = c;
		return old;
	}

	function __compare(other:Loader) {
		return untyped __dollar__compare(this.l, other.l);
	}

	/**
		Loads a neko primitive. By default, the name is of the form `[library@method]`.
		The primitive might not be used directly in Haxe since some of the Neko values
		needs an object wrapper in Haxe.
	**/
	public function loadPrimitive(prim:String, nargs:Int):Dynamic {
		return untyped l.loadprim(prim.__s, nargs);
	}

	/**
		Loads a Module with the given name. If `loader` is defined, this will be
		this Module loader, else this loader will be inherited. When loaded this
		way, the module is directly executed.
	**/
	public function loadModule(modName:String, ?loader:Loader):Module {
		var exp = untyped l.loadmodule(modName.__s, if (loader == null) l else loader.l);
		return new Module(exp.__module);
	}

	/**
		Returns the local Loader. This is the loader that was used to load the
		module in which the code is defined.
	**/
	public static function local() {
		return new Loader(untyped __dollar__loader);
	}

	/**
		Creates a loader using two methods. This loader will not have an accessible cache or path,
		although you can implement such mechanism in the methods body.
	**/
	public static function make(loadPrim:String->Int->Dynamic, loadModule:String->Loader->Module) {
		var l = {
			loadprim: function(prim, nargs) {
				return loadPrim(new String(prim), nargs);
			},
			loadmodule: function(mname, loader) {
				return loadModule(new String(mname), new Loader(loader)).exportsTable();
			},
			args: untyped __dollar__amake(0),
			cache: {},
		};
		return new Loader(cast l);
	}
}