#include <psemek/ui/default_element_factory.hpp>

#include <psemek/ui/box_shape.hpp>

namespace psemek::ui
{

	namespace
	{

		struct button_impl
			: button
		{
			button_impl()
			{
				set_label(std::make_shared<struct label>());
				label()->set_valign(label::valignment::center);
				label()->set_halign(label::halignment::center);
				label()->set_overflow(label::overflow_mode::drop);
				label()->set_multiline(label::multiline_mode::none);
			}

			struct shape const & shape() const override
			{
				return shape_;
			}

			void reshape(geom::box<float, 2> const & bbox) override
			{
				shape_.box = bbox;
				auto s = style();
				if (s && label()) label()->reshape(geom::shrink(bbox, 1.f * (s->border_width + s->inner_margin)));
			}

			void draw(painter & p) const override
			{
				auto s = style();
				if (!s) return;

				if (s->shadow_offset != geom::vector{0, 0})
					p.draw_rect(shape_.box + geom::cast<float>(s->shadow_offset), s->shadow_color);

				if (s->border_width > 0)
					p.draw_rect(shape_.box, s->border_color);

				gfx::color_rgba color = s->fg_color;
				if (state() == state_t::mouseover)
					color = s->highlight_color;
				else if (state() == state_t::mousedown)
					color = s->action_color;
				p.draw_rect(geom::shrink(shape_.box, 1.f * s->border_width), color);
			}

			geom::box<float, 2> size_constraints() const override
			{
				auto sc = element::size_constraints();

				if (label())
					sc = label()->size_constraints();

				if (auto st = style())
				{
					float extra = 2.f * (st->border_width + st->inner_margin);
					sc[0] += extra;
					sc[1] += extra;
				}

				return sc;
			}

		private:
			box_shape shape_;
		};

		struct frame_impl
			: frame
		{

			struct shape const & shape() const override { return shape_; }

			void reshape(geom::box<float, 2> const & bbox) override
			{
				shape_.box = bbox;
				auto st = style();
				if (!st) return;
				for (auto c : children())
					if (c) c->reshape(geom::shrink(bbox, 1.f * (st->border_width + st->outer_margin)));
			}

			geom::box<float, 2> size_constraints() const override
			{
				geom::box<float, 2> r = element::size_constraints();

				for (auto c : children())
					if (c)
						r = c->size_constraints();

				auto st = style();
				if (st)
				{
					float extra = 2.f * (st->border_width + st->outer_margin);
					r[0] += extra;
					r[1] += extra;
				}

				if (min_size())
				{
					auto s = *min_size();
					r[0].min = s[0];
					r[1].min = s[1];
				}

				if (max_size())
				{
					auto s = *max_size();
					r[0].max = s[0];
					r[1].max = s[1];
				}

				return r;
			}

			void draw(painter & p) const override
			{
				auto st = style();
				if (!st) return;

				if (st->shadow_offset != geom::vector{0, 0})
					p.draw_rect(shape_.box + geom::cast<float>(st->shadow_offset), st->shadow_color);

				if (st->border_width > 0)
					p.draw_rect(shape_.box, st->border_color);
				p.draw_rect(geom::shrink(shape_.box, 1.f * st->border_width), st->bg_color);
			}

		private:
			box_shape shape_;
		};

	}

	struct default_element_factory::impl
	{};

	default_element_factory::default_element_factory()
		: pimpl_{make_impl()}
	{}

	default_element_factory::~default_element_factory() = default;

	std::shared_ptr<button> default_element_factory::make_button(std::string text)
	{
		auto r = std::make_shared<button_impl>();
		r->label()->set_text(std::move(text));
		return r;
	}

	std::shared_ptr<label> default_element_factory::make_label(std::string text)
	{
		auto r = std::make_shared<label>();
		r->set_text(text);
		return r;
	}

	std::shared_ptr<frame> default_element_factory::make_frame()
	{
		return std::make_shared<frame_impl>();
	}

	std::shared_ptr<screen> default_element_factory::make_screen()
	{
		return std::make_shared<screen>();
	}

	std::shared_ptr<grid_layout> default_element_factory::make_grid_layout()
	{
		return std::make_shared<grid_layout>();
	}

}