cc-haxe/src/lib/KVStore.hx

package lib;

import haxe.Serializer;
import haxe.Unserializer;
import kernel.fs.FS;
import haxe.ds.StringMap;

/**
	Key value store with persistence.
**/
class KVStore {
	private var kvStore: StringMap<Dynamic> = new StringMap();
	public final namespace:String;

	public function new(namespace: String) {
		this.namespace = namespace;
	}

	public static function removeNamespace(namespace: String): Void {
		var nsFile = getNamespaceFile(namespace);
		FS.delete(nsFile);
	}

	public static function getStoreForClass(?pos:haxe.PosInfos) {
		var className = pos.className;
		return new KVStore(className);
	}

	private static function getNamespaceFile(namespace: String): String {
		return '/var/ns/$namespace';
	}

	public function load() {
		var nsFile = getNamespaceFile(this.namespace);
		if (FS.exists(nsFile)){
			var handle = FS.openRead(nsFile);
			parseFile(handle.readAll());
		}
	}

	public function save() {
		var handle = FS.openWrite(getNamespaceFile(this.namespace));

		handle.write(Serializer.run(this.kvStore));

		handle.close();
	}

	private function parseFile(content: String) {
		var unserializer = new Unserializer(content);
		this.kvStore = unserializer.unserialize();
	}

	public inline function set(key: String, value: Dynamic) {
		this.kvStore.set(key,value);
	}

	public inline function get<T>(key: String,?orElse:T = null): Null<T> {
		return this.kvStore.get(key) ?? orElse;
	}

	public inline function exists(key: String): Bool {
		return this.kvStore.exists(key);
	}

	public inline function clear() {
		this.kvStore.clear();
	}

	public inline function remove(key: String): Bool {
		return this.kvStore.remove(key);
	}

	public inline function keys(): Iterator<String> {
		return this.kvStore.keys();
	}

	public inline function keyValueIterator():KeyValueIterator<String, Dynamic> {
		return this.kvStore.keyValueIterator();
	}
}