/*
 * This file is part of libgim.
 *
 * libgim is free software: you can redistribute it and/or modify it under the
 * terms of the GNU General Public License as published by the Free Software
 * Foundation, either version 3 of the License, or (at your option) any later
 * version.
 * 
 * libgim is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with libgim.  If not, see <http://www.gnu.org/licenses/>.
 *
 * Copyright 2010 Danny Robson <danny@blubinc.net>
 */


#include "json.hpp"
#include "maths.hpp"
#include "io.hpp"

#include <cassert>
#include <deque>
#include <stdexcept>
#include <cstdlib>
#include <algorithm>
#include <sstream>

#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>


using namespace std;
using namespace util;

/*
 * Parsing
 */

struct parse_context {
    parse_context(json::node *_root):
        root  (_root),
        value (NULL),
        key   (NULL),
        start (NULL),
        stop  (NULL)
    { ; }

    json::node *root,
               *value,
               *key;
    const char *start,
               *stop;
};


%%{
    machine json;

    ## Record whether parsing was successful for future use
    action success
    { __success = true; }

    action failure {
        __success = false;
        /*std::cerr << std::endl
                  << "Failure on: '" << fc << "' in level " << top << " at " << fpc - p
                  << std::endl;
        */
    }


    action new_object { nodestack.push_back (parse_context(new json::object));  }
    action new_array  { nodestack.push_back (parse_context(new json::array));  }

    action new_object_value {
        assert (nodestack.back ().root->is_object ());
        assert (nodestack.back ().key);
        assert (nodestack.back ().value);

        if (!nodestack.back ().key->is_string ())
            throw error ("object keys must be strings");

        json::object *object = (json::object *)nodestack.back ().root;
        object->insert (nodestack.back ().key->to_string (),
                        nodestack.back ().value);
        nodestack.back ().key   = NULL;
        nodestack.back ().value = NULL;
    }

    action new_array_value {
        assert (nodestack.back ().root->is_array ());
        assert (nodestack.back ().value);

        json::array *array = (json::array *)nodestack.back ().root;
        array->insert (nodestack.back ().value);
        nodestack.back ().value = NULL;
    }

    action new_string  {
        assert (!nodestack.empty ());
        assert (!nodestack.back ().value);

        std::string value (std::string (nodestack.back ().start,
                                        nodestack.back ().stop));
        nodestack.back ().value = new json::string(value);
    }

    action new_boolean {
        assert (!nodestack.empty ());
        assert (!nodestack.back ().value);

        throw error ("unable to parse boolean");
    }

    action new_number  {
        assert (!nodestack.empty ());
        assert (!nodestack.back ().value);
        
        errno = 0;
        double value = strtod (nodestack.back ().start, NULL);
        if (errno)
            throw error ("unable to parse number");
        nodestack.back ().value = new json::number (value);
    }

    action new_null    {
        assert (!nodestack.empty ());
        assert (!nodestack.back ().value);

        nodestack.back().value = new json::null ();
    }

    action new_object_key {
        assert (!nodestack.empty ());
        assert (nodestack.back ().root->is_object ());
        assert (nodestack.back ().value);
        assert (!nodestack.back ().key);

        nodestack.back ().key   = nodestack.back ().value;
        nodestack.back ().value = NULL;
    }

    prepush {
        fsmstack.push_back  (0);
    }

    postpop {
        fsmstack.pop_back ();
        __root = nodestack.back ().root;
        if (nodestack.size () > 1)
            (nodestack.rbegin () + 1)->value = nodestack.back ().root;
        nodestack.pop_back ();
    }

    variable stack fsmstack;

    alphtype char;

    ## numerical
    exp      =   [eE]('-' | '+')? digit+;
    frac     =   '.' digit+;
    int      =   '-'? [1-9]? digit+;

    number   =   int (  frac
                      | exp
                      | frac exp)?;

    ## textual
    char     =
          any - (cntrl | '\"' | '\\')
        | '\\\"'
        | '\\\\'
        | '\\/'
        | '\\b'
        | '\\f'
        | '\\n'
        | '\\r'
        | '\\t'
        | '\\u' xdigit{4};

    string = ('"'    
              char* >{ nodestack.back ().start = fpc; }
                    %{ nodestack.back ().stop  = fpc; })
              '"' 
              @new_string;

    ## other
    boolean =
          'true'  @{ nodestack.back ().value = new json::boolean ( true); }
        | 'false' @{ nodestack.back ().value = new json::boolean (false); };

    ## components
    object   = '{' @{ fhold; fcall _object; } '}';
    array    = '[' @{ fhold; fcall  _array; } ']';

    value    =
          string  
        | boolean 
        | number  >{ nodestack.back ().start = fpc; } %{ nodestack.back ().stop = fpc; } %new_number
        | object  
        | array  
        | 'null'  %new_null;

    ## compound data types
    _array   := ('[' @new_array
                  space* ((value %new_array_value space* ',' space*)* value %new_array_value space*)?
              ']')
              $!failure
              @{ fhold; fret; };

    pair     = string %new_object_key space* ':' space* value %new_object_value;
    _object := ('{' @new_object
                 space* ((pair space* ',' space*)* pair space*)?
             '}')
             $!failure
             @{ fhold; fret; };

    json := (space* object space*)
            $!failure
            %success
            >{ __success = false; };

    write data;
}%%
    

