/++
    Simple JSON wrappers around Phobos' [std.json] to make keeping JSON storages easier.
    This is not a replacement for [std.json]; it merely extends it.

    Example:
    ---
    JSONStorage json;
    assert(json.storage.type == JSONType.null_);

    json.load("somefile.json");
    assert(json.storage.type == JSONType.object);

    json.serialiseInto!(JSONStorage.KeyOrderStrategy.inGivenOrder)
        (stdout.lockingTextWriter, [ "foo", "bar", "baz" ]);

    // Printed to screen, regardless how `.toPrettyString` would have ordered it:
    /*
        {
            "foo": {
                1,
                2,
            },
            "bar": {
                3,
                4,
            },
            "baz": {
                5,
                6,
            }
        }
    */

    // Prints keys in sorted order.
    json.serialiseInto!(JSONStorage.KeyOrderStrategy.sorted)(stdout.lockingTextWriter)

    // Use a [std.array.Appender|Appender] to serialise into a string.

    // Adding and removing values still needs the same dance as with std.json.
    // Room for future improvement.
    json.storage["qux"] = null;
    json.storage["qux"].array = null;
    json.storage["qux"].array ~= 7;
    json.storage["qux"].array ~= 8;

    json.save("somefile.json");
    ---
 +/
module lu.json;

private:

import std.json : JSONValue;
import std.range.primitives : isOutputRange;
import std.traits : isMutable;
import std.typecons : Flag, No, Yes;

public:


// JSONStorage
/++
    A wrapped [std.json.JSONValue|JSONValue] with helper functions.

    Example:
    ---
    JSONStorage s;

    s.reset();  // not always necessary

    s.storage["foo"] = null;  // JSONValue quirk
    s.storage["foo"]["abc"] = JSONValue(42);
    s.storage["foo"]["def"] = JSONValue(3.14f);
    s.storage["foo"]["ghi"] = JSONValue([ "bar", "baz", "qux" ]);
    s.storage["bar"] = JSONValue("asdf");

    assert(s.storage.length == 2);
    ---
 +/
struct JSONStorage
{
private:
    import std.json : JSONValue, parseJSON;

public:
    /// The underlying [std.json.JSONValue|JSONValue] storage of this [JSONStorage].
    JSONValue storage;

    alias storage this;

    /++
        Strategy in which to sort object-type JSON keys when we format/serialise
        the stored [storage] to string.
     +/
    enum KeyOrderStrategy
    {
        /++
            Order is as [std.json.JSONValue.toPrettyString|JSONValue.toPrettyString]
            formats it.
         +/
        passthrough,
        sorted,   /// Sorted by key.
        reverse,  /// Reversely sorted by key.

        /++
            Keys are listed in the order given in a passed `string[]` array.

            Actual keys not present in the array are not included in the output,
            and keys not existing yet present in the array are added as empty.
         +/
        inGivenOrder,
    }

    // reset
    /++
        Initialises and clears the [std.json.JSONValue|JSONValue], preparing
        it for object storage.
     +/
    void reset() @safe pure nothrow @nogc
    {
        storage.object = null;
    }

    // load
    /++
        Loads JSON from disk.

        In the case where the file doesn't exist or is otherwise invalid, then
        [std.json.JSONValue|JSONValue] is initialised to null (by way of
        [JSONStorage.reset]).

        Params:
            filename = Filename of file to read from.

        Throws:
            Whatever [std.file.readText|readText] and/or
            [std.json.parseJSON|parseJSON] throws.

            [lu.common.FileTypeMismatchException] if the filename exists
            but is not a file.
     +/
    void load(const string filename) @safe
    in (filename.length, "Tried to load an empty filename into a JSON storage")
    {
        import lu.common : FileTypeMismatchException;
        import std.file : exists, getAttributes, isFile, readText;
        import std.path : baseName;

        if (!filename.exists)
        {
            return reset();
        }
        else if (!filename.isFile)
        {
            reset();
            throw new FileTypeMismatchException("File exists but is not a file.",
                filename.baseName, cast(ushort)getAttributes(filename));
        }

        immutable fileContents = readText(filename);
        storage = parseJSON(fileContents.length ? fileContents : "{}");
    }


    // save
    /++
        Saves the JSON storage to disk. Formatting is done as specified by the
        passed [KeyOrderStrategy] argument.

        Merely leverages [serialiseInto] and [std.stdio.writeln|writeln].

