#include "dialog_saveas.h"
#include "ui_dialog_saveas.h"
#include "dialog_writesettings.h"
#include "base/HGBase.h"
#include "HGUIGlobal.h"
#include "app_cfg.h"
#include <assert.h>
#include <qabstractproxymodel.h>
#include <QDateTime>

class del_delegate : public QAbstractProxyModel
{
    QAbstractProxyModel* prev_ = nullptr;
    QFileDialog*    dlg_ = nullptr;

public:
    del_delegate(QFileDialog* fd) : dlg_(fd)
    {
        prev_ = dlg_->proxyModel();
        dlg_->setProxyModel(this);
        dlg_->setOption(QFileDialog::Option::ReadOnly, true);
    }
    ~del_delegate()
    {
        dlg_->setProxyModel(prev_);
        prev_ = nullptr;
    }

public:
    virtual bool removeRows(int row, int count, const QModelIndex &parent = QModelIndex()) override
    {
        return prev_->removeRows(row, count, parent);
    }
    Q_INVOKABLE virtual QModelIndex mapToSource(const QModelIndex &proxyIndex) const override
    {
        return prev_->mapToSource(proxyIndex);
    }
    Q_INVOKABLE virtual QModelIndex mapFromSource(const QModelIndex &sourceIndex) const override
    {
        return prev_->mapFromSource(sourceIndex);
    }
    Q_INVOKABLE virtual int rowCount(const QModelIndex &parent = QModelIndex()) const override
    {
        return prev_->rowCount(parent);
    }
    Q_INVOKABLE virtual int columnCount(const QModelIndex &parent = QModelIndex()) const override
    {
        return prev_->columnCount(parent);
    }
    Q_INVOKABLE virtual QModelIndex index(int row, int column,
                              const QModelIndex &parent = QModelIndex()) const override
    {
        return prev_->index(row, column, parent);
    }
    Q_INVOKABLE virtual QModelIndex parent(const QModelIndex &child) const override
    {
        return prev_->parent(child);
    }

};

Dialog_SaveAs::Dialog_SaveAs(bool isSaveAs, QWidget *parent) :
    QDialog(parent)
    , ui(new Ui::Dialog_SaveAs)
    , m_isSaveAs(isSaveAs)
{
    ui->setupUi(this);
    setWindowFlags(Qt::Dialog | Qt::WindowCloseButtonHint);

    ui->fileDialog->setAcceptMode(QFileDialog::AcceptSave);
    ui->fileDialog->setWindowFlags(ui->fileDialog->windowFlags() & ~Qt::Dialog);
    ui->fileDialog->setOption(QFileDialog::DontUseNativeDialog,  true);
    ui->fileDialog->setOption(QFileDialog::ReadOnly,  true);    // disable 'Delete' menu item
    ui->fileDialog->setSizeGripEnabled(false);

    QString filter = "JPG - JPEG / JFIF(*.jpg)"
                     ";;BMP - Windows Bitmap(*.bmp)"
                     ";;PNG - Portable Network Graphics(*.png)"
                     ";;PPM - Portable PixMap(*.ppm)"
                     ";;PGM - Portable GreyMap(*.pgm)"
                     ";;PBM - Portable BitMap(*.pbm)"
                     ";;TIF - TIFF Revision 6(*.tif)"
                     ";;PDF - Portable Document Format(*.pdf)"
                     ";;OFD - Open Fixed-layout Document(*.ofd)"
                     ";;GIF - Graphics Interchange Format(*.gif)";
#if 0
                     if (!getOcrPath().empty())
                        filter += ";;OCR->PDF - Portable Document Format(*.pdf)";
#elif !defined (x86_64) && !defined (loongarch64)
                     filter += ";;OCR->PDF - Portable Document Format(*.pdf)";
                     filter += ";;OCR->OFD - Open Fixed-layout Document(*.ofd)";
                     if (m_isSaveAs)
                        filter += ";;OCR->RTF - Rich Text Format(*.rtf)";
#endif

    ui->fileDialog->setNameFilter(filter);

    connect(ui->fileDialog, SIGNAL(accepted()), this, SLOT(on_dialog_accepted()));
    connect(ui->fileDialog, SIGNAL(rejected()), this, SLOT(close()));
    connect(ui->fileDialog, SIGNAL(filterSelected(const QString&)), this, SLOT(on_filterSelected(const QString&)));

    m_suffix = getCfgValue("saveAs", "suffix", 0);
    if (m_suffix > 10)
        m_suffix = 10;
    else if (m_suffix < 0)
        m_suffix = 0;
    ui->fileDialog->selectNameFilter(ui->fileDialog->nameFilters().at(m_suffix));
    ui->btn_option->setEnabled(0 == m_suffix || 6 == m_suffix);
    ui->cbox_subFolder->setChecked(false);

    ui->cbox_subFolder->setChecked(getCfgValue("saveAs", "subFolderByTime", false));

    ui->label_ocrType->setVisible(false);
    ui->comboBox_ocrType->setVisible(false);

#if 0
    if (!getOcrPath().empty())
    {
        ui->label_ocrType->setVisible(isOcr());
        ui->comboBox_ocrType->setVisible(isOcr());
        QStringList ocrType = { tr("English"), tr("Simplified Chinese"), tr("Traditional Chinese"), tr("Japanese"), tr("Korean") };
        ui->comboBox_ocrType->addItems(ocrType);
        ui->comboBox_ocrType->setCurrentIndex(getCfgValue("saveAs", "ocrLanguage", 0));
    }

#endif
}

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

