/*
Copyright (c) 2013 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.

Author: Leonardo de Moura
*/
#include <cstring>
#include <sstream>
#include <string>
#include "util/test.h"
#include "util/name.h"
#include "util/name_set.h"
#include "util/init_module.h"
using namespace lean;

static void tst1() {
    name n("foo");
    lean_assert(n == name("foo"));
    lean_assert(name(n, 1) != name(n, 2));
    lean_assert(name(n, 1) == name(n, 1));
    lean_assert(name(name(n, 1), 2) != name(name(n, 1), 1));
    lean_assert(name(name(n, 1), 1) == name(name(n, 1), 1));
    lean_assert(name(name(n, 2), 1) != name(name(n, 1), 1));
    lean_assert(name(name(n, "bla"), 1) == name(name(n, "bla"), 1));
    lean_assert(name(name(n, "foo"), 1) != name(name(n, "bla"), 1));
    lean_assert(name(name(name("f"), "bla"), 1) != name(name(n, "bla"), 1));
    lean_assert(n != name());
    lean_assert(name().kind() == name_kind::ANONYMOUS);
    lean_assert(name(name(), "foo") == name("foo"));

    lean_assert(name(n, 1) < name(n, 2));
    std::cout << "cmp(" << name(n, 1) << ", " << name(n, 2) << ") = " << cmp(name(n, 1), name(n, 2)) << "\n";
    lean_assert(!(name(n, 1) >= name(n, 2)));
    lean_assert(name(n, 1) < name(name(n, 1), 1));
    lean_assert(n < name(n, 1));
    lean_assert(name(n, 2) > name(name(n, 1), 1));
    lean_assert(name(name("aa"), 2) < name(name(n, 1), 1));
    lean_assert(name(n, "aaa") < name(n, "xxx"));
    lean_assert(name(n, 1) < name(n, "xxx"));
    lean_assert(name(n, 1) < name(name(n, "xxx"), 1));
    lean_assert(name() < name(name(n, "xxx"), 1));
}

static name mk_big_name(unsigned num) {
    name n("foo");
    for (unsigned i = 0; i < num; i++) {
        n = name(n, "bla");
    }
    return n;
}

static void tst2() {
    name n1 = mk_big_name(2000);
    name n2 = mk_big_name(2000);
    lean_assert(n1.hash() == n2.hash());
    for (unsigned i = 0; i < 10000; i++)
        n1.hash();
    std::cout << "n1.hash(): " << n1.hash() << "\n";
}

static void tst3() {
    name n{"foo", "bla", "tst"};
    std::cout << n << "\n";
    lean_assert(n == name(name(name("foo"), "bla"), "tst"));
}

static void tst4() {
    lean_assert(is_prefix_of(name{"foo", "bla"}, name{"foo", "bla"}));
    lean_assert(is_prefix_of(name{"foo", "bla"}, name{"foo", "bla", "foo"}));
    lean_assert(is_prefix_of(name{"foo"}, name{"foo", "bla", "foo"}));
    lean_assert(!is_prefix_of(name{"foo"}, name{"fo", "bla", "foo"}));
    lean_assert(!is_prefix_of(name{"foo", "bla", "foo"}, name{"foo", "bla"}));
    lean_assert(is_prefix_of(name{"foo", "bla"}, name(name{"foo", "bla"}, 0u)));
    lean_assert(is_prefix_of(name("aaa"), name("aaa")));
    lean_assert(!is_prefix_of(name("aaa"), name("aaab")));
    lean_assert(!is_prefix_of(name("aaab"), name("aaa")));
    lean_assert(!is_prefix_of(name{"foo", "bla"}, name{"foo"}));
    lean_assert(is_prefix_of(name{"foo"}, name{"foo", "bla"}));
}

static void tst5() {
    lean_assert(name().size() > 0);
    lean_assert(name({"foo", "bla", "boing"}).get_prefix() == name({"foo", "bla"}));
    lean_assert(!name({"foo", "bla", "boing"}).is_atomic());
    lean_assert(name({"foo"}).is_atomic());
    lean_assert(name({"foo", "bla", "boing"}).get_string() == "boing");
    lean_assert(name(name("foo"), 1u).get_numeral() == nat(1));
    lean_assert(name::anonymous().is_anonymous());
    name n1("foo");
    name n2 = n1;
    lean_assert(n2 == n1);
    std::cout << name::anonymous() << "\n";
    std::cout << name({"foo", "bla", "boing"}).get_prefix() << "\n";
    lean_assert(name("foo").is_string());
    lean_assert(name(name("boo"), "foo").is_string());
    lean_assert(name(name("foo"), 0u).is_numeral());
    lean_assert(name(name("foo"), 0u).get_prefix().is_string());
}

