#include <platform.h>

#include <kernel/config.h>
#include <kernel/region.h>
#include <kernel/save.h>
#include <kernel/unit.h>
#include <kernel/attrib.h>
#include <util/rng.h>
#include <kernel/gamedata.h>
#include <util/message.h>
#include <binarystore.h>
#include <filestream.h>
#include <memstream.h>
#include <storage.h>
#include <stream.h>
#include <tests.h>

#include "curse.h"

#include <stdlib.h>
#include <string.h>
#include <ctype.h>

#include <CuTest.h>


static void test_curse(CuTest * tc)
{
    attrib *attrs = NULL;
    curse *c, *result;
    int cid;

    curse_type ct_dummy = { "dummy", CURSETYP_NORM, 0, M_SUMEFFECT, NULL };

    test_setup();
    c = create_curse(NULL, &attrs, &ct_dummy, 1.0, 1, 1, 1);
    cid = c->no;
    result = findcurse(cid);
    CuAssertPtrEquals(tc, c, result);
    a_remove(&attrs, attrs);
    result = findcurse(cid);
    CuAssertPtrEquals(tc, NULL, result);
    test_teardown();
}

typedef struct {
    curse *c;
    region *r;
    unit *u;
} curse_fixture;

static void setup_curse(curse_fixture *fix, const char *name) {
    test_setup();
    fix->r = test_create_region(0, 0, NULL);
    fix->u = test_create_unit(test_create_faction(NULL), fix->r);
    fix->c = create_curse(fix->u, &fix->r->attribs, ct_find(name), 1.0, 1, 1.0, 0);
}

static void teardown_curse(curse_fixture *fix) {
    test_teardown();
}

static void test_magicstreet(CuTest *tc) {
    curse_fixture fix;
    message *msg;
    setup_curse(&fix, "magicstreet");
    fix.c->duration = 2;
    msg = fix.c->type->curseinfo(fix.r, TYP_REGION, fix.c, 0);
    CuAssertStrEquals(tc, "curseinfo::magicstreet", test_get_messagetype(msg));
    msg_release(msg);
    teardown_curse(&fix);
}

static void test_magicstreet_warning(CuTest *tc) {
    curse_fixture fix;
    message *msg;
    setup_curse(&fix, "magicstreet");
    fix.c->duration = 1;
    msg = fix.c->type->curseinfo(fix.r, TYP_REGION, fix.c, 0);
    CuAssertStrEquals(tc, "curseinfo::magicstreetwarn", test_get_messagetype(msg));
    msg_release(msg);
    teardown_curse(&fix);
}

static void test_good_dreams(CuTest *tc) {
    curse_fixture fix;
    message *msg;
    setup_curse(&fix, "gbdream");
    fix.c->effect = 1;
    msg = fix.c->type->curseinfo(fix.r, TYP_REGION, fix.c, 0);
    CuAssertStrEquals(tc, "curseinfo::gooddream", test_get_messagetype(msg));
    msg_release(msg);
    teardown_curse(&fix);
}

static void test_bad_dreams(CuTest *tc) {
    curse_fixture fix;
    message *msg;
    setup_curse(&fix, "gbdream");
    fix.c->effect = -1;
    msg = fix.c->type->curseinfo(fix.r, TYP_REGION, fix.c, 0);
    CuAssertStrEquals(tc, "curseinfo::baddream", test_get_messagetype(msg));
    msg_release(msg);
    teardown_curse(&fix);
}

static void test_memstream(CuTest *tc) {
    storage store;
    stream out = { 0 };
    char buf[1024];
    int val = 0;

    mstream_init(&out);
    binstore_init(&store, &out);
    store.handle.data = &out;

    WRITE_INT(&store, 999999);
    WRITE_TOK(&store, "fortytwo");
    WRITE_INT(&store, 42);

    out.api->rewind(out.handle);
    READ_INT(&store, &val);
    READ_TOK(&store, buf, 1024);
    CuAssertIntEquals(tc, 999999, val);
    CuAssertStrEquals(tc, "fortytwo", buf);
    READ_INT(&store, &val);
    CuAssertIntEquals(tc, 42, val);
    mstream_done(&out);
}