void Dialog_SaveAs::set_current_directory(const QString& dir)
{
    ui->fileDialog->setDirectory(dir);
}

QString Dialog_SaveAs::getSavePath()
{
    return m_savePath;
}

int Dialog_SaveAs::getJpegQuality()
{
    return getCfgValue("saveParam", "jpegQuality", 80);
}

int Dialog_SaveAs::getTiffCompressionBW()
{
    return getCfgValue("saveParam", "tiffCompBW", 1);
}

int Dialog_SaveAs::getTiffCompression()
{
    return getCfgValue("saveParam", "tiffCompClr", 1);
}

int Dialog_SaveAs::getTiffQuality()
{
    return getCfgValue("saveParam", "tiffQuality", 80);
}

bool Dialog_SaveAs::isOcr()
{
    return (m_suffix >= 10);
}

void Dialog_SaveAs::on_dialog_accepted()
{
    QString extName;
    switch (m_suffix)
    {
    case 0:
        extName = ".jpg";
        break;
    case 1:
        extName = ".bmp";
        break;
    case 2:
        extName = ".png";
        break;
    case 3:
        extName = ".ppm";
        break;
    case 4:
        extName = ".pgm";
        break;
    case 5:
        extName = ".pbm";
        break;
    case 6:
        extName = ".tif";
        break;
    case 7:
        extName = ".pdf";
        break;
    case 8:
        extName = ".ofd";
        break;
    case 9:
        extName = ".gif";
        break;
    case 10:
        extName = ".pdf";
        break;
    case 11:
        extName = ".ofd";
        break;
    case 12:
        extName = ".rtf";
        break;
    }

    assert(!extName.isEmpty());

    QString selectedFile = ui->fileDialog->selectedFiles()[0];
    int pos = selectedFile.lastIndexOf('/');
    if (-1 != pos)
    {
        QString path = selectedFile.left(pos + 1);
        QString name = selectedFile.right(selectedFile.count() - pos - 1);

        bool checked = ui->cbox_subFolder->isChecked();

        if(checked)
        {
            QString dirName = NULL;
            dirName = path + QDateTime::currentDateTime().toString("yy-MM-dd hh.mm.ss");
            QDir dir(dirName);
            if(!dir.exists())
            {
                dir.mkdir(dirName);
            }
            path = dirName;
        }

        if(name.endsWith(extName))
            m_savePath = getStdFileName(path + "/" + name);
        else
            m_savePath = getStdFileName(path +  "/" + name + extName);
    }

#if 0
    int ocrLanguage = ui->comboBox_ocrType->currentIndex();
    saveCfgValue("saveAs", "ocrLanguage", ocrLanguage);
#endif

    saveCfgValue("saveAs", "suffix", m_suffix);
    saveCfgValue("saveAs", "subFolderByTime", ui->cbox_subFolder->isChecked());
    accept();
}

void Dialog_SaveAs::on_filterSelected(const QString& filterName)
{
    m_suffix = ui->fileDialog->nameFilters().indexOf(filterName);
    ui->btn_option->setEnabled(0 == m_suffix || 6 == m_suffix || 7 == m_suffix || 8 == m_suffix);

#if 0
    ui->label_ocrType->setVisible(isOcr());
    ui->comboBox_ocrType->setVisible(isOcr());
#endif
}

void Dialog_SaveAs::on_btn_option_clicked()
{
    Dialog_WriteSettings dlg(m_suffix, this);
    dlg.exec();
}