/*
    * Copyright (c) 2016 ingenieux Labs
    *
    * 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 br.com.ingenieux.mojo.beanstalk;
  
  import com.amazonaws.auth.AWSCredentials;
  import com.amazonaws.services.elasticbeanstalk.AWSElasticBeanstalk;
  import com.amazonaws.services.elasticbeanstalk.AWSElasticBeanstalkClient;
  import com.amazonaws.services.elasticbeanstalk.model.DescribeEnvironmentsResult;
  import com.amazonaws.services.elasticbeanstalk.model.EnvironmentDescription;
  import com.amazonaws.services.elasticbeanstalk.model.TerminateEnvironmentRequest;
  
  import org.apache.commons.lang.text.StrSubstitutor;
  import org.apache.maven.plugin.testing.AbstractMojoTestCase;
  import org.codehaus.plexus.configuration.PlexusConfiguration;
  import org.junit.Ignore;
  
  import java.io.File;
  import java.io.FileInputStream;
  import java.net.URISyntaxException;
  import java.util.Properties;
  
  import br.com.ingenieux.mojo.aws.AbstractAWSMojo;
  import br.com.ingenieux.mojo.beanstalk.app.CreateApplicationMojo;
  import br.com.ingenieux.mojo.beanstalk.bundle.UploadSourceBundleMojo;
  import br.com.ingenieux.mojo.beanstalk.config.CreateConfigurationTemplateMojo;
  import br.com.ingenieux.mojo.beanstalk.config.DescribeConfigurationTemplatesMojo;
  import br.com.ingenieux.mojo.beanstalk.dns.CheckAvailabilityMojo;
  import br.com.ingenieux.mojo.beanstalk.env.CreateEnvironmentMojo;
  import br.com.ingenieux.mojo.beanstalk.env.TerminateEnvironmentMojo;
  import br.com.ingenieux.mojo.beanstalk.env.UpdateEnvironmentMojo;
  import br.com.ingenieux.mojo.beanstalk.env.WaitForEnvironmentMojo;
  import br.com.ingenieux.mojo.beanstalk.version.CreateApplicationVersionMojo;
  
  @Ignore
  public abstract class BeanstalkTestBase extends AbstractMojoTestCase {
  
    public static final String PROP_VERSION_LABEL = "versionLabel";
  
    public static final String PROP_S3_KEY_MASK = "s3KeyMask";
  
    public static final String PROP_S3_BUCKET = "s3Bucket";
  
    Properties properties;
  
    StrSubstitutor strSub;
  
    CheckAvailabilityMojo checkAvailabilityMojo;
  
    CreateApplicationMojo createAppMojo;
  
    CreateApplicationVersionMojo createAppVersionMojo;
  
    UploadSourceBundleMojo uploadSourceBundleMojo;
  
    CreateEnvironmentMojo createEnvMojo;
  
    TerminateEnvironmentMojo termEnvMojo;
  
    WaitForEnvironmentMojo waitForEnvMojo;
  
    UpdateEnvironmentMojo updateEnvMojo;
  
    DescribeConfigurationTemplatesMojo describeConfigTemplatesMojo;
  
    CreateConfigurationTemplateMojo createConfigurationTemplateMojo;
  
    String versionLabel;
  
    AWSCredentials credentials;
  
    AWSElasticBeanstalk service;
  
    public BeanstalkTestBase() {
      super();
    }
  
    @Override
    protected void setUp() throws Exception {
      super.setUp();
  
      Properties properties = new Properties();
  
      properties.load(new FileInputStream("test.properties"));
  
      this.properties = properties;
 
     configureMojos();
 
     this.credentials = uploadSourceBundleMojo.getAWSCredentials().getCredentials();
 
     this.service = new AWSElasticBeanstalkClient(credentials);
   }
 
   protected void configureMojos() throws Exception {
     createAppMojo = getMojo(CreateApplicationMojo.class);
 
     createAppVersionMojo = getMojo(CreateApplicationVersionMojo.class);
 
     uploadSourceBundleMojo = getMojo(UploadSourceBundleMojo.class);
 
     createAppMojo = getMojo(CreateApplicationMojo.class);
 
     createAppVersionMojo = getMojo(CreateApplicationVersionMojo.class);
 
     createEnvMojo = getMojo(CreateEnvironmentMojo.class);
 
     waitForEnvMojo = getMojo(WaitForEnvironmentMojo.class);
 
     termEnvMojo = getMojo(TerminateEnvironmentMojo.class);
 
     updateEnvMojo = getMojo(UpdateEnvironmentMojo.class);
 
     describeConfigTemplatesMojo = getMojo(DescribeConfigurationTemplatesMojo.class);
 
     versionLabel = String.format("test-%08X", System.currentTimeMillis());
 
     checkAvailabilityMojo = getMojo(CheckAvailabilityMojo.class);
 
     createConfigurationTemplateMojo = getMojo(CreateConfigurationTemplateMojo.class);
   }
 
   protected File getBasePom(String pomName) {
     return new File(getBasedir(), "target/test-classes/br/com/ingenieux/mojo/beanstalk/" + pomName);
   }
 
   @SuppressWarnings("unchecked")
   protected <T extends AbstractAWSMojo<?>> T getMojo(Class<T> mojoClazz) throws Exception {
     File testPom = this.getBasePom("pom.xml");
 
     PlexusConfiguration pluginConfiguration = extractPluginConfiguration("beanstalk-maven-plugin", testPom);
 
     return (T) configureMojo(mojoClazz.newInstance(), pluginConfiguration);
   }
 
   protected File getWarFile() throws URISyntaxException {
     return new File(BeanstalkTestBase.class.getResource("test-war.war").toURI());
   }
 
   protected String getS3Path() {
     properties.put(PROP_VERSION_LABEL, this.versionLabel);
 
     strSub = new StrSubstitutor(properties);
     return strSub.replace(properties.get(PROP_S3_KEY_MASK));
   }
 
   protected String getS3Bucket() {
     return properties.getProperty(PROP_S3_BUCKET);
   }
 
   public void clearEnvironments() {
     DescribeEnvironmentsResult environments = service.describeEnvironments();
 
     for (EnvironmentDescription d : environments.getEnvironments()) {
       service.terminateEnvironment(new TerminateEnvironmentRequest().withEnvironmentId(d.getEnvironmentId()).withTerminateResources(true));
     }
   }
 }