#include <platform.h>

#include <kernel/ally.h>
#include <kernel/alliance.h>
#include <kernel/calendar.h>
#include <kernel/faction.h>
#include <kernel/item.h>
#include <kernel/plane.h>
#include <kernel/race.h>
#include <kernel/region.h>
#include <kernel/unit.h>
#include <kernel/config.h>
#include <util/goodies.h>
#include <util/language.h>
#include <util/password.h>

#include <attributes/racename.h>

#include "monsters.h"
#include <CuTest.h>
#include <tests.h>
#include <selist.h>

#include <assert.h>
#include <stdio.h>

static void test_destroyfaction_allies(CuTest *tc) {
    faction *f1, *f2;
    region *r;
    ally *al;

    test_setup();
    r = test_create_region(0, 0, NULL);
    f1 = test_create_faction(NULL);
    test_create_unit(f1, r);
    f2 = test_create_faction(NULL);
    al = ally_add(&f1->allies, f2);
    al->status = HELP_FIGHT;
    CuAssertIntEquals(tc, HELP_FIGHT, alliedgroup(0, f1, f2, f1->allies, HELP_ALL));
    CuAssertPtrEquals(tc, f2, f1->next);
    destroyfaction(&f1->next);
    CuAssertIntEquals(tc, false, faction_alive(f2));
    CuAssertIntEquals(tc, 0, alliedgroup(0, f1, f2, f1->allies, HELP_ALL));
    test_teardown();
}

static void test_remove_empty_factions_alliance(CuTest *tc) {
    faction *f;
    struct alliance *al;

    test_setup();
    f = test_create_faction(NULL);
    al = makealliance(0, "Hodor");
    setalliance(f, al);
    CuAssertPtrEquals(tc, f, alliance_get_leader(al));
    CuAssertIntEquals(tc, 1, selist_length(al->members));
    remove_empty_factions();
    CuAssertPtrEquals(tc, 0, al->_leader);
    CuAssertIntEquals(tc, 0, selist_length(al->members));
    test_teardown();
}

static void test_remove_empty_factions(CuTest *tc) {
    faction *f, *fm;
    int fno;

    test_setup();
    fm = get_or_create_monsters();
    assert(fm);
    f = test_create_faction(NULL);
    fno = f->no;
    remove_empty_factions();
    CuAssertIntEquals(tc, false, f->_alive);
    CuAssertPtrEquals(tc, fm, factions);
    CuAssertPtrEquals(tc, NULL, fm->next);
    CuAssertPtrEquals(tc, 0, findfaction(fno));
    CuAssertPtrEquals(tc, fm, get_monsters());
    test_teardown();
}

static void test_remove_dead_factions(CuTest *tc) {
    faction *f, *fm;
    region *r;
    int fno;

    test_setup();
    r = test_create_region(0, 0, NULL);
    fm = get_or_create_monsters();
    f = test_create_faction(NULL);
    assert(fm && r && f);
    test_create_unit(f, r);
    test_create_unit(fm, r);
    remove_empty_factions();
    CuAssertPtrEquals(tc, f, findfaction(f->no));
    CuAssertPtrNotNull(tc, get_monsters());
    fm->units = 0;
    f->_alive = false;
    fno = f->no;
    remove_empty_factions();
    CuAssertPtrEquals(tc, 0, findfaction(fno));
    CuAssertPtrEquals(tc, fm, get_monsters());
    test_teardown();
}

static void test_addfaction(CuTest *tc) {
    faction *f = 0;
    const struct race *rc;
    const struct locale *lang;

    test_setup();
    rc = rc_get_or_create("human");
    lang = test_create_locale();
    f = addfaction("test@eressea.de", "hurrdurr", rc, lang, 1234);
    CuAssertPtrNotNull(tc, f);
    CuAssertPtrNotNull(tc, f->name);
    CuAssertPtrEquals(tc, NULL, (void *)f->units);
    CuAssertPtrEquals(tc, NULL, (void *)f->next);
    CuAssertPtrEquals(tc, NULL, (void *)f->banner);
    CuAssertPtrEquals(tc, NULL, (void *)f->spellbook);
    CuAssertPtrEquals(tc, NULL, (void *)f->ursprung);
    CuAssertPtrEquals(tc, (void *)factions, (void *)f);
    CuAssertStrEquals(tc, "test@eressea.de", f->email);
    CuAssertTrue(tc, checkpasswd(f, "hurrdurr"));
    CuAssertPtrEquals(tc, (void *)lang, (void *)f->locale);
    CuAssertIntEquals(tc, 1234, f->subscription);
    CuAssertIntEquals(tc, FFL_ISNEW|FFL_PWMSG, f->flags);
    CuAssertIntEquals(tc, 0, f->age);
    CuAssertTrue(tc, faction_alive(f));
    CuAssertIntEquals(tc, M_GRAY, f->magiegebiet);
    CuAssertIntEquals(tc, turn, f->lastorders);
    CuAssertPtrEquals(tc, f, findfaction(f->no));
    test_teardown();
}

