Qt XDF阅读管理

XDF是道客巴巴文件下载的文档格式，类似PDF。文档使用过程中，需要对不同文档进行分类以方便阅读，Qt XDF阅读管理提供文档阅读、分类和文件名查找，文本文件存储相关数据。

 

 mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include <QTimer>
#include <QDir>
#include <QListWidget>
#include <QListWidgetItem>
#include <QDesktopServices>
#include <QUrl>
#include <QTextStream>
#include <QMenu>
#include <QAction>
#include <QCollator>

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    ui->setupUi(this);
    QFont font;
    font.setPixelSize(26);
    setFont(font);
    setWindowTitle(QStringLiteral("XDF阅读管理"));

    mDirPath = QStringLiteral("D:\\123");

    ui->comboBox_type->setSizeAdjustPolicy(QComboBox::AdjustToContents);

    connect(ui->listWidget, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(readXDF(QModelIndex)));
    ui->listWidget->setSelectionBehavior(QAbstractItemView::SelectItems);
    ui->listWidget->setSelectionMode(QAbstractItemView::ExtendedSelection);
    ui->listWidget->setContextMenuPolicy(Qt::CustomContextMenu);
    connect(ui->listWidget, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(listWidgetMenu(QPoint)));

    getTypes();
    getDetails();

    QTimer::singleShot(0, this, SLOT(readFiles()));
}

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

void MainWindow::closeEvent(QCloseEvent *event)
{
    Q_UNUSED(event);
}

void MainWindow::readFiles()
{
    QDir dir(mDirPath);
    if (!dir.exists())
    {
        return;
    }

    dir.setFilter(QDir::Files);
    dir.setSorting(QDir::Name);
    QFileInfoList fileInfoLst = dir.entryInfoList();
    if (fileInfoLst.isEmpty())
    {
        return;
    }

    mNameLst.clear();
    mFileMap.clear();

    foreach (QFileInfo fileInfo, fileInfoLst)
    {
        if ((fileInfo.suffix().toLower() == "xdf") || (fileInfo.suffix().toLower() == "pdf"))
        {
            QString name = fileInfo.fileName();
            QString path = fileInfo.absoluteFilePath();
            mNameLst.append(name);
            mFileMap.insert(name, path);
        }
    }

    QLocale cn(QLocale::Chinese);
    QCollator collator(cn);
    std::sort(mNameLst.begin(), mNameLst.end(), collator);

    foreach (QString name, mNameLst)
    {
        addListWidgetItem(name, mFileMap.value(name));
    }

    updateLabel();
}

void MainWindow::readXDF(const QModelIndex &index)
{
    Q_UNUSED(index);

    QString file = ui->listWidget->currentItem()->data(Qt::UserRole).toString();
    QDesktopServices::openUrl(QUrl::fromLocalFile(file));
}

void MainWindow::getTypes()
{
    ui->comboBox_type->addItem(QStringLiteral("无"));

    QStringList dataLst = openReadFile("xdf_types.txt");
    foreach (QString data, dataLst)
    {
        if (!data.isEmpty())
        {
            ui->comboBox_type->addItem(data);
        }
    }
}

void MainWindow::getDetails()
{
    QStringList tmpLst;
    QStringList dataLst = openReadFile("xdf_details.txt");
    foreach (QString data, dataLst)
    {
        if (!data.isEmpty())
        {
            if (!tmpLst.contains(data))
            {
                tmpLst.append(data);
            }
        }
    }

    mDetailMap.clear();

    dataLst = tmpLst;
    foreach (QString data, dataLst)
    {
        if (!data.isEmpty())
        {
            QStringList lst = data.split(":");
            if (!mDetailMap.contains(lst[0]))
            {
                QStringList valueLst;
                valueLst.append(lst[1]);
                mDetailMap.insert(lst[0], valueLst);
            }
            else
            {
                QStringList valueLst = mDetailMap.value(lst[0]);
                valueLst.append(lst[1]);
                mDetailMap.insert(lst[0], valueLst);
            }
        }
    }
}

