FileIndexation/DiscFunctions.cpp at master · olehpidhi/FileIndexation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#include "DiscFunctions.h"
#include "mainwindow.h"
#include "ui_mainwindow.h"

void findAllFilesInDirectory(const CHAR* dirPath, std::list<FileInfo>& filesData, QLabel* currentDirectory)
{
    //copy directory path to
    CHAR directoryPath[MAX_PATH];
    StringCchCopyA(directoryPath, MAX_PATH, dirPath);

    //get length of directory path
    size_t pathLength = strlen(dirPath);
    //StringCchLengthA(directoryPath, MAX_PATH, &pathLength);

    //check if length of path is valid
    if (pathLength > (MAX_PATH - 3))
    {
        //show error message here
        return;
    }
    //make string compatilbel with FundFile function

    StringCchCatA(directoryPath, MAX_PATH, "\\*");

    //find first file in directory
    WIN32_FIND_DATAA foundFileData;
    HANDLE foundFileHandler = FindFirstFileA(directoryPath, &foundFileData);

    if (foundFileHandler == INVALID_HANDLE_VALUE)
    {
        //show error message
        return;
    }

    do
    {
        char* filename = foundFileData.cFileName;
        if(!(foundFileData.dwFileAttributes & (FILE_ATTRIBUTE_NOT_CONTENT_INDEXED))
                && !(foundFileData.dwFileAttributes & (FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM))
                && (filename[strlen(filename)-1] != '.'))
        {
            CHAR absolutePath[MAX_PATH];
            StringCchCopyA(absolutePath, MAX_PATH, dirPath);
            StringCchCatA(absolutePath, MAX_PATH, foundFileData.cFileName);

            StringCchCatA(absolutePath, MAX_PATH, "\\\0");
            QString txt = QString(absolutePath);
            if(txt == "C:\\Program Files\\Common Files\\Adobe\\Adobe Photoshop CS5\\AMT_Driver")
            {
                int a =9;
            }
            currentDirectory->setText(txt);

            if (foundFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
            {
                findAllFilesInDirectory(absolutePath, filesData, currentDirectory);
            }
            else
            {
                filesData.push_back(FileInfo(foundFileData, std::string(absolutePath)));
            }
        }
    } while (FindNextFileA(foundFileHandler, &foundFileData) != 0);


    if (GetLastError() == ERROR_NO_MORE_FILES)
    {
        //showError
    }
    FindClose(foundFileHandler);
}

void DisplayErrorBox(LPTSTR lpszFunction)
{
    // Retrieve the system error message for the last-error code

    LPVOID lpMsgBuf;
    LPVOID lpDisplayBuf;
    DWORD dw = GetLastError();

    FormatMessage(
        FORMAT_MESSAGE_ALLOCATE_BUFFER |
        FORMAT_MESSAGE_FROM_SYSTEM |
        FORMAT_MESSAGE_IGNORE_INSERTS,
        NULL,
        dw,
        MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
        (LPTSTR)&lpMsgBuf,
        0, NULL);

    // Display the error message and clean up

    lpDisplayBuf = (LPVOID)LocalAlloc(LMEM_ZEROINIT,
        (lstrlen((LPCTSTR)lpMsgBuf) + lstrlen((LPCTSTR)lpszFunction) + 40)*sizeof(TCHAR));
    StringCchPrintf((LPTSTR)lpDisplayBuf,
        LocalSize(lpDisplayBuf) / sizeof(TCHAR),
        TEXT("%s failed with error %d: %s"),
        lpszFunction, dw, lpMsgBuf);
    MessageBox(NULL, (LPCTSTR)lpDisplayBuf, TEXT("Error"), MB_OK);

    LocalFree(lpMsgBuf);
    LocalFree(lpDisplayBuf);
}

std::list<std::string> getDiscsLetters()
{
    CHAR driveStrings[MAX_PATH];
    DWORD result = GetLogicalDriveStringsA(result, driveStrings);
    std::list<std::string> drivesLetters;
    if(result > 0 && result <= MAX_PATH)
    {
        char* singleDrive = driveStrings;
        while(*singleDrive)
        {
            drivesLetters.push_back(std::string(singleDrive));
            singleDrive += strlen(singleDrive) + 1;
        }
    }
    return drivesLetters;
}