清理文件2

// FileCleanerDlg.h : 头文件
//

#pragma once
#include <vector>
using std::vector;

// CFileCleanerDlg 对话框
class CFileCleanerDlg : public CDialogEx
{
// 构造
public:
    CFileCleanerDlg(CWnd* pParent = NULL);    // 标准构造函数

// 对话框数据
    enum { IDD = IDD_FILECLEANER_DIALOG };

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
    HICON m_hIcon;

    // 生成的消息映射函数
    virtual BOOL OnInitDialog();
    afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
    afx_msg void OnPaint();
    afx_msg HCURSOR OnQueryDragIcon();
    DECLARE_MESSAGE_MAP()
public:
    afx_msg void OnBnClickedButton1();
    CString m_szPath;
    CString m_szFormat;
    CString m_szClearFile;
    void ErgodicFile(TCHAR  szPath[MAX_PATH]);
    vector<CString> m_vecPath;

    afx_msg void OnBnClickedButton2();
    afx_msg void OnBnClickedButton3();
};

// FileCleanerDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "FileCleaner.h"
#include "FileCleanerDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialogEx
{
public:
    CAboutDlg();

// 对话框数据
    enum { IDD = IDD_ABOUTBOX };

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()

// CFileCleanerDlg 对话框

//, m_strPath(_T(""))
//, m_strShow(_T(""))
//, m_Filter(L".txt.cpp.exe")

//变量的初始化，过滤文件的格式初始化
CFileCleanerDlg::CFileCleanerDlg(CWnd* pParent /*=NULL*/)
    : CDialogEx(CFileCleanerDlg::IDD, pParent)
    , m_szPath(_T(""))
    , m_szFormat(L".txt.exe")
    , m_szClearFile(_T(""))
{
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CFileCleanerDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
    DDX_Text(pDX, IDC_EDIT1, m_szPath);
    DDX_Text(pDX, IDC_EDIT3, m_szFormat);
    DDX_Text(pDX, IDC_EDIT2, m_szClearFile);
}

BEGIN_MESSAGE_MAP(CFileCleanerDlg, CDialogEx)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_BN_CLICKED(IDC_BUTTON1, &CFileCleanerDlg::OnBnClickedButton1)
    ON_BN_CLICKED(IDC_BUTTON2, &CFileCleanerDlg::OnBnClickedButton2)
    ON_BN_CLICKED(IDC_BUTTON3, &CFileCleanerDlg::OnBnClickedButton3)
END_MESSAGE_MAP()

// CFileCleanerDlg 消息处理程序

BOOL CFileCleanerDlg::OnInitDialog()
{
    CDialogEx::OnInitDialog();

    // 将“关于...”菜单项添加到系统菜单中。

    // IDM_ABOUTBOX 必须在系统命令范围内。
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

    CMenu* pSysMenu = GetSystemMenu(FALSE);
    if (pSysMenu != NULL)
    {
        BOOL bNameValid;
        CString strAboutMenu;
        bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
        ASSERT(bNameValid);
        if (!strAboutMenu.IsEmpty())
        {
            pSysMenu->AppendMenu(MF_SEPARATOR);
            pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
        }
    }

    // 设置此对话框的图标。  当应用程序主窗口不是对话框时，框架将自动
    //  执行此操作
    SetIcon(m_hIcon, TRUE);            // 设置大图标
    SetIcon(m_hIcon, FALSE);        // 设置小图标

    // TODO:  在此添加额外的初始化代码

    return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void CFileCleanerDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
    if ((nID & 0xFFF0) == IDM_ABOUTBOX)
    {
        CAboutDlg dlgAbout;
        dlgAbout.DoModal();
    }
    else
    {
        CDialogEx::OnSysCommand(nID, lParam);
    }
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CFileCleanerDlg::OnPaint()
{
    if (IsIconic())
    {
        CPaintDC dc(this); // 用于绘制的设备上下文

        SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

        // 使图标在工作区矩形中居中
        int cxIcon = GetSystemMetrics(SM_CXICON);
        int cyIcon = GetSystemMetrics(SM_CYICON);
        CRect rect;
        GetClientRect(&rect);
        int x = (rect.Width() - cxIcon + 1) / 2;
        int y = (rect.Height() - cyIcon + 1) / 2;

        // 绘制图标
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CDialogEx::OnPaint();
    }
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CFileCleanerDlg::OnQueryDragIcon()
{
    return static_cast<HCURSOR>(m_hIcon);
}

//选择文件夹
void CFileCleanerDlg::OnBnClickedButton1()
{
    // TODO:  在此添加控件通知处理程序代码
    TCHAR strPath[MAX_PATH];         //设置变量保存路径，此路径显示在弹出框顶部
    BROWSEINFO bi = {};
    bi.hwndOwner = m_hWnd;           //
    bi.pszDisplayName = strPath;
    LPCITEMIDLIST   pid = SHBrowseForFolder(&bi);
    SHGetPathFromIDList(pid, strPath);
    m_szPath = strPath;
    UpdateData(FALSE);
    CoTaskMemFree((LPVOID)pid);

}

//遍历文件
void CFileCleanerDlg::ErgodicFile(TCHAR  szPath[MAX_PATH])
{
    CString strPath = szPath;
    strPath += L"\\*";
    WIN32_FIND_DATA fi = {};
    HANDLE hFile = FindFirstFile(strPath, &fi);
    if (hFile == INVALID_HANDLE_VALUE)
    {
        return;
    }
    do
    {
        if (!lstrcmp(fi.cFileName, L".") || !lstrcmp(fi.cFileName, L".."))
        {
            continue;
        }
        TCHAR *p = PathFindExtension(fi.cFileName);
        if (-1 == m_szFormat.Find(p))
        {
            continue;
        }
        if (fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
        {
            CString strDIr = szPath;
            strDIr += L"\\";
            strDIr += fi.cFileName;
            ErgodicFile(strDIr.GetBuffer());
            continue;
        }
        CString strDIr = szPath;
        strDIr += L"\\";
        strDIr += fi.cFileName;
        m_vecPath.push_back(strDIr);
    } while (FindNextFile(hFile, &fi));

}

//获取文件夹中的文件
void CFileCleanerDlg::OnBnClickedButton2()
{
    // TODO:  在此添加控件通知处理程序代码
    /*m_szClearFile = L"";*/
    m_vecPath.clear();
    ErgodicFile(m_szPath.GetBuffer());
    for (DWORD i = 0; i < m_vecPath.size(); i++)
    {
        m_szClearFile += m_vecPath[i];
        m_szClearFile += L"\r\n";
    }
    UpdateData(FALSE);
}

//删除文件
void CFileCleanerDlg::OnBnClickedButton3()
{
    for (int i = 0; i < m_vecPath.size(); i++)
        DeleteFile(m_vecPath[i].GetString());
    MessageBox(TEXT("删除成功！"), NULL, 0);

    m_szClearFile = "";
    UpdateData(FALSE);
}