//===========================================
//  Lumina-desktop source code
//  Copyright (c) 2017, Ken Moore
//  Available under the 3-clause BSD license
//  See the LICENSE file for full details
//===========================================
#include "RootDesktopObject.h"
#include <QQmlEngine>
#include <QApplication>
#include <QScreen>
#include <global-objects.h>
#include <QDebug>

// === PUBLIC ===
RootDesktopObject::RootDesktopObject(QObject *parent) : QObject(parent){
  last_window_up = 0;
  updateScreens(); //make sure the internal list is updated right away
  connect(this, SIGNAL(changePanels(QStringList)), this, SLOT(setPanels(QStringList)) );
}

RootDesktopObject::~RootDesktopObject(){

}

void RootDesktopObject::RegisterType(){
  static bool done = false;
  if(done){ return; }
  done=true;
  qmlRegisterType<RootDesktopObject>("Lumina.Backend.RootDesktopObject", 2, 0, "RootDesktopObject");
  //Also register any types that are needed by this class
  ScreenObject::RegisterType();
  NativeWindowObject::RegisterType();
}

RootDesktopObject* RootDesktopObject::instance(){
  static RootDesktopObject* r_obj = new RootDesktopObject();
  return r_obj;
}

//QML Read Functions
QStringList RootDesktopObject::screens(){
  //qDebug() << "Request Screens:" << s_objects.length();
  QStringList names;
  for(int i=0; i<s_objects.length(); i++){ names << s_objects[i]->name(); }
  return names;
}

ScreenObject* RootDesktopObject::screen(QString id){
  //qDebug() << "Got Screen Request:" << id;
  for(int i=0; i<s_objects.length(); i++){
    if(s_objects[i]->name()==id){ return s_objects[i]; }
  }
  return 0;
}

QStringList RootDesktopObject::panels(){
  //qDebug() << "Request Panels:" << panel_objects.length();
  QStringList names;
  for(int i=0; i<panel_objects.length(); i++){ names << panel_objects[i]->name(); }
  return names;
}

PanelObject* RootDesktopObject::panel(QString id){
  //qDebug() << "Got Panel Request:" << id;
  for(int i=0; i<panel_objects.length(); i++){
    if(panel_objects[i]->name()==id){ return panel_objects[i]; }
  }
  return 0;
}

QStringList RootDesktopObject::windows(){
  //qDebug() << "Request Panels:" << panel_objects.length();
  QStringList names;
  for(int i=0; i<window_objects.length(); i++){ names << QString::number(window_objects[i]->id()); }
  return names;
}

NativeWindowObject* RootDesktopObject::window(QString id){
  //qDebug() << "Got Panel Request:" << id;
  WId chk = id.toInt(); //numerical ID's in this case
  for(int i=0; i<window_objects.length(); i++){
    if(window_objects[i]->id()==chk){ return window_objects[i]; }
  }
  return 0;
}

void RootDesktopObject::setPanels(QList<PanelObject*> list){
  panel_objects = list;
  emit panelsChanged();
}

void RootDesktopObject::setPanels(QStringList ids){
  //Make this thread-safe for object creation
  if(this->thread() != QThread::currentThread()){
    //use internal signal/slot combo to change threads
    this->emit changePanels(ids);
    return;
  }
  //qDebug() << "GOT PANEL CHANGE:" << ids;
  //Get the current bounding rectangle for the session
  QRect total;
  bool change = false;
  for(int i=0; i<=s_objects.length(); i++){
    QRect geom;
    QString prefix;
    if(i==s_objects.length()){
      geom = total; //session geometry
      prefix="session/";
    }else{
      geom = s_objects[i]->geometry();
      total = total.united(geom);
      prefix=s_objects[i]->name()+"/";
    }
    QStringList newids = ids.filter(prefix);
    //qDebug() << " Check Panel IDs:" << prefix << newids << ids;
    //First update/remove any current panel objects
    for(int i=0; i<panel_objects.length(); i++){
      if(newids.contains(panel_objects[i]->name()) ){
        //qDebug() << " - Update Existing Panel:" << panel_objects[i]->name();
        newids.removeAll(panel_objects[i]->name()); //already handled
        panel_objects[i]->syncWithSettings(geom);
      }else if(panel_objects[i]->name().startsWith(prefix) ){
        //qDebug() << " - Remove Existing Panel:" << panel_objects[i]->name();
        panel_objects.takeAt(i)->deleteLater();
        i--;
        change = true; //list changed
      }
    }
    //Now create any new panel objects as needed
    for(int i=0; i<newids.length(); i++){
      //qDebug() << " - Create Panel:" << newids[i];
      PanelObject *tmp = new PanelObject(newids[i], this);
      tmp->syncWithSettings(geom);
      panel_objects << tmp;
      change = true; //list changed
    }
  } //end loop over screens+session
  if(change){ emit panelsChanged(); }
}

void RootDesktopObject::setWindows(QList<NativeWindowObject*> list){
  window_objects = list;
  emit windowsChanged();
  mousePositionChanged(true);
}

void RootDesktopObject::logout(){
  emit startLogout();
}

void RootDesktopObject::lockscreen(){
  emit lockScreen();
}

void RootDesktopObject::mousePositionChanged(bool lowerall){
  emit mouseMoved();
  // Go through the transparent windows (in order of high->low in stack)
  // and raise/lower the transparent overlays as needed
  QPoint pos = QCursor::pos();
  for(int i=window_objects.length()-1; i>=0; i--){
    if(window_objects[i]->geometry().contains(pos) ){
      if(last_window_up!= window_objects[i]){
        if(last_window_up!=0){ Lumina::NWS->lowerWindow(last_window_up); }
	Lumina::NWS->raiseWindow(window_objects[i]);
        last_window_up = window_objects[i];
      }
      if(!lowerall){ return; } //found the currently-hovered window
    }else if(lowerall){
      Lumina::NWS->lowerWindow(window_objects[i]);
    }
  }
  //failover for when no window has the mouse over it (lower all of them)
  if(last_window_up!=0 && !lowerall){ Lumina::NWS->lowerWindow(last_window_up); }
}

void RootDesktopObject::launchApp(QString appOrPath){
  emit launchApplication(appOrPath);
}

// === PUBLIC SLOTS ===
void RootDesktopObject::updateScreens(){
  QList<QScreen*> scrns = QApplication::screens();
  QList<ScreenObject*> tmp; //copy of the internal array initially
  for(int i=0; i<scrns.length(); i++){
    bool found = false;
    for(int j=0; j<s_objects.length() && !found; j++){
      if(s_objects[j]->name()==scrns[i]->name()){ found = true; tmp << s_objects.takeAt(j); }
    }
    if(!found){ tmp << new ScreenObject(scrns[i], this); }
  }
  //Delete any leftover objects
  for(int i=0; i<s_objects.length(); i++){ s_objects[i]->deleteLater(); }
  s_objects = tmp;
  emit screensChanged();
  for(int i=0; i<s_objects.length(); i++){
    s_objects[i]->emit geomChanged();
  }
}

void RootDesktopObject::ChangeWallpaper(QString screen, QString value){
  for(int i=0; i<s_objects.length(); i++){
    if(s_objects[i]->name()==screen){ s_objects[i]->setBackground(value); break; }
  }
}

QString RootDesktopObject::CurrentWallpaper(QString screen){
  for(int i=0; i<s_objects.length(); i++){
    if(s_objects[i]->name()==screen){ return s_objects[i]->background();}
  }
  return ""; //unknown
}



// === PRIVATE ===