OperateFileTransfer.cpp

// Copyright Copyright (c) Kang Lin studio, All Rights Reserved
// Author: Kang Lin <kl222@126.com>
 
#include <QLoggingCategory>
#include <QRegularExpression>
#include <QClipboard>
#include <QApplication>
#include <QMimeData>
#include <QUrl>
 
#include "BackendFileTransfer.h"
#include "OperateFileTransfer.h"
#include "Plugin.h"
#include "DlgFileTransfer.h"
 
static Q_LOGGING_CATEGORY(log, "Plugin.FileTransfer.Operate")
 
COperateFileTransfer::COperateFileTransfer(CPlugin *plugin) : COperate(plugin)
    , m_frmFileTransfer(nullptr)
    , m_pThread(nullptr)
{
    qDebug(log) << Q_FUNC_INFO;
    bool check = connect(&m_Parameter, &CParameter::sigChanged,
                         this, [&](){
                             emit this->sigUpdateParameters(this);
                         });
    Q_ASSERT(check);
}
 
COperateFileTransfer::~COperateFileTransfer()
{
    qDebug(log) << Q_FUNC_INFO;
}
 
CBackend* COperateFileTransfer::InstanceBackend()
{
    return new CBackendFileTransfer(this);
}
 
CParameterFileTransfer* COperateFileTransfer::GetParameter()
{
    return &m_Parameter;
}
 
const qint16 COperateFileTransfer::Version() const
{
    return GetPlugin()->Version().toInt();
}
 
QWidget *COperateFileTransfer::GetViewer()
{
    return m_frmFileTransfer;
}
 
int COperateFileTransfer::Initial()
{
    int nRet = 0;
    nRet = COperate::Initial();
    if(nRet)
        return nRet;
 
    if(m_pActionSettings) {
        m_Menu.addSeparator();
        m_Menu.addAction(m_pActionSettings);
    }
 
    m_frmFileTransfer = new CFrmFileTransfer();
    if(m_frmFileTransfer) {
        bool check = connect(m_frmFileTransfer,
                             SIGNAL(sigCopyUrlToClipboard(const QString&)),
                             this, SLOT(slotCopyUrlToClipboard(const QString&)));
        Q_ASSERT(check);
    }
 
    return nRet;
}
 
int COperateFileTransfer::Clean()
{
    int nRet = 0;
    if(m_frmFileTransfer) {
        m_frmFileTransfer->deleteLater();
        m_frmFileTransfer = nullptr;
    }
    nRet = COperate::Clean();
    return nRet;
}
 
int COperateFileTransfer::Start()
{
    int nRet = 0;
    m_pThread = new CBackendThread(this);
    if(!m_pThread) {
        qCritical(log) << "new CBackendThread fail";
        return -2;
    }
    m_pThread->start();
    return nRet;
}
 
int COperateFileTransfer::Stop()
{
    int nRet = 0;
    if(m_pThread)
    {
        m_pThread->quit();
        //Don't delete m_pThread, See CBackendThread
        m_pThread = nullptr;
    }
    return nRet;
}
 
int COperateFileTransfer::SetPluginParameters(CParameterPlugin *pPara)
{
    return m_Parameter.SetPluginParameters(pPara);
}
 
QDialog *COperateFileTransfer::OnOpenDialogSettings(QWidget *parent)
{
    return new CDlgFileTransfer(&m_Parameter, parent);
}
 
int COperateFileTransfer::Load(QSettings &set)
{
    if(m_frmFileTransfer)
        m_frmFileTransfer->Load(set);
    return m_Parameter.Load(set);
}
 
int COperateFileTransfer::Save(QSettings &set)
{
    if(m_frmFileTransfer)
        m_frmFileTransfer->Save(set);
    return m_Parameter.Save(set);
}
 
