#include <platform.h>
#include <stdlib.h>
#include "move.h"

#include "keyword.h"
#include "lighthouse.h"

#include <kernel/config.h>
#include <kernel/ally.h>
#include <kernel/building.h>
#include <kernel/faction.h>
#include <kernel/region.h>
#include <kernel/ship.h>
#include <kernel/terrain.h>
#include <kernel/item.h>
#include <kernel/unit.h>
#include <kernel/race.h>
#include <kernel/order.h>

#include <util/attrib.h>
#include <util/language.h>
#include <util/message.h>
#include <util/base36.h>
#include <util/parser.h>

#include <CuTest.h>
#include <tests.h>
#include <assert.h>

static void test_ship_not_allowed_in_coast(CuTest * tc)
{
    region *r1, *r2;
    ship * sh;
    terrain_type *ttype, *otype;
    ship_type *stype;

    test_cleanup();
    ttype = test_create_terrain("glacier", LAND_REGION | ARCTIC_REGION | WALK_INTO);
    otype = test_create_terrain("ocean", SEA_REGION);
    stype = test_create_shiptype("derp");
    free(stype->coasts);
    stype->coasts = (struct terrain_type **)calloc(2, sizeof(struct terrain_type *));

    r1 = test_create_region(0, 0, ttype);
    r2 = test_create_region(1, 0, otype);
    sh = test_create_ship(0, stype);

    CuAssertIntEquals(tc, SA_COAST, check_ship_allowed(sh, r2));
    CuAssertIntEquals(tc, SA_NO_COAST, check_ship_allowed(sh, r1));
    stype->coasts[0] = ttype;
    CuAssertIntEquals(tc, SA_COAST, check_ship_allowed(sh, r1));
    test_cleanup();
}

typedef struct move_fixture {
    region *r;
    ship *sh;
    building * b;
    unit *u;
} move_fixture;

static void setup_harbor(move_fixture *mf) {
    region *r;
    ship * sh;
    terrain_type * ttype;
    building_type * btype;
    building * b;
    unit *u;

    test_cleanup();

    ttype = test_create_terrain("glacier", LAND_REGION | ARCTIC_REGION | WALK_INTO);
    btype = test_create_buildingtype("harbour");

    sh = test_create_ship(0, 0);
    r = test_create_region(0, 0, ttype);

    b = test_create_building(r, btype);
    b->flags |= BLD_MAINTAINED;

    u = test_create_unit(test_create_faction(0), r);
    u->ship = sh;
    ship_set_owner(u);

    mf->r = r;
    mf->u = u;
    mf->sh = sh;
    mf->b = b;
}

static void test_ship_allowed_without_harbormaster(CuTest * tc)
{
    move_fixture mf;

    setup_harbor(&mf);

    CuAssertIntEquals(tc, SA_HARBOUR, check_ship_allowed(mf.sh, mf.r));
}

static void test_ship_blocked_by_harbormaster(CuTest * tc) {
    unit *u;
    move_fixture mf;

    setup_harbor(&mf);

    u = test_create_unit(test_create_faction(0), mf.r);
    u->building = mf.b;
    building_set_owner(u);

    CuAssertIntEquals_Msg(tc, "harbor master must contact ship", SA_NO_COAST, check_ship_allowed(mf.sh, mf.r));
    test_cleanup();
}

static void test_ship_has_harbormaster_contact(CuTest * tc) {
    unit *u;
    move_fixture mf;

    setup_harbor(&mf);

    u = test_create_unit(test_create_faction(0), mf.r);
    u->building = mf.b;
    building_set_owner(u);
    usetcontact(mf.b->_owner, mf.sh->_owner);

    CuAssertIntEquals(tc, SA_HARBOUR, check_ship_allowed(mf.sh, mf.r));
    test_cleanup();
}

static void test_ship_has_harbormaster_same_faction(CuTest * tc) {
    unit *u;
    move_fixture mf;

    setup_harbor(&mf);

    u = test_create_unit(mf.u->faction, mf.r);
    u->building = mf.b;
    building_set_owner(u);

    CuAssertIntEquals(tc, SA_HARBOUR, check_ship_allowed(mf.sh, mf.r));
    test_cleanup();
}

static void test_ship_has_harbormaster_ally(CuTest * tc) {
    unit *u;
    move_fixture mf;
    ally *al;

    setup_harbor(&mf);

    u = test_create_unit(test_create_faction(0), mf.r);
    u->building = mf.b;
    building_set_owner(u);
    al = ally_add(&u->faction->allies, mf.u->faction);
    al->status = HELP_GUARD;

    CuAssertIntEquals(tc, SA_HARBOUR, check_ship_allowed(mf.sh, mf.r));
    test_cleanup();
}

