CustomNPC-Plus/src/main/java/noppes/npcs/client/AssetsBrowser.java

package noppes.npcs.client;

import cpw.mods.fml.common.Loader;
import cpw.mods.fml.common.ModContainer;
import cpw.mods.fml.common.ObfuscationReflectionHelper;
import net.minecraft.client.Minecraft;
import net.minecraft.client.resources.*;
import org.apache.commons.lang3.StringUtils;

import java.io.File;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

public class AssetsBrowser {
    public boolean isRoot;
    private int depth;
    private String folder;
    public HashSet<String> folders = new HashSet<String>();
    public HashSet<String> files = new HashSet<String>();
    private String[] extensions;

    public AssetsBrowser(String folder, String[] extensions) {
        this.extensions = extensions;
        this.setFolder(folder);
    }

    public void setFolder(String folder) {
        if (!folder.endsWith("/"))
            folder += "/";
        isRoot = folder.length() <= 1;
        this.folder = "/assets" + folder;
        depth = StringUtils.countMatches(this.folder, "/");
        getFiles();
    }

    public AssetsBrowser(String[] extensions) {
        this.extensions = extensions;
    }

    private void getFiles() {
        folders.clear();
        files.clear();

        ResourcePackRepository repos = Minecraft.getMinecraft().getResourcePackRepository();
        SimpleReloadableResourceManager simplemanager = (SimpleReloadableResourceManager) Minecraft.getMinecraft().getResourceManager();

        Map<String, IResourceManager> map = ObfuscationReflectionHelper.getPrivateValue(SimpleReloadableResourceManager.class, simplemanager, 2);
        HashSet<String> set = new HashSet<String>();
        for (String name : map.keySet()) {
            if (!(map.get(name) instanceof FallbackResourceManager))
                continue;
            FallbackResourceManager manager = (FallbackResourceManager) map.get(name);
            List<IResourcePack> list = ObfuscationReflectionHelper.getPrivateValue(FallbackResourceManager.class, manager, 0);
            for (IResourcePack pack : list) {
                if (pack instanceof AbstractResourcePack) {
                    AbstractResourcePack p = (AbstractResourcePack) pack;
                    File file = NPCResourceHelper.getPackFile(p);
                    if (file != null)
                        set.add(file.getAbsolutePath());
                }
            }
        }

        for (String file : set) {
            progressFile(new File(file));
        }
        for (ModContainer mod : Loader.instance().getModList()) {
            if (mod.getSource().exists())
                progressFile(mod.getSource());
        }
//		List<ResourcePackRepository.Entry> list = repos.getRepositoryEntries();
//		System.out.println(repos.rprDefaultResourcePack);
//		for(ResourcePackRepository.Entry entry : list) {
//			System.out.println(entry.getResourcePack());
//			File file = new File(repos.getDirResourcepacks(),entry.getResourcePackName());
//			if(file.exists()){
//				progressFile(file);
//			}
//			checkFolder(new File(CustomNpcs.Dir,"assets"),CustomNpcs.Dir.getAbsolutePath().length());
//		}
//		File file = new File(repos.getDirResourcepacks(),repos.getResourcePackName());
//		if(file.exists()){
//			progressFile(file);
//		}
//		checkFolder(new File(CustomNpcs.Dir,"assets"),CustomNpcs.Dir.getAbsolutePath().length());
        //TODO fix
    }

    private void progressFile(File file) {
        try {
            if (!file.isDirectory() && (file.getName().endsWith(".jar") || file.getName().endsWith(".zip"))) {
                ZipFile zip = new ZipFile(file);
                Enumeration<? extends ZipEntry> entries = zip.entries();
                while (entries.hasMoreElements()) {
                    ZipEntry zipentry = entries.nextElement();
                    String entryName = zipentry.getName();
                    checkFile(entryName);

                }
                zip.close();
            } else if (file.isDirectory()) {
                int length = file.getAbsolutePath().length();
                checkFolder(file, length);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private void checkFolder(File file, int length) {
        File[] files = file.listFiles();
        if (files == null) {
            return;
        }
        for (File f : files) {
            String name = f.getAbsolutePath().substring(length);
            name = name.replace("\\", "/");
            if (!name.startsWith("/"))
                name = "/" + name;
            if (f.isDirectory() && (folder.startsWith(name) || name.startsWith(folder))) {
                checkFile(name + "/");
                checkFolder(f, length);
            } else
                checkFile(name);
        }
    }

    private void checkFile(String name) {
        if (!name.startsWith("/"))
            name = "/" + name;
        if (!name.startsWith(this.folder)) {
            return;
        }
        String[] split = name.split("/");
        int count = split.length;
        if (count == depth + 1) {
            if (validExtension(name))
                files.add(split[depth]);
        } else if (depth + 1 < count) {
            folders.add(split[depth]);
        }
    }

    private boolean validExtension(String entryName) {
        int index = entryName.indexOf(".");
        if (index < 0)
            return false;
        String extension = entryName.substring(index + 1);
        for (String ex : extensions) {
            if (ex.equalsIgnoreCase(extension))
                return true;
        }
        return false;
    }

    public String getAsset(String asset) {
        String[] split = folder.split("/");
        if (split.length < 3)
            return null;
        String texture = split[2] + ":";
        texture += folder.substring(texture.length() + 8) + asset;
        return texture;
    }

    public static String getRoot(String asset) {
        String mod = "minecraft";
        int index = asset.indexOf(":");
        if (index > 0) {
            mod = asset.substring(0, index);
            asset = asset.substring(index + 1);
        }
        if (asset.startsWith("/"))
            asset = asset.substring(1);
        String location = "/" + mod + "/" + asset;
        index = location.lastIndexOf("/");
        if (index > 0)
            location = location.substring(0, index);
        return location;
    }
}