// SPDX-FileCopyrightText: 2022 - 2023 UnionTech Software Technology Co., Ltd.
//
// SPDX-License-Identifier: GPL-3.0-or-later

#include "basicstatusbar.h"
#include "private/basicstatusbar_p.h"
#include "utils/fileutils.h"

#include "dfm-base/base/application/application.h"

#include <DAnchors>

#include <QHBoxLayout>
#include <QLabel>

DWIDGET_USE_NAMESPACE
using namespace dfmbase;

BasicStatusBar::BasicStatusBar(QWidget *parent)
    : QFrame(parent),
      d(new BasicStatusBarPrivate(this))
{
    initUI();
}

QSize BasicStatusBar::sizeHint() const
{
    QSize size = QFrame::sizeHint();

    size.setHeight(qMax(25, size.height()));

    return size;
}

void BasicStatusBar::clearLayoutAndAnchors()
{
    while (d->layout->count() > 0)
        delete d->layout->takeAt(0);

    DAnchorsBase::clearAnchors(this);
}

void BasicStatusBar::itemSelected(const QList<AbstractFileInfo *> &infoList)
{
    if (!d->tip)
        return;

    d->fileCount = 0;
    d->fileSize = 0;
    d->folderCount = 0;
    d->folderContains = 0;

    QList<QUrl> dirUrlList;
    for (const AbstractFileInfo *info : infoList) {
        if (info->isAttributes(OptInfoType::kIsDir)) {
            d->folderCount += 1;
            dirUrlList << info->urlOf(UrlInfoType::kUrl);
        } else {
            d->fileCount += 1;
            d->fileSize += info->size();
        }
    }

    d->showContains = true;
    const bool dirUrlsEmpty = dirUrlList.isEmpty();
    if (!dirUrlsEmpty) {
        // check mtp setting
        const bool showInfo = Application::instance()->genericAttribute(Application::GenericAttribute::kMTPShowBottomInfo).toBool();
        if (!showInfo) {
            bool isMtp = FileUtils::isMtpFile(dirUrlList.first());
            if (isMtp) {
                d->showContains = false;
            } else {
                d->calcFolderContains(dirUrlList);
            }
        } else {
            d->calcFolderContains(dirUrlList);
        }
    }

    updateStatusMessage();
}

void BasicStatusBar::itemCounted(const int count)
{
    d->discardCurrentJob();

    d->tip->setText(d->counted.arg(QString::number(count)));
}

void BasicStatusBar::updateStatusMessage()
{
    QString selectedFolders;

    if (d->folderCount == 1 && d->folderContains == 1) {
        selectedFolders = d->selectOnlyOneFolder.arg(QString::number(d->folderCount), d->onlyOneItemCounted.arg(d->folderContains));
    } else if (d->folderCount == 1 && d->folderContains != 1) {
        selectedFolders = d->selectOnlyOneFolder.arg(QString::number(d->folderCount), d->counted.arg(d->folderContains));
    } else if (d->folderCount > 1 && d->folderContains == 1) {
        selectedFolders = d->selectFolders.arg(QString::number(d->folderCount), d->onlyOneItemCounted.arg(d->folderContains));
    } else if (d->folderCount > 1 && d->folderContains != 1) {
        selectedFolders = d->selectFolders.arg(QString::number(d->folderCount), d->counted.arg(d->folderContains));
    } else {
        selectedFolders = "";
    }
    if (!selectedFolders.isEmpty() && !d->showContains)
        selectedFolders = d->selectedNetworkOnlyOneFolder.arg(QString::number(d->folderCount));

    QString selectedFiles;

    if (d->fileCount == 1) {
        selectedFiles = d->selectOnlyOneFile.arg(QString::number(d->fileCount), FileUtils::formatSize(d->fileSize));
    } else if (d->fileCount > 1) {
        selectedFiles = d->selectFiles.arg(QString::number(d->fileCount), FileUtils::formatSize(d->fileSize));
    } else {
        selectedFiles = "";
    }
    if (!selectedFiles.isEmpty() && !d->showContains)
        selectedFiles = d->fileCount > 1 ? (d->selected.arg(d->fileCount)) : (d->onlyOneItemSelected.arg(d->fileCount));

    if (selectedFolders.isEmpty()) {
        d->tip->setText(QString("%1").arg(selectedFiles));
    } else if (selectedFiles.isEmpty()) {
        d->tip->setText(QString("%1").arg(selectedFolders));
    } else {
        d->tip->setText(QString("%1,%2").arg(selectedFolders, selectedFiles));
    }
}

void BasicStatusBar::insertWidget(const int index, QWidget *widget, int stretch, Qt::Alignment alignment)
{
    if (d->layout)
        d->layout->insertWidget(index, widget, stretch, alignment);
}

void BasicStatusBar::addWidget(QWidget *widget, int stretch, Qt::Alignment alignment)
{
    if (d->layout)
        d->layout->addWidget(widget, stretch, alignment);
}

void BasicStatusBar::setTipText(const QString &tip)
{
    if (d->tip)
        d->tip->setText(tip);
}

void BasicStatusBar::initUI()
{
    setBackgroundRole(QPalette::Window);
    setFocusPolicy(Qt::NoFocus);

    d->initTipLabel();
    d->initLayout();
}