static void tst6() {
    lean_assert(name({"foo", "bla"}).is_safe_ascii());
    lean_assert(!name({"foo", "b∀aaa"}).is_safe_ascii());
    lean_assert(!name({"∀", "boo"}).is_safe_ascii());
    lean_assert(!name(name(name("baa"), "bla∀"), "foo").is_safe_ascii());
}

static void tst7() {
    lean_assert(name("foo") + name("bla") == name({"foo", "bla"}));
    lean_assert(name("foo") + name({"bla", "test"}) == name({"foo", "bla", "test"}));
    lean_assert(name({"foo", "hello"}) + name({"bla", "test"}) == name({"foo", "hello", "bla", "test"}));
    lean_assert(name("foo") + (name("bla") + name({"bla", "test"})) == name(name(name(name("foo"), "bla"), "bla"), "test"));
    lean_assert(name() + name({"bla", "test"}) == name({"bla", "test"}));
    lean_assert(name({"bla", "test"}) + name() == name({"bla", "test"}));
}

static void tst8() {
    name u1 = name::mk_internal_unique_name();
    name u2 = name::mk_internal_unique_name();
    lean_assert(u1 != u2);
    std::cout << u1 << " " << u2 << "\n";
}

static void tst11() {
    name n1("a");
    name n2("b");
    swap(n1, n2);
    lean_assert(n1 == name("b"));
    lean_assert(n2 == name("a"));
}

static void tst12() {
    name_set s;
    s.insert(name("foo"));
    s.insert(name(name("foo"), 1));
    s.insert(name("bla"));
    lean_assert(mk_unique(s, name("boo")) == name("boo"));
    lean_assert(mk_unique(s, name("bla")) == name(name("bla"), 1));
    lean_assert(mk_unique(s, name("foo")) == name(name("foo"), 2));
}

static void tst13() {
    std::ostringstream ostrm;
    serializer out(ostrm);
    name s1({"a", "b"});
    out << s1;
    std::cout << "------\n";
    unsigned i = 0;
    for (unsigned char c : ostrm.str()) {
        std::cout << i << ": " << static_cast<unsigned>(c) << "\n";
        i++;
    }
    std::istringstream istrm(ostrm.str());
    deserializer in(istrm);
    name s2;
    in >> s2;
    std::cout << "h1: " << s1.hash() << "\nh2: " << s2.hash() << "\n";
    std::cout << "s1>> " << s1 << "\ns2>> " << s2 << "\n";
    lean_assert(s1 == s2);
}

static void tst14() {
    name a("a");
    name b(a, nat("100000000000000000000000000"));
    lean_assert(b.get_numeral() == nat("100000000000000000000000000"));
}

static void tst15() {
    std::string s("a");
    s.push_back(0);
    s.push_back('b');
    std::cout << ">> " << static_cast<unsigned>(s[0]) << "\n";
    std::cout << ">> " << static_cast<unsigned>(s[1]) << "\n";
    std::cout << ">> " << static_cast<unsigned>(s[2]) << "\n";
    std::cout << "------\n";
    string_ref v(s);
    std::cout << ">> " << static_cast<unsigned>(v.data()[0]) << "\n";
    std::cout << ">> " << static_cast<unsigned>(v.data()[1]) << "\n";
    std::cout << ">> " << static_cast<unsigned>(v.data()[2]) << "\n";
    lean_assert(v.to_std_string() == s);
    name a(s);
    lean_assert(a.get_string().to_std_string() == s);
    lean_assert(a.get_string().num_bytes() == 3);
}

int main() {
    save_stack_info();
    initialize_util_module();
    tst1();
    tst2();
    tst3();
    tst4();
    tst5();
    tst6();
    tst7();
    tst8();
    tst11();
    tst12();
    tst13();
    tst14();
    tst15();
    finalize_util_module();
    return has_violations() ? 1 : 0;
}