static void test_walkingcapacity(CuTest *tc) {
    unit *u;
    int cap;
    const struct item_type *itype;

    test_setup();
    init_resources();

    u = test_create_unit(test_create_faction(NULL), test_create_region(0, 0, NULL));
    cap = u->number * (u->_race->capacity + u->_race->weight);
    CuAssertIntEquals(tc, cap, walkingcapacity(u));
    scale_number(u, 2);
    cap = u->number * (u->_race->capacity + u->_race->weight);
    CuAssertIntEquals(tc, cap, walkingcapacity(u));

    itype = it_find("horse");
    assert(itype);
    i_change(&u->items, itype, 1);
    cap += itype->capacity;
    CuAssertIntEquals(tc, cap, walkingcapacity(u));
    i_change(&u->items, itype, 1);
    cap += itype->capacity;
    CuAssertIntEquals(tc, cap, walkingcapacity(u));

    itype = test_create_itemtype("cart");
    assert(itype);
    i_change(&u->items, itype, 1);
    CuAssertIntEquals(tc, cap, walkingcapacity(u));
    set_level(u, SK_RIDING, 1);
    cap += itype->capacity;
    CuAssertIntEquals(tc, cap, walkingcapacity(u));

    itype = test_create_itemtype("trollbelt");
    assert(itype);
    i_change(&u->items, itype, 1);
    CuAssertIntEquals(tc, cap + (STRENGTHMULTIPLIER-1) * u->_race->capacity, walkingcapacity(u));
    config_set("rules.trollbelt.multiplier", "5");
    CuAssertIntEquals(tc, cap + 4 * u->_race->capacity, walkingcapacity(u));

    test_cleanup();
}

static void test_ship_trails(CuTest *tc) {
    ship *sh;
    region *r1, *r2, *r3;
    terrain_type *otype;
    region_list *route = 0;

    test_cleanup();
    otype = test_create_terrain("ocean", SEA_REGION);
    r1 = test_create_region(0, 0, otype);
    r2 = test_create_region(1, 0, otype);
    r3 = test_create_region(2, 0, otype);
    sh = test_create_ship(r1, 0);
    move_ship(sh, r1, r3, 0);
    CuAssertPtrEquals(tc, r3, sh->region);
    CuAssertPtrEquals(tc, sh, r3->ships);
    CuAssertPtrEquals(tc, 0, r1->ships);
    CuAssertPtrEquals(tc, 0, a_find(r1->attribs, &at_shiptrail));
    CuAssertPtrEquals(tc, 0, a_find(r3->attribs, &at_shiptrail));
    add_regionlist(&route, r3);
    add_regionlist(&route, r2);
    move_ship(sh, r3, r1, route);
    CuAssertPtrEquals(tc, r1, sh->region);
    CuAssertPtrEquals(tc, sh, r1->ships);
    CuAssertPtrEquals(tc, 0, r3->ships);
    CuAssertPtrEquals(tc, 0, a_find(r1->attribs, &at_shiptrail));
    CuAssertPtrNotNull(tc, a_find(r2->attribs, &at_shiptrail));
    CuAssertPtrNotNull(tc, a_find(r3->attribs, &at_shiptrail));
    free_regionlist(route);
    test_cleanup();
}

static void test_age_trails(CuTest *tc) {
    region_list *route = 0;
    region *r1, *r2;
    ship *sh;

    test_cleanup();
    r1 = test_create_region(0, 0, 0);
    r2 = test_create_region(1, 0, 0);
    sh = test_create_ship(r1, 0);
    add_regionlist(&route, r1);
    add_regionlist(&route, r2);
    move_ship(sh, r1, r2, route);

    CuAssertPtrNotNull(tc, r1->attribs);
    a_age(&r1->attribs, r1);
    CuAssertPtrNotNull(tc, r1->attribs);
    a_age(&r1->attribs, r1);
    CuAssertPtrEquals(tc, 0, r1->attribs);
    free_regionlist(route);
    test_cleanup();
}

struct drift_fixture {
    faction *f;
    region *r;
    unit *u;
    terrain_type *t_ocean;
    ship_type *st_boat;
    ship *sh;

};

void setup_drift (struct drift_fixture *fix) {
    test_setup();
    test_create_locale();
    config_set("rules.ship.storms", "0");

    fix->st_boat = test_create_shiptype("boat");
    fix->st_boat->cabins = 20000;

    test_create_ocean(0, 0);
    fix->u = test_create_unit(fix->f = test_create_faction(0), fix->r = test_create_ocean(-1, 0));
    assert(fix->r && fix->u && fix->f);
    set_level(fix->u, SK_SAILING, fix->st_boat->sumskill);
    u_set_ship(fix->u, fix->sh = test_create_ship(fix->u->region, fix->st_boat));
    assert(fix->sh);
}

