/**
	Defines a string based multi-map with conserved insertion order.

	Copyright: © 2012-2014 Sönke Ludwig
	License: Subject to the terms of the MIT license, as written in the included LICENSE.txt file.
	Authors: Sönke Ludwig
*/
module vibe.utils.dictionarylist;

import vibe.utils.string : icmp2;
import std.exception : enforce;


/**
	Behaves similar to $(D VALUE[string]) but the insertion order is not changed
	and multiple values per key are supported.

	This kind of map is used for MIME headers (e.g. for HTTP, see
	vibe.inet.message.InetHeaderMap), or for form data
	(vibe.inet.webform.FormFields). Note that the map can contain fields with
	the same key multiple times if addField is used for insertion. Insertion
	order is preserved.

	Note that despite case not being relevant for matching keyse, iterating
	over the map will yield	the original case of the key that was put in.

	Insertion and lookup has O(n) complexity.
*/
struct DictionaryList(VALUE, bool case_sensitive = true, size_t NUM_STATIC_FIELDS = 32, bool USE_HASHSUM = false) {
	import std.typecons : Tuple;

	private {
		alias KeyValue = Tuple!(string, "key", ValueType, "value");

		static struct Field {
			static if (USE_HASHSUM) uint keyCheckSum;
			else {
				enum keyCheckSum = 0;
				this(uint, string key, VALUE value) { this.key = key; this.value = value; }
			}
			KeyValue tuple;
			@property ref inout(string) key() inout { return tuple.key; }
			@property ref inout(VALUE) value() inout { return tuple.value; }
		}
		Field[NUM_STATIC_FIELDS] m_fields;
		size_t m_fieldCount = 0;
		Field[] m_extendedFields;

		enum bool safeValueCopy = __traits(compiles, (VALUE v) @safe { VALUE vc; vc = v; });
		template typedGet(T) {
			enum typedGet = __traits(compiles, (VALUE v) { return v.get!T; });
		}
		template canAssign(T) {
			enum canAssign = __traits(compiles, (T t) { VALUE v = t; });
		}
	}

	alias ValueType = VALUE;

	struct FieldTuple { string key; ValueType value; }

	/** The number of fields present in the map.
	*/
	@property size_t length() const { return m_fieldCount + m_extendedFields.length; }

	/// Supports serialization using vibe.data.serialization.
	static DictionaryList fromRepresentation(FieldTuple[] array)
	{
		DictionaryList ret;
		foreach (ref v; array) ret.addField(v.key, v.value);
		return ret;
	}
	/// ditto
	FieldTuple[] toRepresentation() const {
		FieldTuple[] ret;
		foreach (k, ref v; this.byKeyValue) ret ~= FieldTuple(k, v);
		return ret;
	}

	/** Generates an associative-array equivalent string representation of the dictionary.
	*/
	void toString(scope void delegate(const(char)[] str) @safe sink)
	const {
		sink("[");
		bool first = true;

		foreach (k, v; this.byKeyValue) {
			if (!first) sink(", ");
			else first = false;
			() @trusted {
				import std.format : formattedWrite;
				string[] ka = (&k)[0 .. 1];
				ValueType[] va = (&v)[0 .. 1];
				sink.formattedWrite("%(%s%): %(%s%)", ka, va);
			} ();
		}
		sink("]");
	}
	/// ditto
	void toString(scope void delegate(const(char)[] str) @system sink)
	const {
		toString(cast(void delegate(const(char)[]) @safe)sink);
	}
	/// ditto
	string toString()
	const {
		import std.array : appender;
		auto ret = appender!string();
		toString((s) { ret.put(s); });
		return ret.data;
	}

	/** Removes the first field that matches the given key.
	*/
	void remove(string key)
	{
		static if (USE_HASHSUM) auto keysum = computeCheckSumI(key);
		enum keysum = 0;
		auto idx = getIndex(m_fields[0 .. m_fieldCount], key, keysum);
		if( idx >= 0 ){
			auto slice = m_fields[0 .. m_fieldCount];
			removeFromArrayIdx(slice, idx);
			m_fieldCount--;
		} else {
			idx = getIndex(m_extendedFields, key, keysum);
			enforce(idx >= 0);
			removeFromArrayIdx(m_extendedFields, idx);
		}
	}

	/** Removes all fields that matches the given key.
	*/
	void removeAll(string key)
	{
		static if (USE_HASHSUM) auto keysum = computeCheckSumI(key);
		else enum keysum = 0;
		for (size_t i = 0; i < m_fieldCount;) {
			if (m_fields[i].keyCheckSum == keysum && matches(m_fields[i].key, key)) {
				auto slice = m_fields[0 .. m_fieldCount];
				removeFromArrayIdx(slice, i);
				m_fieldCount--;
			} else i++;
		}

		for (size_t i = 0; i < m_extendedFields.length;) {
			if (m_fields[i].keyCheckSum == keysum && matches(m_fields[i].key, key))
				removeFromArrayIdx(m_extendedFields, i);
			else i++;
		}
	}

