ArchiveProcessor.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2011 Red Hat Inc. and/or its affiliates and other contributors
 * as indicated by the @authors tag. All rights reserved.
 * 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.jboss.arquillian.showcase.extension.systemproperties.client;

import java.io.ByteArrayOutputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

import org.jboss.arquillian.config.descriptor.api.ArquillianDescriptor;
import org.jboss.arquillian.config.descriptor.api.ExtensionDef;
import org.jboss.arquillian.container.test.spi.client.deployment.ApplicationArchiveProcessor;
import org.jboss.arquillian.core.api.Instance;
import org.jboss.arquillian.core.api.annotation.Inject;
import org.jboss.arquillian.showcase.extension.systemproperties.SystemProperties;
import org.jboss.arquillian.test.spi.TestClass;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.shrinkwrap.api.asset.StringAsset;
import org.jboss.shrinkwrap.api.container.ResourceContainer;

/**
 * ArchiveProcessor
 *
 * @author <a href="mailto:aslak@redhat.com">Aslak Knutsen</a>
 * @version $Revision: $
 */
public class ArchiveProcessor implements ApplicationArchiveProcessor
{
   @Inject
   private Instance<ArquillianDescriptor> descriptor;
   
   @Override
   public void process(Archive<?> applicationArchive, TestClass testClass)
   {
      String prefix = getPrefix();
      if(prefix != null)
      {
         if(applicationArchive instanceof ResourceContainer)
         {
            ResourceContainer<?> container = (ResourceContainer<?>)applicationArchive;
            container.addAsResource(
                  new StringAsset(
                        toString(
                              filterSystemProperties(prefix))), SystemProperties.FILE_NAME);
         }
      }
   }

   private String getPrefix()
   {
      return getConfiguration().get(SystemProperties.CONFIG_PREFIX);
   }

   private Map<String, String> getConfiguration()
   {
      for(ExtensionDef def : descriptor.get().getExtensions())
      {
         if(SystemProperties.EXTENSION_NAME.equalsIgnoreCase(def.getExtensionName()))
         {
            return def.getExtensionProperties();
         }
      }
      return new HashMap<String, String>();
   }

   private Properties filterSystemProperties(String prefix)
   {
      Properties filteredProps = new Properties();
      Properties sysProps = System.getProperties();
      for (Map.Entry<Object, Object> entry: sysProps.entrySet())
      {
         if(entry.getKey().toString().startsWith(prefix))
         {
            String newKey = entry.getKey().toString().replaceFirst(prefix, "");
            filteredProps.put(newKey, entry.getValue());
         }
      }
      return filteredProps;
   }
   
   private String toString(Properties props)
   {
      try
      {
         ByteArrayOutputStream out = new ByteArrayOutputStream();
         props.store(out, "Arquillian SystemProperties Extension");
         return out.toString();
      }
      catch (Exception e)
      {
         throw new RuntimeException("Could not store properties", e);
      }
   }
}