All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.arangodb.shaded.vertx.core.file.impl.FileSystemImpl Maven / Gradle / Ivy

There is a newer version: 7.8.0
Show newest version
/*
 * Copyright (c) 2011-2019 Contributors to the Eclipse Foundation
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
 * which is available at https://www.apache.org/licenses/LICENSE-2.0.
 *
 * SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
 */

package com.arangodb.shaded.vertx.core.file.impl;

import com.arangodb.shaded.vertx.codegen.annotations.Nullable;
import com.arangodb.shaded.vertx.core.AsyncResult;
import com.arangodb.shaded.vertx.core.Future;
import com.arangodb.shaded.vertx.core.Handler;
import com.arangodb.shaded.vertx.core.Promise;
import com.arangodb.shaded.vertx.core.buffer.Buffer;
import com.arangodb.shaded.vertx.core.file.AsyncFile;
import com.arangodb.shaded.vertx.core.file.CopyOptions;
import com.arangodb.shaded.vertx.core.file.FileProps;
import com.arangodb.shaded.vertx.core.file.FileSystem;
import com.arangodb.shaded.vertx.core.file.FileSystemException;
import com.arangodb.shaded.vertx.core.file.FileSystemProps;
import com.arangodb.shaded.vertx.core.file.OpenOptions;
import com.arangodb.shaded.vertx.core.impl.ContextInternal;
import com.arangodb.shaded.vertx.core.impl.VertxInternal;

import java.io.File;
import java.io.FilenameFilter;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.nio.file.CopyOption;
import java.nio.file.FileAlreadyExistsException;
import java.nio.file.FileStore;
import java.nio.file.FileVisitOption;
import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.LinkOption;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.StandardCopyOption;
import java.nio.file.attribute.BasicFileAttributes;
import java.nio.file.attribute.FileAttribute;
import java.nio.file.attribute.GroupPrincipal;
import java.nio.file.attribute.PosixFileAttributeView;
import java.nio.file.attribute.PosixFilePermission;
import java.nio.file.attribute.PosixFilePermissions;
import java.nio.file.attribute.UserPrincipal;
import java.nio.file.attribute.UserPrincipalLookupService;
import java.util.ArrayList;
import java.util.EnumSet;
import java.util.HashSet;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.regex.Pattern;

/**
 *
 * This class is thread-safe
 *
 * @author Tim Fox
 */
public class FileSystemImpl implements FileSystem {

  private static final CopyOptions DEFAULT_OPTIONS = new CopyOptions();

  protected final VertxInternal vertx;

  public FileSystemImpl(VertxInternal vertx) {
    this.vertx = vertx;
  }

  public FileSystem copy(String from, String to, Handler> handler) {
    return copy(from, to, DEFAULT_OPTIONS, handler);
  }

  @Override
  public Future copy(String from, String to) {
    return copy(from, to, DEFAULT_OPTIONS);
  }

  @Override
  public FileSystem copy(String from, String to, CopyOptions options, Handler> handler) {
    copy(from, to, options).onComplete(handler);
    return this;
  }

  @Override
  public Future copy(String from, String to, CopyOptions options) {
    return copyInternal(from, to, options).run();
  }

  public FileSystem copyBlocking(String from, String to) {
    copyInternal(from, to, DEFAULT_OPTIONS).perform();
    return this;
  }

  public FileSystem copyRecursive(String from, String to, boolean recursive, Handler> handler) {
    copyRecursive(from, to, recursive).onComplete(handler);
    return this;
  }

  @Override
  public Future copyRecursive(String from, String to, boolean recursive) {
    return copyRecursiveInternal(from, to, recursive).run();
  }

  public FileSystem copyRecursiveBlocking(String from, String to, boolean recursive) {
    copyRecursiveInternal(from, to, recursive).perform();
    return this;
  }

  public FileSystem move(String from, String to, Handler> handler) {
    return move(from, to, DEFAULT_OPTIONS, handler);
  }

  @Override
  public Future move(String from, String to) {
    return move(from, to, DEFAULT_OPTIONS);
  }

  @Override
  public FileSystem move(String from, String to, CopyOptions options, Handler> handler) {
    move(from, to, options).onComplete(handler);
    return this;
  }