static void test_check_passwd(CuTest *tc) {
    faction *f;
    
    f = test_create_faction(NULL);
    faction_setpassword(f, password_encode("password", PASSWORD_DEFAULT));
    CuAssertTrue(tc, checkpasswd(f, "password"));
    CuAssertTrue(tc, !checkpasswd(f, "assword"));
    CuAssertTrue(tc, !checkpasswd(f, "PASSWORD"));
}

static void test_get_monsters(CuTest *tc) {
    faction *f;

    test_setup();
    CuAssertPtrNotNull(tc, (f = get_monsters()));
    CuAssertPtrEquals(tc, f, get_monsters());
    CuAssertIntEquals(tc, 666, f->no);
    CuAssertStrEquals(tc, "Monster", f->name);
    test_teardown();
}

static void test_set_origin(CuTest *tc) {
    faction *f;
    int x = 0, y = 0;
    plane *pl;

    test_setup();
    pl = create_new_plane(0, "", 0, 19, 0, 19, 0);
    f = test_create_faction(NULL);
    CuAssertPtrEquals(tc, 0, f->ursprung);
    faction_setorigin(f, 0, 1, 1);
    CuAssertIntEquals(tc, 0, f->ursprung->id);
    CuAssertIntEquals(tc, 1, f->ursprung->x);
    CuAssertIntEquals(tc, 1, f->ursprung->y);
    faction_getorigin(f, 0, &x, &y);
    CuAssertIntEquals(tc, 1, x);
    CuAssertIntEquals(tc, 1, y);
    adjust_coordinates(f, &x, &y, pl);
    CuAssertIntEquals(tc, -9, x);
    CuAssertIntEquals(tc, -9, y);
    adjust_coordinates(f, &x, &y, 0);
    CuAssertIntEquals(tc, -10, x);
    CuAssertIntEquals(tc, -10, y);
    test_teardown();
}

static void test_set_origin_bug(CuTest *tc) {
    faction *f;
    plane *pl;
    int x = 17, y = 10;

    test_setup();
    pl = create_new_plane(0, "", 0, 19, 0, 19, 0);
    f = test_create_faction(NULL);
    faction_setorigin(f, 0, -10, 3);
    faction_setorigin(f, 0, -13, -4);
    adjust_coordinates(f, &x, &y, pl);
    CuAssertIntEquals(tc, 0, f->ursprung->id);
    CuAssertIntEquals(tc, -9, x);
    CuAssertIntEquals(tc, 2, y);
    test_teardown();
}

static void test_max_migrants(CuTest *tc) {
    faction *f;
    unit *u;
    race *rc;

    test_setup();
    rc = test_create_race("human");
    f = test_create_faction(rc);
    u = test_create_unit(f, test_create_region(0, 0, NULL));
    CuAssertIntEquals(tc, 0, count_maxmigrants(f));
    rc->flags |= RCF_MIGRANTS;
    CuAssertIntEquals(tc, 0, count_maxmigrants(f));
    scale_number(u, 250);
    CuAssertIntEquals(tc, 13, count_maxmigrants(f));
    test_teardown();
}

static void test_valid_race(CuTest *tc) {
    race * rc1, *rc2;
    faction *f;

    test_setup();
    rc1 = test_create_race("human");
    rc2 = test_create_race("elf");
    f = test_create_faction(rc1);
    CuAssertTrue(tc, valid_race(f, rc1));
    CuAssertTrue(tc, !valid_race(f, rc2));
    rc_set_param(rc1, "other_race", "elf");
    CuAssertTrue(tc, valid_race(f, rc1));
    CuAssertTrue(tc, valid_race(f, rc2));
    test_teardown();
}

