/**
    * Copyright (C) 2011 White Source Ltd.
    *
    * 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.whitesource.maven;
  
  import org.apache.commons.lang.StringUtils;
  import org.apache.maven.plugin.MojoExecutionException;
  import org.apache.maven.plugin.MojoFailureException;
  import org.apache.maven.plugins.annotations.LifecyclePhase;
  import org.apache.maven.plugins.annotations.Mojo;
  import org.apache.maven.plugins.annotations.Parameter;
  import org.apache.maven.plugins.annotations.ResolutionScope;
  import org.apache.maven.project.MavenProject;
  import org.whitesource.agent.api.dispatch.CheckPoliciesResult;
  import org.whitesource.agent.api.dispatch.UpdateInventoryResult;
  import org.whitesource.agent.api.model.AgentProjectInfo;
  import org.whitesource.agent.client.WssServiceException;
  
  import java.util.Collection;
  import java.util.Map;
  
  /**
   * Send updates of open source software usage information to White Source.
   *
   * <p>
   *     Further documentation for the plugin and its usage can be found in the
   *     <a href="http://docs.whitesourcesoftware.com/display/serviceDocs/Maven+plugin">online documentation</a>.
   * </p>
   *
   * @author Edo.Shor
   */
  @Mojo(name = "update",
          defaultPhase = LifecyclePhase.PACKAGE,
          requiresDependencyResolution = ResolutionScope.TEST,
          aggregator = true )
  public class UpdateMojo extends AgentMojo {
  
      /**
       * Optional. Set to true to check policies.
       */
      @Parameter( alias = "checkPolicies", property = Constants.CHECK_POLICIES, required = false, defaultValue = "false")
      private boolean checkPolicies;
  
      /* --- Constructors --- */
  
      public UpdateMojo() {
      }
  
      /* --- Concrete implementation methods --- */
  
      @Override
      public void doExecute() throws MojoExecutionException, MojoFailureException {
          if (reactorProjects == null) {
              info("No projects found. Skipping update");
              return;
          }
  
          // initialize
          init();
  
          // Collect OSS usage information
          Collection<AgentProjectInfo> projectInfos = extractProjectInfos();
  
          // send to white source
          if (projectInfos == null || projectInfos.isEmpty()) {
              info("No open source information found.");
          } else {
              sendUpdate(projectInfos);
          }
      }
  
      /* --- Private methods --- */
  
      private void init() {
          // copy token for modules with special names into moduleTokens.
          for (Map.Entry<Object, Object> entry : specialModuleTokens.entrySet()) {
              moduleTokens.put(entry.getKey().toString(), entry.getValue().toString());
          }
  
          // take product name and version from top level project
          MavenProject topLevelProject = session.getTopLevelProject();
          if (topLevelProject != null) {
              if (StringUtils.isBlank(product)) {
                  product = topLevelProject.getName();
              }
              if (StringUtils.isBlank(product)) {
                  product = topLevelProject.getArtifactId();
             }
         }
     }
 
     private void sendUpdate(Collection<AgentProjectInfo> projectInfos) throws MojoFailureException, MojoExecutionException {
         try {
             UpdateInventoryResult updateResult;
             if (checkPolicies) {
                 info("Checking policies...");
                 CheckPoliciesResult result = service.checkPolicies(orgToken, product, productVersion, projectInfos);
 
                 if (outputDirectory == null ||
                         (!outputDirectory.exists() && !outputDirectory.mkdirs())) {
                     warn("Output directory doesn't exist. Skipping policies check report.");
                 } else {
                     generateReport(result);
                 }
 
                 if (result.hasRejections()) {
                     String msg = "Some dependencies were rejected by the organization's policies.";
                     throw new MojoExecutionException(msg); // this is handled in base class
                 } else {
                     info("All dependencies conform with the organization's policies.");
                     info("Sending updates to White Source");
                     updateResult = service.update(orgToken, product, productVersion, projectInfos);
                 }
             } else {
                 info("Sending updates to White Source");
                 updateResult = service.update(orgToken, product, productVersion, projectInfos);
             }
             logResult(updateResult);
         } catch (WssServiceException e) {
             throw new MojoExecutionException(Constants.ERROR_SERVICE_CONNECTION + e.getMessage(), e);
         }
     }
 
     private void logResult(UpdateInventoryResult result) {
         info("Inventory update results for " + result.getOrganization());
 
         // newly created projects
         Collection<String> createdProjects = result.getCreatedProjects();
         if (createdProjects.isEmpty()) {
             info("No new projects found.");
         } else {
             info("Newly created projects:");
             for (String projectName : createdProjects) {
                 info(projectName);
             }
         }
 
         // updated projects
         Collection<String> updatedProjects = result.getUpdatedProjects();
         if (updatedProjects.isEmpty()) {
             info("No projects were updated.");
         } else {
             info("Updated projects:");
             for (String projectName : updatedProjects) {
                 info(projectName);
             }
         }
     }
 
 }