	/// Used by `remove` and `removeAll`
	private static void removeFromArrayIdx(ref Field[] array, size_t idx)
	{
		foreach( j; idx+1 .. array.length)
			array[j-1] = array[j];
		array.length = array.length-1;
	}

	/** Adds a new field to the map.

		The new field will be added regardless of any existing fields that
		have the same key, possibly resulting in duplicates. Use opIndexAssign
		if you want to avoid duplicates.
	*/
	void addField(string key, ValueType value)
	{
		static if (USE_HASHSUM) auto keysum = computeCheckSumI(key);
		else enum keysum = 0;
		if (m_fieldCount < m_fields.length)
			m_fields[m_fieldCount++] = Field(keysum, key, value);
		else m_extendedFields ~= Field(keysum, key, value);
	}

	void addField(T)(string key, T value)
	if (canAssign!T)
	{
		ValueType convertedValue = value;
		addField(key, convertedValue);
	}

	/** Returns the first field that matches the given key.

		If no field is found, def_val is returned.
	*/
	inout(ValueType) get(string key, lazy inout(ValueType) def_val = ValueType.init)
	inout {
		if (auto pv = key in this) return *pv;
		return def_val;
	}

	// DMD bug: cannot set T.init as default value for def_val parameter,
	// because compilation fails with message:
	//      Error: undefined identifier 'T'
	/// ditto
	inout(T) get(T)(string key, lazy inout(T) def_val)
	inout if (typedGet!T) {
		if (auto pv = key in this) return (*pv).get!T;
		return def_val;
	}

	/// ditto
	inout(T) get(T)(string key) // Work around DMD bug
	inout if (typedGet!T) {
		return get!T(key, inout(T).init);
	}

	/** Returns all values matching the given key.

		Note that the version returning an array will allocate for each call.
	*/
	const(ValueType)[] getAll(string key)
	const @trusted { // appender
		import std.array;
		auto ret = appender!(const(ValueType)[])();
		getAll(key, (v) @trusted { ret.put(v); });
		return ret.data;
	}
	/// ditto
	void getAll(string key, scope void delegate(const(ValueType)) @safe nothrow del)
	const {
		getAllImpl(key, del);
	}
	/// ditto
	void getAll(string key, scope void delegate(const(ValueType)) @safe del)
	const {
		getAllImpl(key, del);
	}

	private void getAllImpl(C)(string key, scope C del)
	const {
		static if (USE_HASHSUM) uint keysum = computeCheckSumI(key);
		else enum keysum = 0;
		foreach (ref f; m_fields[0 .. m_fieldCount]) {
			static if (USE_HASHSUM)
				if (f.keyCheckSum != keysum) continue;
			if (matches(f.key, key)) del(f.value);
		}
		foreach (ref f; m_extendedFields) {
			static if (USE_HASHSUM)
				if (f.keyCheckSum != keysum) continue;
			if (matches(f.key, key)) del(f.value);
		}
	}

	/** Returns the first value matching the given key.
	*/
	inout(ValueType) opIndex(string key)
	inout {
		auto pitm = key in this;
		enforce(pitm !is null, "Accessing non-existent key '"~key~"'.");
		return *pitm;
	}

	/** Adds or replaces the given field with a new value.
	*/
	ValueType opIndexAssign(ValueType val, string key)
	{
		static if (USE_HASHSUM) auto keysum = computeCheckSumI(key);
		else enum keysum = 0;
		auto pitm = key in this;
		if (pitm) *pitm = val;
		else if (m_fieldCount < m_fields.length) m_fields[m_fieldCount++] = Field(keysum, key, val);
		else m_extendedFields ~= Field(keysum, key, val);
		return val;
	}

	/// ditto
	ValueType opIndexAssign(T)(T val, string key)
	if (canAssign!T)
	{
		ValueType convertedVal = val;
		return opIndexAssign(convertedVal, key);
	}

	/** Returns a pointer to the first field that matches the given key.
	*/
	inout(ValueType)* opBinaryRight(string op)(string key) inout if(op == "in")
	{
		static if (USE_HASHSUM) uint keysum = computeCheckSumI(key);
		else enum keysum = 0;
		auto idx = getIndex(m_fields[0 .. m_fieldCount], key, keysum);
		if (idx >= 0) return &m_fields[idx].tuple[1];
		idx = getIndex(m_extendedFields, key, keysum);
		if (idx >= 0) return &m_extendedFields[idx].tuple[1];
		return null;
	}
	/// ditto
	bool opBinaryRight(string op)(string key) inout if(op == "!in") {
		return !(key in this);
	}

