//===========================================
//  Lumina-DE source code
//  Copyright (c) 2014, Ken Moore
//  Available under the 3-clause BSD license
//  See the LICENSE file for full details
//===========================================
#include "LTaskButton.h"
#include "LSession.h"

#ifndef DEBUG
#define DEBUG 0
#endif

LTaskButton::LTaskButton(QWidget *parent, bool smallDisplay) : LTBWidget(parent){
  actMenu = new QMenu(this);
  winMenu = new QMenu(this);
  UpdateMenus();
  showText = !smallDisplay;
  this->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
  this->setAutoRaise(false); //make sure these always look like buttons
  this->setContextMenuPolicy(Qt::CustomContextMenu);
  this->setFocusPolicy(Qt::NoFocus);
  this->setSizePolicy(QSizePolicy::Maximum, QSizePolicy::Maximum);
  winMenu->setContextMenuPolicy(Qt::CustomContextMenu);
  connect(this, SIGNAL(customContextMenuRequested(const QPoint&)), this, SLOT(openActionMenu()) );
  connect(this, SIGNAL(clicked()), this, SLOT(buttonClicked()) );
  connect(winMenu, SIGNAL(customContextMenuRequested(const QPoint&)), this, SLOT(openActionMenu()) );
  connect(winMenu, SIGNAL(triggered(QAction*)), this, SLOT(winClicked(QAction*)) );
  connect(winMenu, SIGNAL(aboutToHide()), this, SIGNAL(MenuClosed()));
  connect(actMenu, SIGNAL(aboutToHide()), this, SIGNAL(MenuClosed()));
}

LTaskButton::~LTaskButton(){
	
}

//===========
//      PUBLIC
//===========
QList<WId> LTaskButton::windows(){
  QList<WId> list;
  for(int i=0; i<WINLIST.length(); i++){
    list << WINLIST[i].windowID();
  }
  return list;
}

QString LTaskButton::classname(){
  return cname;
}

bool LTaskButton::isActive(){
  return cstate == LXCB::ACTIVE;
}

void LTaskButton::addWindow(WId win){
  WINLIST << LWinInfo(win);
  UpdateButton();
}

void LTaskButton::rmWindow(WId win){
  for(int i=0; i<WINLIST.length(); i++){
    if(WINLIST[i].windowID() == win){
      WINLIST.removeAt(i);
      break;
    }
  }
  UpdateButton();
}

//==========
//    PRIVATE
//==========
LWinInfo LTaskButton::currentWindow(){
  if(WINLIST.length() == 1 || cWin.windowID()==0){
    return WINLIST[0]; //only 1 window - this must be it
  }else{
    return cWin;
  }
}

