/* This file is part of Clementine.
   Copyright 2010, David Sansome <me@davidsansome.com>

   Clementine is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   Clementine is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with Clementine.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "playlistheader.h"
#include "playlistview.h"

#include <QtDebug>
#include <QContextMenuEvent>
#include <QMenu>
#include <QSettings>

PlaylistHeader::PlaylistHeader(Qt::Orientation orientation, PlaylistView* view,
                               QWidget* parent)
    : StretchHeaderView(orientation, parent),
      view_(view),
      menu_(new QMenu(this)) {
  hide_action_ = menu_->addAction(tr("&Hide..."), this, SLOT(HideCurrent()));
  stretch_action_ = menu_->addAction(tr("&Stretch columns to fit window"), this,
                                     SLOT(ToggleStretchEnabled()));
  rating_lock_ = menu_->addAction(tr("&Lock Rating"), this, 
                                  SLOT(ToggleRatingEditStatus()));
  rating_lock_->setCheckable(true);
  QSettings s;
  s.beginGroup("Playlist");
  rating_lock_->setChecked(s.value("RatingLocked", false).toBool());
  s.endGroup();
  menu_->addSeparator();

  QMenu* align_menu = new QMenu(tr("&Align text"), this);
  QActionGroup* align_group = new QActionGroup(this);
  align_left_action_ = new QAction(tr("&Left"), align_group);
  align_center_action_ = new QAction(tr("&Center"), align_group);
  align_right_action_ = new QAction(tr("&Right"), align_group);

  align_left_action_->setCheckable(true);
  align_center_action_->setCheckable(true);
  align_right_action_->setCheckable(true);
  align_menu->addActions(align_group->actions());

  connect(align_group, SIGNAL(triggered(QAction*)),
          SLOT(SetColumnAlignment(QAction*)));

  menu_->addMenu(align_menu);
  menu_->addSeparator();

  stretch_action_->setCheckable(true);
  stretch_action_->setChecked(is_stretch_enabled());

  connect(this, SIGNAL(StretchEnabledChanged(bool)), stretch_action_,
          SLOT(setChecked(bool)));
}

void PlaylistHeader::contextMenuEvent(QContextMenuEvent* e) {
  menu_section_ = logicalIndexAt(e->pos());

  if (menu_section_ == -1 ||
      (menu_section_ == logicalIndex(0) && logicalIndex(1) == -1))
    hide_action_->setVisible(false);
  else {
    hide_action_->setVisible(true);

    QString title(
        model()->headerData(menu_section_, Qt::Horizontal).toString());
    hide_action_->setText(tr("&Hide %1").arg(title));
    
    // show rating_lock action only for ratings section
    rating_lock_->setVisible(menu_section_ == Playlist::Column_Rating);

    Qt::Alignment alignment = view_->column_alignment(menu_section_);
    if (alignment & Qt::AlignLeft)
      align_left_action_->setChecked(true);
    else if (alignment & Qt::AlignHCenter)
      align_center_action_->setChecked(true);
    else if (alignment & Qt::AlignRight)
      align_right_action_->setChecked(true);
  }

  qDeleteAll(show_actions_);
  show_actions_.clear();
  for (int i = 0; i < count(); ++i) {
    AddColumnAction(i);
  }

  menu_->popup(e->globalPos());
}

void PlaylistHeader::AddColumnAction(int index) {
#ifndef HAVE_MOODBAR
  if (index == Playlist::Column_Mood) {
    return;
  }
#endif

  QString title(model()->headerData(index, Qt::Horizontal).toString());

  QAction* action = menu_->addAction(title);
  action->setCheckable(true);
  action->setChecked(!isSectionHidden(index));
  show_actions_ << action;

  connect(action, &QAction::triggered, [this, index]() { ToggleVisible(index); } );

}

void PlaylistHeader::HideCurrent() {
  if (menu_section_ == -1) return;

  SetSectionHidden(menu_section_, true);
}

void PlaylistHeader::SetColumnAlignment(QAction* action) {
  Qt::Alignment alignment = Qt::AlignVCenter;

  if (action == align_left_action_) alignment |= Qt::AlignLeft;
  if (action == align_center_action_) alignment |= Qt::AlignHCenter;
  if (action == align_right_action_) alignment |= Qt::AlignRight;

  view_->SetColumnAlignment(menu_section_, alignment);
}

void PlaylistHeader::ToggleVisible(int section) {
  SetSectionHidden(section, !isSectionHidden(section));
  emit SectionVisibilityChanged(section, !isSectionHidden(section));
}

void PlaylistHeader::enterEvent(QEvent*) { emit MouseEntered(); }

void PlaylistHeader::ToggleRatingEditStatus() {
  emit SectionRatingLockStatusChanged(rating_lock_->isChecked());
}