zz.filecollector.FileWorkerThread.java Source code

Introduction

Here is the source code for zz.filecollector.FileWorkerThread.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package zz.filecollector;

import java.io.File;
import java.util.Collection;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.apache.commons.io.filefilter.TrueFileFilter;
import zz.filecollector.fileprocessor.FileProcessorRegister;

/**
 *
 * @author zhan
 */
public class FileWorkerThread extends Thread {
    public static final int ACTION_PREVIEW = 0;
    public static final int ACTION_MOVE = 1;
    public static final int ACTION_COPY = 2;
    public static final int ACTION_MOVEDEL = 3;

    private File srcDir;
    private File destDir;
    private final FileCollector photoCollector;
    private int actionType;
    private volatile boolean keepDoing = true;

    private int filesCopied;
    private int filesMoved;
    private int filesDeleted;
    private int filesTotal;

    public FileWorkerThread(FileCollector photoCollector) {
        this.srcDir = photoCollector.getSrcDir();
        this.destDir = photoCollector.getDestDir();
        this.photoCollector = photoCollector;
    }

    @Override
    public void run() {
        this.infoln("" + srcDir.getAbsolutePath() + "  " + destDir.getAbsolutePath());
        this.photoCollector.disableButtons();
        Collection<File> srcFiles = FileUtils.listFiles(srcDir, ExtensionRegister.getInstance().getFileFilter(),
                TrueFileFilter.TRUE);
        for (File file : srcFiles) {
            if (keepDoing) {
                processFile(file);
            } else {
                this.infoln("!");
                break;
            }
        }
        this.infoln(String.format(
                " %d . ? %d .  %d .  %d ",
                this.filesTotal, this.filesCopied, this.filesMoved, this.filesDeleted));
        this.photoCollector.enableButtons();
    }

    private void processFile(File file) {
        FileInfo fileInfo = FileProcessorRegister.extractFileInfo(file);
        if (fileInfo.isValid()) {
            this.filesTotal++;
            this.info(file.getAbsolutePath());

            fileInfo.setDupIndex(0);
            String newName = getNewFilePath(this.destDir.getAbsolutePath(), fileInfo);
            File newFile = new File(newName);
            while (newFile.exists()) {
                FileInfo newInfo = FileProcessorRegister.extractFileInfo(newFile);
                if (newInfo.isSameFile(fileInfo)) {
                    handleDuplicatedFile(file);
                    return;
                } else {
                    newInfo.setDupIndex(newInfo.getDupIndex() + 1);
                    newName = getNewFilePath(this.destDir.getAbsolutePath(), newInfo);
                    newFile = new File(newName);
                }
            }
            handleNewFile(file, newFile);
        } else {
            this.error("? - " + file.getAbsolutePath());
        }
    }

    private void handleDuplicatedFile(File file) {
        this.infoln("... ??.");
        if (actionType == ACTION_MOVEDEL) {
            try {
                if (file.delete()) {
                    this.filesDeleted++;
                    this.infoln(".");
                } else {
                    this.infoln(".");
                }
            } catch (Exception e) {
                this.error(e.getMessage());
            }
        }
    }

    private void handleNewFile(File file, File newFile) {
        this.infoln(" --> " + newFile.getAbsolutePath());

        switch (actionType) {
        case ACTION_COPY:
            try {
                FileUtils.copyFile(file, newFile, true);
                this.filesCopied++;
                this.infoln("?.");
            } catch (Exception ex) {
                this.error(ex.getMessage());
            }
            break;
        case ACTION_MOVE:
        case ACTION_MOVEDEL:
            try {
                FileUtils.moveFile(file, newFile);
                this.filesMoved++;
                this.infoln("?.");
            } catch (Exception ex) {
                this.error(ex.getMessage());
            }
            break;
        default:
            break;
        }
    }

    private String getNewFilePath(String basePath, FileInfo fileInfo) {
        StringBuilder path = new StringBuilder();
        switch (fileInfo.getFileType()) {
        case FileInfo.PHOTO:
            path.append("/");
            break;
        case FileInfo.VIDEO:
            path.append("/");
            break;
        default:
            path.append("/");
        }
        path.append(fileInfo.getYear()).append("/");
        if (fileInfo.getFileType() == FileInfo.PHOTO) {
            path.append(fileInfo.getMonth()).append("/");
        }
        path.append(fileInfo.getNormalizedName());
        return FilenameUtils.concat(basePath, path.toString());
    }

    public void stopWorking() {
        this.keepDoing = false;
    }

    public void setActionType(int actionType) {
        this.actionType = actionType;
    }

    private void info(String msg) {
        this.photoCollector.displayInfo(msg);
    }

    private void infoln(String msg) {
        this.photoCollector.displayInfo(msg + "\n");
    }

    private void error(String msg) {
        this.photoCollector.displayInfo("ERROR: " + msg);
        this.photoCollector.displayInfo("\n");
    }
}