#include "printergui.h"
#include "ui_printergui.h"
#include <QMessageBox>
#include <unistd.h>
#include <QTimer>
#include <pwd.h>

// ____________________________________________________________________________
PrinterGui::PrinterGui(char *argv[], QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::PrinterGui),
    bgTimeout(-1)
{
  // When called it is guaranteed that argv has (at least) 3 elements

  // Set username
  // Do not use getlogin[_r] as it doesn't fucking work!
  struct passwd *pw = getpwuid(getuid());
  if (pw != NULL && pw->pw_name != NULL) {
    // Detect real name
    this->user = strdup(pw->pw_name);
  } else {
    // Fallback to what command line says
    this->user = strdup(argv[1]);
  }

  // Filename
  this->file = new char[strlen(argv[2]) + 10]; // + 10 in case we rename (ghostscript)
  strcpy(this->file, argv[2]);

  // Initialize cups
  num_dests = cupsGetDests(&dests);

  // Initialize UI
  initializeUI();

  // Timer
  this->bgTimer = new QTimer(this);
  this->bgTimer->setInterval(1000);
  connect(bgTimer, SIGNAL(timeout()), this, SLOT(on_bgTimer_timeout()));
}

// ____________________________________________________________________________
PrinterGui::~PrinterGui()
{
  cupsFreeDests(num_dests, dests);
  free(this->user);
  delete[] this->file;
  delete this->ui;
}

// ____________________________________________________________________________
void PrinterGui::initializeUI()
{
  ui->setupUi(this);
  ui->horizontalLayoutButtons->setAlignment(Qt::AlignRight);

  /*  Initialize Treeview  */

  ui->printerList->setColumnCount(3);
  ui->printerList->setSizePolicy(QSizePolicy::Minimum,QSizePolicy::Minimum);

  // Rename headers
  QStringList h;
  h.append("Drucker");
  h.append("Information");
  h.append("Standort");
  ui->printerList->setHeaderLabels(h);

  // Fill treewidget with data from cups dests;
  cups_dest_t *dest = dests;
  for (int i = num_dests; i>0; ++dest, --i )
    if (dest->instance == NULL) {
      QTreeWidgetItem *wi = new QTreeWidgetItem();
      wi->setText(0, QString::fromUtf8(dest->name));
      wi->setText(1, QString::fromUtf8(cupsGetOption("printer-info", dest->num_options, dest->options)));
      wi->setText(2, QString::fromUtf8(cupsGetOption("printer-location", dest->num_options, dest->options)));
      ui->printerList->addTopLevelItem(wi);
      if (dest->is_default)
        ui->printerList->setCurrentItem(wi);
    }

  // Resize columns to contents
  for (int i = 0; i < 3; ++i)
    ui->printerList->resizeColumnToContents(i);

  /*  Main Window properties  */

  // Disable close button
  this->setWindowFlags((this->windowFlags() & ~Qt::WindowCloseButtonHint) | Qt::WindowStaysOnTopHint);
  // center dialog on screen center
  QRect desktopRect = QApplication::desktop()->screenGeometry(this);
  this->move( desktopRect.width()/2-this->width()/2,
              desktopRect.height()/2-this->height()/2);
  this->setWindowTitle(QString::fromUtf8("Drucken - %1").arg(this->user));
}