  @Override
  public Future move(String from, String to, CopyOptions options) {
    return moveInternal(from, to, options).run();
  }

  public FileSystem moveBlocking(String from, String to) {
    moveInternal(from, to, DEFAULT_OPTIONS).perform();
    return this;
  }

  public FileSystem truncate(String path, long len, Handler> handler) {
    truncate(path, len).onComplete(handler);
    return this;
  }

  @Override
  public Future truncate(String path, long len) {
    return truncateInternal(path, len).run();
  }

  public FileSystem truncateBlocking(String path, long len) {
    truncateInternal(path, len).perform();
    return this;
  }

  public FileSystem chmod(String path, String perms, Handler> handler) {
    chmod(path, perms).onComplete(handler);
    return this;
  }

  @Override
  public Future chmod(String path, String perms) {
    return chmodInternal(path, perms).run();
  }

  public FileSystem chmodBlocking(String path, String perms) {
    chmodInternal(path, perms).perform();
    return this;
  }

  public FileSystem chmodRecursive(String path, String perms, String dirPerms, Handler> handler) {
    chmodRecursive(path, perms, dirPerms).onComplete(handler);
    return this;
  }

  @Override
  public Future chmodRecursive(String path, String perms, String dirPerms) {
    return chmodInternal(path, perms, dirPerms).run();
  }

  public FileSystem chmodRecursiveBlocking(String path, String perms, String dirPerms) {
    chmodInternal(path, perms, dirPerms).perform();
    return this;
  }

  public FileSystem chown(String path, String user, String group, Handler> handler) {
    chown(path, user, group).onComplete(handler);
    return this;
  }

  @Override
  public Future chown(String path, @Nullable String user, @Nullable String group) {
    return chownInternal(path, user, group).run();
  }

  public FileSystem chownBlocking(String path, String user, String group) {
    chownInternal(path, user, group).perform();
    return this;
  }

  public FileSystem props(String path, Handler> handler) {
    props(path).onComplete(handler);
    return this;
  }

  @Override
  public Future props(String path) {
    return propsInternal(path).run();
  }

  public FileProps propsBlocking(String path) {
    return propsInternal(path).perform();
  }

  public FileSystem lprops(String path, Handler> handler) {
    lprops(path).onComplete(handler);
    return this;
  }

  @Override
  public Future lprops(String path) {
    return lpropsInternal(path).run();
  }

  public FileProps lpropsBlocking(String path) {
    return lpropsInternal(path).perform();
  }

  public FileSystem link(String link, String existing, Handler> handler) {
    link(link, existing).onComplete(handler);
    return this;
  }

  @Override
  public Future link(String link, String existing) {
    return linkInternal(link, existing).run();
  }

  public FileSystem linkBlocking(String link, String existing) {
    linkInternal(link, existing).perform();
    return this;
  }

  public FileSystem symlink(String link, String existing, Handler> handler) {
    symlink(link, existing).onComplete(handler);
    return this;
  }

  @Override
  public Future symlink(String link, String existing) {
    return symlinkInternal(link, existing).run();
  }

  public FileSystem symlinkBlocking(String link, String existing) {
    symlinkInternal(link, existing).perform();
    return this;
  }

  public FileSystem unlink(String link, Handler> handler) {
    unlink(link).onComplete(handler);
    return this;
  }

  @Override
  public Future unlink(String link) {
    return unlinkInternal(link).run();
  }

  public FileSystem unlinkBlocking(String link) {
    unlinkInternal(link).perform();
    return this;
  }

  public FileSystem readSymlink(String link, Handler> handler) {
    readSymlink(link).onComplete(handler);
    return this;
  }

  @Override
  public Future readSymlink(String link) {
    return readSymlinkInternal(link).run();
  }

  public String readSymlinkBlocking(String link) {
    return readSymlinkInternal(link).perform();
  }

  public FileSystem delete(String path, Handler> handler) {
    delete(path).onComplete(handler);
    return this;
  }

  @Override
  public Future delete(String path) {
    return deleteInternal(path).run();
  }

  public FileSystem deleteBlocking(String path) {
    deleteInternal(path).perform();
    return this;
  }

  public FileSystem deleteRecursive(String path, boolean recursive, Handler> handler) {
    deleteRecursive(path, recursive).onComplete(handler);
    return this;
  }