static void test_ship_no_overload(CuTest *tc) {
    struct drift_fixture fix;

    test_cleanup();

    setup_drift(&fix);

    fix.u->number = 2;
    movement();
    CuAssertPtrEquals(tc, fix.u->region, findregion(-1,0));
    CuAssertIntEquals(tc, 0, fix.sh->damage);

    test_cleanup();
}

static void test_ship_empty(CuTest *tc) {
    struct drift_fixture fix;

    test_cleanup();

    setup_drift(&fix);
    fix.u->ship = NULL;
    ship_update_owner(fix.sh);

    movement();
    CuAssertPtrEquals(tc, fix.sh->region, findregion(0, 0));
    CuAssertIntEquals(tc, 2, ship_damage_percent(fix.sh));
    CuAssertPtrEquals(tc, 0, test_find_messagetype(fix.f->msgs, "ship_drift"));

    test_cleanup();
}

static void test_ship_normal_overload(CuTest *tc) {
    struct drift_fixture fix;

    test_cleanup();

    setup_drift(&fix);

    fix.u->number = 21;
    movement();
    CuAssertPtrEquals(tc, fix.u->region, findregion(0, 0));
    CuAssertIntEquals(tc, 2, ship_damage_percent(fix.sh));
    CuAssertPtrNotNull(tc, test_find_messagetype(fix.f->msgs, "ship_drift"));

    test_cleanup();
}

static void test_ship_big_overload(CuTest *tc) {
    struct drift_fixture fix;

    test_cleanup();

    setup_drift(&fix);

    fix.u->number = 22;
    movement();
    CuAssertPtrEquals(tc, fix.u->region, findregion(-1, 0));
    CuAssertIntEquals(tc, 5, ship_damage_percent(fix.sh));
    CuAssertPtrNotNull(tc, test_find_messagetype(fix.f->msgs, "massive_overload"));

    test_cleanup();
}

static void test_ship_no_real_overload(CuTest *tc) {
    struct drift_fixture fix;

    test_cleanup();

    setup_drift(&fix);

    fix.u->number = 21;
    damage_ship(fix.sh, .80);
    movement();
    CuAssertPtrEquals(tc, fix.u->region, findregion(0, 0));
    CuAssertIntEquals(tc, 82, ship_damage_percent(fix.sh));
    CuAssertPtrEquals(tc, 0, test_find_messagetype(fix.f->msgs, "massive_overload"));

    test_cleanup();
}

static void test_ship_ridiculous_overload(CuTest *tc) {
    struct drift_fixture fix;

    test_cleanup();

    setup_drift(&fix);

    fix.u->number = 500;
    movement();
    CuAssertIntEquals(tc, 37, ship_damage_percent(fix.sh));
    CuAssertPtrNotNull(tc, test_find_messagetype(fix.f->msgs, "massive_overload"));

    test_cleanup();
}

static void test_ship_ridiculous_overload_no_captain(CuTest *tc) {
    struct drift_fixture fix;

    test_cleanup();

    setup_drift(&fix);
    set_level(fix.u, SK_SAILING, 0);

    fix.u->number = 500;
    movement();
    CuAssertIntEquals(tc, 37, ship_damage_percent(fix.sh));
    CuAssertPtrNotNull(tc, test_find_messagetype(fix.f->msgs, "massive_overload"));

    test_cleanup();
}

static void test_ship_ridiculous_overload_bad(CuTest *tc) {
    struct drift_fixture fix;

    test_cleanup();
    setup_drift(&fix);

    fix.u->number = 500;
    config_set("rules.ship.overload.damage.max", "1");
    movement();
    CuAssertTrue(tc, ship_damage_percent(fix.sh) > 99);
    CuAssertPtrNotNull(tc, test_find_messagetype(fix.f->msgs, "massive_overload"));
    CuAssertPtrEquals(tc, 0, fix.sh->region);
    CuAssertPtrNotNull(tc, test_find_messagetype(fix.f->msgs, "shipsink"));
    test_cleanup();
}

extern double damage_overload(double overload);

