/*
Console File Splitter v 0.2

Divide i file in pi� parti e all'occorrenza li ricompone.
Utile per qualsiasi esigenza di spazio.

Autore:             __GiReX__
Linguaggio:         C++
Sistema operativo:  Qualsiasi
Data:               26/11/07 20.47
*/

#include <iostream>
#include <fstream>
#include <cstring>
#include <ctime>
#include <cmath>
#define SPLIT_SUCCESS 0
#define MERGE_SUCCESS 4
using namespace std;

void usage(char *prog);
int split(char *nome_file, int parti);
int merge(char *nome_file, int parti);
void name_part(int n_part, char *nome_file, char *tmp);

int main(int argc, char *argv[])
{
  int parti, error;  clock_t sec;
  char *nome_file = NULL;
    
    if(argc < 3 || argc > 4) {
        usage(argv[0]);
        return 1;
    }
    
    if(argc == 4 && strcmp(argv[1], "-r")) {
        cout << "\n\"" << argv[1] << "\" Parametro non riconosciuto!\n\n";  
        usage(argv[0]);
        return 2;
    }                        
  
    if(argc == 3) { 
        nome_file = argv[1];
        parti = atoi(argv[2]);   
        error = split(nome_file, parti);                 // Splitto   
   } else {
        nome_file = argv[2];
        parti = atoi(argv[3]);
        error = merge(nome_file, parti);                 // Mergo
   }
   sec = clock() / CLOCKS_PER_SEC;

      switch(error) {
          case SPLIT_SUCCESS:
                cout << "Il file e' stato diviso in " << parti << " parti con successo!\n\n"
                     << "Tempo impiegato: " << sec << " sec\n";                
                break;
          case 1:
                cout << "Errore nell'apertura del file!\n\n"
                     << "Controllare che il nome o percorso siano corretti!\n";
                break;
          case 2:
                cout << "Errore: Numero di parti troppo elevato!\n";
                break;
          case 3:
                cout << "Errore nella creazione dei file part!\n";
                break;
          case -1:
                cout << "Errore nella creazione del file di destinazione!\n\n"
                     << "Controllare che il file originale non risieda nella stessa cartella!";
                break;
          case -2:
                cout << "Impossibile aprire i file part!\n\n"
                     << "Controllare che il nome del file corrisponda al nome del file originario\n"
                     << "e che il numero di parti dichiarato corrisponda!\n";
                break;
          case MERGE_SUCCESS:
                cout << "Il file e' stato ricomposto con successo!\n\n"
                     << "Tempo impiegato: " << sec << " sec\n";
                break;
          default:
                cout << "Errore sconosciuto!\n";
          }
      

      cout << "\nPremere un tasto per uscire...\n";    
    cin.ignore();
 return 0;
}
    
void name_part(int n_part, char *nome_file, char *tmp)
{  
 /* char buff[4] = {0};
     
   strcpy(tmp, nome_file);
   strcat(tmp, ".part_");
   strcat(tmp, itoa(n_part, buff, 10));                   // Funzione per convertire int to char*
  */
   sprintf(tmp, "%s.part_%d", nome_file, n_part);
}

int split(char *nome_file, int parti) 
{ 
  ifstream *file = new ifstream(nome_file,ios::binary);  // Apro file in modalit� binaria
  ofstream *file_part = new ofstream;  
  
   long double size, byte_file, last_file_size;          // Byte del file, e byte per ogni file
   char nome_file_part[80] = {0};                        // Nome dei file part
    
   if(!file)   return 1;    
       
   file->seekg(0, ios::end);
   size = file->tellg();                                 // Prendo lunghezza file
   
   cout << "\nFile di: " << size << " byte" << endl;
   if(size < parti*2 || parti > 30) return 2;
   
   file->seekg(0, ios::beg);                             // Risposto il cursore all'inizio del file
   byte_file = ceil(size / parti);                       // Byte per ogni file part arrotondati per eccesso
   last_file_size = (size - (byte_file*parti)) + byte_file;
                                                        // Byte dell'ultimo file che differisce dagli altri
   cout << "Splitting in corso...\n"; 
   
   for(int i = 0; i < parti; i++) 
   { 
    
     name_part(i, nome_file, nome_file_part);           // Prendo nome dei file part
     file_part->open(nome_file_part, ios::binary);      // Apro i file in modalit� binaria
     
     if(!file_part) return 3;
   
     if( i == (parti - 1)) {                            // Se � l'ultimo file..
         for(register int i2 = 0; i2 < last_file_size; i2++) 
              file_part->put(file->get()); 
     } else {
         for(register int i2 = 0; i2 < byte_file; i2++) 
             file_part->put(file->get());        // Un byte alla volta copio n byte per file nel file part
     } 
           
    file_part->close(); 
   }
 
  file->close();
 return SPLIT_SUCCESS;
}

int merge(char *nome_file, int parti)
{
  char bit, nome_part[80] = {0};
   ofstream *file = new ofstream(nome_file, ios::binary);
   ifstream *file_part = new ifstream;
    
    if(!file) return -1;

    cout << "\nRicomponimento (merge) in corso...\n";    
    
     for(int i = 0; i < parti; i++) 
     {
       name_part(i, nome_file, nome_part);                  // Prendo il nome dei file part
       file_part->open(nome_part, ios::binary);             // Apro i file part
       
       if(!file_part)  return -2;

       while(!file_part->eof())   
       {                       // Se npn � un eofbit lo copio nel file
           file_part->get(bit);
           if(!(file_part->rdstate() & ifstream::eofbit)) file->put(bit);  
			       // Copio i file part nel file di destinazione
     }
     
       file_part->clear();
       file_part->close();                                              // Chiudo file part
     }
   
    file->close();
 return MERGE_SUCCESS;
}

void usage(char *prog) 
{
     cout << "Usage:\n\n"
          << "Per dividere: " << prog << " <file_da_dividere> <numero_parti>\n\n"
          << "Per ricomporre: " << prog << " -r <file_origine> <numero_parti>\n\n"
          << "Dove <file_origine> e' il nome del file che e' stato diviso\n"
          << "e <numero_parti> le parti in cui e' stato diviso!\n";
}