Add build cache utility.
This commit is contained in:
56
src/tink/macro/BuildCache.hx
Normal file
56
src/tink/macro/BuildCache.hx
Normal file
@@ -0,0 +1,56 @@
|
||||
package tink.macro;
|
||||
|
||||
import haxe.macro.Context;
|
||||
import haxe.macro.Expr;
|
||||
import haxe.macro.Type;
|
||||
import tink.macro.TypeMap;
|
||||
|
||||
class BuildCache {
|
||||
|
||||
static var cache = init();
|
||||
|
||||
static function init() {
|
||||
|
||||
function refresh() {
|
||||
cache = new Map();
|
||||
return true;
|
||||
}
|
||||
|
||||
Context.onMacroContextReused(refresh);
|
||||
refresh();
|
||||
|
||||
return cache;
|
||||
}
|
||||
|
||||
static public function getType(name, ?type, ?pos:Position, build:Position->Type->Array<TypePath>->String->TypeDefinition) {
|
||||
|
||||
if (pos == null)
|
||||
pos = Context.currentPos();
|
||||
|
||||
if (type == null)
|
||||
switch Context.getLocalType() {
|
||||
case TInst(_.toString() == name => true, [v]):
|
||||
type = v;
|
||||
default:
|
||||
throw 'assert';
|
||||
}
|
||||
|
||||
var forName =
|
||||
switch cache[name] {
|
||||
case null: cache[name] = new TypeMap();
|
||||
case v: v;
|
||||
}
|
||||
|
||||
if (!forName.exists(type)) {
|
||||
var path = '$name${Lambda.count(forName)}',
|
||||
usings = [];
|
||||
|
||||
var def = build(pos, type, usings, path.split('.').pop());
|
||||
|
||||
Context.defineModule(path, [def], usings);
|
||||
forName.set(type, Context.getType(path));
|
||||
}
|
||||
|
||||
return forName.get(type);
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user