se.trixon.toolbox.checksum.FileVisitor.java Source code

Introduction

Here is the source code for se.trixon.toolbox.checksum.FileVisitor.java

Source

/* 
 * Copyright 2015 Patrik Karlsson.
 *
 * 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
 *
 *      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 se.trixon.toolbox.checksum;

import java.io.File;
import java.io.IOException;
import java.nio.file.FileVisitResult;
import java.nio.file.Path;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.ArrayList;
import java.util.List;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.SystemUtils;
import org.openide.awt.StatusDisplayer;
import se.trixon.toolbox.core.Toolbox;

/**
 *
 * @author Patrik Karlsson <patrik@trixon.se>
 */
public class FileVisitor extends SimpleFileVisitor<Path> {

    private final File mBaseDir;
    private boolean mInterrupted;
    private List<String> mRelativePaths = new ArrayList<>();

    public FileVisitor(File baseDir, List<String> relativePaths) {
        mBaseDir = baseDir;
        mRelativePaths = relativePaths;
    }

    public boolean isInterrupted() {
        return mInterrupted;
    }

    @Override
    public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throws IOException {
        if (Thread.interrupted()) {
            mInterrupted = true;
            return FileVisitResult.TERMINATE;
        }

        Toolbox.setStatusText(dir.toFile().getAbsolutePath(), StatusDisplayer.IMPORTANCE_ERROR_HIGHLIGHT);

        String[] filePaths = dir.toFile().list();

        if (filePaths != null && filePaths.length > 0) {
            for (String fileName : filePaths) {
                File file = new File(dir.toFile(), fileName);
                if (file.isFile()) {
                    mRelativePaths.add(getRelativePath(file, mBaseDir));
                }
            }
        }

        return FileVisitResult.CONTINUE;
    }

    @Override
    public FileVisitResult visitFileFailed(Path file, IOException exc) {
        return FileVisitResult.CONTINUE;
    }

    private String getRelativePath(File file, File base) {
        return StringUtils.substringAfter(file.getAbsolutePath(),
                base.getAbsolutePath() + SystemUtils.FILE_SEPARATOR);
    }
}