// This file is distributed under the BSD License.
// See LICENSE.TXT for details.

#ifndef CHAISCRIPT_ENGINE_HPP_
#define CHAISCRIPT_ENGINE_HPP_

#include <exception>
#include <fstream>


#include "chaiscript_prelude.hpp"
#include "chaiscript_parser.hpp"

namespace chaiscript
{
    template <typename Eval_Engine>
    class ChaiScript_System {
        Eval_Engine engine;
        ChaiScript_Parser parser;

    public:
        ChaiScript_System()  {

        }

        Eval_Engine &get_eval_engine() {
            return engine;
        }

        void debug_print(TokenPtr t, std::string prepend = "") {
            std::cout << prepend << "(" << token_type_to_string(t->identifier) << ") " << t->text << " : " << t->start.line << ", " << t->start.column << std::endl;
            for (unsigned int j = 0; j < t->children.size(); ++j) {
                debug_print(t->children[j], prepend + "  ");
            }
        }

        const dispatchkit::Boxed_Value eval(const std::vector<dispatchkit::Boxed_Value> &vals) {
            std::string val;

            try {
                val = dispatchkit::boxed_cast<std::string &>(vals[0]);
            }
            catch (EvalError &ee) {
                throw EvalError("Can not evaluate string: " + val + " reason: " + ee.reason, TokenPtr());
            }
            catch (std::exception &e) {
                throw EvalError("Can not evaluate string: " + val, TokenPtr());
            }
            return evaluate_string(val);
        }

        std::string load_file(const char *filename) {
            std::ifstream infile (filename, std::ios::in | std::ios::ate);

            if (!infile.is_open()) {
                std::string fname = filename;
                throw std::runtime_error("Can not open: " + fname);
            }

            std::streampos size = infile.tellg();
            infile.seekg(0, std::ios::beg);

            std::vector<char> v(size);
            infile.read(&v[0], size);

            std::string ret_val (v.empty() ? std::string() : std::string (v.begin(), v.end()).c_str());

            return ret_val;
        }

        void build_eval_system() {
            dispatchkit::Bootstrap::bootstrap(engine);
            dispatchkit::bootstrap_vector<std::vector<dispatchkit::Boxed_Value> >(engine, "Vector");
            dispatchkit::bootstrap_map<std::map<std::string, dispatchkit::Boxed_Value> >(engine, "Map");
            dispatchkit::bootstrap_pair<std::pair<dispatchkit::Boxed_Value, dispatchkit::Boxed_Value > >(engine, "Pair");

            engine.register_function(boost::shared_ptr<dispatchkit::Proxy_Function>(
                  new dispatchkit::Dynamic_Proxy_Function(boost::bind(&ChaiScript_System<Eval_Engine>::eval, boost::ref(*this), _1), 1)), "eval");

            evaluate_string(chaiscript_prelude);
        }

        dispatchkit::Boxed_Value evaluate_string(const std::string &input, const char *filename = "__EVAL__") {
            //debug_print(tokens);
            dispatchkit::Boxed_Value value;

            try {
                parser.clear_match_stack();
                if (parser.parse(input, filename)) {
                    //show_match_stack();
                    value = eval_token<Eval_Engine>(engine, parser.ast());
                    //clear_match_stack();
                }
            }
            catch (const ReturnValue &rv) {
                value = rv.retval;
            }
            catch (Parse_Error &pe) {
                std::cout << pe.reason << " in " << pe.filename << " at " << pe.position.line << ", " << pe.position.column << std::endl;
                parser.clear_match_stack();
            }
            catch (EvalError &ee) {
                if (filename != std::string("__EVAL__")) {
                    std::cout << "Eval error: \"" << ee.reason << "\" in '" << ee.location->filename << "' line: " << ee.location->start.line+1 << std::endl;
                }
                else {
                    std::cout << "Eval error: \"" << ee.reason << "\"" << std::endl;
                }
            }
            catch (std::exception &e) {
                std::cout << "Exception: " << e.what() << std::endl;
            }

            return value;
        }

        dispatchkit::Boxed_Value evaluate_file(const char *filename) {
            return evaluate_string(load_file(filename), filename);
        }
    };

    typedef ChaiScript_System<dispatchkit::Dispatch_Engine> ChaiScript_Engine;
}
#endif /* CHAISCRIPT_ENGINE_HPP_ */