  @Override
  public Future deleteRecursive(String path, boolean recursive) {
    return deleteInternal(path, recursive).run();
  }

  public FileSystem deleteRecursiveBlocking(String path, boolean recursive) {
    deleteInternal(path, recursive).perform();
    return this;
  }

  public FileSystem mkdir(String path, Handler> handler) {
    mkdir(path).onComplete(handler);
    return this;
  }

  @Override
  public Future mkdir(String path) {
    return mkdirInternal(path).run();
  }

  public FileSystem mkdirBlocking(String path) {
    mkdirInternal(path).perform();
    return this;
  }

  public FileSystem mkdirs(String path, Handler> handler) {
    mkdirs(path).onComplete(handler);
    return this;
  }

  @Override
  public Future mkdirs(String path) {
    return mkdirInternal(path, true).run();
  }

  public FileSystem mkdirsBlocking(String path) {
    mkdirInternal(path, true).perform();
    return this;
  }

  public FileSystem mkdir(String path, String perms, Handler> handler) {
    mkdir(path, perms).onComplete(handler);
    return this;
  }

  @Override
  public Future mkdir(String path, String perms) {
    return mkdirInternal(path, perms).run();
  }

  public FileSystem mkdirBlocking(String path, String perms) {
    mkdirInternal(path, perms).perform();
    return this;
  }

  public FileSystem mkdirs(String path, String perms, Handler> handler) {
    mkdirs(path, perms).onComplete(handler);
    return this;
  }

  @Override
  public Future mkdirs(String path, String perms) {
    return mkdirInternal(path, perms, true).run();
  }

  public FileSystem mkdirsBlocking(String path, String perms) {
    mkdirInternal(path, perms, true).perform();
    return this;
  }

  public FileSystem readDir(String path, Handler>> handler) {
    readDir(path).onComplete(handler);
    return this;
  }

  @Override
  public Future> readDir(String path) {
    return readDirInternal(path).run();
  }

  public List readDirBlocking(String path) {
    return readDirInternal(path).perform();
  }

  public FileSystem readDir(String path, String filter, Handler>> handler) {
    readDir(path, filter).onComplete(handler);
    return this;
  }

  @Override
  public Future> readDir(String path, String filter) {
    return readDirInternal(path, filter).run();
  }

  public List readDirBlocking(String path, String filter) {
    return readDirInternal(path, filter).perform();
  }

  public FileSystem readFile(String path, Handler> handler) {
    readFile(path).onComplete(handler);
    return this;
  }

  @Override
  public Future readFile(String path) {
    return readFileInternal(path).run();
  }

  public Buffer readFileBlocking(String path) {
    return readFileInternal(path).perform();
  }

  public FileSystem writeFile(String path, Buffer data, Handler> handler) {
    writeFile(path, data).onComplete(handler);
    return this;
  }

  @Override
  public Future writeFile(String path, Buffer data) {
    return writeFileInternal(path, data).run();
  }

  public FileSystem writeFileBlocking(String path, Buffer data) {
    writeFileInternal(path, data).perform();
    return this;
  }

  public FileSystem open(String path, OpenOptions options, Handler> handler) {
    open(path, options).onComplete(handler);
    return this;
  }

  @Override
  public Future open(String path, OpenOptions options) {
    return openInternal(path, options).run();
  }

  public AsyncFile openBlocking(String path, OpenOptions options) {
    return openInternal(path, options).perform();
  }

  public FileSystem createFile(String path, Handler> handler) {
    createFile(path).onComplete(handler);
    return this;
  }

  @Override
  public Future createFile(String path) {
    return createFileInternal(path).run();
  }

  public FileSystem createFileBlocking(String path) {
    createFileInternal(path).perform();
    return this;
  }

  public FileSystem createFile(String path, String perms, Handler> handler) {
    createFile(path, perms).onComplete(handler);
    return this;
  }

  @Override
  public Future createFile(String path, String perms) {
    return createFileInternal(path, perms).run();
  }

  public FileSystem createFileBlocking(String path, String perms) {
    createFileInternal(path, perms).perform();
    return this;
  }

  public FileSystem exists(String path, Handler> handler) {
    exists(path).onComplete(handler);
    return this;
  }

