proj/VTBFileUtil/VTBFileUtil.cpp

// VTB File Utility
//
// Douglas Thrift
//
// VTBFileUtil.cpp

#ifdef _DEBUG
#pragma warning(disable:4786)
#endif // _DEBUG

#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <cctype>
#include <cstdio>
#include <cstdlib>

#include <sys/stat.h>

using namespace std;

const unsigned LENGTH = 1024;
const char* CFG = "VTBFileUtil.cfg";

string filename(string& scan);
void rename(string& scan, string& good);
void move(string& good, string& directory);
void size(string& scan);
void quit(int code);

int main(int argc, char* argv[])
{
        string directory;

        if (argc > 1)
        {
                ifstream fin(CFG);
                getline(fin, directory);
                fin.close();
        }
        else
        {
                cout << "VTB File Utility\n"
                        << "Usage: VTBFileUtil [pdfscan ...]\n";
                return 0;
        }

        if (directory == "")
        {
                do
                {
                        cout << "No directory specified.\nDirectory: ";

                        char* cstr = new char[LENGTH];
                        cin.getline(cstr, LENGTH);

                        directory = cstr;

                        delete [] cstr;
                }
                while (directory == "");

                ofstream fout(CFG);
                fout << directory << "\n";
                fout.close();
        }

        for (unsigned index = 1; index < argc; index++)
        {
                if (strpbrk(argv[index], "*?")) continue;

                string extension;
                for (int number = 4; number > 0; number--)
                {
                        extension += tolower(argv[index][strlen(argv[index]) - number]);
                }
                if (extension != ".pdf")
                {
                        cout << argv[index] << " is not a PDF.\n";
                        continue;
                }

                ifstream tester(argv[index]);
                if (!tester.is_open())
                {
                        cout << argv[index] << " does not exist.\n";
                        continue;
                }
                tester.close();

                string scan = argv[index];
                string command = "start " + scan;
                system(command.c_str());

                string good;
                bool correct = false;
                do
                {
                        good = filename(scan);

                        cout << "Correct (y/n): ";

                        string answer;

                        char* cstr = new char[LENGTH];
                        cin.getline(cstr, LENGTH);

                        answer = cstr;

                        delete [] cstr;

                        for (int index = 0; index < answer.length(); index++)
                        {
                                answer[index] = tolower(answer[index]);
                        }

                        if (answer == "yes" || answer == "y")
                        {
                                correct = true;
                        }
                        else
                        {
                                cout << "Continue (y/n): ";
                                
                                cstr = new char[LENGTH];
                                cin.getline(cstr, LENGTH);

                                answer = cstr;

                                delete [] cstr;

                                if (answer != "yes" && answer != "y") quit(0);
                        }
                }
                while (!correct);

                rename(scan, good);

                move(good, directory);

                tester.open(directory.c_str());
                if (tester.is_open())
                {
                        tester.close();
                        rename(directory, good);

                        command = "mkdir " + directory;
                        system(command.c_str());

                        move(good, directory);
                }

                size(directory + "\\" + good);
        }

        return 0;
}

string filename(string& scan)
{
        string name, file;
        unsigned number;

        cout << "Scan PDF: " << scan << "\n"
                << "Client Name: ";

        char* cstr = new char[LENGTH];
        cin.getline(cstr, LENGTH);

        name = cstr;

        delete [] cstr;

        cout << "Client Number: ";

        cstr = new char[LENGTH];
        cin.getline(cstr, LENGTH);

        number = strtoul(cstr, 0, 0);

        delete [] cstr;

        cstr = new char[LENGTH];
        sprintf(cstr, "%u_", number);

        file += cstr;

        delete [] cstr;

        for (unsigned index = 0; index < name.length(); index++)
        {
                if (name[index] == ',' || name[index] == '.')
                {
                }
                else if (isspace(name[index]))
                {
                        file += '_';
                }
                else
                {
                        file += tolower(name[index]);
                }
        }

        file += ".pdf";

        cout << "Filename: " << file << "\n";

        return file;
}