//=============
//   PUBLIC SLOTS
//=============
void LTaskButton::UpdateButton(){
  if(winMenu->isVisible()){ return; } //skip this if the window menu is currently visible for now
  bool statusOnly = (WINLIST.length() == LWINLIST.length());
  LWINLIST = WINLIST;
  
  winMenu->clear();
  LXCB::WINDOWVISIBILITY showstate = LXCB::IGNORE;
  for(int i=0; i<WINLIST.length(); i++){
    if(WINLIST[i].windowID() == 0){
      WINLIST.removeAt(i);
      i--;
      continue;
    }
    if(i==0 && !statusOnly){
      //Update the button visuals from the first window
      this->setIcon(WINLIST[i].icon(noicon));
      cname = WINLIST[i].Class();
      if(cname.isEmpty()){ 
	//Special case (chrome/chromium does not register *any* information with X except window title)
	cname = WINLIST[i].text();
	if(cname.contains(" - ")){ cname = cname.section(" - ",-1); }
      }
      this->setToolTip(cname);
    }
    bool junk;
    QAction *tmp = winMenu->addAction( WINLIST[i].icon(junk), WINLIST[i].text() );
      tmp->setData(i); //save which number in the WINLIST this entry is for
    LXCB::WINDOWVISIBILITY stat = WINLIST[i].status(true); //update the saved state for the window
    if(stat<LXCB::ACTIVE && WINLIST[i].windowID() == LSession::handle()->activeWindow()){ stat = LXCB::ACTIVE; }
    if(stat > showstate){ showstate = stat; } //higher priority
  }
  //Now setup the button appropriately
  // - visibility
  if(showstate == LXCB::IGNORE || WINLIST.length() < 1){ this->setVisible(false); }
  else{ this->setVisible(true); }
  // - functionality
  if(WINLIST.length() == 1){
    //single window
    this->setPopupMode(QToolButton::DelayedPopup);
    this->setMenu(actMenu);
    if(showText){ 
      QString txt = WINLIST[0].text();
      if(txt.length()>30){ txt.truncate(27); txt.append("..."); }
      else if(txt.length()<30){ txt = txt.leftJustified(30, ' '); }
      this->setToolButtonStyle(Qt::ToolButtonTextBesideIcon); this->setText(txt);
     }else if(noicon){ this->setToolButtonStyle(Qt::ToolButtonTextBesideIcon); this->setText( cname ); }
    else{ this->setToolButtonStyle(Qt::ToolButtonIconOnly); this->setText(""); }
    this->setToolTip(WINLIST[0].text());
  }else if(WINLIST.length() > 1){
    //multiple windows
    this->setPopupMode(QToolButton::InstantPopup);
    this->setMenu(winMenu);
    this->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
    if(noicon || showText){ "("+QString::number(WINLIST.length())+") "+cname; }
    else{ this->setText("("+QString::number(WINLIST.length())+")"); }
  }
  this->setState(showstate); //Make sure this is after the button setup so that it properly sets the margins/etc
  cstate = showstate; //save this for later
}

void LTaskButton::UpdateMenus(){
  //Action menu should be auto-created for the state of the current window (cWin/cstate)
  actMenu->clear();
  if(cstate!=LXCB::ACTIVE){
    actMenu->addAction( LXDG::findIcon("edit-select",""), tr("Activate Window"), this, SLOT(triggerWindow()) );
  }
  if(cstate!=LXCB::INVISIBLE){
    actMenu->addAction( LXDG::findIcon("view-close",""), tr("Minimize Window"), this, SLOT(minimizeWindow()) );
    if(LSession::handle()->XCB->WindowIsMaximized(cWin.windowID()) ){
      actMenu->addAction( LXDG::findIcon("view-restore",""), tr("Restore Window"), this, SLOT(maximizeWindow()) );
    }else{
      actMenu->addAction( LXDG::findIcon("view-fullscreen",""), tr("Maximize Window"), this, SLOT(maximizeWindow()) );
    }
  }
  actMenu->addAction( LXDG::findIcon("window-close",""), tr("Close Window"), this, SLOT(closeWindow()) );
  if(WINLIST.length()>1 && !winMenu->isVisible()){
    actMenu->addSeparator();
     actMenu->addAction( LXDG::findIcon("layer-visible-on",""), tr("Show All Windows"), this, SLOT(showAllWindows()) );
     actMenu->addAction( LXDG::findIcon("layer-visible-off",""), tr("Minimize All Windows"), this, SLOT(hideAllWindows()) );
     actMenu->addAction( LXDG::findIcon("window-close",""), tr("Close All Windows"), this, SLOT(closeAllWindows()) );
  }
}

//=============
//   PRIVATE SLOTS
//=============
void LTaskButton::buttonClicked(){
  if(WINLIST.length() > 1){
    winMenu->popup(QCursor::pos());
  }else{
    triggerWindow(); 
  }
}

void LTaskButton::closeWindow(){
  if(DEBUG){ qDebug() << "Close Window:" << this->text(); }
  if(winMenu->isVisible()){ winMenu->hide(); }
  LWinInfo win = currentWindow();
  LSession::handle()->XCB->CloseWindow(win.windowID());
  cWin = LWinInfo(); //clear the current
}