	/** Iterates over all fields, including duplicates.
	*/
	auto byKeyValue() {return Rng!false(&this, 0); }
	/// ditto
	auto byKeyValue() const { return Rng!true(&this, 0); }
	/// ditto
	auto byKey() inout { import std.algorithm.iteration : map; return byKeyValue().map!(p => p[0]); }
	/// ditto
	auto byValue() inout { import std.algorithm.iteration : map; return byKeyValue().map!(p => p[1]); }

	static if (is(typeof({ const(ValueType) v; ValueType w; w = v; }))) {
		/** Duplicates the header map.
		*/
		@property DictionaryList dup()
		const {
			DictionaryList ret;
			ret.m_fields[0 .. m_fieldCount] = m_fields[0 .. m_fieldCount];
			ret.m_fieldCount = m_fieldCount;
			ret.m_extendedFields = m_extendedFields.dup;
			return ret;
		}
	}

	private ptrdiff_t getIndex(in Field[] map, string key, uint keysum)
	const {
		foreach (i, ref const(Field) entry; map) {
			static if (USE_HASHSUM) if (entry.keyCheckSum != keysum) continue;
			if (matches(entry.key, key)) return i;
		}
		return -1;
	}

	private static bool matches(string a, string b)
	{
		static if (case_sensitive) return a == b;
		else return a.length == b.length && icmp2(a, b) == 0;
	}

	// very simple check sum function with a good chance to match
	// strings with different case equal
	static if (USE_HASHSUM) private static uint computeCheckSumI(string s)
	@trusted {
		uint csum = 0;
		immutable(char)* pc = s.ptr, pe = s.ptr + s.length;
		for (; pc != pe; pc++) {
			static if (case_sensitive) csum ^= *pc;
			else csum ^= *pc & 0x1101_1111;
			csum = (csum << 1) | (csum >> 31);
		}
		return csum;
	}

	private static struct Rng(bool CONST) {
	@safe nothrow @nogc:
		static if (CONST) {
			alias KVT = const(KeyValue);
			const(DictionaryList)* list;
		} else {
			alias KVT = KeyValue;
			DictionaryList* list;
		}
		size_t idx;

		@property Rng save() { return this; }

		@property bool empty() const { return idx >= list.length; }
		@property ref KVT front() {
			if (idx < list.m_fieldCount)
				return list.m_fields[idx].tuple;
			return list.m_extendedFields[idx - list.m_fieldCount].tuple;
		}
		void popFront() { idx++; }
	}
}

@safe unittest
{
	assert(DictionaryList!(string, true, 2).safeValueCopy);
}

@safe unittest {
	DictionaryList!(int, true) a;
	a.addField("a", 1);
	a.addField("a", 2);
	assert(a["a"] == 1);
	assert(a.getAll("a") == [1, 2]);
	a["a"] = 3;
	assert(a["a"] == 3);
	assert(a.getAll("a") == [3, 2]);
	a.removeAll("a");
	assert(a.getAll("a").length == 0);
	assert(a.get("a", 4) == 4);
	a.addField("b", 2);
	a.addField("b", 1);
	a.remove("b");
	assert(a.getAll("b") == [1]);

	DictionaryList!(int, false) b;
	b.addField("a", 1);
	b.addField("A", 2);
	assert(b["A"] == 1);
	assert(b.getAll("a") == [1, 2]);
}

@safe nothrow unittest {
	DictionaryList!(int, true) a;
	a.addField("a", 1);
	a.addField("a", 2);
	assert(a.getAll("a") == [1, 2]);
	assert("a" in a);
	a["a"] = 3;
	assert(a.getAll("a") == [3, 2]);
	a.removeAll("a");
	assert(a.getAll("a").length == 0);
}

unittest {
	import std.variant : Variant;
	DictionaryList!(Variant) c;
	c["a"] = true;
	c["b"] = "Hello";
	assert(c.get("a").type == typeid(bool));
	assert(c.get!string("b") == "Hello");
	assert(c.get!int("c") == int.init);
	c.addField("d", 9);
	c.addField("d", "bar");
	assert(c.getAll("d") == [ cast(Variant) 9, cast(Variant) "bar" ]);
}

@safe unittest {
	import std.conv : text;
	DictionaryList!int l;
	l["foo"] = 42;
	l["bar"] = 43;
	assert(text(l) == `["foo": 42, "bar": 43]`, text(l));
	assert(l.toString() == `["foo": 42, "bar": 43]`, l.toString());
}

// Issue 2004
unittest {
	import std.variant : Variant;
	DictionaryList!Variant l;
	class C {
		int x = 123;
	}
	l["foo"] = new C;
	auto c = l.get!C("foo");
	assert(c.x == 123);
}