com.chiorichan.ResourceLoader.java Source code

Introduction

Here is the source code for com.chiorichan.ResourceLoader.java

Source

/**
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * Copyright 2016 Chiori Greene a.k.a. Chiori-chan <me@chiorichan.com>
 * All Right Reserved.
 */
package com.chiorichan;

import java.awt.Image;
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.zip.ZipEntry;
import java.util.zip.ZipException;
import java.util.zip.ZipFile;

import javax.imageio.ImageIO;

import com.chiorichan.util.FileFunc;
import com.google.common.io.CharStreams;

public class ResourceLoader {
    /*
     * Provide a folder or zip file that contains the resources.
     */
    public static ResourceLoader buildLoader(String resource) {
        File workingWith = FileFunc.isAbsolute(resource) ? new File(resource)
                : new File(AppConfig.get().getDirectory().getAbsolutePath(), resource);

        if (!workingWith.exists())
            return null;

        try {
            return new ResourceLoader(workingWith);
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
    }

    private final File resourcePath;
    private final ZipFile zipLib;

    private final boolean isZipFile;

    public ResourceLoader(File path) throws IOException {
        resourcePath = path;
        isZipFile = path.getAbsolutePath().endsWith(".zip");

        if (isZipFile)
            zipLib = new ZipFile(resourcePath);
        else
            zipLib = null;
    }

    public Image getImage(String relPath) {
        try {
            InputStream is = getInputStream(relPath);

            BufferedInputStream in = new BufferedInputStream(is);

            return ImageIO.read(in);
        } catch (IOException e) {
            return null;
        }
    }

    public InputStream getInputStream(String relPath) throws ZipException, IOException {
        if (isZipFile) {
            ZipEntry entry = zipLib.getEntry(relPath);
            if (entry == null)
                throw new IOException("No idea what went wrong but the Zip Library returned a null file header.");

            if (entry.isDirectory())
                throw new IOException("Can not get an InputStream on a folder.");

            return zipLib.getInputStream(entry);
        } else {
            File file = new File(resourcePath.getAbsolutePath() + File.pathSeparator + relPath);

            if (!file.exists())
                throw new FileNotFoundException();

            if (file.isDirectory())
                throw new IOException("Can not get an InputStream on a folder.");

            return new FileInputStream(file);
        }
    }

    public String getText(String relPath) {
        try {
            InputStream is = getInputStream(relPath);

            return CharStreams.toString(new InputStreamReader(is, "UTF-8"));
        } catch (IOException e) {
            return null;
        }
    }
}