const QString COperateFileTransfer::Name()
{
    QString szName;
    CParameterNet* pNet = nullptr;
    switch((CParameterFileTransfer::Protocol)m_Parameter.GetProtocol()) {
    case CParameterFileTransfer::Protocol::SFTP:
        pNet = &m_Parameter.m_SSH.m_Net;
        break;
    case CParameterFileTransfer::Protocol::FTP:
        pNet = &m_Parameter.m_Net;
        break;
    }
    if(!pNet->GetHost().isEmpty()) {
        if(m_Parameter.GetPluginParameters()
            && m_Parameter.GetPluginParameters()->GetNameStyles() == CParameterPlugin::NameStyle::Protocol)
            szName = Protocol() + ": ";
        szName += pNet->GetHost()
                  + ":" + QString::number(pNet->GetPort());
    }
    if(szName.isEmpty())
        szName = COperate::Name();
    return szName;
}
 
const QString COperateFileTransfer::Description()
{
    QString szDescription;
    if(!Name().isEmpty())
        szDescription = tr("Name: ") + Name() + "\n";
 
    if(!GetTypeName().isEmpty())
        szDescription += tr("Type: ") + GetTypeName() + "\n";
 
    if(!Protocol().isEmpty()) {
        szDescription += tr("Protocol: ") + Protocol();
#ifdef DEBUG
        if(!GetPlugin()->DisplayName().isEmpty())
            szDescription += " - " + GetPlugin()->DisplayName();
#endif
        szDescription += "\n";
    }
 
    auto &net = m_Parameter.m_Net;
    if(!net.GetHost().isEmpty())
        szDescription += tr("Server address: ") + net.GetHost()
                         + ":" + QString::number(net.GetPort()) + "\n";
 
    CSecurityLevel sl(GetSecurityLevel());
    if(!(GetSecurityLevel() & CSecurityLevel::Level::No)) {
        szDescription += tr("Security level: ");
        if(!sl.GetUnicodeIcon().isEmpty())
            szDescription += sl.GetUnicodeIcon() + " ";
        szDescription += sl.GetString() + "\n";
    }
 
    if(!GetPlugin()->Description().isEmpty())
        szDescription += tr("Description: ") + GetPlugin()->Description();
 
    return szDescription;
}
 
const QString COperateFileTransfer::Protocol() const
{
    switch(m_Parameter.GetProtocol()) {
    case CParameterFileTransfer::Protocol::FTP:
        return "FTP";
    case CParameterFileTransfer::Protocol::SFTP:
        return "SFTP";
    }
    return QString();
}
 
const QString COperateFileTransfer::Id()
{
    QString szId = COperate::Id();
    CParameterNet* pNet = nullptr;
    switch(m_Parameter.GetProtocol()) {
    case CParameterFileTransfer::Protocol::SFTP:
        pNet = &m_Parameter.m_SSH.m_Net;
        break;
    case CParameterFileTransfer::Protocol::FTP:
        pNet = &m_Parameter.m_Net;
        break;
    default:
        break;
    }
    if(pNet && !pNet->GetHost().isEmpty())
        szId += "_" + pNet->GetHost()
                + "_" + QString::number(pNet->GetPort());
    
    static QRegularExpression exp("[-@:/#%!^&* \\.]");
    szId = szId.replace(exp, "_");
    return szId;
}
 
void COperateFileTransfer::slotCopyUrlToClipboard(const QString &szPath)
{
    QUrl url;
    url.setPath(szPath);
    CParameterNet* pNet = nullptr;
    if(m_Parameter.GetProtocol() == CParameterFileTransfer::Protocol::SFTP) {
        pNet = &m_Parameter.m_SSH.m_Net;
        if(pNet->GetHost().isEmpty()) {
            return;
        }
        url.setScheme("sftp");
    }
    if(m_Parameter.GetProtocol() == CParameterFileTransfer::Protocol::FTP) {
        pNet = &m_Parameter.m_Net;
        if(pNet->GetHost().isEmpty()) {
            return;
        }
        url.setScheme("ftp");
    }
    if(!pNet) return;
    url.setUserName(pNet->m_User.GetUser());
    //url.setPassword(pNet->m_User.GetPassword());
    url.setHost(pNet->GetHost());
    url.setPort(pNet->GetPort());
    QClipboard* pClipboard = QApplication::clipboard();
    pClipboard->setText(url.toString());
    // QMimeData* d = new QMimeData();
    // d->setUrls(QList<QUrl>() << QUrl(url));
    // pClipboard->setMimeData(d);
}