  @Override
  public Future exists(String path) {
    return existsInternal(path).run();
  }

  public boolean existsBlocking(String path) {
    return existsInternal(path).perform();
  }

  public FileSystem fsProps(String path, Handler> handler) {
    fsProps(path).onComplete(handler);
    return this;
  }

  @Override
  public Future fsProps(String path) {
    return fsPropsInternal(path).run();
  }

  public FileSystemProps fsPropsBlocking(String path) {
    return fsPropsInternal(path).perform();
  }

  @Override
  public FileSystem createTempDirectory(String prefix, Handler> handler) {
    createTempDirectory(prefix).onComplete(handler);
    return this;
  }

  @Override
  public Future createTempDirectory(String prefix) {
    return createTempDirectoryInternal(null, prefix, null).run();
  }

  @Override
  public String createTempDirectoryBlocking(String prefix) {
    return createTempDirectoryInternal(null, prefix, null).perform();
  }

  @Override
  public FileSystem createTempDirectory(String prefix, String perms, Handler> handler) {
    createTempDirectory(prefix, perms).onComplete(handler);
    return this;
  }

  @Override
  public Future createTempDirectory(String prefix, String perms) {
    return createTempDirectoryInternal(null, prefix, perms).run();
  }

  @Override
  public String createTempDirectoryBlocking(String prefix, String perms) {
    return createTempDirectoryInternal(null, prefix, perms).perform();
  }

  @Override
  public FileSystem createTempDirectory(String dir, String prefix, String perms, Handler> handler) {
    createTempDirectory(dir, prefix, perms).onComplete(handler);
    return this;
  }

  @Override
  public Future createTempDirectory(String dir, String prefix, String perms) {
    return createTempDirectoryInternal(dir, prefix, perms).run();
  }

  @Override
  public String createTempDirectoryBlocking(String dir, String prefix, String perms) {
    return createTempDirectoryInternal(dir, prefix, perms).perform();
  }

  @Override
  public FileSystem createTempFile(String prefix, String suffix, Handler> handler) {
    createTempFile(prefix, suffix).onComplete(handler);
    return this;
  }

  @Override
  public Future createTempFile(String prefix, String suffix) {
    return createTempFileInternal(null, prefix, suffix, null).run();
  }

  @Override
  public String createTempFileBlocking(String prefix, String suffix) {
    return createTempFileInternal(null, prefix, suffix, null).perform();
  }

  @Override
  public FileSystem createTempFile(String prefix, String suffix, String perms, Handler> handler) {
    createTempFile(prefix, suffix, perms).onComplete(handler);
    return this;
  }

  @Override
  public Future createTempFile(String prefix, String suffix, String perms) {
    return createTempFileInternal(null, prefix, suffix, perms).run();
  }

  @Override
  public String createTempFileBlocking(String prefix, String suffix, String perms) {
    return createTempFileInternal(null, prefix, suffix, perms).perform();
  }


  @Override
  public FileSystem createTempFile(String dir, String prefix, String suffix, String perms, Handler> handler) {
    createTempFile(dir, prefix, suffix, perms).onComplete(handler);
    return this;
  }

  @Override
  public Future createTempFile(String dir, String prefix, String suffix, String perms) {
    return createTempFileInternal(dir, prefix, suffix, perms).run();
  }

  @Override
  public String createTempFileBlocking(String dir, String prefix, String suffix, String perms) {
    return createTempFileInternal(dir, prefix, suffix, perms).perform();
  }

  static String getFileAccessErrorMessage(String action, String path) {
    return "Unable to " + action + " file at path '" + path + "'";
  }

  static String getFolderAccessErrorMessage(String action, String path) {
    return "Unable to " + action + " folder at path '" + path + "'";
  }

  static String getFileCopyErrorMessage(String from, String to) {
    return getFileDualOperationErrorMessage("copy", from, to);
  }

  static String getFileMoveErrorMessage(String from, String to) {
    return getFileDualOperationErrorMessage("move", from, to);
  }

  static String getFileDualOperationErrorMessage(String action, String from, String to) {
    return "Unable to " + action + " file from '" + from + "' to '" + to + "'";
  }

