/*

 *   <Graphical interface>

 *   Copyright (C) <2011>  <José Antonio Sánchez Reynaga>, <Lisa "shainer" Vitolo>

 * 

 *   This library is free software; you can redistribute it and/or

 *   modify it under the terms of the GNU Lesser General Public

 *   License as published by the Free Software Foundation; either

 *   version 2.1 of the License, or (at your option) any later version.

 * 

 *   This library 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

 *   Lesser General Public License for more details.

 * 

 *   You should have received a copy of the GNU Lesser General Public

 *   License along with this library; if not, write to the Free Software

 *   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA

 */

#include "window.h"

#include "ui_window.h"

#include <QDBusMessage>

#include <QDBusArgument>

Window::Window(DeviceList& dl, QWidget *parent) :

QWidget(parent), devList(dl),

ui(new Ui::Window)

{

  ui->setupUi(this);

  threadFormat = new ThreadFormat(this);

  /* Device slot */

  connect(&devList, SIGNAL(refreshDevices(USBDevice, bool)), this, SLOT(refreshListDevices(USBDevice, bool)));

  devList.initOperation();

  /* GUI slots */

  connect(ui->cb_devices, SIGNAL(activated(QString)), this, SLOT(newDeviceProperties(QString)));

  connect(ui->checkBox, SIGNAL(stateChanged(int)), this, SLOT(controlLabel(int)));

  /* "Thread" slots */

  connect(threadFormat, SIGNAL(started()), this, SLOT(disableGUI()));

  connect(threadFormat, SIGNAL(finished()), this, SLOT(enableGUI()));

  connect(threadFormat, SIGNAL(finished()), this, SLOT(checkFormat()));

  /* Button slots */

  connect(ui->but_format, SIGNAL(clicked(bool)), this, SLOT(preFormatOperations()));

  connect(ui->but_exit, SIGNAL(clicked(bool)), this, SLOT(close()));

  /* Available filesystems will appear here */

  ui->cb_filesystem->addItem("fat32");

  ui->cb_filesystem->addItem("ntfs");

  ui->cb_filesystem->addItem("ext4");

  ui->cb_filesystem->addItem("ext3");

  ui->cb_filesystem->addItem("ext2");

}

Window::~Window() {}

/*

 * Called when a new USB device dev is added (added=true) or removed

 * (added=false). Update lists accordingly.

 */

void Window::refreshListDevices(USBDevice dev, bool added)

{

  /* No changes while formatting is running */

  if(threadFormat->isRunning())

  {

    return;

  }

  if(added == true)

  {

    ui->cb_devices->insertItem(0, dev.getPath());

  }

  else

  {

    int i = ui->cb_devices->findText(dev.getPath());

    /*

     * **HACK ALERT**

     * Of course any attempt of removal is made with present devices, so i==-1 (device not present in list)

     * should never be true. However, see DeviceList implementation for an explanation.

     * In addition, when called from device notifier only the "caller" device is added to the list, so this

     * may happen anyway.

     */

    if (i != -1)

    {

      ui->cb_devices->removeItem(i);

    }

    /* No more devices! */

    if (ui->cb_devices->count() == 0)

    {

      ui->lb_product->setText(i18n("Product: "));

      ui->lb_size->setText(i18n("Capacity: "));

      ui->lb_filesystem->setText(i18n("Current filesystem: "));

      ui->lb_notice->setText(i18n("No devices found"));

      disableGUI();

      return;

    }

  }

  ui->cb_devices->setCurrentIndex(0);

  newDeviceProperties(ui->cb_devices->currentText());

}

/*

 * Refresh properties of a device when uses selects it on the combobox

 */

void Window::refreshDeviceProperties()

{  

  QString str_device = ui->cb_devices->currentText();

  newDeviceProperties(str_device);

}

void Window::newDeviceProperties(QString devname)

{

  USBDevice usb = devList.findDeviceFromPath(devname);

  QString size;

  size.setNum(usb.getCapacity());

  ui->lb_product->setText(i18n("Product: ") + usb.getVendor());

  ui->lb_size->setText(i18n("Capacity: ") + size + " MB");

  ui->lb_filesystem->setText(i18n("Current filesystem: ") +  usb.getFilesystem());

  enableGUI();

  ui->lb_notice->setText("");

}

Device Window::getCurrentDevice()

{

  QString path = ui->cb_devices->currentText();

  USBDevice ud = devList.findDeviceFromPath(path);

  return ud.getDevice();

}

void Window::preFormatOperations()

{

  /* Get confirmation from user */

  int rp = QMessageBox::question(this, i18n("Warning"), i18n("Do you really want to format the device?"), QMessageBox::Yes, QMessageBox::Cancel);

  if(rp == QMessageBox::Cancel)

  {

    return;

  }

  Device actual = getCurrentDevice();

  if(!actual.as<Block>()->device().contains(QRegExp("([0-9])$")))

  {

    qDebug() << "WARNING: It is needed to create a partition";

    QMessageBox::warning(this, i18n("Warning"), i18n("It is needed to create a partition"), QMessageBox::Ok);

    return;

  }

  applyFormat();

}

void Window::applyFormat()

{ 

   /* Saves arguments from the GUI */

   threadFormat->setFilesystem(ui->cb_filesystem->currentText());

   threadFormat->setDirDev(ui->cb_devices->currentText());

   ui->lb_notice->setText(i18n("Formatting USB device..."));

   threadFormat->start();

}

void Window::controlLabel(int state)

{

	if (state == 2)

	{

		ui->le_label_device->setEnabled(true);

	}

	else

	{

		ui->le_label_device->setDisabled(true);

	}

}	

void Window::disableGUI()

{

  ui->widget_2->setDisabled(true);

  ui->but_format->setDisabled(true);

  ui->widget_4->setDisabled(true);

  ui->widget_5->setDisabled(true);

}

void Window::enableGUI()

{

  ui->widget_2->setEnabled(true);

  ui->but_format->setEnabled(true);

  ui->widget_4->setEnabled(true);

  ui->widget_5->setEnabled(true);

}

/*

 * Automatically called when format ends (see the activated slots in the constructor)

 * checks if everything went okay and restores previous situation

 */

void Window::checkFormat()

{

	if (threadFormat->isOk())

	{

		ui->lb_notice->setText("Device successfully formatted");

	}

	else

	{

		ui->lb_notice->setText("An error occurred while formatting");

	}

	enableGUI();

}

Ui::Window* Window::getUi()

{

  return ui;

}

#include "window.moc"