/*
    * 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.eclipse.aether.transport.file;
  
  import javax.inject.Named;
  
  import java.io.IOException;
  import java.io.UncheckedIOException;
  import java.net.URI;
  import java.nio.file.FileSystem;
  import java.nio.file.FileSystemNotFoundException;
  import java.nio.file.FileSystems;
  import java.nio.file.Path;
  import java.nio.file.Paths;
  import java.util.HashMap;
  import java.util.Map;
  
  import org.eclipse.aether.RepositorySystemSession;
  import org.eclipse.aether.repository.RemoteRepository;
  import org.eclipse.aether.repository.RepositoryUriUtils;
  import org.eclipse.aether.spi.connector.transport.Transporter;
  import org.eclipse.aether.spi.connector.transport.TransporterFactory;
  import org.eclipse.aether.transfer.NoTransporterException;
  
  import static java.util.Objects.requireNonNull;
  
  /**
   * A transporter factory for repositories using the {@code file:} or {@code bundle:} protocol.
   */
  @Named(FileTransporterFactory.NAME)
  public final class FileTransporterFactory implements TransporterFactory {
      public static final String NAME = "file";
  
      private float priority;
  
      @Override
      public float getPriority() {
          return priority;
      }
  
      /**
       * Sets the priority of this component.
       *
       * @param priority The priority.
       * @return This component for chaining, never {@code null}.
       */
      public FileTransporterFactory setPriority(float priority) {
          this.priority = priority;
          return this;
      }
  
      /**
       * Creates new instance of {@link FileTransporter}.
       *
       * @param session The session.
       * @param repository The remote repository.
       */
      @Override
      public Transporter newInstance(RepositorySystemSession session, RemoteRepository repository)
              throws NoTransporterException {
          requireNonNull(session, "session cannot be null");
          requireNonNull(repository, "repository cannot be null");
  
          String repositoryUrl = repository.getUrl();
          if (repositoryUrl.startsWith("bundle:")) {
              try {
                  repositoryUrl = repositoryUrl.substring("bundle:".length());
                  URI bundlePath = URI.create("jar:"
                          + Paths.get(RepositoryUriUtils.toUri(repositoryUrl))
                                  .toAbsolutePath()
                                  .toUri()
                                  .toASCIIString());
                  Map<String, String> env = new HashMap<>();
                  FileSystem fileSystem = FileSystems.newFileSystem(bundlePath, env);
                  return new FileTransporter(
                          fileSystem,
                          true,
                          false,
                          fileSystem.getPath(fileSystem.getSeparator()),
                          FileTransporter.WriteOp.COPY);
              } catch (IOException e) {
                  throw new UncheckedIOException(e); // hard failure; most probably user error (ie wrong path or perm)
              }
         } else {
             // special case in file: transport: to support custom FS providers (like JIMFS), we cannot
             // cover all possible protocols (to throw NoTransporterEx), hence we rely on FS rejecting the URI
             FileTransporter.WriteOp writeOp = FileTransporter.WriteOp.COPY;
             if (repositoryUrl.startsWith("symlink+")) {
                 writeOp = FileTransporter.WriteOp.SYMLINK;
                 repositoryUrl = repositoryUrl.substring("symlink+".length());
             } else if (repositoryUrl.startsWith("hardlink+")) {
                 writeOp = FileTransporter.WriteOp.HARDLINK;
                 repositoryUrl = repositoryUrl.substring("hardlink+".length());
             }
             try {
                 Path basePath =
                         Paths.get(RepositoryUriUtils.toUri(repositoryUrl)).toAbsolutePath();
                 return new FileTransporter(
                         basePath.getFileSystem(),
                         false,
                         !basePath.getFileSystem().isReadOnly(),
                         basePath,
                         writeOp);
             } catch (FileSystemNotFoundException | IllegalArgumentException e) {
                 throw new NoTransporterException(repository, e);
             }
         }
     }
 }