org.arquillian.spacelift.task.os.SpawnProcessTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arquillian-spacelift Show documentation
Arquillian Process and Package Manager
There is a newer version: 1.0.2
/*
 * JBoss, Home of Professional Open Source
 * Copyright 2014, Red Hat Middleware LLC, and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.arquillian.spacelift.task.os;

import java.io.File;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.arquillian.spacelift.execution.Execution;
import org.arquillian.spacelift.execution.ExecutionException;
import org.arquillian.spacelift.process.Command;
import org.arquillian.spacelift.task.Task;

/**
 * @author Karel Piwko
 */
class SpawnProcessTask extends Task {

    private Command command;
    private boolean redirectErrorStream;
    private List allowedExitCodes;
    private File workingDirectory;
    private Map environment;
    private boolean runsAsDaemon;

    public SpawnProcessTask redirectErrorStream(boolean redirectErrorStream) {
        this.redirectErrorStream = redirectErrorStream;
        this.environment = new HashMap();
        return this;
    }

    public SpawnProcessTask command(Command command) {
        this.command = command;
        return this;
    }

    public SpawnProcessTask workingDirectory(File workingDirectory) {
        this.workingDirectory = workingDirectory;
        return this;
    }

    public SpawnProcessTask addEnvironment(Map environment) {
        this.environment.putAll(environment);
        return this;
    }

    public SpawnProcessTask shouldExitWith(List allowedExitCodes) {
        this.allowedExitCodes = allowedExitCodes;
        return this;
    }

    public SpawnProcessTask runAsDaemon(boolean isDaemon) {
        this.runsAsDaemon = isDaemon;
        return this;
    }

    @Override
    public Execution execute() throws ExecutionException {

        // here we rewrap future based execution into process based execution to get better details about execution
        Execution processFutureExecution = super.execute();

        Process process = processFutureExecution.await();
        ProcessReference ref = new ProcessReference(command.getProgramName());
        ref.setProcess(process);

        ProcessBasedExecution execution = new ProcessBasedExecution(processFutureExecution, ref,
            command.getProgramName(),
            allowedExitCodes);

        // register shutdown hook
        if (!runsAsDaemon) {
            execution.registerShutdownHook();
        }

        return execution;
    }

    @Override
    protected Process process(Object input) throws Exception {

        Validate.executionNotNull(command, "Command must not be null");
        Validate.executionNotNull(command.getProgramName(), "Command program name must not be null");

        ProcessBuilder builder = new ProcessBuilder(command.getFullCommand());
        builder.directory(workingDirectory);

        builder.environment().putAll(environment);
        builder.redirectErrorStream(redirectErrorStream);
        return builder.start();
    }
}