platform/script-debugger/protocol/protocol-model-generator/src/org/jetbrains/protocolReader/FileSet.java

package org.jetbrains.protocolReader;

import gnu.trove.THashSet;
import gnu.trove.TObjectProcedure;
import org.jetbrains.annotations.NotNull;

import java.io.IOException;
import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;

/**
 * Records a list of files in the root directory and deletes files that were not re-generated.
 */
class FileSet {
  private final Path rootDir;
  @SuppressWarnings("MismatchedQueryAndUpdateOfCollection")
  private final THashSet<Path> unusedFiles;

  FileSet(Path rootDir) throws IOException {
    this.rootDir = rootDir;
    unusedFiles = new THashSet<Path>();
    Files.walkFileTree(rootDir, new SimpleFileVisitor<Path>() {
      @Override
      public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throws IOException {
        return Files.isHidden(dir) ? FileVisitResult.SKIP_SUBTREE : FileVisitResult.CONTINUE;
      }

      @NotNull
      @Override
      public FileVisitResult visitFile(Path path, BasicFileAttributes attrs) throws IOException {
        if (!Files.isHidden(path)) {
          unusedFiles.add(path);
        }
        return FileVisitResult.CONTINUE;
      }
    });
  }

  FileUpdater createFileUpdater(String filePath) {
    Path file = rootDir.resolve(filePath);
    unusedFiles.remove(file);
    return new FileUpdater(file);
  }

  void deleteOtherFiles() {
    unusedFiles.forEach(new TObjectProcedure<Path>() {
      @Override
      public boolean execute(Path path) {
        try {
          if (Files.deleteIfExists(path)) {
            Path parent = path.getParent();
            try (DirectoryStream<Path> stream = Files.newDirectoryStream(parent)) {
              if (!stream.iterator().hasNext()) {
                Files.delete(parent);
              }
            }
          }
        }
        catch (IOException e) {
          throw new RuntimeException(e);
        }
        return true;
      }
    });
  }
}