// FileName: Download.h
#pragma once
#include "lib\XLDownload.h"
#include "lib\XLError.h"
#include <vector>

// 下載隊(duì)列的大小，決定同時(shí)開(kāi)啟下載線(xiàn)程的數(shù)量
const int LIMIT = 2;

struct Down
{
    // 解析出來(lái)的下載地址
    CString strDownURL;
    // 保存的路徑
    CString strSavePath;
};

struct Flag
{
    DWORD dwFlag;            // 記錄下載線(xiàn)程實(shí)時(shí)狀態(tài)
    LONG lTaskId;            // 記錄對(duì)應(yīng)任務(wù)ID號(hào)
    ULONGLONG ullFileSize;    // 下載文件大小
    ULONGLONG ullRecvSize;    // 已下載大小
};

class CDownload
{
private:

    // 保存下載的日志信息
    std::vector<CString> strLogArr;

    // 解析XML文件
    //void Analysis (LPCTSTR lpPathXML, LPCTSTR lpBegin, LPCTSTR lpEnd);
    void Analysis (LPCTSTR lpPathXML);

    // 開(kāi)始下載
    void Start (void);

    // 調(diào)用迅雷的下載接口
    void CreateDownThread (int n);

    // 下載任務(wù)查詢(xún)監(jiān)控線(xiàn)程
    static void DownQury (LPVOID ptr);

    // 保存日志
    void SaveLog (LPCTSTR lpPathAndName);

public:

    // 保存解析出來(lái)的下載隊(duì)列
    std::vector<Down> strDownArr;

    // 記錄最新的下載位置
    int m_nIndex;

    // 下載隊(duì)列的大小，對(duì)應(yīng)位為1時(shí)，該下載位為空，可以啟動(dòng)新的下載線(xiàn)程
    struct Flag Flags[LIMIT];

    // 總共的下載數(shù)
    int m_nTotal;

    // 失敗的下載數(shù)
    int m_nFail;

    // 成功的下載數(shù)
    int m_nSuccess;

    CDownload();
    ~CDownload();
    BOOL DownloadFilesByXML(LPCTSTR lpPahtXML);
};

// FileName: Download.cpp
#include "stdafx.h"
#include "Download.h"
#include <Shlwapi.h>
#include <locale.h>
#pragma comment(lib, "lib\\XLDownload.lib")

// 包含XTP界面庫(kù)頭文件
#include "XTToolkitPro.h"

CDownload::CDownload()
{
    // 初始化
    m_nIndex = 0;

    // 初始化為沒(méi)有下載線(xiàn)程
    for (int n=0; n<LIMIT; n++)
    {
        Flags[n].dwFlag = 1;
        Flags[n].lTaskId = n;
    }

    m_nSuccess = 0;
    m_nTotal = 0;
    m_nFail = 0;
}

CDownload::~CDownload()
{

}

BOOL CDownload::DownloadFilesByXML(LPCTSTR lpPathXML)
{
    DWORD dwRet = XL_SUCCESS;

    // 檢測(cè)XML文件是否存在
    if (!PathFileExists(lpPathXML))
    {
        ::MessageBox(NULL, _T("XML文件不存在!"), _T("友情提示"), MB_OK|MB_ICONEXCLAMATION);   
        return FALSE;
    }

    // 解析XML文件
    Analysis(lpPathXML);

    // 統(tǒng)計(jì)下載鏈接數(shù)
    m_nTotal = strDownArr.size();

    // 開(kāi)始下載
    Start();

    return TRUE;
}