  private BlockingAction copyInternal(String from, String to, CopyOptions options) {
    Objects.requireNonNull(from);
    Objects.requireNonNull(to);
    Objects.requireNonNull(options);
    Set copyOptionSet = toCopyOptionSet(options);
    CopyOption[] copyOptions = copyOptionSet.toArray(new CopyOption[0]);
    return new BlockingAction() {
      public Void perform() {
        try {
          Path source = vertx.resolveFile(from).toPath();
          Path target = vertx.resolveFile(to).toPath();
          Files.copy(source, target, copyOptions);
        } catch (IOException e) {
          throw new FileSystemException(getFileCopyErrorMessage(from, to), e);
        }
        return null;
      }
    };
  }

  private BlockingAction copyRecursiveInternal(String from, String to, boolean recursive) {
    Objects.requireNonNull(from);
    Objects.requireNonNull(to);
    return new BlockingAction() {
      public Void perform() {
        try {
          Path source = vertx.resolveFile(from).toPath();
          Path target = vertx.resolveFile(to).toPath();
          if (recursive) {
            Files.walkFileTree(source, EnumSet.of(FileVisitOption.FOLLOW_LINKS), Integer.MAX_VALUE,
              new SimpleFileVisitor() {
                public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs)
                  throws IOException {
                  Path targetDir = target.resolve(source.relativize(dir));
                  try {
                    Files.copy(dir, targetDir);
                  } catch (FileAlreadyExistsException e) {
                    if (!Files.isDirectory(targetDir)) {
                      throw e;
                    }
                  }
                  return FileVisitResult.CONTINUE;
                }

                @Override
                public FileVisitResult visitFile(Path file, BasicFileAttributes attrs)
                  throws IOException {
                  Files.copy(file, target.resolve(source.relativize(file)));
                  return FileVisitResult.CONTINUE;
                }
              });
          } else {
            Files.copy(source, target);
          }
        } catch (IOException e) {
          throw new FileSystemException(getFileCopyErrorMessage(from, to), e);
        }
        return null;
      }
    };
  }

  private BlockingAction moveInternal(String from, String to, CopyOptions options) {
    Objects.requireNonNull(from);
    Objects.requireNonNull(to);
    Objects.requireNonNull(options);
    Set copyOptionSet = toCopyOptionSet(options);
    CopyOption[] copyOptions = copyOptionSet.toArray(new CopyOption[0]);
    return new BlockingAction() {
      public Void perform() {
        try {
          Path source = vertx.resolveFile(from).toPath();
          Path target = vertx.resolveFile(to).toPath();
          Files.move(source, target, copyOptions);
        } catch (IOException e) {
          throw new FileSystemException(getFileMoveErrorMessage(from, to), e);
        }
        return null;
      }
    };
  }

  private BlockingAction truncateInternal(String p, long len) {
    Objects.requireNonNull(p);
    return new BlockingAction() {
      public Void perform() {
        try {
          String path = vertx.resolveFile(p).getAbsolutePath();
          if (len < 0) {
            throw new FileSystemException("Cannot truncate file to size < 0");
          }
          if (!Files.exists(Paths.get(path))) {
            throw new FileSystemException("Cannot truncate file " + path + ". Does not exist");
          }
          try (RandomAccessFile raf = new RandomAccessFile(path, "rw")) {
            raf.setLength(len);
          }
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("truncate", p) ,e);
        }
        return null;
      }
    };
  }

  private BlockingAction chmodInternal(String path, String perms) {
    return chmodInternal(path, perms, null);
  }

  protected BlockingAction chmodInternal(String path, String perms, String dirPerms) {
    Objects.requireNonNull(path);
    Set permissions = PosixFilePermissions.fromString(perms);
    Set dirPermissions = dirPerms == null ? null : PosixFilePermissions.fromString(dirPerms);
    return new BlockingAction() {
      public Void perform() {
        try {
          Path target = vertx.resolveFile(path).toPath();
          if (dirPermissions != null) {
            Files.walkFileTree(target, new SimpleFileVisitor() {
              public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throws IOException {
                //The directory entries typically have different permissions to the files, e.g. execute permission
                //or can't cd into it
                Files.setPosixFilePermissions(dir, dirPermissions);
                return FileVisitResult.CONTINUE;
              }

              public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
                Files.setPosixFilePermissions(file, permissions);
                return FileVisitResult.CONTINUE;
              }
            });
          } else {
            Files.setPosixFilePermissions(target, permissions);
          }
        } catch (SecurityException e) {
          throw new FileSystemException("Accessed denied for chmod on " + path);
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("chmod", path), e);
        }
        return null;
      }
    };
  }

  protected BlockingAction chownInternal(String path, String user, String group) {
    Objects.requireNonNull(path);
    return new BlockingAction() {
      public Void perform() {
        try {
          Path target = vertx.resolveFile(path).toPath();
          UserPrincipalLookupService service = target.getFileSystem().getUserPrincipalLookupService();
          UserPrincipal userPrincipal = user == null ? null : service.lookupPrincipalByName(user);
          GroupPrincipal groupPrincipal = group == null ? null : service.lookupPrincipalByGroupName(group);
          if (groupPrincipal != null) {
            PosixFileAttributeView view = Files.getFileAttributeView(target, PosixFileAttributeView.class, LinkOption.NOFOLLOW_LINKS);
            if (view == null) {
              throw new FileSystemException("Change group of file not supported");
            }
            view.setGroup(groupPrincipal);

          }
          if (userPrincipal != null) {
            Files.setOwner(target, userPrincipal);
          }
        } catch (SecurityException e) {
          throw new FileSystemException("Accessed denied for chown on " + path);
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("crown", path), e);
        }
        return null;
      }
    };
  }

  private BlockingAction propsInternal(String path) {
    return props(path, true);
  }

  private BlockingAction lpropsInternal(String path) {
    return props(path, false);
  }

  private BlockingAction props(String path, boolean followLinks) {
    Objects.requireNonNull(path);
    return new BlockingAction() {
      public FileProps perform() {
        try {
          Path target = vertx.resolveFile(path).toPath();
          BasicFileAttributes attrs;
          if (followLinks) {
            attrs = Files.readAttributes(target, BasicFileAttributes.class);
          } else {
            attrs = Files.readAttributes(target, BasicFileAttributes.class, LinkOption.NOFOLLOW_LINKS);
          }
          return new FilePropsImpl(attrs);
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("analyse", path), e);
        }
      }
    };
  }

  private BlockingAction linkInternal(String link, String existing) {
    return link(link, existing, false);
  }

  private BlockingAction symlinkInternal(String link, String existing) {
    return link(link, existing, true);
  }

  private BlockingAction link(String link, String existing, boolean symbolic) {
    Objects.requireNonNull(link);
    Objects.requireNonNull(existing);
    return new BlockingAction() {
      public Void perform() {
        try {
          Path source = vertx.resolveFile(link).toPath();
          Path target = vertx.resolveFile(existing).toPath();
          if (symbolic) {
            Files.createSymbolicLink(source, target);
          } else {
            Files.createLink(source, target);
          }
        } catch (IOException e) {
          final String message = "Unable to link existing file '" + existing
            + "' to '" + link
            + "'";
          throw new FileSystemException(message, e);
        }
        return null;
      }
    };
  }

  private BlockingAction unlinkInternal(String link) {
    return deleteInternal(link);
  }

  private BlockingAction readSymlinkInternal(String link) {
    Objects.requireNonNull(link);
    return new BlockingAction() {
      public String perform() {
        try {
          Path source = vertx.resolveFile(link).toPath();
          return Files.readSymbolicLink(source).toString();
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("read", link), e);
        }
      }
    };
  }

  private BlockingAction deleteInternal(String path) {
    return deleteInternal(path, false);
  }

  private BlockingAction deleteInternal(String path, boolean recursive) {
    Objects.requireNonNull(path);
    return new BlockingAction() {
      public Void perform() {
        try {
          Path source = vertx.resolveFile(path).toPath();
          delete(source, recursive);
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("delete", path), e);
        }
        return null;
      }
    };
  }

  public static void delete(Path path, boolean recursive) throws IOException {
    if (recursive) {
      Files.walkFileTree(path, new SimpleFileVisitor() {
        public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
          Files.delete(file);
          return FileVisitResult.CONTINUE;
        }
        public FileVisitResult postVisitDirectory(Path dir, IOException e) throws IOException {
          if (e == null) {
            Files.delete(dir);
            return FileVisitResult.CONTINUE;
          } else {
            throw e;
          }
        }
      });
    } else {
      Files.delete(path);
    }
  }

  private BlockingAction mkdirInternal(String path) {
    return mkdirInternal(path, null, false);
  }

  private BlockingAction mkdirInternal(String path, boolean createParents) {
    return mkdirInternal(path, null, createParents);
  }

  private BlockingAction mkdirInternal(String path, String perms) {
    return mkdirInternal(path, perms, false);
  }

  protected BlockingAction mkdirInternal(String path, String perms, boolean createParents) {
    Objects.requireNonNull(path);
    FileAttribute attrs = perms == null ? null : PosixFilePermissions.asFileAttribute(PosixFilePermissions.fromString(perms));
    return new BlockingAction() {
      public Void perform() {
        try {
          Path source = vertx.resolveFile(path).toPath();
          if (createParents) {
            if (attrs != null) {
              Files.createDirectories(source, attrs);
            } else {
              Files.createDirectories(source);
            }
          } else {
            if (attrs != null) {
              Files.createDirectory(source, attrs);
            } else {
              Files.createDirectory(source);
            }
          }
        } catch (IOException e) {
          throw new FileSystemException(getFolderAccessErrorMessage("create", path), e);
        }
        return null;
      }
    };
  }

  protected BlockingAction createTempDirectoryInternal(String parentDir, String prefix, String perms) {
    FileAttribute attrs = perms == null ? null : PosixFilePermissions.asFileAttribute(PosixFilePermissions.fromString(perms));
    return new BlockingAction() {
      public String perform() {
        try {
          Path tmpDir;
          if (parentDir != null) {
            Path dir = vertx.resolveFile(parentDir).toPath();
            if (attrs != null) {
              tmpDir = Files.createTempDirectory(dir, prefix, attrs);
            } else {
              tmpDir = Files.createTempDirectory(dir, prefix);
            }
          } else {
            if (attrs != null) {
              tmpDir = Files.createTempDirectory(prefix, attrs);
            } else {
              tmpDir = Files.createTempDirectory(prefix);
            }
          }
          return tmpDir.toFile().getAbsolutePath();
        } catch (IOException e) {
          throw new FileSystemException(getFolderAccessErrorMessage("create subfolder of", parentDir), e);
        }
      }
    };
  }

  protected BlockingAction createTempFileInternal(String parentDir, String prefix, String suffix, String perms) {
    FileAttribute attrs = perms == null ? null : PosixFilePermissions.asFileAttribute(PosixFilePermissions.fromString(perms));
    return new BlockingAction() {
      public String perform() {
        try {
          Path tmpFile;
          if (parentDir != null) {
            Path dir = vertx.resolveFile(parentDir).toPath();
            if (attrs != null) {
              tmpFile = Files.createTempFile(dir, prefix, suffix, attrs);
            } else {
              tmpFile = Files.createTempFile(dir, prefix, suffix);
            }
          } else {
            if (attrs != null) {
              tmpFile = Files.createTempFile(prefix, suffix, attrs);
            } else {
              tmpFile = Files.createTempFile(prefix, suffix);
            }
          }
          return tmpFile.toFile().getAbsolutePath();
        } catch (IOException e) {
          final String message = "Unable to create temporary file with prefix '" + prefix
            + "' and suffix '" + suffix
            + "' at " + parentDir;
          throw new FileSystemException(message, e);
        }
      }
    };
  }

  private BlockingAction> readDirInternal(String path) {
    return readDirInternal(path, null);
  }

  private BlockingAction> readDirInternal(String p, String filter) {
    Objects.requireNonNull(p);
    return new BlockingAction>() {
      public List perform() {
        try {
          File file = vertx.resolveFile(p);
          if (!file.exists()) {
            throw new FileSystemException("Cannot read directory " + file + ". Does not exist");
          }
          if (!file.isDirectory()) {
            throw new FileSystemException("Cannot read directory " + file + ". It's not a directory");
          } else {
            FilenameFilter fnFilter;
            if (filter != null) {
              fnFilter = new FilenameFilter() {
                public boolean accept(File dir, String name) {
                  return Pattern.matches(filter, name);
                }
              };
            } else {
              fnFilter = null;
            }
            File[] files;
            if (fnFilter == null) {
              files = file.listFiles();
            } else {
              files = file.listFiles(fnFilter);
            }
            List ret = new ArrayList<>(files.length);
            for (File f : files) {
              ret.add(f.getCanonicalPath());
            }
            return ret;
          }
        } catch (IOException e) {
          throw new FileSystemException(getFolderAccessErrorMessage("read", p), e);
        }
      }
    };
  }

  private BlockingAction readFileInternal(String path) {
    Objects.requireNonNull(path);
    return new BlockingAction() {
      public Buffer perform() {
        try {
          Path target = vertx.resolveFile(path).toPath();
          byte[] bytes = Files.readAllBytes(target);
          return Buffer.buffer(bytes);
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("read", path), e);
        }
      }
    };
  }

  private BlockingAction writeFileInternal(String path, Buffer data) {
    Objects.requireNonNull(path);
    Objects.requireNonNull(data);
    return new BlockingAction() {
      public Void perform() {
        try {
          Path target = vertx.resolveFile(path).toPath();
          Files.write(target, data.getBytes());
          return null;
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("write", path), e);
        }
      }
    };
  }

  private BlockingAction openInternal(String p, OpenOptions options) {
    Objects.requireNonNull(p);
    Objects.requireNonNull(options);
    return new BlockingAction() {
      public AsyncFile perform() {
        String path = vertx.resolveFile(p).getAbsolutePath();
        return doOpen(path, options, context);
      }
    };
  }

  protected AsyncFile doOpen(String path, OpenOptions options, ContextInternal context) {
    return new AsyncFileImpl(vertx, path, options, context);
  }

  private BlockingAction createFileInternal(String path) {
    return createFileInternal(path, null);
  }

  protected BlockingAction createFileInternal(String p, String perms) {
    Objects.requireNonNull(p);
    FileAttribute attrs = perms == null ? null : PosixFilePermissions.asFileAttribute(PosixFilePermissions.fromString(perms));
    return new BlockingAction() {
      public Void perform() {
        try {
          Path target = vertx.resolveFile(p).toPath();
          if (attrs != null) {
            Files.createFile(target, attrs);
          } else {
            Files.createFile(target);
          }
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("create", p), e);
        }
        return null;
      }
    };
  }

  private BlockingAction existsInternal(String path) {
    Objects.requireNonNull(path);
    return new BlockingAction() {
      public Boolean perform() {
        File file = vertx.resolveFile(path);
        return file.exists();
      }
    };
  }

  private BlockingAction fsPropsInternal(String path) {
    Objects.requireNonNull(path);
    return new BlockingAction() {
      public FileSystemProps perform() {
        try {
          Path target = vertx.resolveFile(path).toPath();
          FileStore fs = Files.getFileStore(target);
          return new FileSystemPropsImpl(fs.getTotalSpace(), fs.getUnallocatedSpace(), fs.getUsableSpace());
        } catch (IOException e) {
          throw new FileSystemException(getFileAccessErrorMessage("analyse", path), e);
        }
      }
    };
  }

  protected abstract class BlockingAction implements Handler> {

    protected final ContextInternal context;

    protected BlockingAction() {
      this.context = vertx.getOrCreateContext();
    }

    /**
     * Run the blocking action using a thread from the worker pool.
     */
    public Future run() {
      return context.executeBlockingInternal(this);
    }

    @Override
    public void handle(Promise fut) {
      try {
        T result = perform();
        fut.complete(result);
      } catch (Exception e) {
        fut.fail(e);
      }
    }

    public abstract T perform();

  }

  // Visible for testing
  static Set toCopyOptionSet(CopyOptions copyOptions) {
    Set copyOptionSet = new HashSet<>();
    if (copyOptions.isReplaceExisting()) copyOptionSet.add(StandardCopyOption.REPLACE_EXISTING);
    if (copyOptions.isCopyAttributes()) copyOptionSet.add(StandardCopyOption.COPY_ATTRIBUTES);
    if (copyOptions.isAtomicMove()) copyOptionSet.add(StandardCopyOption.ATOMIC_MOVE);
    if (copyOptions.isNofollowLinks()) copyOptionSet.add(LinkOption.NOFOLLOW_LINKS);
    return copyOptionSet;
  }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy