#include "npc.hpp"
//    ast/npc.cpp - Structure of non-player characters (including mobs).
//
//    Copyright © 2014 Ben Longbons <b.r.longbons@gmail.com>
//
//    This file is part of The Mana World (Athena server)
//
//    This program 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.
//
//    This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.

#include "../compat/memory.hpp"

#include "../io/cxxstdio.hpp"
#include "../io/extract.hpp"

#include "../mmo/extract_enums.hpp"

#include "../high/extract_mmo.hpp"

#include "../poison.hpp"


namespace tmwa
{
namespace npc
{
namespace parse
{
    // separate file because virtual
    TopLevel::~TopLevel() {}

#define TRY_EXTRACT(bit, var) ({ if (!extract(bit.data, &var.data)) return Err(bit.span.error_str("failed to extract"_s #var)); var.span = bit.span; })

    static
    Result<Warp> parse_warp(io::LineSpan span, std::vector<Spanned<std::vector<Spanned<RString>>>>& bits)
    {
        if (bits.size() != 4)
        {
            return Err(span.error_str("expect 4 |component|s"_s));
        }
        if (bits[0].data.size() != 3)
        {
            return Err(bits[0].span.error_str("expect 3 ,component,s"_s));
        }
        assert(bits[1].data.size() == 1);
        assert(bits[1].data[0].data == "warp"_s);
        if (bits[2].data.size() != 1)
        {
            return Err(bits[2].span.error_str("expect 1 ,component,s"_s));
        }
        if (bits[3].data.size() != 5)
        {
            return Err(bits[3].span.error_str("expect 5 ,component,s"_s));
        }

        Warp warp;
        warp.span = span;
        TRY_EXTRACT(bits[0].data[0], warp.m);
        TRY_EXTRACT(bits[0].data[1], warp.x);
        TRY_EXTRACT(bits[0].data[2], warp.y);
        warp.key_span = bits[1].data[0].span;
        TRY_EXTRACT(bits[2].data[0], warp.name);
        TRY_EXTRACT(bits[3].data[0], warp.xs);
        TRY_EXTRACT(bits[3].data[1], warp.ys);
        TRY_EXTRACT(bits[3].data[2], warp.to_m);
        TRY_EXTRACT(bits[3].data[3], warp.to_x);
        TRY_EXTRACT(bits[3].data[4], warp.to_y);
        return Ok(std::move(warp));
    }
    static
    Result<Shop> parse_shop(io::LineSpan span, std::vector<Spanned<std::vector<Spanned<RString>>>>& bits)
    {
        if (bits.size() != 4)
        {
            return Err(span.error_str("expect 4 |component|s"_s));
        }
        if (bits[0].data.size() != 4)
        {
            return Err(bits[0].span.error_str("expect 4 ,component,s"_s));
        }
        assert(bits[1].data.size() == 1);
        assert(bits[1].data[0].data == "shop"_s);
        if (bits[2].data.size() != 1)
        {
            return Err(bits[2].span.error_str("expect 1 ,component,s"_s));
        }
        if (bits[3].data.size() < 2)
        {
            return Err(bits[3].span.error_str("expect at least 2 ,component,s"_s));
        }

        Shop shop;
        shop.span = span;
        TRY_EXTRACT(bits[0].data[0], shop.m);
        TRY_EXTRACT(bits[0].data[1], shop.x);
        TRY_EXTRACT(bits[0].data[2], shop.y);
        TRY_EXTRACT(bits[0].data[3], shop.d);
        shop.key_span = bits[1].data[0].span;
        TRY_EXTRACT(bits[2].data[0], shop.name);
        TRY_EXTRACT(bits[3].data[0], shop.npc_class);
        shop.items.span = bits[3].span;
        shop.items.span.begin = bits[3].data[1].span.begin;
        for (size_t i = 1; i < bits[3].data.size(); ++i)
        {
            shop.items.data.emplace_back();
            auto& item = shop.items.data.back();
            auto& data = bits[3].data[i];
            assert(data.span.begin.line == data.span.end.line);
            item.span = data.span;

            XString name;
            if (!extract(data.data, record<':'>(&name, &item.data.value.data)))
                return Err(data.span.error_str("Failed to split item:value"_s));
            item.data.name.span = item.span;
            item.data.name.span.end.column = item.data.name.span.begin.column + name.size() - 1;
            item.data.value.span = item.span;
            item.data.value.span.begin.column = item.data.name.span.begin.column + name.size() + 1;
            if (name.endswith(' '))
            {
                item.data.name.span.warning("Shop item has useless space before the colon"_s);
                name = name.rstrip();
            }
            if (name.is_digit10())
            {
                item.data.name.span.warning("Shop item is an id; should be a name"_s);
            }
            if (!extract(name, &item.data.name.data))
                return Err("item name problem (too long?)"_s);
        }
        return Ok(std::move(shop));
    }
    static
    Result<Monster> parse_monster(io::LineSpan span, std::vector<Spanned<std::vector<Spanned<RString>>>>& bits)
    {
        if (bits.size() != 4)
        {
            return Err(span.error_str("expect 4 |component|s"_s));
        }
        if (bits[0].data.size() != 3 && bits[0].data.size() != 5)
        {
            return Err(bits[0].span.error_str("expect 3 or 5 ,component,s"_s));
        }
        assert(bits[1].data.size() == 1);
        assert(bits[1].data[0].data == "monster"_s);
        if (bits[2].data.size() != 1)
        {
            return Err(bits[2].span.error_str("expect 1 ,component,s"_s));
        }
        if (bits[3].data.size() != 2 && bits[3].data.size() != 4 && bits[3].data.size() != 5)
        {
            return Err(bits[3].span.error_str("expect 2, 4, or 5 ,component,s"_s));
        }

        Monster mob;
        mob.span = span;
        TRY_EXTRACT(bits[0].data[0], mob.m);
        TRY_EXTRACT(bits[0].data[1], mob.x);
        TRY_EXTRACT(bits[0].data[2], mob.y);
        if (bits[0].data.size() >= 5)
        {
            TRY_EXTRACT(bits[0].data[3], mob.xs);
            TRY_EXTRACT(bits[0].data[4], mob.ys);
        }
        else
        {
            mob.xs.data = 0;
            mob.xs.span = bits[0].data[2].span;
            mob.xs.span.end.column++;
            mob.xs.span.begin.column = mob.xs.span.end.column;
            mob.ys.data = 0;
            mob.ys.span = mob.xs.span;
        }
        mob.key_span = bits[1].data[0].span;
        TRY_EXTRACT(bits[2].data[0], mob.name);
        TRY_EXTRACT(bits[3].data[0], mob.mob_class);
        TRY_EXTRACT(bits[3].data[1], mob.num);
        if (bits[3].data.size() >= 4)
        {
            static_assert(std::is_same<decltype(mob.delay1.data)::period, std::chrono::milliseconds::period>::value, "delay1 is milliseconds");
            static_assert(std::is_same<decltype(mob.delay2.data)::period, std::chrono::milliseconds::period>::value, "delay2 is milliseconds");
            if (bits[3].data[2].data.is_digit10())
                bits[3].data[2].span.warning("delay1 lacks units; defaulting to ms"_s);
            if (bits[3].data[3].data.is_digit10())
                bits[3].data[3].span.warning("delay2 lacks units; defaulting to ms"_s);
            TRY_EXTRACT(bits[3].data[2], mob.delay1);
            TRY_EXTRACT(bits[3].data[3], mob.delay2);
            if (bits[3].data.size() >= 5)
            {
                TRY_EXTRACT(bits[3].data[4], mob.event);
            }
            else
            {
                mob.event.data = NpcEvent();
                mob.event.span = bits[3].data[3].span;
                mob.event.span.end.column++;
                mob.event.span.begin.column = mob.event.span.end.column;
            }
        }
        else
        {
            mob.delay1.data = std::chrono::milliseconds::zero();
            mob.delay1.span = bits[3].data[1].span;
            mob.delay1.span.end.column++;
            mob.delay1.span.begin.column = mob.delay1.span.end.column;
            mob.delay2.data = std::chrono::milliseconds::zero();
            mob.delay2.span = mob.delay1.span;
            mob.event.data = NpcEvent();
            mob.event.span = mob.delay1.span;
        }
        return Ok(std::move(mob));
    }
    static
    Result<MapFlag> parse_mapflag(io::LineSpan span, std::vector<Spanned<std::vector<Spanned<RString>>>>& bits)
    {
        if (bits.size() != 3 && bits.size() != 4)
        {
            return Err(span.error_str("expect 3 or 4 |component|s"_s));
        }
        if (bits[0].data.size() != 1)
        {
            return Err(bits[0].span.error_str("expect 1 ,component,s"_s));
        }
        assert(bits[1].data.size() == 1);
        assert(bits[1].data[0].data == "mapflag"_s);
        if (bits[2].data.size() != 1)
        {
            return Err(bits[2].span.error_str("expect 1 ,component,s"_s));
        }
        if (bits.size() >= 4)
        {
            if (bits[3].data.size() != 1)
            {
                return Err(bits[3].span.error_str("expect 1 ,component,s"_s));
            }
        }

        MapFlag mapflag;
        mapflag.span = span;
        TRY_EXTRACT(bits[0].data[0], mapflag.m);
        mapflag.key_span = bits[1].data[0].span;
        TRY_EXTRACT(bits[2].data[0], mapflag.name);
        if (bits.size() >= 4)
        {
            TRY_EXTRACT(bits[3].data[0], mapflag.opt_extra);
        }
        else
        {
            mapflag.opt_extra.data = ""_s;
            mapflag.opt_extra.span = bits[2].span;
            mapflag.opt_extra.span.end.column++;
            mapflag.opt_extra.span.begin.column = mapflag.opt_extra.span.end.column;
        }
        return Ok(std::move(mapflag));
    }
    static
    Result<ScriptFunction> parse_script_function(io::LineSpan span, std::vector<Spanned<std::vector<Spanned<RString>>>>& bits, io::LineCharReader& lr)
    {
        //  ScriptFunction:     function|script|Fun Name{code}
        if (bits.size() != 3)
        {
            return Err(span.error_str("expect 3 |component|s"_s));
        }
        assert(bits[0].data.size() == 1);
        assert(bits[0].data[0].data == "function"_s);
        assert(bits[1].data.size() == 1);
        assert(bits[1].data[0].data == "script"_s);
        if (bits[2].data.size() != 1)
        {
            return Err(bits[2].span.error_str("expect 1 ,component,s"_s));
        }

        ScriptFunction script_function;
        script_function.span = span;
        script_function.key1_span = bits[0].data[0].span;
        script_function.key_span = bits[1].data[0].span;
        TRY_EXTRACT(bits[2].data[0], script_function.name);
        // also expect '{' and parse real script
        script_function.body = TRY(script::parse::parse_script_body(lr));
        return Ok(std::move(script_function));
    }
    static
    Result<ScriptNone> parse_script_none(io::LineSpan span, std::vector<Spanned<std::vector<Spanned<RString>>>>& bits, io::LineCharReader& lr)
    {
        //  ScriptNone:         -|script|script name|-1{code}
        if (bits.size() != 4)
        {
            return Err(span.error_str("expect 4 |component|s"_s));
        }
        assert(bits[0].data.size() == 1);
        assert(bits[0].data[0].data == "-"_s);
        assert(bits[1].data.size() == 1);
        assert(bits[1].data[0].data == "script"_s);
        if (bits[2].data.size() != 1)
        {
            return Err(bits[2].span.error_str("expect 1 ,component,s"_s));
        }
        assert(bits[3].data[0].data == "-1"_s);
        if (bits[3].data.size() != 1)
        {
            return Err(bits[2].span.error_str("last |component| should be just -1"_s));
        }

        ScriptNone script_none;
        script_none.span = span;
        script_none.key1_span = bits[0].data[0].span;
        script_none.key_span = bits[1].data[0].span;
        TRY_EXTRACT(bits[2].data[0], script_none.name);
        script_none.key4_span = bits[3].data[0].span;
        // also expect '{' and parse real script
        script_none.body = TRY(script::parse::parse_script_body(lr));
        return Ok(std::move(script_none));
    }
    static
    Result<ScriptMapNone> parse_script_map_none(io::LineSpan span, std::vector<Spanned<std::vector<Spanned<RString>>>>& bits, io::LineCharReader& lr)
    {
        //  ScriptMapNone:      m,x,y,d|script|script name|-1{code}
        if (bits.size() != 4)
        {
            return Err(span.error_str("expect 4 |component|s"_s));
        }
        if (bits[0].data.size() != 4)
        {
            return Err(bits[0].span.error_str("expect 3 ,component,s"_s));
        }
        assert(bits[1].data.size() == 1);
        assert(bits[1].data[0].data == "script"_s);
        if (bits[2].data.size() != 1)
        {
            return Err(bits[2].span.error_str("expect 1 ,component,s"_s));
        }
        if (bits[3].data.size() != 1 || bits[3].data[0].data != "-1"_s)
        {
            return Err(bits[2].span.error_str("last |component| should be just -1"_s));
        }

        ScriptMapNone script_map_none;
        script_map_none.span = span;
        TRY_EXTRACT(bits[0].data[0], script_map_none.m);
        TRY_EXTRACT(bits[0].data[1], script_map_none.x);
        TRY_EXTRACT(bits[0].data[2], script_map_none.y);
        TRY_EXTRACT(bits[0].data[3], script_map_none.d);
        script_map_none.key_span = bits[1].data[0].span;
        TRY_EXTRACT(bits[2].data[0], script_map_none.name);
        script_map_none.key4_span = bits[3].data[0].span;
        // also expect '{' and parse real script
        script_map_none.body = TRY(script::parse::parse_script_body(lr));
        return Ok(std::move(script_map_none));
    }
    static
    Result<ScriptMap> parse_script_map(io::LineSpan span, std::vector<Spanned<std::vector<Spanned<RString>>>>& bits, io::LineCharReader& lr)
    {
        //  ScriptMap:          m,x,y,d|script|script name|class,xs,ys{code}
        if (bits.size() != 4)
        {
            return Err(span.error_str("expect 4 |component|s"_s));
        }
        if (bits[0].data.size() != 4)
        {
            return Err(bits[0].span.error_str("expect 3 ,component,s"_s));
        }
        assert(bits[1].data.size() == 1);
        assert(bits[1].data[0].data == "script"_s);
        if (bits[2].data.size() != 1)
        {
            return Err(bits[2].span.error_str("expect 1 ,component,s"_s));
        }
        if (bits[3].data.size() != 3)
        {
            return Err(bits[3].span.error_str("expect 3 ,component,s"_s));
        }

        ScriptMap script_map;
        script_map.span = span;
        TRY_EXTRACT(bits[0].data[0], script_map.m);
        TRY_EXTRACT(bits[0].data[1], script_map.x);
        TRY_EXTRACT(bits[0].data[2], script_map.y);
        TRY_EXTRACT(bits[0].data[3], script_map.d);
        script_map.key_span = bits[1].data[0].span;
        TRY_EXTRACT(bits[2].data[0], script_map.name);
        TRY_EXTRACT(bits[3].data[0], script_map.npc_class);
        TRY_EXTRACT(bits[3].data[1], script_map.xs);
        TRY_EXTRACT(bits[3].data[2], script_map.ys);
        // also expect '{' and parse real script
        script_map.body = TRY(script::parse::parse_script_body(lr));
        return Ok(std::move(script_map));
    }
    static
    Result<std::unique_ptr<Script>> parse_script_any(io::LineSpan span, std::vector<Spanned<std::vector<Spanned<RString>>>>& bits, io::LineCharReader& lr)
    {
        std::unique_ptr<Script> rv;
        // 4 cases:
        //  ScriptFunction:     function|script|Fun Name{code}
        //  ScriptNone:         -|script|script name|-1{code}
        //  ScriptMapNone:      m,x,y,d|script|script name|-1{code}
        //  ScriptMap:          m,x,y,d|script|script name|class,xs,ys{code}
        if (bits[0].data[0].data == "function"_s)
        {
            rv = make_unique<ScriptFunction>(TRY(parse_script_function(span, bits, lr)));
        }
        else if (bits[0].data[0].data == "-"_s)
        {
            rv = make_unique<ScriptNone>(TRY(parse_script_none(span, bits, lr)));
        }
        else if (bits.size() >= 4 && bits[3].data[0].data == "-1"_s)
        {
            rv = make_unique<ScriptMapNone>(TRY(parse_script_map_none(span, bits, lr)));
        }
        else
        {
            rv = make_unique<ScriptMap>(TRY(parse_script_map(span, bits, lr)));
        }
        return Ok(std::move(rv));
    }

