// 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>

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

struct B : A
{
};

struct C
{
	int a;
};

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

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

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

    void setA(A* 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(); }

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)
	];

    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);
}