package me.km;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;

public class PackageHelper 
{
    public static ArrayList<Class> getClasses(String packageName)
    {
        URL jar = getPath();
        if (jar != null)
        {
            try
            {
                ZipInputStream zip = new ZipInputStream(jar.openStream());
                ArrayList<Class> classes = new ArrayList<>();
                String path = packageName.replace(".", "/");
                while(true) 
                {
                    ZipEntry e = zip.getNextEntry();
                    if (e == null)
                    {
                        break;
                    }
                    String name = e.getName();
                    if(name.startsWith(path))
                    {
                        name = name.replace("/", ".").substring(0, e.getName().length() - 6);
                        try 
                        {
                            classes.add(Class.forName(name));
                        } 
                        catch (ClassNotFoundException ex) 
                        {
                        } 
                    }
                }
                return classes;
            }
            catch(IOException ex)
            {
            }
        } 
        return new ArrayList<>();
    }
    
    private static URL getPath()
    {
        try
        {
            String s = KajetansMod.class.getProtectionDomain().getCodeSource().getLocation().getPath();
            s = s.substring(5, s.length() - 25);
            
            File f = new File(s);
            if(!f.exists())
            {
                for(int i = 0; i < 20; i++)
                {
                    System.out.println(f + " does not exist");
                }
            }
            
            return new URL("file:" + s);
        }
        catch(Exception ex)
        {
            System.out.println(ex);
        }
        return null;
    }
}