    /// Try to extract a top-level token
    /// Return None at EOL, or Some(span)
    /// This will alternate betweeen returning words and separators
    static
    Option<Spanned<RString>> lex(io::LineCharReader& lr, bool first)
    {
        io::LineChar c;
        // at start of line, skip whitespace
        if (first)
        {
            while (lr.get(c) && (/*c.ch() == ' ' ||*/ c.ch() == '\n'))
            {
                lr.adv();
            }
        }
        // at end of file, end of line, or start of script, signal end
        if (!lr.get(c) || c.ch() == '\n' || c.ch() == '{')
        {
            return None;
        }
        // separators are simple
        if (c.ch() == '|' || c.ch() == ',')
        {
            Spanned<RString> bit;
            bit.span.begin = c;
            bit.span.end = c;
            bit.data = RString(VString<1>(c.ch()));
            lr.adv();
            return Some(bit);
        }
        io::LineSpan span;
        MString accum;
        accum += c.ch();
        span.begin = c;
        span.end = c;
        lr.adv();
        if (c.ch() != '/')
            first = false;

        // if one-char token followed by an end-of-line or separator, stop
        if (!lr.get(c) || c.ch() == '\n' || c.ch() == '{' || c.ch() == ',' || c.ch() == '|')
        {
            Spanned<RString> bit;
            bit.span = span;
            bit.data = RString(accum);
            return Some(bit);
        }

        accum += c.ch();
        span.end = c;
        lr.adv();

        // if first token on line, can get comment
        if (first && c.ch() == '/')
        {
            while (lr.get(c) && c.ch() != '\n' && c.ch() != '{')
            {
                accum += c.ch();
                span.end = c;
                lr.adv();
            }
            Spanned<RString> bit;
            bit.span = span;
            bit.data = RString(accum);
            return Some(bit);
        }
        // otherwise, collect until an end of line or separator
        while (lr.get(c) && c.ch() != '\n' && c.ch() != '{' && c.ch() != ',' && c.ch() != '|')
        {
            accum += c.ch();
            span.end = c;
            lr.adv();
        }
        Spanned<RString> bit;
        bit.span = span;
        bit.data = RString(accum);
        return Some(std::move(bit));
    }