static void test_ship_damage_overload(CuTest *tc) {
    CuAssertDblEquals(tc, 0.0, damage_overload(1), ASSERT_DBL_DELTA);
    CuAssertDblEquals(tc, 0.05, damage_overload(1.1), ASSERT_DBL_DELTA);
    CuAssertDblEquals(tc, 0.05, damage_overload(1.5), ASSERT_DBL_DELTA);
    CuAssertDblEquals(tc, 0.21, damage_overload(3.25), ASSERT_DBL_DELTA);
    CuAssertDblEquals(tc, 0.37, damage_overload(5), ASSERT_DBL_DELTA);
}

typedef struct traveldir {
    int no;
    direction_t dir;
    int age;
} traveldir;

static void test_follow_ship_msg(CuTest * tc) {
    region *r;
    ship * sh;
    faction *f;
    unit *u;
    const ship_type *stype;
    message *msg;
    order *ord;
    traveldir *td = NULL;
    attrib *a;
    void *p;
    
    test_setup();
    init_resources();

    f = test_create_faction(0);
    r = test_create_plain(0, 0);
    test_create_ocean(-1, 1); /* D_NORTHWEST */

    stype = st_find("boat");
    sh = test_create_ship(r, stype);
    u = test_create_unit(f, r);
    u->ship = sh;
    ship_set_owner(u);
    set_level(u, SK_SAILING, 10);

    ord = create_order(K_FOLLOW, f->locale, "SHIP 2");
    assert(ord);
    unit_addorder(u, ord);

    set_money(u, 999999); /* overloaded ship */

    a = a_add(&(r->attribs), a_new(&at_shiptrail));
    td = (traveldir *)a->data.v;
    td->no = 2;
    td->dir = D_NORTHWEST;
    td->age = 2;

    mt_register(mt_new_va("error18", "unit:unit", "region:region", "command:order", 0));

    init_order_depr(ord);
    getstrtoken();

    follow_ship(u, ord);

    CuAssertPtrNotNull(tc, msg = test_find_messagetype(u->faction->msgs, "error18"));
    p = msg->parameters[2].v;
    CuAssertPtrNotNull(tc, p);
    CuAssertIntEquals(tc, K_FOLLOW, getkeyword((order *)p));

    test_cleanup();
}

static void test_drifting_ships(CuTest *tc) {
    ship *sh;
    region *r;
    terrain_type *t_ocean, *t_plain;
    ship_type *st_boat;
    test_setup();
    t_ocean = test_create_terrain("ocean", SEA_REGION);
    t_plain = test_create_terrain("plain", LAND_REGION);
    r = test_create_region(0, 0, t_ocean);
    test_create_region(1, 0, t_ocean);
    st_boat = test_create_shiptype("boat");
    sh = test_create_ship(r, st_boat);
    CuAssertIntEquals(tc, D_EAST, drift_target(sh));
    test_create_region(-1, 0, t_plain);
    CuAssertIntEquals(tc, D_WEST, drift_target(sh));
    test_cleanup();
}

static void test_ship_leave_trail(CuTest *tc) {
    ship *s1, *s2;
    region *r1, *r2;
    terrain_type *t_ocean;
    ship_type *st_boat;
    region_list *route = NULL;

    test_setup();
    t_ocean = test_create_terrain("ocean", SEA_REGION);
    r1 = test_create_region(0, 0, t_ocean);
    add_regionlist(&route, test_create_region(2, 0, t_ocean));
    add_regionlist(&route, r2 = test_create_region(1, 0, t_ocean));
    st_boat = test_create_shiptype("boat");
    s1 = test_create_ship(r1, st_boat);
    s2 = test_create_ship(r1, st_boat);
    leave_trail(s1, r1, route);
    a_add(&r1->attribs, a_new(&at_lighthouse));
    leave_trail(s2, r1, route);
    a_add(&r2->attribs, a_new(&at_lighthouse));
    CuAssertPtrEquals(tc, &at_shiptrail, (void *)r1->attribs->type);
    CuAssertPtrEquals(tc, &at_shiptrail, (void *)r1->attribs->next->type);
    CuAssertPtrEquals(tc, &at_lighthouse, (void *)r1->attribs->next->next->type);
    CuAssertPtrEquals(tc, &at_shiptrail, (void *)r2->attribs->type);
    CuAssertPtrEquals(tc, &at_shiptrail, (void *)r2->attribs->next->type);
    CuAssertPtrEquals(tc, &at_lighthouse, (void *)r2->attribs->next->next->type);
    free_regionlist(route);
    test_cleanup();
}

