File-Organizer/main.cpp at main · Diwak4r/File-Organizer · 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
121
122
123
124
125
126
127
128
#include "organizer.h"

void showMenu() {
    cout << "\n===============================================" << endl;
    cout << "         FILE ORGANIZER" << endl;
    cout << "===============================================" << endl;
    cout << "1. Create Test Files" << endl;
    cout << "2. Show Categories" << endl;
    cout << "3. Organize Files" << endl;
    cout << "4. Show Statistics" << endl;
    cout << "5. Undo" << endl;
    cout << "6. Exit" << endl;
    cout << "===============================================" << endl;
    cout << "Choice: ";
}

int main() {
    string src = "test_files";
    string dst = "organized_files";

    cout << "Welcome to File Organizer!" << endl;
    cout << "This program helps you organize your files." << endl;

    FileOrganizer organizer(src, dst);

    int choice;
    bool running = true;

    while (running) {
        showMenu();
        cin >> choice;

        // Input validation - handle non-numeric input FIRST
        if (cin.fail()) {
            cin.clear();              // clear error state
            cin.ignore(1000, '\n');   // discard invalid input
            cout << "\nPlease enter a valid number!" << endl;
            cout << "Press Enter...";
            cin.get();
            system("cls");
            continue;
        }

        if (choice <= 0 || choice >= 7) {
            cout << "\nInvalid choice!" << endl;
            cout << "Press Enter...";
            cin.ignore();
            cin.get();
            system("cls");
            continue;
        }

        switch (choice) {
            case 1: {
                organizer.createTestFiles();
                cout << "\nPress Enter...";
                cin.ignore();
                cin.get();
                break;
            }

            case 2: {
                organizer.showCategories();
                cout << "\nPress Enter...";
                cin.ignore();
                cin.get();
                break;
            }

            case 3: {
                cout << "\nEnter source folder path: ";
                cin.ignore();
                getline(cin, src);
                cout << "Enter destination folder path: ";
                getline(cin, dst);
                organizer = FileOrganizer(src, dst);

                organizer.organizeFiles();
                cout << "\nPress Enter...";
                cin.get();
                break;
            }

            case 4: {
                organizer.displayStatistics();
                cout << "\nPress Enter...";
                cin.ignore();
                cin.get();
                break;
            }

            case 5: {
                char ans;
                cout << "\nThis will restore files to original location." << endl;
                cout << "Sure? (y/n): ";
                cin >> ans;

                if (ans == 'y' || ans == 'Y') {
                    organizer.undoLastOperation();
                }
                cout << "\nPress Enter...";
                cin.ignore();
                cin.get();
                break;
            }

            case 6: {
                cout << "\nThanks for using File Organizer!" << endl;
                running = false;
                break;
            }

            default: {
                cout << "\nInvalid choice!" << endl;
                cout << "Press Enter...";
                cin.ignore();
                cin.get();
            }
        }

        // clear screen
        if (running) {
            system("cls");
        }
    }

    return 0;
}