    Result<std::unique_ptr<TopLevel>> parse_top(io::LineCharReader& in)
    {
        std::unique_ptr<TopLevel> rv;
        Spanned<std::vector<Spanned<std::vector<Spanned<RString>>>>> bits;

        {
            Spanned<RString> mayc = TRY_UNWRAP(lex(in, true),
                    {
                        io::LineChar c;
                        if (in.get(c) && c.ch() == '{')
                        {
                            return Err(c.error_str("Unexpected script open"_s));
                        }
                        return Ok(std::move(rv));
                    });
            if (mayc.data.startswith("//"_s))
            {
                Comment com;
                com.comment = std::move(mayc.data);
                com.span = std::move(mayc.span);
                rv = make_unique<Comment>(std::move(com));
                return Ok(std::move(rv));
            }

            if (mayc.data == "|"_s || mayc.data == ","_s)
                return Err(mayc.span.error_str("Unexpected separator"_s));
            bits.span = mayc.span;
            bits.data.emplace_back();
            bits.data.back().span = mayc.span;
            bits.data.back().data.push_back(mayc);
        }

        while (true)
        {
            Spanned<RString> sep = TRY_UNWRAP(lex(in, false),
                    break);
            if (sep.data == "|"_s)
            {
                bits.data.emplace_back();
            }
            else if (sep.data != ","_s)
            {
                return Err(sep.span.error_str("Expected separator"_s));
            }

            Spanned<RString> word = TRY_UNWRAP(lex(in, false),
                    return Err(sep.span.error_str("Expected word after separator"_s)));
            if (bits.data.back().data.empty())
                bits.data.back().span = word.span;
            else
                bits.data.back().span.end = word.span.end;
            bits.span.end = word.span.end;
            bits.data.back().data.push_back(std::move(word));
        }

        if (bits.data.size() < 2)
            return Err(bits.span.error_str("Expected a line with |s in it"_s));
        for (auto& bit : bits.data)
        {
            if (bit.data.empty())
                return Err(bit.span.error_str("Empty components are not cool"_s));
        }
        if (bits.data[1].data.size() != 1)
            return Err(bits.data[1].span.error_str("Expected a single word in type position"_s));
        Spanned<RString>& w2 = bits.data[1].data[0];
        if (w2.data == "warp"_s)
        {
            rv = make_unique<Warp>(TRY(parse_warp(bits.span, bits.data)));
        }
        else if (w2.data == "shop"_s)
        {
            rv = make_unique<Shop>(TRY(parse_shop(bits.span, bits.data)));
        }
        else if (w2.data == "monster"_s)
        {
            rv = make_unique<Monster>(TRY(parse_monster(bits.span, bits.data)));
        }
        else if (w2.data == "mapflag"_s)
        {
            rv = make_unique<MapFlag>(TRY(parse_mapflag(bits.span, bits.data)));
        }
        else if (w2.data == "script"_s)
        {
            rv = TRY_MOVE(parse_script_any(bits.span, bits.data, in));
        }
        else
        {
            return Err(w2.span.error_str("Unknown type"_s));
        }
        return Ok(std::move(rv));
    }
} // namespace parse
} // namespace npc
} // namespace tmwa