/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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.apache.maven.internal.impl;
  
  import javax.inject.Inject;
  import javax.inject.Named;
  
  import java.nio.file.Path;
  import java.util.Map;
  import java.util.Objects;
  import java.util.Optional;
  import java.util.concurrent.ConcurrentHashMap;
  import java.util.stream.Stream;
  
  import org.apache.maven.api.Artifact;
  import org.apache.maven.api.ProducedArtifact;
  import org.apache.maven.api.annotations.Nonnull;
  import org.apache.maven.api.di.SessionScoped;
  import org.apache.maven.api.services.ArtifactManager;
  import org.apache.maven.project.MavenProject;
  import org.eclipse.sisu.Typed;
  
  import static org.apache.maven.internal.impl.Utils.nonNull;
  
  @Named
  @Typed
  @SessionScoped
  public class DefaultArtifactManager implements ArtifactManager {
  
      @Nonnull
      private final InternalMavenSession session;
  
      private final Map<String, Path> paths = new ConcurrentHashMap<>();
  
      @Inject
      public DefaultArtifactManager(@Nonnull InternalMavenSession session) {
          this.session = session;
      }
  
      @Nonnull
      @Override
      public Optional<Path> getPath(@Nonnull Artifact artifact) {
          String id = id(nonNull(artifact, "artifact"));
          if (session.getMavenSession().getAllProjects() != null) {
              for (MavenProject project : session.getMavenSession().getAllProjects()) {
                  if (id.equals(id(project.getArtifact()))
                          && project.getArtifact().getFile() != null) {
                      return Optional.of(project.getArtifact().getFile().toPath());
                  }
              }
          }
          Path path = paths.get(id);
          if (path == null && artifact instanceof DefaultArtifact) {
              path = ((DefaultArtifact) artifact).getArtifact().getPath();
          }
          return Optional.ofNullable(path);
      }
  
      @Override
      public void setPath(@Nonnull ProducedArtifact artifact, Path path) {
          String id = id(nonNull(artifact, "artifact"));
          if (session.getMavenSession().getAllProjects() != null) {
              session.getMavenSession().getAllProjects().stream()
                      .flatMap(this::getProjectArtifacts)
                      .filter(a -> Objects.equals(id, id(a)))
                      .forEach(a -> a.setFile(path != null ? path.toFile() : null));
          }
          if (path == null) {
              paths.remove(id);
          } else {
              paths.put(id, path);
          }
      }
  
      /**
       * Retrieve a stream of the project's artifacts.
       * Do not include the POM artifact as the file can't be set anyway.
       */
      private Stream<org.apache.maven.artifact.Artifact> getProjectArtifacts(MavenProject project) {
          return Stream.concat(Stream.of(project.getArtifact()), project.getAttachedArtifacts().stream());
      }
  
      private String id(org.apache.maven.artifact.Artifact artifact) {
         return artifact.getGroupId()
                 + ":" + artifact.getArtifactId()
                 + ":" + artifact.getArtifactHandler().getExtension()
                 + (artifact.getClassifier() == null || artifact.getClassifier().isEmpty()
                         ? ""
                         : ":" + artifact.getClassifier())
                 + ":" + artifact.getVersion();
     }
 
     private String id(Artifact artifact) {
         return artifact.key();
     }
 }