static void test_movement_speed(CuTest *tc) {
    unit * u;
    race * rc;
    const struct item_type *it_horse;

    test_setup();
    it_horse = test_create_horse();
    rc = test_create_race(NULL);
    u = test_create_unit(test_create_faction(rc), test_create_region(0, 0, NULL));

    rc->speed = 1.0;
    CuAssertIntEquals(tc, BP_WALKING, movement_speed(u));

    rc->speed = 2.0;
    CuAssertIntEquals(tc, 2 * BP_WALKING, movement_speed(u));

    set_level(u, SK_RIDING, 1);
    i_change(&u->items, it_horse, 1);
    CuAssertIntEquals(tc, BP_RIDING, movement_speed(u));

    test_cleanup();
}

static void test_route_cycle(CuTest *tc) {
    unit *u;
    region *r;
    struct locale *lang;
    char buffer[32];

    test_setup();
    test_create_region(1, 0, NULL);
    r = test_create_region(2, 0, NULL);
    lang = test_create_locale();
    CuAssertPtrNotNull(tc, LOC(lang, shortdirections[D_WEST]));
    u = test_create_unit(test_create_faction(NULL), r);
    u->faction->locale = lang;
    CuAssertIntEquals(tc, RCF_WALK, u->_race->flags & RCF_WALK);
    u->orders = create_order(K_ROUTE, u->faction->locale, "WEST EAST NW");
    CuAssertStrEquals(tc, "route WEST EAST NW", get_command(u->orders, lang, buffer, sizeof(buffer)));
    init_order(u->orders, u->faction->locale);
    move_cmd(u, u->orders);
    CuAssertIntEquals(tc, 1, u->region->x);
    CuAssertStrEquals(tc, "route east nw west", get_command(u->orders, lang, buffer, sizeof(buffer)));
    test_cleanup();
}

static void test_route_pause(CuTest *tc) {
    unit *u;
    region *r;
    struct locale *lang;
    char buffer[32];

    test_setup();
    test_create_region(1, 0, NULL);
    r = test_create_region(2, 0, NULL);
    lang = test_create_locale();
    CuAssertPtrNotNull(tc, LOC(lang, shortdirections[D_WEST]));
    u = test_create_unit(test_create_faction(NULL), r);
    u->faction->locale = lang;
    CuAssertIntEquals(tc, RCF_WALK, u->_race->flags & RCF_WALK);
    u->orders = create_order(K_ROUTE, u->faction->locale, "PAUSE EAST NW");
    CuAssertStrEquals(tc, "route PAUSE EAST NW", get_command(u->orders, lang, buffer, sizeof(buffer)));
    init_order(u->orders, u->faction->locale);
    move_cmd(u, u->orders);
    CuAssertIntEquals(tc, 2, u->region->x);
    CuAssertStrEquals(tc, "route PAUSE EAST NW", get_command(u->orders, lang, buffer, sizeof(buffer)));
    test_cleanup();
}

CuSuite *get_move_suite(void)
{
    CuSuite *suite = CuSuiteNew();
    SUITE_ADD_TEST(suite, test_movement_speed);
    SUITE_ADD_TEST(suite, test_walkingcapacity);
    SUITE_ADD_TEST(suite, test_ship_not_allowed_in_coast);
    SUITE_ADD_TEST(suite, test_ship_leave_trail);
    SUITE_ADD_TEST(suite, test_ship_allowed_without_harbormaster);
    SUITE_ADD_TEST(suite, test_ship_blocked_by_harbormaster);
    SUITE_ADD_TEST(suite, test_ship_has_harbormaster_contact);
    SUITE_ADD_TEST(suite, test_ship_has_harbormaster_ally);
    SUITE_ADD_TEST(suite, test_ship_has_harbormaster_same_faction);
    SUITE_ADD_TEST(suite, test_ship_trails);
    SUITE_ADD_TEST(suite, test_age_trails);
    SUITE_ADD_TEST(suite, test_ship_no_overload);
    SUITE_ADD_TEST(suite, test_ship_empty);
    SUITE_ADD_TEST(suite, test_ship_normal_overload);
    SUITE_ADD_TEST(suite, test_ship_no_real_overload);
    SUITE_ADD_TEST(suite, test_ship_big_overload);
    SUITE_ADD_TEST(suite, test_ship_ridiculous_overload);
    SUITE_ADD_TEST(suite, test_ship_ridiculous_overload_bad);
    SUITE_ADD_TEST(suite, test_ship_ridiculous_overload_no_captain);
    SUITE_ADD_TEST(suite, test_ship_damage_overload);
    SUITE_ADD_TEST(suite, test_follow_ship_msg);
    SUITE_ADD_TEST(suite, test_drifting_ships);
    SUITE_ADD_TEST(suite, test_route_cycle);
    SUITE_ADD_TEST(suite, test_route_pause);
    return suite;
}