diff --git a/common/codechicken/core/laumch/DepLoader.java b/common/codechicken/core/laumch/DepLoader.java new file mode 100644 index 000000000..c5f23025c --- /dev/null +++ b/common/codechicken/core/laumch/DepLoader.java @@ -0,0 +1,682 @@ +package codechicken.core.laumch; + +import java.awt.Desktop; +import java.awt.Dialog.ModalityType; +import java.awt.Dimension; +import java.awt.event.WindowAdapter; +import java.awt.event.WindowEvent; +import java.beans.PropertyChangeEvent; +import java.beans.PropertyChangeListener; +import java.io.File; +import java.io.FileOutputStream; +import java.io.IOException; +import java.io.InputStream; +import java.io.InputStreamReader; +import java.io.InterruptedIOException; +import java.net.MalformedURLException; +import java.net.URL; +import java.net.URLConnection; +import java.nio.ByteBuffer; +import java.util.Arrays; +import java.util.HashMap; +import java.util.HashSet; +import java.util.Iterator; +import java.util.LinkedList; +import java.util.List; +import java.util.Map; +import java.util.Map.Entry; +import java.util.regex.Matcher; +import java.util.regex.Pattern; +import java.util.zip.ZipEntry; +import java.util.zip.ZipFile; + +import javax.swing.Box; +import javax.swing.JDialog; +import javax.swing.JEditorPane; +import javax.swing.JLabel; +import javax.swing.JOptionPane; +import javax.swing.JProgressBar; +import javax.swing.WindowConstants; +import javax.swing.event.HyperlinkEvent; +import javax.swing.event.HyperlinkListener; + +import net.minecraft.launchwrapper.LaunchClassLoader; +import argo.jdom.JdomParser; +import argo.jdom.JsonNode; +import argo.jdom.JsonRootNode; +import argo.jdom.JsonStringNode; +import argo.saj.InvalidSyntaxException; +import cpw.mods.fml.common.versioning.ComparableVersion; +import cpw.mods.fml.relauncher.FMLInjectionData; +import cpw.mods.fml.relauncher.FMLLaunchHandler; +import cpw.mods.fml.relauncher.IFMLCallHook; +import cpw.mods.fml.relauncher.IFMLLoadingPlugin; + +/** + * For autodownloading stuff. + * This is really unoriginal, mostly ripped off FML, credits to cpw. + */ +public class DepLoader implements IFMLLoadingPlugin, IFMLCallHook +{ + private static ByteBuffer downloadBuffer = ByteBuffer.allocateDirect(1 << 23); + private static final String owner = "CB's DepLoader"; + private static DepLoadInst inst; + + public interface IDownloadDisplay + { + void resetProgress(int sizeGuess); + + void setPokeThread(Thread currentThread); + + void updateProgress(int fullLength); + + boolean shouldStopIt(); + + void updateProgressString(String string, Object ... data); + + Object makeDialog(); + + void showErrorDialog(String name, String url); + } + + @SuppressWarnings("serial") + public static class Downloader extends JOptionPane implements IDownloadDisplay + { + private JDialog container; + private JLabel currentActivity; + private JProgressBar progress; + boolean stopIt; + Thread pokeThread; + + private Box makeProgressPanel() + { + Box box = Box.createVerticalBox(); + box.add(Box.createRigidArea(new Dimension(0,10))); + JLabel welcomeLabel = new JLabel(""+owner+" is setting up your minecraft environment"); + box.add(welcomeLabel); + welcomeLabel.setAlignmentY(LEFT_ALIGNMENT); + welcomeLabel = new JLabel("Please wait, "+owner+" has some tasks to do before you can play"); + welcomeLabel.setAlignmentY(LEFT_ALIGNMENT); + box.add(welcomeLabel); + box.add(Box.createRigidArea(new Dimension(0,10))); + currentActivity = new JLabel("Currently doing ..."); + box.add(currentActivity); + box.add(Box.createRigidArea(new Dimension(0,10))); + progress = new JProgressBar(0, 100); + progress.setStringPainted(true); + box.add(progress); + box.add(Box.createRigidArea(new Dimension(0,30))); + return box; + } + + @Override + public JDialog makeDialog() + { + if(container != null) + return container; + + setMessageType(JOptionPane.INFORMATION_MESSAGE); + setMessage(makeProgressPanel()); + setOptions(new Object[] { "Stop" }); + addPropertyChangeListener(new PropertyChangeListener() + { + @Override + public void propertyChange(PropertyChangeEvent evt) + { + if (evt.getSource() == Downloader.this && evt.getPropertyName()==VALUE_PROPERTY) + { + requestClose("This will stop minecraft from launching\nAre you sure you want to do this?"); + } + } + }); + container = new JDialog(null, "Hello", ModalityType.MODELESS); + container.setResizable(false); + container.setLocationRelativeTo(null); + container.add(this); + this.updateUI(); + container.pack(); + container.setMinimumSize(container.getPreferredSize()); + container.setVisible(true); + container.setDefaultCloseOperation(WindowConstants.DO_NOTHING_ON_CLOSE); + container.addWindowListener(new WindowAdapter() + { + @Override + public void windowClosing(WindowEvent e) + { + requestClose("Closing this window will stop minecraft from launching\nAre you sure you wish to do this?"); + } + }); + return container; + } + + protected void requestClose(String message) + { + int shouldClose = JOptionPane.showConfirmDialog(container, message, "Are you sure you want to stop?", JOptionPane.YES_NO_OPTION, JOptionPane.WARNING_MESSAGE); + if (shouldClose == JOptionPane.YES_OPTION) + container.dispose(); + + stopIt = true; + if (pokeThread != null) + pokeThread.interrupt(); + } + + @Override + public void updateProgressString(String progressUpdate, Object... data) + { + //FMLLog.finest(progressUpdate, data); + if (currentActivity!=null) + currentActivity.setText(String.format(progressUpdate, data)); + } + + @Override + public void resetProgress(int sizeGuess) + { + if (progress!=null) + progress.getModel().setRangeProperties(0, 0, 0, sizeGuess, false); + } + + @Override + public void updateProgress(int fullLength) + { + if (progress!=null) + progress.getModel().setValue(fullLength); + } + + @Override + public void setPokeThread(Thread currentThread) + { + this.pokeThread = currentThread; + } + + @Override + public boolean shouldStopIt() + { + return stopIt; + } + + @Override + public void showErrorDialog(String name, String url) { + JEditorPane ep = new JEditorPane("text/html", + "" + + owner+" was unable to download required library "+name + + "
Check your internet connection and try restarting or download it manually from" + + "
"+url+" and put it in your mods folder" + + ""); + + ep.setEditable(false); + ep.setOpaque(false); + ep.addHyperlinkListener(new HyperlinkListener() + { + @Override + public void hyperlinkUpdate(HyperlinkEvent event) + { + try + { + if (event.getEventType().equals(HyperlinkEvent.EventType.ACTIVATED)) + Desktop.getDesktop().browse(event.getURL().toURI()); + } + catch(Exception e) + {} + } + }); + + JOptionPane.showMessageDialog(null, ep, "A download error has occured", JOptionPane.ERROR_MESSAGE); + } + } + + public static class DummyDownloader implements IDownloadDisplay + { + @Override + public void resetProgress(int sizeGuess) + { + } + + @Override + public void setPokeThread(Thread currentThread) + { + } + + @Override + public void updateProgress(int fullLength) + { + } + + @Override + public boolean shouldStopIt() + { + return false; + } + + @Override + public void updateProgressString(String string, Object... data) + { + } + + @Override + public Object makeDialog() + { + return null; + } + + @Override + public void showErrorDialog(String name, String url) { + } + } + + public static class Dependancy + { + public String url; + public String[] filesplit; + public ComparableVersion version; + + public String existing; + /** + * Flag set to add this dep to the classpath immediately because it is required for a coremod. + */ + public boolean coreLib; + + public Dependancy(String url, String[] filesplit, boolean coreLib) + { + this.url = url; + this.filesplit = filesplit; + this.coreLib = coreLib; + version = new ComparableVersion(filesplit[1]); + } + + public String getName() + { + return filesplit[0]; + } + + public String fileName() + { + return filesplit[0]+filesplit[1]+filesplit[2]; + } + } + + public static class DepLoadInst + { + private File modsDir; + private File v_modsDir; + private IDownloadDisplay downloadMonitor; + private JDialog popupWindow; + + private Map depMap = new HashMap(); + private HashSet depSet = new HashSet(); + + public DepLoadInst() + { + String mcVer = (String) FMLInjectionData.data()[4]; + File mcDir = (File) FMLInjectionData.data()[6]; + + modsDir = new File(mcDir, "mods"); + v_modsDir = new File(mcDir, "mods/"+mcVer); + if(!v_modsDir.exists()) + v_modsDir.mkdirs(); + } + + private void addClasspath(String name) + { + try + { + ((LaunchClassLoader)DepLoader.class.getClassLoader()).addURL(new File(v_modsDir, name).toURI().toURL()); + } + catch(MalformedURLException e) + { + throw new RuntimeException(e); + } + } + + private void download(Dependancy dep) + { + popupWindow = (JDialog) downloadMonitor.makeDialog(); + File libFile = new File(v_modsDir, dep.fileName()); + try + { + URL libDownload = new URL(dep.url+'/'+dep.fileName()); + downloadMonitor.updateProgressString("Downloading file %s", libDownload.toString()); + System.out.format("Downloading file %s\n", libDownload.toString()); + URLConnection connection = libDownload.openConnection(); + connection.setConnectTimeout(5000); + connection.setReadTimeout(5000); + connection.setRequestProperty("User-Agent", ""+owner+" Downloader"); + int sizeGuess = connection.getContentLength(); + download(connection.getInputStream(), sizeGuess, libFile); + downloadMonitor.updateProgressString("Download complete"); + System.out.println("Download complete"); + + scanDepInfo(libFile); + } + catch (Exception e) + { + libFile.delete(); + if (downloadMonitor.shouldStopIt()) + { + System.err.println("You have stopped the downloading operation before it could complete"); + System.exit(1); + return; + } + downloadMonitor.showErrorDialog(dep.fileName(), dep.url+'/'+dep.fileName()); + throw new RuntimeException("A download error occured", e); + } + } + + private void download(InputStream is, int sizeGuess, File target) throws Exception + { + if (sizeGuess > downloadBuffer.capacity()) + throw new Exception(String.format("The file %s is too large to be downloaded by "+owner+" - the download is invalid", target.getName())); + + downloadBuffer.clear(); + + int bytesRead, fullLength = 0; + + downloadMonitor.resetProgress(sizeGuess); + try + { + downloadMonitor.setPokeThread(Thread.currentThread()); + byte[] smallBuffer = new byte[1024]; + while ((bytesRead = is.read(smallBuffer)) >= 0) { + downloadBuffer.put(smallBuffer, 0, bytesRead); + fullLength += bytesRead; + if (downloadMonitor.shouldStopIt()) + { + break; + } + downloadMonitor.updateProgress(fullLength); + } + is.close(); + downloadMonitor.setPokeThread(null); + downloadBuffer.limit(fullLength); + downloadBuffer.position(0); + } + catch (InterruptedIOException e) + { + // We were interrupted by the stop button. We're stopping now.. clear interruption flag. + Thread.interrupted(); + throw new Exception("Stop"); + } + catch (IOException e) + { + throw e; + } + + try + { + /*String cksum = generateChecksum(downloadBuffer); + if (cksum.equals(validationHash)) + {*/ + if(!target.exists()) + target.createNewFile(); + + + downloadBuffer.position(0); + FileOutputStream fos = new FileOutputStream(target); + fos.getChannel().write(downloadBuffer); + fos.close(); + /*} + else + { + throw new RuntimeException(String.format("The downloaded file %s has an invalid checksum %s (expecting %s). The download did not succeed correctly and the file has been deleted. Please try launching again.", target.getName(), cksum, validationHash)); + }*/ + } + catch (Exception e) + { + throw e; + } + } + + private String checkExisting(String[] dependancy) + { + for(File f : modsDir.listFiles()) + { + String[] split = splitFileName(f.getName()); + if(split == null || !split[0].equals(dependancy[0])) + continue; + + if(f.renameTo(new File(v_modsDir, f.getName()))) + continue; + + if(f.delete()) + continue; + + f.deleteOnExit(); + } + + for(File f : v_modsDir.listFiles()) + { + String[] split = splitFileName(f.getName()); + if(split == null || !split[0].equals(dependancy[0])) + continue; + + ComparableVersion found = new ComparableVersion(split[1]); + ComparableVersion requested = new ComparableVersion(dependancy[1]); + + int cmp = found.compareTo(requested); + if(cmp < 0) + { + System.out.println("Deleted old version "+f.getName()); + f.delete(); + return null; + } + if(cmp > 0) + { + System.err.println("Warning: version of "+dependancy[0]+", "+split[1]+" is newer than request "+dependancy[1]); + return f.getName(); + } + return f.getName();//found dependancy + } + return null; + } + + public void load() + { + scanDepInfos(); + if(depMap.isEmpty()) + return; + + loadDeps(); + activateDeps(); + } + + private void activateDeps() + { + for(Dependancy dep : depMap.values()) + if(dep.coreLib) + addClasspath(dep.existing); + } + + private void loadDeps() + { + downloadMonitor = FMLLaunchHandler.side().isClient() ? new Downloader() : new DummyDownloader(); + try + { + while(!depSet.isEmpty()) + { + Iterator it = depSet.iterator(); + Dependancy dep = depMap.get(it.next()); + it.remove(); + load(dep); + } + } + finally + { + if (popupWindow != null) + { + popupWindow.setVisible(false); + popupWindow.dispose(); + } + } + } + + private void load(Dependancy dep) + { + dep.existing = checkExisting(dep.filesplit); + if(dep.existing == null)//download dep + { + download(dep); + dep.existing = dep.fileName(); + } + } + + private List modFiles() + { + List list = new LinkedList(); + list.addAll(Arrays.asList(modsDir.listFiles())); + list.addAll(Arrays.asList(v_modsDir.listFiles())); + return list; + } + + private void scanDepInfos() + { + for(File file : modFiles()) + { + if(!file.getName().endsWith(".jar") && !file.getName().endsWith(".zip")) + continue; + + scanDepInfo(file); + } + } + + private void scanDepInfo(File file) + { + try + { + ZipFile zip = new ZipFile(file); + ZipEntry e = zip.getEntry("dependancies.info"); + if(e != null) + loadJSon(zip.getInputStream(e)); + zip.close(); + } + catch(Exception e) + { + System.err.println("Failed to load dependancies.info from "+file.getName()+" as JSON"); + e.printStackTrace(); + } + } + + private void loadJSon(InputStream input) throws IOException, InvalidSyntaxException + { + InputStreamReader reader = new InputStreamReader(input); + JsonRootNode root = new JdomParser().parse(reader); + if(root.hasElements()) + loadJSonArr(root); + else + loadJson(root); + reader.close(); + } + + private void loadJSonArr(JsonRootNode root) throws IOException + { + for(JsonNode node : root.getElements()) + loadJson(node); + } + + private void loadJson(JsonNode node) throws IOException + { + boolean obfuscated = ((LaunchClassLoader)DepLoader.class.getClassLoader()) + .getClassBytes("net.minecraft.world.World") == null; + + String testClass = node.getStringValue("class"); + if(DepLoader.class.getResource("/"+testClass.replace('.', '/')+".class") != null) + return; + + String repo = node.getStringValue("repo"); + String file = node.getStringValue("file"); + if(!obfuscated && node.isNode("dev")) + file = node.getStringValue("dev"); + + boolean coreLib = node.isNode("coreLib") && node.getBooleanValue("coreLib"); + + List reserved = Arrays.asList("repo", "file", "class", "dev", "coreLib"); + for(Entry e : node.getFields().entrySet()) + { + String s = e.getKey().getText(); + if(!e.getValue().hasText() || reserved.contains(s)) + continue; + + file = file.replaceAll("@"+s.toUpperCase()+"@", e.getValue().getText()); + } + + String[] split = splitFileName(file); + if(split == null) + throw new RuntimeException("Invalid filename format for dependancy: "+file); + + addDep(new Dependancy(repo, split, coreLib)); + } + + private void addDep(Dependancy newDep) + { + if(mergeNew(depMap.get(newDep.getName()), newDep)) + { + depMap.put(newDep.getName(), newDep); + depSet.add(newDep.getName()); + } + } + + private boolean mergeNew(Dependancy oldDep, Dependancy newDep) { + if(oldDep == null) + return true; + + Dependancy newest = newDep.version.compareTo(oldDep.version) > 0 ? newDep : oldDep; + newest.coreLib = newDep.coreLib || oldDep.coreLib; + + return newest == newDep; + } + } + + public static void load() + { + if(inst == null) + { + inst = new DepLoadInst(); + inst.load(); + } + } + + private static String[] splitFileName(String filename) + { + Pattern p = Pattern.compile("(.+?)([\\d\\.\\w]+)(\\.[^\\d]+)"); + Matcher m = p.matcher(filename); + if(!m.matches()) + return null; + + return new String[]{m.group(1), m.group(2), m.group(3)}; + } + + @Override + public String[] getASMTransformerClass() + { + return null; + } + + @Override + public String getModContainerClass() + { + return null; + } + + @Override + public String getSetupClass() + { + return getClass().getName(); + } + + @Override + public void injectData(Map data) + { + } + + @Override + public Void call() + { + load(); + + return null; + } + + @Override + public String[] getLibraryRequestClass() + { + return null; + } +} diff --git a/etc/core/dependancies.info b/etc/core/dependancies.info new file mode 100755 index 000000000..e9bff4de9 --- /dev/null +++ b/etc/core/dependancies.info @@ -0,0 +1,19 @@ +{ + "repo": "http://files.minecraftforge.net/ForgeMultipart/", + "file": "ForgeMultipart-universal-@MCVERSION@-@VERSION@.jar", + "dev": "ForgeMultipart-dev-@MCVERSION@-@VERSION@.jar", + "class": "codechicken.lib.asm.ASMHelper", + "version": "1.0.0.219", + "mcversion": "1.6.4", + "coreLib": false +} + +{ + "repo": "http://files.minecraftforge.net/CodeChickenLib/", + "file": "CodeChickenLib-universal-@MCVERSION@-@VERSION@.jar", + "dev": "CodeChickenLib-dev-@MCVERSION@-@VERSION@.jar", + "class": "codechicken.lib.asm.ASMHelper", + "version": "1.0.0.45", + "mcversion": "1.6.4", + "coreLib": true +} \ No newline at end of file