static void test_write_flag(CuTest *tc) {
    curse_fixture fix;
    gamedata data;
    storage store;
    region * r;
    curse * c;
    int uid;

    mstream_init(&data.strm);
    gamedata_init(&data, &store, RELEASE_VERSION);

    setup_curse(&fix, "magicwalls");
    c = fix.c;
    r = fix.r;
    uid = r->uid;
    c_setflag(c, CURSE_ISNEW);
    c_setflag(c, CURSE_IMMUNE);
    CuAssertIntEquals(tc, c->type->flags | CURSE_IMMUNE | CURSE_ISNEW, c_flags(c));
    write_game(&data);
    free_gamedata();
    data.strm.api->rewind(data.strm.handle);
    read_game(&data);
    r = findregionbyid(uid);
    CuAssertPtrNotNull(tc, r);
    CuAssertPtrNotNull(tc, r->attribs);
    c = (curse *)r->attribs->data.v;
    CuAssertIntEquals(tc, c->type->flags | CURSE_IMMUNE | CURSE_ISNEW, c_flags(c));

    mstream_done(&data.strm);
    gamedata_done(&data);
    teardown_curse(&fix);
}

static void test_curse_ids(CuTest *tc) {
    const curse_type ct_dummy = { "dummy", CURSETYP_NORM, 0, M_SUMEFFECT, NULL };
    curse *c1, *c2;
    attrib *a1 = 0, *a2 = 0;

    test_setup();
    rng_init(0);
    c1 = create_curse(NULL, &a1, &ct_dummy, 1, 1, 1, 1);
    rng_init(0);
    c2 = create_curse(NULL, &a2, &ct_dummy, 1, 1, 1, 1);
    CuAssertTrue(tc, c1->no != c2->no);
    a_remove(&a1, a1);
    a_remove(&a2, a2);
    test_teardown();
}

static void test_curse_flags(CuTest *tc) {
    const curse_type ct_dummy = { "dummy", CURSETYP_NORM, 0, M_SUMEFFECT, NULL };
    curse *c1, *c2;
    unit *u;

    test_setup();
    u = test_create_unit(test_create_faction(NULL), test_create_region(0, 0, NULL));
    c1 = create_curse(u, &u->attribs, &ct_dummy, 1, 1, 1, 0);
    CuAssertPtrEquals(tc, u, c1->magician);
    CuAssertIntEquals(tc, 1, (int)c1->effect);
    CuAssertIntEquals(tc, 1, (int)c1->vigour);
    CuAssertIntEquals(tc, 1, c1->duration);
    c2 = create_curse(u, &u->attribs, &ct_dummy, 1, 1, 1, 0);
    CuAssertPtrEquals(tc, c1, c2);
    CuAssertPtrEquals(tc, u, c1->magician);
    CuAssertIntEquals(tc, 2, (int)c1->effect);
    CuAssertIntEquals(tc, 1, (int)c1->vigour);
    CuAssertIntEquals(tc, 1, c1->duration);
    test_teardown();
}

CuSuite *get_curse_suite(void)
{
    CuSuite *suite = CuSuiteNew();
    SUITE_ADD_TEST(suite, test_curse);
    SUITE_ADD_TEST(suite, test_magicstreet);
    SUITE_ADD_TEST(suite, test_magicstreet_warning);
    SUITE_ADD_TEST(suite, test_good_dreams);
    SUITE_ADD_TEST(suite, test_bad_dreams);
    SUITE_ADD_TEST(suite, test_memstream);
    SUITE_ADD_TEST(suite, test_write_flag);
    SUITE_ADD_TEST(suite, test_curse_flags);
    SUITE_ADD_TEST(suite, test_curse_ids);
    return suite;
}