com.dragovorn.dragonbot.api.bot.file.FileManager.java Source code

Introduction

Here is the source code for com.dragovorn.dragonbot.api.bot.file.FileManager.java

Source

/*
 * Copyright (c) 2016. Andrew Burr
 *  Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
 *  associated documentation files (the "Software"), to deal in the Software without restriction,
 *  including without limitation the rights to use, copy, modify, merge, publish, distribute,
 *  sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
 *  furnished to do so, subject to the following conditions:
 *
 *  The above copyright notice and this permission notice shall be included in all copies or
 *  substantial portions of the Software.
 *
 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
 *  INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
 *  PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
 *  COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 *  ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH
 *  THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

package com.dragovorn.dragonbot.api.bot.file;

import org.apache.commons.io.FileUtils;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

// FIXME: 11/27/16 make this into an object in DragonBot
public class FileManager {

    private static File directory = new File("Dragon Bot");
    private static File config = new File(directory, "config.yml");
    private static File logs = new File(directory, "logs");
    private static File plugins = new File(directory, "plugins");
    private static File updater = new File(directory, "updater.jar");
    public static final File dir = new File(System.getProperty("user.home"), ".dragonbot");

    private static List<File> pluginAddedFiles = new ArrayList<>();

    public static void reloadFiles() {
        FileManager.config = new File(directory, "config.yml");
        FileManager.logs = new File(directory, "logs");
        FileManager.plugins = new File(directory, "plugins");
        FileManager.updater = new File(directory, "updater.jar");

        List<File> toCopy = new ArrayList<>();

        toCopy.addAll(pluginAddedFiles);

        pluginAddedFiles.clear();

        toCopy.forEach(file -> {
            try {
                if (file.isDirectory()) {
                    FileUtils.copyDirectory(file, new File(directory, file.getName()));
                } else {
                    FileUtils.copyFile(file, new File(directory, file.getName()));
                }

                file.delete();

                pluginAddedFiles.add(new File(directory, file.getName()));
            } catch (IOException exception) {
                exception.printStackTrace();
            }
        });
    }

    public static File getFile(String name) {
        for (File file : pluginAddedFiles) {
            if (file.getName().indexOf(".") > 0) {
                if (file.getName().substring(0, file.getName().lastIndexOf(".")).equals(name)) {
                    return file;
                }
            }
        }

        return null;
    }

    public static File addFile(File file) {
        if (!file.exists()) {
            if (file.isDirectory()) {
                file.mkdirs();
            } else {
                try {
                    file.createNewFile();
                } catch (IOException exception) {
                    exception.printStackTrace();
                }
            }
        }

        pluginAddedFiles.add(file);

        return file;
    }

    public static void setDirectory(String directory) {
        File file = new File(directory, "Dragon Bot");

        if (FileManager.directory.exists() && !file.exists()) {
            if (file.mkdirs()) {
                try {
                    FileUtils.copyDirectory(FileManager.directory, file);
                } catch (IOException exception) {
                    exception.printStackTrace();
                }
            } else {
                throw new RuntimeException("Unable to create file: " + file.getName()); // FIXME: 11/23/16 make custom exception
            }

            try {
                FileUtils.deleteDirectory(FileManager.directory);
            } catch (IOException exception) {
                exception.printStackTrace();
            }
        }

        FileManager.directory = file;

        File path = new File(FileManager.dir, "path");

        try {
            FileWriter fileWriter = new FileWriter(path);

            BufferedWriter writer = new BufferedWriter(fileWriter);

            writer.write(directory);

            writer.close();
            fileWriter.close();
        } catch (IOException exception) {
            exception.printStackTrace();
        }

        reloadFiles();
    }

    public static File getDirectory() {
        return directory;
    }

    public static File getConfig() {
        return config;
    }

    public static File getLogs() {
        return logs;
    }

    public static File getPlugins() {
        return plugins;
    }

    public static File getUpdater() {
        return updater;
    }
}