void LTaskButton::maximizeWindow(){
  if(DEBUG){ qDebug() << "Maximize Window:" << this->text(); }
  if(winMenu->isVisible()){ winMenu->hide(); }
  LWinInfo win = currentWindow();
  LSession::handle()->XCB->MaximizeWindow(win.windowID());
  //LSession::handle()->adjustWindowGeom(win.windowID(), true); //run this for now until the WM works properly
  cWin = LWinInfo(); //clear the current 
}

void LTaskButton::minimizeWindow(){
  if(DEBUG){ qDebug() << "Minimize Window:" << this->text(); }
  if(winMenu->isVisible()){ winMenu->hide(); }
  LWinInfo win = currentWindow();
  LSession::handle()->XCB->MinimizeWindow(win.windowID());
  cWin = LWinInfo(); //clear the current 
  QTimer::singleShot(100, this, SLOT(UpdateButton()) ); //make sure to update this button if losing active status
}

void LTaskButton::showAllWindows(){
  for(int i=WINLIST.length()-1; i>=0; i--){
    if(WINLIST[i].status()==LXCB::INVISIBLE){
      LSession::handle()->XCB->RestoreWindow(WINLIST[i].windowID());
    }
  }
}

void LTaskButton::hideAllWindows(){
  for(int i=WINLIST.length()-1; i>=0; i--){
    LXCB::WINDOWVISIBILITY state = WINLIST[i].status();
    if(state==LXCB::VISIBLE || state==LXCB::ACTIVE){
      LSession::handle()->XCB->MinimizeWindow(WINLIST[i].windowID());
    }
  }
}

void LTaskButton::closeAllWindows(){
  for(int i=WINLIST.length()-1; i>=0; i--){
    LSession::handle()->XCB->CloseWindow(WINLIST[i].windowID());
  }
}

void LTaskButton::triggerWindow(){
  LWinInfo win = currentWindow();
  //Check which state the window is currently in and flip it to the other
  //LXCB::WINDOWSTATE state = cstate;
  //if(DEBUG){ qDebug() << "Window State: " << state; }
  if(cstate == LXCB::ACTIVE){
    if(DEBUG){ qDebug() << "Minimize Window:" << this->text(); }
    LSession::handle()->XCB->MinimizeWindow(win.windowID());
    QTimer::singleShot(100, this, SLOT(UpdateButton()) ); //make sure to update this button if losing active status
  }else{
    if(DEBUG){ qDebug() << "Activate Window:" << this->text(); }
    LSession::handle()->XCB->ActivateWindow(win.windowID());
  }
  cWin = LWinInfo(); //clear the current
}

void LTaskButton::winClicked(QAction* act){
  //Get the window from the action
  if(act->data().toInt() < WINLIST.length()){
    cWin = WINLIST[act->data().toInt()];
    cstate = cWin.status(); 
  }else{ cWin = LWinInfo(); } //clear it
  //Now trigger the window
  triggerWindow();
}

void LTaskButton::openActionMenu(){
  //Get the Window the mouse is currently over
  QPoint curpos = QCursor::pos();
  QAction *act = winMenu->actionAt(winMenu->mapFromGlobal(curpos));
  //qDebug() << "Button Context Menu:" << curpos.x() << curpos.y() << winMenu->geometry().x() << winMenu->geometry().y() << winMenu->geometry().width() << winMenu->geometry().height();
  cWin = WINLIST[0]; //default to the first window
  if( act != 0 && winMenu->isVisible() ){
    //Get the window from the action
    //qDebug() << "Found Action:" << act->data().toInt();
    if(act->data().toInt() < WINLIST.length()){
      cWin = WINLIST[act->data().toInt()];
    }
  }
  cstate = cWin.status();
  //Now show the action menu
  UpdateMenus();
  actMenu->popup(QCursor::pos());
}