static void test_set_email(CuTest *tc) {
    faction *f;
    char email[10];
    test_setup();
    CuAssertIntEquals(tc, 0, check_email("enno@eressea.de"));
    CuAssertIntEquals(tc, 0, check_email("bugs@eressea.de"));
    CuAssertIntEquals(tc, -1, check_email("bad@@eressea.de"));
    CuAssertIntEquals(tc, -1, check_email("eressea.de"));
    CuAssertIntEquals(tc, -1, check_email("eressea@"));
    CuAssertIntEquals(tc, -1, check_email(""));
    CuAssertIntEquals(tc, -1, check_email(NULL));
    f = test_create_faction(NULL);

    sprintf(email, "enno");
    faction_setemail(f, email);
    CuAssertTrue(tc, email != f->email);
    CuAssertStrEquals(tc, "enno", f->email);
    CuAssertStrEquals(tc, "enno", faction_getemail(f));
    faction_setemail(f, "bugs@eressea.de");
    CuAssertStrEquals(tc, "bugs@eressea.de", f->email);
    faction_setemail(f, NULL);
    CuAssertPtrEquals(tc, 0, f->email);
    CuAssertStrEquals(tc, "", faction_getemail(f));
    test_teardown();
}

static void test_save_special_items(CuTest *tc) {
    unit *u, *ug;
    race * rc;
    struct item_type *itype, *it_silver, *it_horse;

    test_setup();
    it_horse = test_create_horse();
    it_silver = test_create_silver();
    itype = test_create_itemtype("banana");
    itype->flags |= ITF_NOTLOST;
    rc = test_create_race("template");
    u = test_create_unit(test_create_faction(NULL), test_create_region(0, 0, NULL));
    i_change(&u->items, itype, 1);

    /* when there is no monster in the region, a ghost of the dead unit is created: */
    save_special_items(u);
    CuAssertPtrNotNull(tc, u->next);
    ug = u->next;
    CuAssertPtrEquals(tc, NULL, ug->next);
    CuAssertPtrEquals(tc, rc, (void *)ug->_race);
    CuAssertIntEquals(tc, 1, u->number);
    CuAssertIntEquals(tc, 0, i_get(u->items, itype));
    CuAssertIntEquals(tc, 1, i_get(ug->items, itype));
    CuAssertStrEquals(tc, "ghost", get_racename(ug->attribs));
    CuAssertStrEquals(tc, u->_name, ug->_name);

    i_change(&u->items, itype, 1);
    /* when there is a monster, it takes all special items: */
    save_special_items(u);
    CuAssertPtrEquals(tc, NULL, ug->next);
    CuAssertIntEquals(tc, 2, i_get(ug->items, itype));
    CuAssertPtrEquals(tc, NULL, u->items);

    i_change(&u->items, itype, 1);
    i_change(&u->items, it_horse, 5);
    i_change(&u->items, it_silver, 10);
    /* horses and money need to go to the region and are not taken: */
    save_special_items(u);
    CuAssertIntEquals(tc, 3, i_get(ug->items, itype));
    CuAssertIntEquals(tc, 5, i_get(u->items, it_horse));
    CuAssertIntEquals(tc, 10, i_get(u->items, it_silver));
    test_teardown();
}

CuSuite *get_faction_suite(void)
{
    CuSuite *suite = CuSuiteNew();
    SUITE_ADD_TEST(suite, test_max_migrants);
    SUITE_ADD_TEST(suite, test_addfaction);
    SUITE_ADD_TEST(suite, test_remove_empty_factions);
    SUITE_ADD_TEST(suite, test_destroyfaction_allies);
    SUITE_ADD_TEST(suite, test_remove_empty_factions_alliance);
    SUITE_ADD_TEST(suite, test_remove_dead_factions);
    SUITE_ADD_TEST(suite, test_get_monsters);
    SUITE_ADD_TEST(suite, test_set_origin);
    SUITE_ADD_TEST(suite, test_set_origin_bug);
    SUITE_ADD_TEST(suite, test_check_passwd);
    SUITE_ADD_TEST(suite, test_valid_race);
    SUITE_ADD_TEST(suite, test_set_email);
    SUITE_ADD_TEST(suite, test_save_special_items);
    return suite;
}