/*
 * Node
 */

std::unique_ptr<json::node>
json::parse (const boost::filesystem::path &path)
    { return parse ((const char *)slurp (path)); }


std::unique_ptr<json::node>
json::parse (const std::string &path)
    { return parse (path.c_str (), path.c_str () + path.size ()); }

std::unique_ptr<json::node>
json::parse (const char *start,
             const char *stop) {
    bool __success = true;
    json::node *__root = NULL;
    size_t top = 0;
    int cs;
    deque <int> fsmstack;
    deque <parse_context> nodestack;

    const char *p   = start,
               *pe  = stop,
               *eof = stop;

    %%write init;
    %%write exec;

    if (!__success) 
        throw error ("unable to parse json");

    //__root->print (cout) << endl;
    assert (*__root == *__root);
    return std::unique_ptr<json::node> (__root);
}


std::unique_ptr<json::node>
json::parse (const char *start)
    { return parse (start, start + strlen (start)); }


const json::object&
json::node::to_object  (void) const
        { throw error ("node is not an object"); }


const json::array&
json::node::to_array   (void) const
        { throw error ("node is not an array"); }


const json::string&
json::node::to_string  (void) const
        { throw error ("node is not a string"); }


const json::number&
json::node::to_number  (void) const
        { throw error ("node is not a number"); }


const json::boolean&
json::node::to_boolean (void) const
        { throw error ("node is not a boolean"); }


bool
json::node::operator!=(const node &rhs) const
    { return !(*this == rhs); }


bool json::node::operator==(const char *rhs) const
    { return to_string () == rhs; }


const json::node&
json::node::operator[] (const std::string &key) const
    { return to_object ()[key]; }


const json::node&
json::node::operator[] (unsigned int idx) const
    { return to_array()[idx]; }

/*
 * Object
 */

json::object::~object () {
    for (auto i = m_values.begin (); i != m_values.end (); ++i)
        delete i->second;
}


bool
json::object::operator ==(const json::object &rhs) const {
    for (auto i = rhs.m_values.begin (), j = m_values.begin ();
         i != rhs.m_values.end () && j != m_values.end ();
         ++i, ++j)
    {
        if (i->first != j->first)
            return false;
        if ((*i->second) != (*j->second))
            return false;
    }

    return true;
}


void
json::object::insert (const std::string &_key, json::node* value)
        { m_values[_key] = value; }


const json::node&
json::object::operator[](const std::string &key) const {
    auto value = m_values.find (key);
    if (value == m_values.end ()) {
        ostringstream ss;
        ss << "no key: " << key;
        throw json::error (ss.str());
    }

    return *value->second;
}


std::ostream&
json::object::write (std::ostream &os) const {
    os << "{\n";
    {
        indenter raii(os);

        for (auto i = m_values.begin (); i != m_values.end ();) {
            os << '"' << i->first << "\" : " << *i->second;

            if (++i != m_values.end ())
                os << ",\n";
        }
    }

    os << "\n}";
    return os;
}


/*
 * Array
 */

json::array::~array() {
    for (auto i = m_values.begin(); i != m_values.end (); ++i)
        delete *i;
}


void
json::array::insert (json::node *_value)
        { m_values.push_back (_value); }


bool
json::array::operator ==(const json::array &rhs) const {
    for (auto i = rhs.m_values.begin (), j = m_values.begin ();
         i != rhs.m_values.end () && j != m_values.end ();
         ++i, ++j)
    { if ((**i) != (**j)) return false; }

    return true;
}


std::ostream&
json::array::write (std::ostream &os) const {
    os << "[\n"; 
    {
        indenter raii(os); 

        for (auto i = m_values.begin (); i != m_values.end (); ++i) {
            (*i)->write (os);

            if (i != m_values.end () - 1)
                os << ",\n";
        }
    }
    os << "\n]";
    return os;
}


/*
 * String
 */

std::ostream&
json::string::write (std::ostream &os) const {
    os << '"' << m_value << '"';
    return os;
}


bool
json::string::operator ==(const json::string &rhs) const
    { return rhs.m_value == m_value; }


/*
 * Number
 */

std::ostream&
json::number::write (std::ostream &os) const {
    os << m_value;
    return os;
}


bool
json::number::operator ==(const json::number &rhs) const 
    { return almost_equal (rhs.m_value, m_value); }


/*
 * Boolean
 */

std::ostream&
json::boolean::write (std::ostream &os) const {
    os << (m_value ? "true" : "false");
    return os;
}

bool
json::boolean::operator ==(const json::boolean &rhs) const 
    { return rhs.m_value == m_value; }


/*
 * Null
 */ 

std::ostream&
json::null::write (std::ostream &os) const {
    os << "null";
    return os;
}

ostream&
operator <<(ostream &os, const json::node &n)
    { return n.write (os); }