SetEnvBuildWrapper.java

package hudson.plugins.setenv;

import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Properties;
import java.io.ByteArrayInputStream;
import java.io.StringReader;

import net.sf.json.JSONArray;
import net.sf.json.JSONObject;

import org.kohsuke.stapler.DataBoundConstructor;
import org.kohsuke.stapler.StaplerRequest;
import org.kohsuke.stapler.StaplerResponse;

import hudson.EnvVars;
import hudson.Extension;
import hudson.Launcher;
import hudson.model.Descriptor;
import hudson.model.AbstractBuild;
import hudson.model.AbstractProject;
import hudson.model.BuildListener;
import hudson.plugins.setenv.Messages;
import hudson.tasks.BuildWrapper;
import hudson.tasks.BuildWrapperDescriptor;

/**
 * This plugin will set user-configured environment variables. 
 *
 * @author Takeshi Wakasugi
 */
public class SetEnvBuildWrapper extends BuildWrapper {
    private String localVarText;

    @DataBoundConstructor
    public SetEnvBuildWrapper(String localVarText){
        this.localVarText = localVarText;
    }
    
    public String getLocalVarText() {
        return localVarText;
    }
    
    public void setLocalVarText(String localVarText) {
        this.localVarText = localVarText;
    }
    
    public Map<String,String> getLocalVarMap() {
        Map<String,String> map = new HashMap<String,String>(0);
        try {
            for (Entry<Object,Object> entry : load(localVarText).entrySet()) {
                map.put(entry.getKey().toString(), entry.getValue().toString());
            }
        }
        catch (IOException e) {
            // Not sure what to do here?
        }

        return map;
    }

    private Properties load(String properties) throws IOException {
        Properties p = new Properties();
        try {
            p.load(new StringReader(properties));
        } catch (NoSuchMethodError e) {
            // load(Reader) method is only available on JDK6.
            // this fall back version doesn't work correctly with non-ASCII characters,
            // but there's no other easy ways out it seems.
            p.load(new ByteArrayInputStream(properties.getBytes()));
        }
        return p;
    }

    /*    public class EnvironmentImpl extends BuildWrapper.Environment{
        public void buildEnvVars(Map<String, String> env) {
            // To set local-configured value
            Map<String,String> localVarMap = getLocalVarMap();
            if(localVarMap != null)env.putAll(localVarMap);
        }
        }*/

    @Override
    public Environment setUp(AbstractBuild build, Launcher launcher,
                             BuildListener listener) throws IOException, InterruptedException {
        class EnvironmentImpl extends BuildWrapper.Environment {
            public void buildEnvVars(Map<String, String> env) {
                // To set local-configured value
                Map<String,String> localVarMap = getLocalVarMap();
                EnvVars.resolve(localVarMap);
                if (localVarMap != null) {
                    env.putAll(localVarMap);
                }
            }
        }
        
        return new EnvironmentImpl();
    }
    
    /**
     * Descriptor for {@link SetEnvBuildWrapper}. Used as a singleton.
     * The class is marked as public so that it can be accessed from views.
     */
    @Extension
    public static final class DescriptorImpl extends BuildWrapperDescriptor {
        @Override
        public String getDisplayName() {
            return Messages.SetEnvBuildWrapper_DisplayName(); 
        }

        @Override
        public boolean isApplicable(AbstractProject item) {
            return true;
        }

    }

}