void MainWindow::updateLabel()
{
    ui->label->setText(QStringLiteral("%1个文件").arg(ui->listWidget->count()));
}

void MainWindow::on_pushButton_search_clicked()
{
    QString keyword = ui->lineEdit_keyword->text();
    if (keyword.isEmpty() || keyword.trimmed().isEmpty())
    {
        return;
    }

    QStringList retLst;
    foreach (QString name, mFileMap.keys())
    {
        if (name.contains(keyword))
        {
            retLst.append(QStringLiteral("%1*%2").arg(name).arg(mFileMap.value(name)));
        }
    }

    if (retLst.isEmpty())
    {
        return;
    }

    ui->listWidget->clear();
    foreach (QString ret, retLst)
    {
        addListWidgetItem(ret.split("*")[0], ret.split("*")[1]);
    }

    updateLabel();
}

void MainWindow::on_pushButton_type_add_clicked()
{
    if (ui->comboBox_type->currentText() == QStringLiteral("无"))
    {
        return;
    }

    if (ui->listWidget->currentItem())
    {
        QFile file(QStringLiteral("xdf_details.txt"));
        if (!file.open(QIODevice::Append | QIODevice::Text))
        {
            return;
        }

        QListWidgetItem *item = ui->listWidget->currentItem();
        QTextStream out(&file);
        out.setCodec("UTF-8");
        out << QStringLiteral("%1:%2\n").arg(ui->comboBox_type->currentText()).arg(item->text());
        file.close();

        getDetails();
    }
}

void MainWindow::listWidgetMenu(QPoint point)
{
    QListWidgetItem *item = ui->listWidget->itemAt(point);
    if (item == nullptr)
    {
        return;
    }

    QMenu *menu = new QMenu(this);
    QAction *addType = new QAction(QStringLiteral("添加分类"), this);
    menu->addAction(addType);

    connect(addType, SIGNAL(triggered(bool)), this, SLOT(addMultiType()));
    menu->exec(QCursor::pos());

    delete addType;
    delete menu;
}

void MainWindow::addMultiType()
{
    if (ui->comboBox_type->currentText() == QStringLiteral("无"))
    {
        return;
    }

    QStringList dataLst;
    QList<QListWidgetItem*> itemLst = ui->listWidget->selectedItems();
    foreach (QListWidgetItem *item, itemLst)
    {
        dataLst.append(QStringLiteral("%1:%2\n").arg(ui->comboBox_type->currentText()).arg(item->text()));
    }

    QFile file(QStringLiteral("xdf_details.txt"));
    if (!file.open(QIODevice::Append | QIODevice::Text))
    {
        return;
    }

    QTextStream out(&file);
    out.setCodec("UTF-8");
    foreach (QString data, dataLst)
    {
        out << data;
    }
    file.close();

    getDetails();
}

QStringList MainWindow::openReadFile(QString fileName)
{
    QStringList dataLst;
    QFile file(fileName);
    if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
    {
        return dataLst;
    }

    QTextStream in(&file);
    in.setCodec("UTF-8");
    dataLst = in.readAll().split("\n");

    return dataLst;
}

void MainWindow::addListWidgetItem(QString name, QString path)
{
    QListWidgetItem *item = new QListWidgetItem;
    item->setText(name);
    item->setData(Qt::UserRole, path);
    ui->listWidget->addItem(item);
}

void MainWindow::on_pushButton_type_search_clicked()
{
    if (ui->comboBox_type->currentText() == QStringLiteral("无"))
    {
        ui->listWidget->clear();
        foreach (QString name, mNameLst)
        {
            addListWidgetItem(name, mFileMap.value(name));
        }

        updateLabel();

        return;
    }

    QStringList nameLst = mDetailMap.value(ui->comboBox_type->currentText(), QStringList());
    if (nameLst.isEmpty())
    {
        return;
    }

    QLocale cn(QLocale::Chinese);
    QCollator collator(cn);
    std::sort(nameLst.begin(), nameLst.end(), collator);

    ui->listWidget->clear();
    foreach (QString name, nameLst)
    {
        addListWidgetItem(name, mFileMap.value(name));
    }

    updateLabel();
}