// ____________________________________________________________________________
void PrinterGui::on_printerList_currentItemChanged(QTreeWidgetItem *current, QTreeWidgetItem *previous)
{
  ui->printerList->setFocus();

  cups_dest_t *dest =cupsGetNamedDest(CUPS_HTTP_DEFAULT, current->text(0).toUtf8().constData(), NULL);


  /* * Check printer properties (auth, color, duplex, copies) * */
  // get printer capabilities


  const char *type = cupsGetOption("printer-type", dest->num_options, dest->options);
  int res = 0;
  if (type != NULL) {
    res = ::atoi(type);
  }

  // Check color capabilities
  if (res & CUPS_PRINTER_COLOR) {
    ui->comboBoxColor->setEnabled(true);
    ui->comboBoxColor->setCurrentIndex(1);
  } else {
    ui->comboBoxColor->setEnabled(false);
    ui->comboBoxColor->setCurrentIndex(0);
  }

  // Check duplex capabilities
  if (res & CUPS_PRINTER_DUPLEX) {
    ui->comboBoxSides->setEnabled(true);
  } else {
    ui->comboBoxSides->setEnabled(false);
    ui->comboBoxSides->setCurrentIndex(0);
  }

  // Check copy capabilities
  if (res & CUPS_PRINTER_COPIES) {
    ui->lineEditCopies->setEnabled(true);
    ui->labelCopies->setEnabled(true);
  } else {
    ui->lineEditCopies->setEnabled(false);
    ui->lineEditCopies->setText("1");
    ui->labelCopies->setEnabled(false);
  }

  // Check availability
  if (res & CUPS_PRINTER_REJECTING) {
    ui->buttonPrint->setEnabled(false);
  } else {
    ui->buttonPrint->setEnabled(true);
  }
}

// ____________________________________________________________________________
void PrinterGui::on_buttonCancel_clicked()
{
  // Quit with code 1
  QCoreApplication::instance()->exit(1);
}

// ____________________________________________________________________________
void PrinterGui::on_buttonPrint_clicked()
{
  QString cmd;

  // Wenn Farbe möglich ist. Aber trotzdem Graustufen gewählt
  // Schieb file durch ghostscript
  if (ui->comboBoxColor->isEnabled() && ui->comboBoxColor->currentIndex() == 0) {
    // Run ghostscript to make file grayscale
    cmd = QString::fromUtf8("gs -sDEVICE=ps2write -sProcessColorModel=DeviceGray -sColorConversionStrategy=/Gray -dNOPAUSE -dBATCH -dQUIET -dSAFER  -sOutputFile=\"%1.conv\" \"%1\"")
        .arg(QString::fromUtf8(this->file));
    if (system(cmd.toUtf8().constData())) {
      QMessageBox::critical(this, "PrinterGUI", "Kann Druckauftrag nicht in Graustufen konvertieren.");
      return;
    } else {
      strcat(this->file, ".conv");
    }

  }

  /* * Print via cups lib * */

  // Destination / Queue
  cups_dest_t *dest = cupsGetDest(
              ui->printerList->currentItem()->text(0).toUtf8().constData(),
              NULL,
              num_dests,
              dests);

  // Duplex
  if (ui->comboBoxSides->isEnabled()) {
    switch (ui->comboBoxSides->currentIndex()) {
    case 0:
      dest->num_options = cupsAddOption ("Duplex",
                                         "None",
                                         dest->num_options,
                                         &(dest->options));
      break;
    case 1:
      dest->num_options = cupsAddOption ("Duplex",
                                         "DuplexNoTumble",
                                         dest->num_options,
                                         &(dest->options));
      break;
    case 2:
      dest->num_options = cupsAddOption ("Duplex",
                                         "DuplexTumble",
                                         dest->num_options,
                                         &(dest->options));
      break;
    }
  }

  // Kopien
  if (ui->lineEditCopies->isEnabled()) {
    dest->num_options = cupsAddOption ("copies",
                                       ui->lineEditCopies->text().toUtf8().constData(),
                                       dest->num_options,
                                       &(dest->options));
  }

  cupsSetUser(this->user);
  char jobtitle[100];
  snprintf(jobtitle, 100, "gui-%d-%s", (int)getpid(), this->user);

  // Drucken
  if( 0 == cupsPrintFile(dest->name,
                         file,
                         jobtitle,
                         dest->num_options,
                         dest->options)) {
    QMessageBox::critical(this, "CUPS Fehler", cupsLastErrorString());
  } else {
    this->bgTimeout = 0;
    this->hide();
  }

}

void PrinterGui::on_bgTimer_timeout()
{
  if (this->bgTimeout == -1) {
    // Could do something here every second....
    return;
  }
  if (++this->bgTimeout > 4) {
    // Job was sent, GUI is invisible, quit after a few seconds
    QCoreApplication::instance()->exit(0);
  }
}