        Params:
            strategy = Key order strategy in which to sort object-type JSON keys.
            filename = Filename of the file to save to.
            givenOrder = The order in which object-type keys should be listed in
                the output file. Non-existent keys are represented as empty. Not
                specified keys are omitted.
     +/
    void save(KeyOrderStrategy strategy = KeyOrderStrategy.passthrough)
        (const string filename,
        const string[] givenOrder = string[].init) @safe
    in (filename.length, "Tried to save a JSON storage to an empty filename")
    {
        import std.array : Appender;
        import std.json : JSONType;
        import std.stdio : File, writeln;

        Appender!(char[]) sink;
        sink.reserve(1024);  // guesstimate

        if (storage.type == JSONType.object)
        {
            static if (strategy == KeyOrderStrategy.inGivenOrder)
            {
                serialiseInto!strategy(sink, givenOrder);
            }
            else
            {
                serialiseInto!strategy(sink);
            }
        }
        else
        {
            serialiseInto!(KeyOrderStrategy.passthrough)(sink);
        }

        File(filename, "w").writeln(sink.data);
    }


    // serialiseInto
    /++
        Formats an object-type JSON storage into an output range sink.

        Top-level keys are sorted as per the passed [KeyOrderStrategy]. This
        overload is specialised for [KeyOrderStrategy.inGivenOrder].

        Params:
            strategy = Order strategy in which to sort top-level keys.
            sink = Output sink to fill with formatted output.
            givenOrder = The order in which object-type keys should be listed in
                the output file. Non-existent keys are represented as empty.
                Not specified keys are omitted.
     +/
    void serialiseInto(KeyOrderStrategy strategy : KeyOrderStrategy.inGivenOrder, Sink)
        (auto ref Sink sink, const string[] givenOrder) @safe
    if (isOutputRange!(Sink, char[]))
    in (givenOrder.length, "Tried to serialise a JSON storage in order given without a given order")
    {
        if (storage.isNull)
        {
            sink.put("{}");
            return;
        }

        sink.put("{\n");

        foreach (immutable i, immutable key; givenOrder)
        {
            import std.format : formattedWrite;

            sink.formattedWrite("    \"%s\": ", key);

            if (const entry = key in storage)
            {
                import lu.string : indentInto;
                entry.toPrettyString.indentInto(sink, 1, 1);
            }
            else
            {
                sink.put("{}");
            }

            sink.put((i+1 < givenOrder.length) ? ",\n" : "\n");
        }

        sink.put("}");
    }


    // serialiseInto
    /++
        Formats an object-type JSON storage into an output range sink.

        Top-level keys are sorted as per the passed [KeyOrderStrategy]. This
        overload is specialised for strategies other than
        [KeyOrderStrategy.inGivenOrder], and as such takes one parameter fewer.

        Params:
            strategy = Order strategy in which to sort top-level keys.
            sink = Output sink to fill with formatted output.
     +/
    void serialiseInto(KeyOrderStrategy strategy = KeyOrderStrategy.passthrough, Sink)
        (auto ref Sink sink) @safe
    if ((strategy != KeyOrderStrategy.inGivenOrder) && isOutputRange!(Sink, char[]))
    {
        if (storage.isNull)
        {
            sink.put("{}");
            return;
        }

        static if (strategy == KeyOrderStrategy.passthrough)
        {
            // Just pass through and save .toPrettyString; keep original behaviour.
            sink.put(storage.toPrettyString);
        }
        else static if ((strategy == KeyOrderStrategy.sorted) ||
            (strategy == KeyOrderStrategy.reverse))
        {
            import std.array : array;
            import std.format : formattedWrite;
            import std.range : enumerate;
            import std.algorithm.sorting : sort;

            auto rawRange = storage
                .objectNoRef
                .byKey
                .array
                .sort;

            static if (strategy == KeyOrderStrategy.reverse)
            {
                import std.range : retro;
                auto range = rawRange.retro;
            }
            else static if (strategy == KeyOrderStrategy.sorted)
            {
                // Already sorted
                alias range = rawRange;
            }
            else
            {
                static assert(0, "Logic error; unexpected `KeyOrderStrategy` " ~
                    "passed to `serialiseInto`");
            }

            sink.put("{\n");

            foreach(immutable i, immutable key; range.enumerate)
            {
                import lu.string : indentInto;
                sink.formattedWrite("    \"%s\": ", key);
                storage[key].toPrettyString.indentInto(sink, 1, 1);
                sink.put((i+1 < range.length) ? ",\n" : "\n");
            }

            sink.put("}");
        }
        else
        {
            static assert(0, "Logic error; invalid `KeyOrderStrategy` " ~
                "passed to `serialiseInto`");
        }
    }

