ConfigureFabricCommand.java

/*
 * Copyright 2014 JBoss Inc
 *
 * 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.overlord.dtgov.karaf.commands;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.util.Properties;

import org.apache.felix.gogo.commands.Command;
import org.overlord.commons.codec.AesEncrypter;
import org.overlord.commons.karaf.commands.configure.AbstractConfigureFabricCommand;

/**
 * Karaf console command for use within JBoss Fuse. It should be used before the
 * fabric is created. It configures s-ramp fabric profiles, including overlord
 * profile and sramp profile.
 *
 * @author David Virgil Naranjo
 */
@Command(scope = "overlord:fabric:dtgov", name = "configure")
public class ConfigureFabricCommand extends AbstractConfigureFabricCommand {


    private static String DTGOV_PROFILE_PATH;

    static {
        if (File.separator.equals("/")) { //$NON-NLS-1$
            DTGOV_PROFILE_PATH = "overlord/dtgov.profile"; //$NON-NLS-1$
        } else {
            DTGOV_PROFILE_PATH = "overlord\\dtgov.profile"; //$NON-NLS-1$
        }
    }

    /*
     * (non-Javadoc)
     *
     * @see org.apache.karaf.shell.console.AbstractAction#doExecute()
     */
    @Override
    protected Object doExecute() throws Exception {
        super.doExecute();
        addHeaderProperties();
        configureDtgovProperties();
        return null;
    }


    private void configureDtgovProperties() throws Exception {
        InputStream is = this.getClass().getResourceAsStream("/" + ConfigureConstants.DTGOV_PROPERTIES_FILE_NAME); //$NON-NLS-1$
        Properties dtgovProps = new Properties();
        dtgovProps.load(is);

        String aesEncryptedValue = AesEncrypter.encrypt(password);
        StringBuilder aesEncrypterBuilder = new StringBuilder();
        aesEncrypterBuilder.append("$\\{crypt:").append(aesEncryptedValue).append("\\}"); //$NON-NLS-1$ //$NON-NLS-2$
        aesEncryptedValue = aesEncrypterBuilder.toString();

        for (Object key : dtgovProps.keySet()) {
            String value = (String) dtgovProps.get(key);
            if (value.contains(ConfigureConstants.DTGOV_WORKFLOW_PASSWORD)) {
                dtgovProps.put(key, aesEncryptedValue);
            }
        }
        File dtgovFile = new File(getDtgovPropertiesFilePath());
        FileOutputStream os = null;
        try {
            os = new FileOutputStream(dtgovFile);
            dtgovProps.store(os, ""); //$NON-NLS-1$
        } finally {
            if (os != null) {
                os.close();
            }
        }

    }

    /**
     * Adds the header properties.
     *
     * @throws Exception
     *             the exception
     */
    private void addHeaderProperties() throws Exception {

        String filePath = getOverlordPropertiesFilePath();

        Properties props = new Properties();
        InputStream is = null;
        try {
            is = new FileInputStream(new File(filePath));
            props.load(is);
        } finally {
            if (is != null) {
                is.close();
            }
        }

        FileOutputStream out = null;
        try {
            out = new FileOutputStream(filePath);
            props.setProperty(ConfigureConstants.DTGOV_HEADER_HREF, ConfigureConstants.DTGOV_HEADER_HREF_VALUE);
            props.setProperty(ConfigureConstants.DTGOV_HEADER_LABEL, ConfigureConstants.DTGOV_HEADER_LABEL_VALUE);
            props.setProperty(ConfigureConstants.DTGOV_HEADER_PRIMARY_BRAND, ConfigureConstants.DTGOV_HEADER_PRIMARY_BRAND_VALUE);
            props.setProperty(ConfigureConstants.DTGOV_HEADER_SECOND_BRAND, ConfigureConstants.DTGOV_HEADER_SECOND_BRAND_VALUE);
            props.store(out, null);
        } finally {
            if (out != null) {
                out.close();
            }
        }
    }

    /**
     * Gets the fabric sramp profile path.
     *
     * @return the fuse config path
     */
    public String getDtgovFabricProfilePath() {
        StringBuilder fuse_config_path = new StringBuilder();
        fuse_config_path.append(getFabricProfilesPath()).append(DTGOV_PROFILE_PATH).append(File.separator);
        return fuse_config_path.toString();
    }

    /**
     * Gets the sramp properties file path.
     *
     * @return the sramp properties file path
     */
    private String getDtgovPropertiesFilePath() {
        StringBuilder fuse_config_path = new StringBuilder();
        fuse_config_path.append(getDtgovFabricProfilePath()).append(ConfigureConstants.DTGOV_PROPERTIES_FILE_NAME);
        return fuse_config_path.toString();
    }

}