| package codechicken.core.launch; |
| |
| 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("<html><b><font size='+1'>"+owner+" is setting up your minecraft environment</font></b></html>"); |
| box.add(welcomeLabel); |
| welcomeLabel.setAlignmentY(LEFT_ALIGNMENT); |
| welcomeLabel = new JLabel("<html>Please wait, "+owner+" has some tasks to do before you can play</html>"); |
| 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; |
| } |
| |
| //TODO @Override |
| public JDialog makeDialog() |
| { |
| if(container != null) |
| return container; |
| |
| setMessageType(JOptionPane.INFORMATION_MESSAGE); |
| setMessage(makeProgressPanel()); |
| setOptions(new Object[] { "Stop" }); |
| addPropertyChangeListener(new PropertyChangeListener() |
| { |
| //TODO @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() |
| { |
| //TODO @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(); |
| } |
| |
| //TODO @Override |
| public void updateProgressString(String progressUpdate, Object... data) |
| { |
| //FMLLog.finest(progressUpdate, data); |
| if (currentActivity!=null) |
| currentActivity.setText(String.format(progressUpdate, data)); |
| } |
| |
| //TODO @Override |
| public void resetProgress(int sizeGuess) |
| { |
| if (progress!=null) |
| progress.getModel().setRangeProperties(0, 0, 0, sizeGuess, false); |
| } |
| |
| //TODO @Override |
| public void updateProgress(int fullLength) |
| { |
| if (progress!=null) |
| progress.getModel().setValue(fullLength); |
| } |
| |
| //TODO @Override |
| public void setPokeThread(Thread currentThread) |
| { |
| this.pokeThread = currentThread; |
| } |
| |
| //TODO @Override |
| public boolean shouldStopIt() |
| { |
| return stopIt; |
| } |
| |
| //TODO @Override |
| public void showErrorDialog(String name, String url) { |
| JEditorPane ep = new JEditorPane("text/html", |
| "<html>" + |
| owner+" was unable to download required library "+name + |
| "<br>Check your internet connection and try restarting or download it manually from" + |
| "<br><a href=\""+url+"\">"+url+"</a> and put it in your mods folder" + |
| "</html>"); |
| |
| ep.setEditable(false); |
| ep.setOpaque(false); |
| ep.addHyperlinkListener(new HyperlinkListener() |
| { |
| //TODO @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 |
| { |
| //TODO @Override |
| public void resetProgress(int sizeGuess) |
| { |
| } |
| |
| //TODO @Override |
| public void setPokeThread(Thread currentThread) |
| { |
| } |
| |
| //TODO @Override |
| public void updateProgress(int fullLength) |
| { |
| } |
| |
| //TODO @Override |
| public boolean shouldStopIt() |
| { |
| return false; |
| } |
| |
| //TODO @Override |
| public void updateProgressString(String string, Object... data) |
| { |
| } |
| |
| //TODO @Override |
| public Object makeDialog() |
| { |
| return null; |
| } |
| |
| //TODO @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<String, Dependancy> depMap = new HashMap<String, Dependancy>(); |
| private HashSet<String> depSet = new HashSet<String>(); |
| |
| 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<String> 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<File> modFiles() |
| { |
| List<File> list = new LinkedList<File>(); |
| 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<String> reserved = Arrays.asList("repo", "file", "class", "dev", "coreLib"); |
| for(Entry<JsonStringNode, JsonNode> 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)}; |
| } |
| |
| //TODO @Override |
| public String[] getASMTransformerClass() |
| { |
| return null; |
| } |
| |
| //TODO @Override |
| public String getModContainerClass() |
| { |
| return null; |
| } |
| |
| //TODO @Override |
| public String getSetupClass() |
| { |
| return getClass().getName(); |
| } |
| |
| //TODO @Override |
| public void injectData(Map<String, Object> data) |
| { |
| } |
| |
| //TODO @Override |
| public Void call() |
| { |
| load(); |
| |
| return null; |
| } |
| |
| //TODO @Override |
| public String[] getLibraryRequestClass() |
| { |
| return null; |
| } |
| } |