#include "collapsible_group.hpp"
#include "ui_collapsible_group.h"

namespace scwx
{
namespace qt
{
namespace ui
{

class CollapsibleGroupImpl
{
public:
   explicit CollapsibleGroupImpl(CollapsibleGroup* self) : self_ {self} {}
   ~CollapsibleGroupImpl() = default;

   void Initialize();
   void SetExpanded(bool expanded);

   CollapsibleGroup* self_;
};

CollapsibleGroup::CollapsibleGroup(QWidget* parent) :
    QFrame(parent),
    p {std::make_unique<CollapsibleGroupImpl>(this)},
    ui(new Ui::CollapsibleGroup)
{
   ui->setupUi(this);
   p->Initialize();
}

CollapsibleGroup::CollapsibleGroup(const QString& title, QWidget* parent) :
    QFrame(parent),
    p {std::make_unique<CollapsibleGroupImpl>(this)},
    ui(new Ui::CollapsibleGroup)
{
   ui->setupUi(this);
   ui->titleButton->setText(title);
   p->Initialize();
}

CollapsibleGroup::~CollapsibleGroup()
{
   delete ui;
}

void CollapsibleGroupImpl::Initialize()
{
   QObject::connect(
      self_->ui->titleButton,
      &QAbstractButton::toggled,
      self_,
      [this](bool checked) { SetExpanded(checked); },
      Qt::DirectConnection);

   self_->Expand();
}

void CollapsibleGroup::SetContentsLayout(QLayout* layout)
{
   ui->contentsFrame->setLayout(layout);
}

void CollapsibleGroup::SetTitle(const QString& title)
{
   ui->titleButton->setText(title);
}

void CollapsibleGroup::Collapse()
{
   // Update the title frame
   if (ui->titleButton->isChecked())
   {
      ui->titleButton->setChecked(false);
   }
}

void CollapsibleGroup::Expand()
{
   // Update the title frame
   if (!ui->titleButton->isChecked())
   {
      ui->titleButton->setChecked(true);
   }
}

void CollapsibleGroupImpl::SetExpanded(bool expanded)
{
   // Update contents visibility
   self_->ui->contentsFrame->setVisible(expanded);
}

} // namespace ui
} // namespace qt
} // namespace scwx