se.somath.publisher.includer.SourceCodeReader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of model Show documentation
There is a newer version: 1.0.5
package se.somath.publisher.includer;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.filefilter.IOFileFilter;
import se.somath.publisher.excpetion.PublishException;

import java.io.File;
import java.io.IOException;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

import static se.somath.publisher.includer.FileFilter.createDirectoryFilter;
import static se.somath.publisher.includer.FileFilter.createFileFilter;

public class SourceCodeReader {

    public List readFile(String root, String fileName) {
        Collection files = listFiles(root);
        File sourceCodeFile = findWantedSourceCodeFile(root, fileName, files);

        return readSourcecodeContent(sourceCodeFile);
    }

    private Collection listFiles(String root) {
        File rootDir = locateRootDirectory(root);
        IOFileFilter fileFilter = createFileFilter();
        IOFileFilter dirFilter = createDirectoryFilter();

        return FileUtils.listFiles(rootDir, fileFilter, dirFilter);
    }

    private File locateRootDirectory(String root) {
        DirectoryLocator directoryLocator = new DirectoryLocator();

        return directoryLocator.locateDirectory(root);
    }

    File findWantedSourceCodeFile(String root, String fileName, Collection files) {
        File sourceCodeFile = null;

        for (File candidate : files) {
            String candidatePath = getCandidatePath(candidate);
            String filePath = getFilePath(root, fileName);

            if (candidatePath.contains(filePath)) {
                return candidate;
            }
        }

        return sourceCodeFile;
    }

    private String getCandidatePath(File candidate) {
        try {
            return candidate.getCanonicalPath();
        } catch (IOException e) {
            throw new PublishException(e);
        }
    }

    private String getFilePath(String root, String fileName) {
        String fixedRoot = removeLeadingFileSeparator(root);
        fixedRoot = addEndingFileSeparator(fixedRoot);

        String fixedFileName = removeLeadingFileSeparator(fileName);

        return fixedRoot + fixedFileName;
    }

    private String addEndingFileSeparator(String root) {
        if (!root.endsWith(File.separator)) {
            root = root + File.separator;
        }

        return root;
    }

    private String removeLeadingFileSeparator(String path) {
        String pathSeparator = File.separator;

        if (path.startsWith("." + pathSeparator)) {
            int beginIndex = 2;
            path = path.substring(beginIndex);
        } else if (path.startsWith(pathSeparator)) {
            int beginIndex = 1;
            path = path.substring(beginIndex);
        }

        return path;
    }

    private List readSourcecodeContent(File sourceCodeFile) {
        if (sourceCodeFile == null) {
            return Collections.emptyList();
        }

        try {
            return FileUtils.readLines(sourceCodeFile);
        } catch (IOException e) {
            throw new PublishException(e);
        }
    }
}