// Copyright (c) 2004 Daniel Wallin and Arvid Norberg

// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
// and/or sell copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following conditions:

// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF
// ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED
// TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
// PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
// SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR
// ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
// ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE
// OR OTHER DEALINGS IN THE SOFTWARE.

#include "test.hpp"
#include <luabind/luabind.hpp>

namespace {

struct A
{
    int get() const
    {
        return 5;
    }
};

struct B : A
{
};

struct C
{
    int a;
};

struct property_test : counted_type<property_test>
{
    property_test(): c_ref(&c), o(6) {}
     ~property_test() { c.a = 0; }

    std::string str_;
    C* c_ref;
    C c;
    float o;
    int a_;
    signed char b;

    void set(int a) { a_ = a; }
    int get() const { return a_; }

    void setA(A*) {}
    A* getA() const { return 0; }

    void set_str(const char* str)
    { str_ = str; }

    const char* get_str() const
    { return str_.c_str(); }
};

COUNTER_GUARD(property_test);

void free_setter(property_test& p, int a)
{ p.set(a); }

int free_getter(const property_test& p)
{ return p.get(); }

} // namespace unnamed
struct borrowed_attribute : counted_type<borrowed_attribute>
{
};

struct attribute_holder : counted_type<attribute_holder>
{
    attribute_holder(borrowed_attribute* b)
      : borrowed(b)
    {}

    borrowed_attribute* borrowed;
};

void test_main(lua_State* L)
{
    using namespace luabind;

    module(L)
    [
        class_<property_test>("property")
            .def(luabind::constructor<>())
            .def("get", &property_test::get)
            .property("a", &property_test::get, &property_test::set)
            .property("str", &property_test::get_str, &property_test::set_str)
            .property("A", &property_test::getA, &property_test::setA)
            .def_readonly("o", &property_test::o)
            .property("free", &free_getter, &free_setter)
            .def_readwrite("b", &property_test::b)
            .def_readwrite("c", &property_test::c)
            .def_readwrite("c_ref", &property_test::c_ref),

        class_<A>("A")
            .def(constructor<>())
            .property("a", &A::get),

        class_<B, A>("B")
            .def(constructor<>())
            .property("x", &A::get),

        class_<C>("C")
            .def(constructor<>())
            .def_readwrite("a", &C::a)
    ];

    DOSTRING(L, "test = property()\n");

    DOSTRING(L,
        "test.a = 5\n"
        "assert(test.a == 5)");

    DOSTRING(L, "assert(test.o == 6)");

    DOSTRING(L,
        "test.new_member = 'foo'\n"
        "assert(test.new_member == 'foo')");

    DOSTRING(L,
        "property.new_member2 = 'bar'\n"
        "assert(property.new_member2 == 'bar')");

    DOSTRING(L,
        "b = property()\n"
        "assert(b.new_member2 == 'bar')");

    DOSTRING(L,
        "test.str = 'foobar'\n"
        "assert(test.str == 'foobar')\n");

    module(L) [
        class_<borrowed_attribute>("borrowed_attribute")
            .def(constructor<>()),

        class_<attribute_holder>("attribute_holder")
            .def(constructor<borrowed_attribute*>())
            .def_readwrite(
                "borrowed", &attribute_holder::borrowed, no_dependency)
    ];

    DOSTRING(L, "test = property()\n");

    DOSTRING(L,
        "test.a = 5\n"
        "assert(test.a == 5)");

    DOSTRING(L, "assert(test.o == 6)");

    DOSTRING(L,
        "test.new_member = 'foo'\n"
        "assert(test.new_member == 'foo')");

    DOSTRING(L,
        "property.new_member2 = 'bar'\n"
        "assert(property.new_member2 == 'bar')");

    DOSTRING(L,
        "b = property()\n"
        "assert(b.new_member2 == 'bar')");

    DOSTRING(L,
        "test.str = 'foobar'\n"
        "assert(test.str == 'foobar')\n");

    DOSTRING(L,
        "test.free = 6\n"
        "assert(test.free == 6)\n");

    DOSTRING(L,
        "test.b = 3\n"
        "assert(test.b == 3)\n");

    DOSTRING(L, "test.c.a = 1\n"
        "assert(test.c.a == 1)\n"
        "assert(test.c_ref.a == 1)");

    DOSTRING(L, "t1 = property()\n"
        "c = t1.c\n"
        "c2 = t1.c_ref\n"
        "c.a = 1\n"
        "t1 = nil\n"
        "collectgarbage()\n"
        "collectgarbage()\n"
        "assert(c.a == 1)\n"
        "assert(c2.a == 1)");

    DOSTRING(L,
        "a = B()\n");

    DOSTRING(L,
        "assert(a.a == 5)\n");

    DOSTRING(L,
        "assert(a.x == 5)\n");

    DOSTRING(L,
        "borrowed = borrowed_attribute()\n"
        "holder = attribute_holder(borrowed)\n"
    );

    TEST_CHECK(borrowed_attribute::count == 1);
    TEST_CHECK(attribute_holder::count == 1);

    DOSTRING(L,
        "holder.borrowed = borrowed\n"
        "borrowed2 = holder.borrowed\n"
        "holder = nil\n"
        "collectgarbage()\n"
    );

    TEST_CHECK(borrowed_attribute::count == 1);
    TEST_CHECK(attribute_holder::count == 0);
}