    ///
    @system unittest
    {
        import std.array : Appender;
        import std.json;

        JSONStorage this_;
        Appender!(char[]) sink;

        // Original JSON
        this_.storage = parseJSON(
`{
"#abc":
{
"hirrsteff" : "o",
"foobar" : "v"
},
"#def":
{
"harrsteff": "v",
"flerpeloso" : "o"
},
"#zzz":
{
"asdf" : "v"
}
}`);

        // KeyOrderStrategy.passthrough
        this_.serialiseInto!(KeyOrderStrategy.passthrough)(sink);
        assert((sink.data ==
`{
    "#abc": {
        "foobar": "v",
        "hirrsteff": "o"
    },
    "#def": {
        "flerpeloso": "o",
        "harrsteff": "v"
    },
    "#zzz": {
        "asdf": "v"
    }
}`), '\n' ~ sink.data);
        sink.clear();

        // KeyOrderStrategy.sorted
        this_.serialiseInto!(KeyOrderStrategy.sorted)(sink);
        assert((sink.data ==
`{
    "#abc": {
        "foobar": "v",
        "hirrsteff": "o"
    },
    "#def": {
        "flerpeloso": "o",
        "harrsteff": "v"
    },
    "#zzz": {
        "asdf": "v"
    }
}`), '\n' ~ sink.data);
        sink.clear();

        // KeyOrderStrategy.reverse
        this_.serialiseInto!(KeyOrderStrategy.reverse)(sink);
        assert((sink.data ==
`{
    "#zzz": {
        "asdf": "v"
    },
    "#def": {
        "flerpeloso": "o",
        "harrsteff": "v"
    },
    "#abc": {
        "foobar": "v",
        "hirrsteff": "o"
    }
}`), '\n' ~ sink.data);
        sink.clear();

        // KeyOrderStrategy.inGivenOrder
        this_.serialiseInto!(KeyOrderStrategy.inGivenOrder)(sink, [ "#def", "#abc", "#foo", "#fighters" ]);
        assert((sink.data ==
`{
    "#def": {
        "flerpeloso": "o",
        "harrsteff": "v"
    },
    "#abc": {
        "foobar": "v",
        "hirrsteff": "o"
    },
    "#foo": {},
    "#fighters": {}
}`), '\n' ~ sink.data);
        sink.clear();

        // Empty JSONValue
        JSONStorage this2;
        this2.serialiseInto(sink);
        assert((sink.data ==
`{}`), '\n' ~ sink.data);
    }
}

///
unittest
{
    import std.conv : text;
    import std.json : JSONValue;

    JSONStorage s;
    s.reset();

    s.storage["key"] = null;
    s.storage["key"]["subkey1"] = "abc";
    s.storage["key"]["subkey2"] = "def";
    s.storage["key"]["subkey3"] = "ghi";
    assert((s.storage["key"].object.length == 3), s.storage["key"].object.length.text);

    s.storage["foo"] = null;
    s.storage["foo"]["arr"] = JSONValue([ "blah "]);
    s.storage["foo"]["arr"].array ~= JSONValue("bluh");
    assert((s.storage["foo"]["arr"].array.length == 2), s.storage["foo"]["arr"].array.length.text);
}


// populateFromJSON
/++
    Recursively populates a passed associative or dynamic array with the
    contents of a [std.json.JSONValue|JSONValue].

    This is used where we want to store information on disk but keep it in
    memory without the overhead of dealing with [std.json.JSONValue|JSONValue]s.

    Note: This only works with [std.json.JSONValue|JSONValue]s that conform to
    arrays and associative arrays, not such that mix element/value types.

    Params:
        target = Reference to target array or associative array to write to.
        json = Source [std.json.JSONValue|JSONValue] to sync the contents with.
        lowercaseKeys = Whether or not to save string keys in lowercase.
        lowercaseValues = Whether or not to save final string values in lowercase.

    Throws:
        [object.Exception|Exception] if the passed [std.json.JSONValue|JSONValue]
        had unexpected types.
 +/
void populateFromJSON(T)
    (ref T target,
    const JSONValue json,
    const Flag!"lowercaseKeys" lowercaseKeys = No.lowercaseKeys,
    const Flag!"lowercaseValues" lowercaseValues = No.lowercaseValues) @safe
