com.consol.citrus.admin.util.FileHelperImpl.java Source code

Introduction

Here is the source code for com.consol.citrus.admin.util.FileHelperImpl.java

Source

/*
 * Copyright 2006-2013 the original author or authors.
 *
 * 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 com.consol.citrus.admin.util;

import com.consol.citrus.admin.exception.CitrusAdminRuntimeException;
import org.apache.commons.io.FilenameUtils;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.io.*;
import java.net.URLDecoder;
import java.util.*;

/**
 * @author Martin Maher, Christoph Deppisch
 * @since 2013.04.22
 */
@Component
public class FileHelperImpl implements FileHelper {

    /**
     * {@inheritDoc}
     */
    public String[] getFolders(File directory) {
        if (directory.exists()) {
            String[] files = directory.list(new FilenameFilter() {
                public boolean accept(File dir, String name) {
                    return name.charAt(0) != '.' && new File(dir, name).isDirectory();
                }
            });

            if (files != null) {
                Arrays.sort(files, String.CASE_INSENSITIVE_ORDER);
                return files;
            } else {
                return new String[] {};
            }
        } else {
            throw new CitrusAdminRuntimeException(
                    "Could not open directory because it does not exist: " + directory);
        }
    }

    /**
     * {@inheritDoc}
     */
    public String[] getFiles(File directory, final String fileExtension) {
        List<String> fileNames = new ArrayList<String>();

        if (!directory.exists()) {
            return new String[] {};
        }

        File[] found = directory.listFiles(new FilenameFilter() {
            public boolean accept(File dir, String name) {
                return name.endsWith(fileExtension);
            }
        });

        for (File file : found) {
            fileNames.add(FilenameUtils.getBaseName(file.getName()));
        }

        return fileNames.toArray(new String[fileNames.size()]);
    }

    /**
     * {@inheritDoc}
     */
    public File findFileInPath(File directory, String filename, boolean recursive) {
        if (!directory.isDirectory()) {
            String msg = String.format("Expected a directory but instead got '%s'", directory.getAbsolutePath());
            throw new UnsupportedOperationException(msg);
        }

        File[] files = directory.listFiles();
        for (File file : files) {
            if (file.isDirectory()) {
                File returnedFile = findFileInPath(file, filename, recursive);
                if (returnedFile != null) {
                    return returnedFile;
                }
            } else {
                if (file.getName().equals(filename)) {
                    return file;
                }
            }
        }
        return null;
    }

    /**
     * {@inheritDoc}
     */
    public String decodeDirectoryUrl(String url, String rootDirectory) {
        String directory = null;

        try {
            directory = URLDecoder.decode(url, "UTF-8"); // TODO use system default encoding?
        } catch (UnsupportedEncodingException e) {
            throw new CitrusAdminRuntimeException("Unable to decode directory URL", e);
        }

        if (directory.equals("/")) {
            if (StringUtils.hasText(rootDirectory)) {
                directory = rootDirectory;
            } else {
                return rootDirectory;
            }
        }

        if (directory.charAt(directory.length() - 1) == '\\') {
            directory = directory.substring(0, directory.length() - 1) + "/";
        } else if (directory.charAt(directory.length() - 1) != '/') {
            directory += "/";
        }

        return directory;
    }
}