#include <TFile.h>
#include <fstream>
#include <sstream>
#include <vector>
#include <ICOMETLog.hxx>
#include <ICOMETLog.hxx>

void Usage(const char* prog){
    COMETLog("Usage: "<<prog<<" [options]\n"
          "\n"
          " -h,-?            Print this help\n"
          " -I <filename>    Text file containing list of root files, one per line\n"
          );
}

bool ReadInputFiles(const char* listfile,std::vector<std::string>& input_files);
bool CheckOneFile(const std::string& filename);

int main(int argc,char** argv){
    COMET::ICOMETLog::Configure();

    typedef std::vector<std::string> FileList;
    FileList input_files;

    for (;;) {
        char c = getopt(argc, argv, "h?I:");
        std::stringstream stream;
        if(optarg) stream.str(optarg);
        if (c<0) break;
        switch (c) {
            case 'I':
                {
                    if(!ReadInputFiles(optarg,input_files)){
                        return 1;
                    }
                    continue;
                }
            case 'h':case '?':
                Usage(argv[0]);
                return 0;
            default:
                COMETError("Unknown option: "<<c<<std::endl);
                Usage(argv[0]);
                return 1;
        }
        if(stream.fail() || ! stream.eof()){
            COMETError("Bad option value: '"<<optarg<<"' for option '"<<c <<"'"<<std::endl);
            Usage(argv[0]);
            return 1;
        }
    }

    for(int i=optind; i<argc;++i) input_files.push_back(argv[i]);

    int failed=0;
    int succeeded=0;
    for(FileList::const_iterator i_file=input_files.begin(); 
            i_file!=input_files.end();++i_file){
        if(CheckOneFile(*i_file))++succeeded;
        else ++failed;
    }
    COMETLog(std::endl<<"Summary");
    COMETLog("Number of successfull files: "<<succeeded);
    COMETLog("Number of failed files: "<<failed);

    return 0;
}

bool ReadInputFiles(const char* listfile_name,std::vector<std::string>& input_files){
    std::ifstream listfile(listfile_name);
    if(!listfile){
        COMETError("Cannot open input text file: "<<listfile_name);
        return false;
    }

    std::string line;
    while(std::getline(listfile,line)){
        input_files.push_back(line);
    }

    return true;
}

bool CheckOneFile(const std::string& filename){
    static int checked=1;
    TFile* file=TFile::Open(filename.c_str());
    bool file_is_good=true;
    if(file->TestBit(TFile::kRecovered)){
        file_is_good=false;
    }
    COMETLog(checked++<<") "<<filename<<": "<<(file_is_good?"good":"bad"));
    file->Close();
    return file_is_good;
}