#include <iostream>
#include <fstream>
#include "KillProcess.cpp"
using namespace std;

int OhNo(string Horrors)
{
    cerr << Horrors << endl;
    return 1;
}

int main()
{
    char fileName[255];
    cout << "Enter file to be securely removed: ";
    cin.getline(fileName,255);
    
    int number = 0; int number2 = 0; int number3 = 0; int number4 = 0; int number5 = 0;
    int number6 = 0; int number7 = 0; int number8 = 0; int number9 = 0; int number10 = 0;
    
    ifstream fin(fileName);
    if (fin)
    {
       cout << "File found." << endl;
       cout << "Now ending process if running..." << endl;
       KillProcess(fileName);
       
       int times, theLimit;
       cout << "Enter times to securely overwrite: ";
       cin >> times;
       if ( !cin.good() ) return OhNo("Error: bad input");
       
       // Overwrite method: Erases all data in file, writes random number. Does this as much as specified.
       for (theLimit = 1; theLimit <= times; theLimit++)
       {
           ofstream fout(fileName);
           number = rand(); number2 = rand(); number3 = rand(); number4 = rand(); number5 = rand();
           number6 = rand(); number7 = rand(); number8 = rand(); number9 = rand(); number10 = rand();
           
           fout << number << number2 << number3 << number4 << number5
           << number7 << number8 << number9 << number10;
           
           fout.close();
           cout << "*";
       }
       
       cout << "\nFile was overwritten " << theLimit << " times." << endl;
       cout << "Now removing file..." << endl;
       fin.close();
       int check = remove(fileName);
       
       if ( check == 0 ) { cout << "Success." << endl; }
       if ( check != 0 ) { cout << "Failure to remove file." << endl; }
    }
    else
    {
        cout << "File not found" << endl;
    }
    cin.sync();
    cin.get();
    return 0;
}