/*
    * Copyright 2017-2022 original authors
    *
    * 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
    *
    * https://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 io.micronaut.build;
  
  import com.github.dockerjava.api.command.PushImageCmd;
  import com.github.dockerjava.api.model.AuthConfig;
  import com.google.cloud.tools.jib.api.Credential;
  import com.google.cloud.tools.jib.maven.MavenProjectProperties;
  import io.micronaut.build.services.ApplicationConfigurationService;
  import io.micronaut.build.services.DockerService;
  import io.micronaut.build.services.JibConfigurationService;
  import org.apache.maven.plugin.MojoExecutionException;
  import org.apache.maven.plugin.MojoFailureException;
  import org.apache.maven.plugins.annotations.Mojo;
  import org.apache.maven.project.MavenProject;
  import org.testcontainers.utility.DockerImageName;
  import org.testcontainers.utility.RegistryAuthLocator;
  
  import javax.inject.Inject;
  import java.util.Set;
  
  /**
   * <p>Implementation of the <code>deploy</code> lifecycle for pushing Docker images</p>
   * <p><strong>WARNING</strong>: this goal is not intended to be executed directly. Instead, Execute the <code>deploy</code>
   * phase specifying the packaging type, eg:</p>
   *
   * <pre>mvn deploy -Dpackaging=docker-native</pre>
   *
   * @author Álvaro Sánchez-Mariscal
   * @since 1.1
   */
  @Mojo(name = "docker-push")
  public class DockerPushMojo extends AbstractDockerMojo {
  
      @Inject
      public DockerPushMojo(MavenProject mavenProject, JibConfigurationService jibConfigurationService,
                               ApplicationConfigurationService applicationConfigurationService, DockerService dockerService) {
          super(mavenProject, jibConfigurationService, applicationConfigurationService, dockerService);
      }
  
      @Override
      public void execute() throws MojoExecutionException, MojoFailureException {
          Packaging packaging = Packaging.of(mavenProject.getPackaging());
          if (packaging == Packaging.DOCKER || packaging == Packaging.DOCKER_NATIVE || packaging == Packaging.DOCKER_CRAC) {
              Set<String> images = getTags();
  
              // getTags() will automatically generate an image name if none is specified
              // To maintain error compatibility, check that an image name has been
              // manually specified.
              if (jibConfigurationService.getToImage().isPresent()) {
                  for (String taggedImage : images) {
                      getLog().info("Pushing image: " + taggedImage);
                      try (PushImageCmd pushImageCmd = dockerService.pushImageCmd(taggedImage)) {
                          AuthConfig defaultAuthConfig = new AuthConfig();
                          if (jibConfigurationService.getCredentials().isPresent()) {
                              Credential credential = jibConfigurationService.getCredentials().get();
                              defaultAuthConfig
                                      .withUsername(credential.getUsername())
                                      .withPassword(credential.getPassword());
                          }
                          RegistryAuthLocator registryAuthLocator = RegistryAuthLocator.instance();
                          DockerImageName dockerImageName = DockerImageName.parse(taggedImage);
                          AuthConfig authConfig = registryAuthLocator.lookupAuthConfig(dockerImageName, defaultAuthConfig);
  
                          pushImageCmd
                                  .withAuthConfig(authConfig)
                                  .start()
                                  .awaitCompletion();
                      } catch (InterruptedException e) {
                          Thread.currentThread().interrupt();
                      } catch (Exception e) {
                          throw new MojoExecutionException(e.getMessage(), e);
                      }
                  }
              } else {
                  throw new MojoFailureException("The plugin " + MavenProjectProperties.PLUGIN_KEY + " is misconfigured. Missing <to> tag");
              }
          } else {
              throw new MojoFailureException("The <packaging> must be set to either [" + Packaging.DOCKER.id() + "] or [" + Packaging.DOCKER_NATIVE.id() + "]");
          }
      }
  }