//void CDownload::Analysis(LPCTSTR lpPathXML, LPCTSTR lpBegin, LPCTSTR lpEnd)
void CDownload::Analysis(LPCTSTR lpPathXML)
{
    // 讀取XML文件
    CXTPPropExchangeXMLNode px(TRUE, NULL, _T("DownLoadInfo"));

    if (!px.LoadFromFile(lpPathXML))
    {

        ::MessageBox(NULL, _T("打開(kāi)XML文件出錯(cuò)!"), _T("友情提示"), MB_OK|MB_ICONEXCLAMATION);
        return;
    }

    if (!px.OnBeforeExchange())
    {
        ::MessageBox(NULL, _T("OnBeforeExchange false!"), _T("友情提示"), MB_OK|MB_ICONEXCLAMATION);
        return;
    }

    px.SetCompactMode(TRUE);

    CXTPPropExchangeEnumeratorPtr enumItem(px.GetEnumerator(_T("Item")));
    POSITION posItem = enumItem->GetPosition();

    CString strLocalName;
    Down down;

    // 遍歷所有的結(jié)點(diǎn)
    while (posItem)
    {
        CXTPPropExchangeSection sec(enumItem->GetNext(posItem));

        if (sec->IsLoading())
        {
            PX_String(sec->GetSection(_T("URL")), NULL, down.strDownURL);
            PX_String(sec->GetSection(_T("LocalPath")), NULL, down.strSavePath);
            PX_String(sec->GetSection(_T("LocalName")), NULL, strLocalName);

            down.strSavePath += strLocalName;
            // 路徑轉(zhuǎn)換
            down.strSavePath.Replace('/', '\\');

            strDownArr.push_back(down);
        }
    }
}

// 下載任務(wù)查詢(xún)
void CDownload::DownQury (LPVOID ptr)
{
    CDownload* pDown = (CDownload*)ptr;
    bool nSign = true;
    DWORD dwRst;
    TCHAR errorBuf[500];
    DWORD dwSize = 500;
    CString strLog;

    // 如果沒(méi)有下載完成一直下載
    while ((pDown->m_nIndex < (pDown->strDownArr).size()) || nSign)
    {
        int n = 0;
        LONG temp = -1;
        nSign = false;

        while (n<LIMIT)
        {
            switch(pDown->Flags[n].dwFlag)
            {
                // 沒(méi)有下載，開(kāi)始新的下載
                case 1 :
                    pDown->CreateDownThread(n);
                    dwRst = XLQueryTaskInfo(pDown->Flags[n].lTaskId, &temp, &(pDown->Flags[n].ullFileSize), &(pDown->Flags[n].ullRecvSize));

                    if (dwRst == XL_SUCCESS && pDown->m_nIndex<pDown->strDownArr.size())
                    {
                        pDown->strLogArr.push_back(pDown->strDownArr[pDown->m_nIndex].strDownURL);
                        pDown->strLogArr.push_back(_T("state:suceess"));
                        pDown->Flags[n].dwFlag = temp;
                    }
                    else
                    {
                        if (pDown->m_nIndex < pDown->strDownArr.size())
                        {
                            pDown->m_nFail += 1;
                            XLGetErrorMsg(dwRst, errorBuf, dwSize);
                            pDown->strLogArr.push_back(pDown->strDownArr[pDown->m_nIndex].strDownURL);
                            strLog.Format(_T("state:%s"), errorBuf);
                            pDown->strLogArr.push_back(strLog);
                        }
                        pDown->m_nIndex += 1;
                    }
                    break;

                // 成功下載完, 開(kāi)始新的下載
                case 11:
                    pDown->CreateDownThread(n);
                    dwRst = XLQueryTaskInfo(pDown->Flags[n].lTaskId, &temp, &(pDown->Flags[n].ullFileSize), &(pDown->Flags[n].ullRecvSize));
                    if (dwRst == XL_SUCCESS && pDown->m_nIndex<pDown->strDownArr.size())
                    {
                        pDown->strLogArr.push_back(pDown->strDownArr[pDown->m_nIndex].strDownURL);
                        pDown->strLogArr.push_back(_T("state:suceess"));
                        pDown->Flags[n].dwFlag = temp;
                    }
                    else
                    {
                        if (pDown->m_nIndex < pDown->strDownArr.size())
                        {
                            pDown->m_nFail += 1;
                            XLGetErrorMsg(dwRst, errorBuf, dwSize);
                            pDown->strLogArr.push_back(pDown->strDownArr[pDown->m_nIndex].strDownURL);
                            strLog.Format(_T("state:%s"), errorBuf);
                            pDown->strLogArr.push_back(strLog);
                        }
                        pDown->m_nIndex += 1;
                    }
                    break;

                // 下載失敗，開(kāi)始新的下載
                case 12:
                    pDown->CreateDownThread(n);
                    dwRst = XLQueryTaskInfo(pDown->Flags[n].lTaskId, &temp, &(pDown->Flags[n].ullFileSize), &(pDown->Flags[n].ullRecvSize));
                    if (dwRst == XL_SUCCESS && pDown->m_nIndex<pDown->strDownArr.size())
                    {
                        pDown->strLogArr.push_back(pDown->strDownArr[pDown->m_nIndex].strDownURL);
                        pDown->strLogArr.push_back(_T("state:suceess"));
                        pDown->Flags[n].dwFlag = temp;
                    }
                    else
                    {
                        if (pDown->m_nIndex < pDown->strDownArr.size())
                        {
                            pDown->m_nFail += 1;
                            XLGetErrorMsg(dwRst, errorBuf, dwSize);
                            pDown->strLogArr.push_back(pDown->strDownArr[pDown->m_nIndex].strDownURL);
                            strLog.Format(_T("state:%s"), errorBuf);
                            pDown->strLogArr.push_back(strLog);
                        }
                        pDown->m_nIndex += 1;
                    }
                    break;

                // 默認(rèn)處理
                default:
                    XLQueryTaskInfo(pDown->Flags[n].lTaskId, &temp, &(pDown->Flags[n].ullFileSize), &(pDown->Flags[n].ullRecvSize));
                    pDown->Flags[n].dwFlag = temp;
                    if (temp == enumTaskStatus_Download)
                    {
                        nSign = true;
                    }
            }
            n++;
        }
        //Sleep(1);
    }

    // 保存日志
    pDown->SaveLog(_T("C:\\down.log"));

    CString strTemp;
    strTemp.Format(_T("下載完成! 下載總數(shù):%d 成功總數(shù)%d 失敗總數(shù)%d 日志存在位置:C:\\down.log"), pDown->m_nTotal, pDown->m_nSuccess, pDown->m_nFail);
    ::MessageBox(NULL, strTemp, _T("友情提示"), MB_OK);

    pDown->m_nSuccess = 0;
    pDown->m_nTotal = 0;
    pDown->m_nFail = 0;

    // 恢復(fù)初始狀態(tài)
    pDown->m_nIndex = 0;

    // 恢復(fù)為下載隊(duì)列為空
    for (int n=0; n<LIMIT; n++)
    {
        pDown->Flags[n].dwFlag = 1;
        pDown->Flags[n].lTaskId = n;
        // 無(wú)論是否下載成功，都必須調(diào)用XLStopTask
        XLStopTask(pDown->Flags[n].lTaskId);
    }

    //釋放資源
    XLUninitDownloadEngine();
}

