package net.sourceforge.seqware.pipeline.workflowV2.engine.oozie.object;

import java.io.File;
import java.util.ArrayList;
import java.util.List;
import static net.sourceforge.seqware.pipeline.workflowV2.engine.oozie.object.OozieJob.file;
import net.sourceforge.seqware.pipeline.workflowV2.model.AbstractJob;
import org.jdom.Element;

/**
 * Container for batching up provision file jobs
 *
 * @author dyuen
 */
public class BatchedOozieProvisionFileJob extends OozieJob {

    private final List<OozieProvisionFileJob> provisionJobs = new ArrayList<>();

    public BatchedOozieProvisionFileJob(AbstractJob job, String name, String oozie_working_dir, boolean useSge, File seqwareJar,
            String threadsSgeParamFormat, String maxMemorySgeParamFormat, StringTruncator truncator) {
        super(job, name, oozie_working_dir, useSge, seqwareJar, threadsSgeParamFormat, maxMemorySgeParamFormat, truncator);
    }

    @Override
    protected Element createSgeElement() {
        File runnerScript = emitRunnerScript();
        File optionsFile = emitOptionsFile();

        Element sge = new Element("sge", SGE_XMLNS);
        add(sge, "script", runnerScript.getAbsolutePath());
        add(sge, "options-file", optionsFile.getAbsolutePath());

        return sge;
    }

    @Override
    protected Element createJavaElement() {
        throw new UnsupportedOperationException();
    }

    public void attachProvisionFileJob(OozieProvisionFileJob job) {
        // mutate longName in order to avoid repeats
        job.setLongName(job.getLongName() + "_" + provisionJobs.size());
        provisionJobs.add(job);
    }

    public int getBatchSize() {
        return provisionJobs.size();
    }

    private File emitRunnerScript() {
        File localFile = file(scriptsDir, runnerFileName(this.getLongName()), true);

        ArrayList<String> args = new ArrayList<>();
        for (OozieProvisionFileJob batchedJob : provisionJobs) {
            batchedJob.setUseCheckFile(true);
            args.add(concat(" ", batchedJob.generateRunnerLine()));
        }

        writeScript(concat("\n", args), localFile);
        return localFile;
    }

    @Override
    public List<String> getAccessionFile() {
        List<String> list = new ArrayList<>();
        for (OozieProvisionFileJob job : this.provisionJobs) {
            List<String> accessionFile = job.getAccessionFile();
            list.addAll(accessionFile);
        }
        return list;
    }
}