if (isMutable!T)
{
    import std.traits : ValueType, isAssociativeArray, isArray, isDynamicArray, isSomeString;
    import std.range : ElementEncodingType;

    static if (isAssociativeArray!T || (isArray!T && !isSomeString!T))
    {
        static if (isAssociativeArray!T)
        {
            const aggregate = json.objectNoRef;
            alias Value = ValueType!T;
        }
        else static if (isArray!T)
        {
            const aggregate = json.arrayNoRef;
            alias Value = ElementEncodingType!T;

            static if (isDynamicArray!T)
            {
                target.reserve(aggregate.length);
            }
        }
        else
        {
            static assert(0, "`populateFromJSON` was passed an unsupported type `" ~ T.stringof ~ "`");
        }

        foreach (ikey, const valJSON; aggregate)
        {
            static if (isAssociativeArray!T)
            {
                static if (isSomeString!Value)
                {
                    if (lowercaseKeys)
                    {
                        import std.uni : toLower;
                        ikey = ikey.toLower;
                    }
                }

                target[ikey] = Value.init;
            }
            else static if (isDynamicArray!T)
            {
                if (ikey >= target.length) target ~= Value.init;
            }

            populateFromJSON(target[ikey], valJSON);
        }

        /*static if (isAssociativeArray!T)
        {
            // This would make it @system.
            target.rehash();
        }*/
    }
    else
    {
        import std.conv : to;
        import std.json : JSONType;

        with (JSONType)
        final switch (json.type)
        {
        case string:
            target = json.str.to!T;

            static if (isSomeString!T)
            {
                if (lowercaseValues)
                {
                    import std.uni : toLower;
                    target = target.toLower;
                }
            }
            break;

        case integer:
            // .integer returns long, keep .to for int compatibility
            target = json.integer.to!T;
            break;

        case uinteger:
            // as above
            target = json.uinteger.to!T;
            break;

        case float_:
            target = json.floating.to!T;
            break;

        case true_:
        case false_:
            target = json.boolean.to!T;
            break;

        case null_:
            // Silently do nothing
            break;

        case object:
        case array:
            import std.format : format;
            throw new Exception("Type mismatch when populating a `%s` with a `%s`"
                .format(T.stringof, json.type));
        }
    }
}

///
unittest
{
    import std.json : JSONType, JSONValue;

    {
        long[string] aa =
        [
            "abc" : 123,
            "def" : 456,
            "ghi" : 789,
        ];

        JSONValue j = JSONValue(aa);
        typeof(aa) fromJSON;

        foreach (immutable key, const value; j.objectNoRef)
        {
            fromJSON[key] = value.integer;
        }

        assert(aa == fromJSON);  // not is

        auto aaCopy = aa.dup;

        aa["jlk"] = 12;
        assert(aa != fromJSON);

        aa = typeof(aa).init;
        populateFromJSON(aa, j);
        assert(aa == aaCopy);
    }
    {
        auto aa =
        [
            "abc" : true,
            "def" : false,
            "ghi" : true,
        ];

        JSONValue j = JSONValue(aa);
        typeof(aa) fromJSON;

        foreach (immutable key, const value; j.objectNoRef)
        {
            if (value.type == JSONType.true_) fromJSON[key] = true;
            else if (value.type == JSONType.false_) fromJSON[key] = false;
            else
            {
                assert(0);
            }
        }

        assert(aa == fromJSON);  // not is

        auto aaCopy = aa.dup;

        aa["jkl"] = false;
        assert(aa != fromJSON);

        aa = typeof(aa).init;
        populateFromJSON(aa, j);
        assert(aa == aaCopy);
    }
    {
        auto arr = [ "abc", "def", "ghi", "jkl" ];

        JSONValue j = JSONValue(arr);
        typeof(arr) fromJSON;

        foreach (const value; j.arrayNoRef)
        {
            fromJSON ~= value.str;
        }

        assert(arr == fromJSON);  // not is

        auto arrCopy = arr.dup;

        arr[0] = "no";
        assert(arr != arrCopy);

        arr = [];
        populateFromJSON(arr, j);
        assert(arr == arrCopy);
    }
    {
        auto aa =
        [
            "abc" : [ "def", "ghi", "jkl" ],
            "def" : [ "MNO", "PQR", "STU" ],
        ];

        JSONValue j = JSONValue(aa);
        typeof(aa)fromJSON;

        foreach (immutable key, const arrJSON; j.objectNoRef)
        {
            foreach (const entry; arrJSON.arrayNoRef)
            {
                fromJSON[key] ~= entry.str;
            }
        }

        assert(aa == fromJSON);  // not is

        auto aaCopy = aa.dup;
        aaCopy["abc"] = aa["abc"].dup;

        aa["abc"][0] = "no";
        aa["ghi"] ~= "VWXYZ";
        assert(aa != fromJSON);

        aa = typeof(aa).init;
        populateFromJSON(aa, j);
        assert(aa == aaCopy);
    }
    {
        int[3] arr = [ 1, 2, 3 ];

        JSONValue j = JSONValue(arr);

        int[3] arr2;
        arr2.populateFromJSON(j);
        assert(arr2 == arr);
    }
}