void rename(string& scan, string& good)
{
        string command = "rename \"" + scan + "\" \"" + good + "\"";

        while (system(command.c_str())) system("pause");
}

void move(string& good, string& directory)
{
        string command = "move \"" + good + "\" \"" + directory + "\"";

        while (system(command.c_str())) system("pause");
}

void size(string& scan)
{
        struct stat status;

        if (!stat(scan.c_str(), &status))
        {
                char* cbytes = new char[LENGTH];
                sprintf(cbytes, "%u", status.st_size);
                
                string bytes = cbytes;
                
                delete [] cbytes;

                for (int index = bytes.length(), number = 0; index > 0; index--,
                        number++)
                {
                        if ((number % 3 == 0) && ((index - 3) > 0))
                        {
                                bytes.insert(index - 3, 1, ',');
                        }
                }

                char* megabytes = new char[LENGTH];
                sprintf(megabytes, "%.2f", (double(status.st_size) / double(1024) /
                        double(1024)));

                cout << "Filesize: " << bytes << " bytes (" << megabytes << " MB)\n";

                delete [] megabytes;
        }
        else
        {
                cerr << "VTBFileUtil: stat(scan.c_str(), &status) failed in void size("
                        << "string& scan).\n";
                exit(1);
        }
}

void quit(int code)
{
        exit(code);
}
Revision:	368
Committed:	2008-08-23T02:44:00-07:00 (16 years, 9 months ago) by douglas
File size:	4450 byte(s)
Log Message:	Rearranged everything else.
#	User	Rev	Content
1	douglas	60	// VTB File Utility
2			//
3			// Douglas Thrift
4			//
5			// VTBFileUtil.cpp
6
7			#ifdef _DEBUG
8			#pragma warning(disable:4786)
9			#endif // _DEBUG
10
11			#include <iostream>
12			#include <fstream>
13			#include <string>
14			#include <vector>
15			#include <cctype>
16			#include <cstdio>
17			#include <cstdlib>
18
19			#include <sys/stat.h>
20
21			using namespace std;
22
23			const unsigned LENGTH = 1024;
24			const char* CFG = "VTBFileUtil.cfg";
25
26			string filename(string& scan);
27			void rename(string& scan, string& good);
28			void move(string& good, string& directory);
29			void size(string& scan);
30			void quit(int code);
31
32			int main(int argc, char* argv[])
33			{
34			string directory;
35
36			if (argc > 1)
37			{
38			ifstream fin(CFG);
39			getline(fin, directory);
40			fin.close();
41			}
42			else
43			{
44			cout << "VTB File Utility\n"
45			<< "Usage: VTBFileUtil [pdfscan ...]\n";
46			return 0;
47			}
48
49			if (directory == "")
50			{
51			do
52			{
53			cout << "No directory specified.\nDirectory: ";
54
55			char* cstr = new char[LENGTH];
56			cin.getline(cstr, LENGTH);
57
58			directory = cstr;
59
60			delete [] cstr;
61			}
62			while (directory == "");
63
64			ofstream fout(CFG);
65			fout << directory << "\n";
66			fout.close();
67			}
68
69			for (unsigned index = 1; index < argc; index++)
70			{
71			if (strpbrk(argv[index], "*?")) continue;
72
73			string extension;
74			for (int number = 4; number > 0; number--)
75			{
76			extension += tolower(argv[index][strlen(argv[index]) - number]);
77			}
78			if (extension != ".pdf")
79			{
80			cout << argv[index] << " is not a PDF.\n";
81			continue;
82			}
83
84			ifstream tester(argv[index]);
85			if (!tester.is_open())
86			{
87			cout << argv[index] << " does not exist.\n";
88			continue;
89			}
90			tester.close();
91
92			string scan = argv[index];
93			string command = "start " + scan;
94			system(command.c_str());
95
96			string good;
97			bool correct = false;
98			do
99			{
100			good = filename(scan);
101
102			cout << "Correct (y/n): ";
103
104			string answer;
105
106			char* cstr = new char[LENGTH];
107			cin.getline(cstr, LENGTH);
108
109			answer = cstr;
110
111			delete [] cstr;
112
113			for (int index = 0; index < answer.length(); index++)
114			{
115			answer[index] = tolower(answer[index]);
116			}
117
118			if (answer == "yes" \|\| answer == "y")
119			{
120			correct = true;
121			}
122			else
123			{
124			cout << "Continue (y/n): ";
125
126			cstr = new char[LENGTH];
127			cin.getline(cstr, LENGTH);
128
129			answer = cstr;
130
131			delete [] cstr;
132
133			if (answer != "yes" && answer != "y") quit(0);
134			}
135			}
136			while (!correct);
137
138			rename(scan, good);
139
140			move(good, directory);
141
142			tester.open(directory.c_str());
143			if (tester.is_open())
144			{
145			tester.close();
146			rename(directory, good);
147
148			command = "mkdir " + directory;
149			system(command.c_str());
150
151			move(good, directory);
152			}
153
154			size(directory + "\\" + good);
155			}
156
157			return 0;
158			}
159
160			string filename(string& scan)
161			{
162			string name, file;
163			unsigned number;
164
165			cout << "Scan PDF: " << scan << "\n"
166			<< "Client Name: ";
167
168			char* cstr = new char[LENGTH];
169			cin.getline(cstr, LENGTH);
170
171			name = cstr;
172
173			delete [] cstr;
174
175			cout << "Client Number: ";
176
177			cstr = new char[LENGTH];
178			cin.getline(cstr, LENGTH);
179
180			number = strtoul(cstr, 0, 0);
181
182			delete [] cstr;
183
184			cstr = new char[LENGTH];
185			sprintf(cstr, "%u_", number);
186
187			file += cstr;
188
189			delete [] cstr;
190
191			for (unsigned index = 0; index < name.length(); index++)
192			{
193			if (name[index] == ',' \|\| name[index] == '.')
194			{
195			}
196			else if (isspace(name[index]))
197			{
198			file += '_';
199			}
200			else
201			{
202			file += tolower(name[index]);
203			}
204			}
205
206			file += ".pdf";
207
208			cout << "Filename: " << file << "\n";
209
210			return file;
211			}
212
213			void rename(string& scan, string& good)
214			{
215			string command = "rename \"" + scan + "\" \"" + good + "\"";
216
217			while (system(command.c_str())) system("pause");
218			}
219
220			void move(string& good, string& directory)
221			{
222			string command = "move \"" + good + "\" \"" + directory + "\"";
223
224			while (system(command.c_str())) system("pause");
225			}
226
227			void size(string& scan)
228			{
229			struct stat status;
230
231			if (!stat(scan.c_str(), &status))
232			{
233			char* cbytes = new char[LENGTH];
234			sprintf(cbytes, "%u", status.st_size);
235
236			string bytes = cbytes;
237
238			delete [] cbytes;
239
240			for (int index = bytes.length(), number = 0; index > 0; index--,
241			number++)
242			{
243			if ((number % 3 == 0) && ((index - 3) > 0))
244			{
245			bytes.insert(index - 3, 1, ',');
246			}
247			}
248
249			char* megabytes = new char[LENGTH];
250			sprintf(megabytes, "%.2f", (double(status.st_size) / double(1024) /
251			double(1024)));
252
253			cout << "Filesize: " << bytes << " bytes (" << megabytes << " MB)\n";
254
255			delete [] megabytes;
256			}
257			else
258			{
259			cerr << "VTBFileUtil: stat(scan.c_str(), &status) failed in void size("
260			<< "string& scan).\n";
261			exit(1);
262			}
263			}
264
265			void quit(int code)
266			{
267			exit(code);
268			}