void CDownload::Start (void)
{
    int nIndex = 0;

    // 1、初始化下載引擎
    if ( FALSE == XLInitDownloadEngine() )
    {
        ::MessageBox(NULL, _T("初始化下載引擎失敗!"), _T("友情提示"), MB_OK|MB_ICONEXCLAMATION);
        return;
    }

    // 下載任務(wù)查詢(xún)監(jiān)控線(xiàn)程
    ::CreateThread (NULL, 0, (LPTHREAD_START_ROUTINE)DownQury, this, 0, NULL);
}

// 調(diào)用迅雷的下載接口
void CDownload::CreateDownThread (int n)
{
    DWORD dwRst = 0;
    if (m_nIndex < strDownArr.size())
    {
        dwRst = XLURLDownloadToFile(strDownArr[m_nIndex].strSavePath, strDownArr[m_nIndex].strDownURL, NULL, Flags[n].lTaskId);
        if (dwRst == XL_SUCCESS)
        {
            Flags[n].dwFlag = enumTaskStatus_Download;
            m_nIndex++;
            m_nSuccess++;
        }
    }
    else
    {
        m_nIndex = strDownArr.size();
    }
}

void CDownload::SaveLog (LPCTSTR lpPathAndName)
{
    CStdioFile file;
    std::vector<CString>::iterator iter;

    // 保存現(xiàn)在的語(yǔ)言環(huán)境
    char * old_locale = _strdup(setlocale(LC_CTYPE,NULL));
    // 設(shè)置中文環(huán)境
    setlocale(LC_CTYPE, "chs");

    if (file.Open(lpPathAndName, CFile::modeCreate|CFile::modeWrite))
    {
        for (iter=strLogArr.begin(); iter!=strLogArr.end(); iter++)
        {
            file.WriteString(*iter);
            file.WriteString(_T("\n"));
        }
    }
    file.Close();
    // 還原語(yǔ)言區(qū)域的設(shè)置
    setlocale(LC_CTYPE, old_locale);
    //  釋放內(nèi)存
    free(old_locale);
}