#include "outputdialog.h"
#include "check/check.h"
#include "output/output_options.h"
#include "ui_outputdialog.h"
#include <QFileDialog>
#include <QMainWindow>
#include <fstream>
#include <iostream>

OutputDialog::OutputDialog(QWidget *parent, Check &check)
    : QDialog(parent), ui(new Ui::OutputDialog), check(check),
    options(*(new OutputOptions())) {
    ui->setupUi(this);
}

OutputDialog::~OutputDialog() { delete ui; }

bool compare_position(Column &c1, Column &c2) {
    return c1.position < c2.position;
}

void OutputDialog::on_buttonBox_accepted() {
    std::ofstream output_file(this->options.get_path());

    for (int i = 0; i < ui->tableWidget->rowCount(); i++) {
        int position = ui->tableWidget->item(i, 0)->text().toInt();
        std::string name = ui->tableWidget->item(i, 1)->text().toStdString();

        Column c;
        c.type = static_cast<ColumnType>(i);
        c.position = position;
        c.name = name;

        this->options.add_or_update_column(c);
    }

    std::sort(this->options.get_columns().begin(),
              this->options.get_columns().end(), compare_position);

    if (options.get_print_header()) {
        for (auto &column : this->options.get_columns()) {
            output_file << column.name
                        << (column.position == this->options.get_columns().size()
                                ? ""
                                : ",");
        }
        output_file << std::endl;
    }

    for (auto goods : this->check.get_goods()) {
        for (auto &column : this->options.get_columns()) {
            std::string output_str;

            switch (column.type) {
            case ColumnType::goods_name:
                output_str = goods.get_name();
                break;
            case ColumnType::goods_price_per_unit:
                output_str = std::to_string(goods.get_price_per_unit());
                break;
            case ColumnType::goods_quantity:
                output_str = std::to_string(goods.get_quantity());
                break;
            case ColumnType::goods_net_weight:
                output_str = "TODO";
                // TODO
                break;
            case ColumnType::goods_total:
                output_str = std::to_string(goods.calculate_total_price());
                break;
            }

            if (column.position != this->options.get_columns().size()) {
                output_str += ",";
            } else {
                output_str += "\n";
            }
            output_file << output_str;
        }
    }

    if (this->options.get_print_total()) {
        output_file << "Total: " << std::to_string(check.calculae_total_price());
    }

    output_file.close();
}

void update_settings(OutputOptions &options, ColumnType t, std::string name,
                     int value) {
    Column column;
    column.type = t;
    column.name = name;
    column.position = value;

    if (value) {
        options.add_or_update_column(column);
    } else {
        options.remove_column(t);
    }
}

void OutputDialog::on_chooseFileButton_clicked() {
    QString filename = QFileDialog::getSaveFileName();
    std::cout << filename.toStdString() << std::endl;
    this->options.set_path(filename.toStdString());
    ui->pathLabel->setText("Path to export: " + filename);
}

void OutputDialog::on_printHeaderCheckBox_stateChanged(int value) {
    this->options.set_print_header(value);
}

void OutputDialog::on_printTotalCheckBox_stateChanged(int value) {
    this->options.set_print_total(value);
}