package org.kit.furia.fragment;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Iterator;
import java.util.List;

import org.apache.log4j.BasicConfigurator;
import org.apache.log4j.FileAppender;
import org.apache.log4j.Level;
import org.apache.log4j.Logger;
import org.apache.log4j.PatternLayout;
import org.apache.log4j.PropertyConfigurator;
import org.kit.furia.FuriaChanConstants;
import org.kit.furia.fragment.asm.FragmentExtractorASM;
import org.kit.furia.fragment.soot.FragmentExtractorSoot;
import org.kit.furia.fragment.soot.NoClassesFound;

/*
 Furia-chan: An Open Source software license violation detector.    
 Copyright (C) 2007 Kyushu Institute of Technology

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

/**
 * FragmentBuilderClientAux fragments a directory of class files. It is designed
 * to be called from the command line. This program is not intended to be called
 * by humans, only intended to be called by other programs. Receives the
 * following parameters:
 * 
 * <pre>
 * [1] The directory that is to be processed. 
 * [2] The output directory in which the fragments will be stored in a &quot;fragments&quot; file.
 * If the filename {@value org.kit.furia.io.AbstractFuriaInput.fragmentFileName}
 * is found in the output directory, this directory is silently skipped.
 * [3] The fragment extractor engine: soot or asm. ASM is faster and works almost in any class files. 
 *  Soot is slower and sometimes fails to load the class files.
 * </pre>
 * 
 * @author Arnoldo Jose Muller Molina TODO: remove the fragments file if there
 *         is an error.
 */

public class FragmentBuilderClientAux {

    private static Logger logger = Logger.getLogger("FragmentBuilderClientAux");

    public static void main(String[] args) throws Exception {

        try {
            File dir = new File(args[0]);
            File output = new File(args[1]);
            if (!dir.exists()) {
                String msg = "Input directory does not exist: " + dir;
                System.err.println(msg);
                throw new IOException(msg);
            }
            output.mkdirs();

            if (!output.exists()) {
                String msg = "Output directory does not exist: " + output;
                System.err.println(msg);
                throw new IOException(msg);
            }

            Logger root = Logger.getRootLogger();
            root.addAppender(new FileAppender(new PatternLayout(
                    PatternLayout.TTCC_CONVERSION_PATTERN), output.toString()
                    + File.separator + "output.txt"));
            root.setLevel(Level.DEBUG);

            File fragmentsFile = new File(output,
                    org.kit.furia.io.AbstractFuriaInput.fragmentFileName);
            if (fragmentsFile.exists()) {
                // nothing to do here. the file was already processed.
                System.exit(0);
            }
            FragmentExtractor fEx = null;
            if (args[2].equals("asm")) {
                fEx = new FragmentExtractorASM();
            } else if (args[2].equals("soot")) {
                fEx = new FragmentExtractorSoot();
            } else {
                throw new Exception(
                        "Must specify fragment extraction engine: soot or asm");
            }

            fEx.extractMethodsFromDirectory(dir.toString(),
                    FuriaChanConstants.MAX_NODES_PER_FRAGMENT,
                    FuriaChanConstants.MIN_NODES_PER_FRAGMENT, output
                            .toString(), fragmentsFile.toString());
            /*
             * Iterator < FragmentBuilder > it = result.iterator(); FileWriter
             * outputFile = new FileWriter(fragmentsFile); while (it.hasNext()) {
             * FragmentBuilder fb = it.next(); StringBuilder tmp = new
             * StringBuilder(); fb.generateString(tmp);
             * outputFile.write(tmp.toString()); } outputFile.close();
             */
            logger.info("Completed Fragmentation for " + args[0]);
        } 
        catch(NoClassesFound e1){
            System.exit(7);
        }
        catch (Exception e) {
            e.printStackTrace();
            logger.fatal("Aborting", e);
            // logger.fatal("Received Env:\n" + System.getenv().toString());
            System.exit(48);
        }

    }
}