path: root/src/lombok/installer/EclipseLocation.java

package lombok.installer;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.List;
import java.util.jar.JarFile;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.zip.ZipEntry;

import javax.swing.JFrame;
import javax.swing.JOptionPane;

final class EclipseLocation {
	private static final String OS_NEWLINE;
	
	static {
		String os = System.getProperty("os.name", "");
		
		if ( "Mac OS".equals(os) ) OS_NEWLINE = "\r";
		else if ( os.toLowerCase().contains("windows") ) OS_NEWLINE = "\r\n";
		else OS_NEWLINE = "\n";
	}
	
	private final File path;
	private volatile boolean hasLombok;
	boolean selected = true;
	
	final class NotAnEclipseException extends Exception {
		private static final long serialVersionUID = 1L;
		
		public NotAnEclipseException(String message, Throwable cause) {
			super(message, cause);
		}
		
		public void showDialog(JFrame appWindow) {
			JOptionPane.showMessageDialog(appWindow, getMessage(), "Cannot configure eclipse installation", JOptionPane.WARNING_MESSAGE);
		}
	}
	
	EclipseLocation(String path) throws NotAnEclipseException {
		if ( path == null ) throw new NullPointerException("path");
		File p = new File(path);
		if ( !p.exists() ) throw new NotAnEclipseException("File does not exist: " + path, null);
		
		final String execName = EclipseFinder.getEclipseExecutableName();
		if ( p.isDirectory() ) {
			for ( File f : p.listFiles() ) {
				if ( f.getName().equalsIgnoreCase(execName) ) {
					p = f;
					break;
				}
			}
		}
		
		this.path = p;
		try {
			this.hasLombok = checkForLombok();
		} catch ( IOException e ) {
			throw new NotAnEclipseException(
					"I can't read the configuration file of the eclipse installed at " + this.path.getAbsolutePath() + "\n" +
					"You may need to run this installer with root privileges if you want to modify that eclipse.", e);
		}
	}
	
	public int hashCode() {
		return path.hashCode();
	}
	
	public boolean equals(Object o) {
		if ( !(o instanceof EclipseLocation) ) return false;
		return ((EclipseLocation)o).path.equals(path);
	}
	
	public String getPath() {
		return path.getAbsolutePath();
	}
	
	public boolean hasLombok() {
		return hasLombok;
	}
	
	private List<File> getTargetDirs() {
		return Arrays.asList(path.getParentFile(), new File(new File(path, "Contents"), "MacOS"));
	}
	
	private boolean checkForLombok() throws IOException {
		for ( File targetDir : getTargetDirs() ) {
			if ( checkForLombok0(targetDir) ) return true;
		}
		
		return false;
	}
	
	private final Pattern JAVA_AGENT_LINE_MATCHER = Pattern.compile(
			"^\\-javaagent\\:.*lombok.*\\.jar$", Pattern.CASE_INSENSITIVE);
	
	private final Pattern BOOTCLASSPATH_LINE_MATCHER = Pattern.compile(
	"^\\-Xbootclasspath\\/a\\:(.*lombok.*\\.jar.*)$", Pattern.CASE_INSENSITIVE);
	
	private boolean checkForLombok0(File dir) throws IOException {
		File iniFile = new File(dir, "eclipse.ini");
		if ( !iniFile.exists() ) return false;
		FileInputStream fis = new FileInputStream(iniFile);
		try {
			BufferedReader br = new BufferedReader(new InputStreamReader(fis));
			String line;
			while ( (line = br.readLine()) != null ) {
				if ( JAVA_AGENT_LINE_MATCHER.matcher(line.trim()).matches() ) return true;
			}
			
			return false;
		} finally {
			fis.close();
		}
	}
	
	public class UninstallException extends Exception {
		private static final long serialVersionUID = 1L;
		
		public UninstallException(String message, Throwable cause) {
			super(message, cause);
		}
	}
	
	public void uninstall() throws UninstallException {
		for ( File dir : getTargetDirs() ) {
			File lombokJar = new File(dir, "lombok.jar");
			if ( lombokJar.exists() ) {
				if ( !lombokJar.delete() ) throw new UninstallException(
						"Can't delete " + lombokJar.getAbsolutePath() +
						" - perhaps the installer does not have the access rights to do so.",
						null);
			}
			
			File agentJar = new File(dir, "lombok.eclipse.agent.jar");
			if ( agentJar.exists() ) {
				if ( !agentJar.delete() ) throw new UninstallException(
						"Can't delete " + agentJar.getAbsolutePath() +
						" - perhaps the installer does not have the access rights to do so.",
						null);
			}
			
			File iniFile = new File(dir, "eclipse.ini");
			StringBuilder newContents = new StringBuilder();
			if ( iniFile.exists() ) {
				try {
					FileInputStream fis = new FileInputStream(iniFile);
					try {
						BufferedReader br = new BufferedReader(new InputStreamReader(fis));
						String line;
						while ( (line = br.readLine()) != null ) {
							if ( JAVA_AGENT_LINE_MATCHER.matcher(line).matches() ) continue;
							Matcher m = BOOTCLASSPATH_LINE_MATCHER.matcher(line);
							if ( m.matches() ) {
								StringBuilder elemBuilder = new StringBuilder();
								elemBuilder.append("-Xbootclasspath/a:");
								boolean first = true;
								for ( String elem : m.group(1).split(Pattern.quote(File.pathSeparator)) ) {
									if ( elem.toLowerCase().endsWith("lombok.jar") ) continue;
									if ( elem.toLowerCase().endsWith("lombok.eclipse.agent.jar") ) continue;
									if ( first ) first = false;
									else elemBuilder.append(File.pathSeparator);
									elemBuilder.append(elem);
								}
								if ( !first ) newContents.append(elemBuilder.toString()).append(OS_NEWLINE);
								continue;
							}
							
							newContents.append(line).append(OS_NEWLINE);
						}
						
					} finally {
						fis.close();
					}
					
					FileOutputStream fos = new FileOutputStream(iniFile);
					try {
						fos.write(newContents.toString().getBytes());
					} finally {
						fos.close();
					}
				} catch ( IOException e ) {
					throw new UninstallException("Cannot uninstall lombok from " + path.getAbsolutePath() +
							" probably because this installer does not have the access rights to do so.", e);
				}
			}
		}
	}
	
	public class InstallException extends Exception {
		private static final long serialVersionUID = 1L;
		
		public InstallException(String message, Throwable cause) {
			super(message, cause);
		}
	}
	
	public void install() throws InstallException {
		for ( File dir : getTargetDirs() ) {
			File iniFile = new File(dir, "eclipse.ini");
			StringBuilder newContents = new StringBuilder();
			if ( iniFile.exists() ) {
				File lombokJar = new File(iniFile.getParentFile(), "lombok.jar");
				File agentJar = new File(iniFile.getParentFile(), "lombok.eclipse.agent.jar");
				
				File ourJar = EclipseFinder.findOurJar();
				byte[] b = new byte[524288];
				boolean readSucceeded = false;
				boolean installSucceeded = false;
				try {
					JarFile jar = new JarFile(ourJar);
					
					try {
						ZipEntry entry = jar.getEntry("lombok.eclipse.agent.jar");
						InputStream in = jar.getInputStream(entry);
						FileOutputStream out = new FileOutputStream(agentJar);
						try {
							while ( true ) {
								int r = in.read(b);
								if ( r == -1 ) break;
								readSucceeded = true;
								out.write(b, 0, r);
							}
						} finally {
							out.close();
						}
					} finally {
						jar.close();
						
					}
					
					FileOutputStream out = new FileOutputStream(lombokJar);
					InputStream in = new FileInputStream(ourJar);
					try {
						while ( true ) {
							int r = in.read(b);
							if ( r == -1 ) break;
							out.write(b, 0, r);
						}
					} finally {
						out.close();
					}
				} catch ( IOException e ) {
					try {
						lombokJar.delete();
						agentJar.delete();
					} catch ( Throwable ignore ) {}
					if ( !readSucceeded ) throw new InstallException("I can't read my own jar file. I think you've found a bug in this installer! I suggest you restart it " +
							"and use the 'what do I do' link, to manually install lombok. And tell us about this. Thanks!", e);
					throw new InstallException("I can't write to your eclipse directory, probably because this installer does not have the access rights.", e);
				}
				
				try {
					FileInputStream fis = new FileInputStream(iniFile);
					try {
						BufferedReader br = new BufferedReader(new InputStreamReader(fis));
						String line;
						while ( (line = br.readLine()) != null ) {
							if ( JAVA_AGENT_LINE_MATCHER.matcher(line).matches() ) continue;
							Matcher m = BOOTCLASSPATH_LINE_MATCHER.matcher(line);
							if ( m.matches() ) {
								StringBuilder elemBuilder = new StringBuilder();
								elemBuilder.append("-Xbootclasspath/a:");
								boolean first = true;
								for ( String elem : m.group(1).split(Pattern.quote(File.pathSeparator)) ) {
									if ( elem.toLowerCase().endsWith("lombok.jar") ) continue;
									if ( elem.toLowerCase().endsWith("lombok.eclipse.agent.jar") ) continue;
									if ( first ) first = false;
									else elemBuilder.append(File.pathSeparator);
									elemBuilder.append(elem);
								}
								if ( !first ) newContents.append(elemBuilder.toString()).append(OS_NEWLINE);
								continue;
							}
							
							newContents.append(line).append(OS_NEWLINE);
						}
						
					} finally {
						fis.close();
					}
					
					newContents.append("-javaagent:lombok.jar").append(OS_NEWLINE);
					newContents.append("-Xbootclasspath/a:lombok.jar" + File.pathSeparator + "lombok.eclipse.agent.jar").append(OS_NEWLINE);
					
					FileOutputStream fos = new FileOutputStream(iniFile);
					try {
						fos.write(newContents.toString().getBytes());
					} finally {
						fos.close();
					}
					installSucceeded = true;
					
				} catch ( IOException e ) {
					throw new InstallException("Cannot install lombok at " + path.getAbsolutePath() +
							" probably because this installer does not have the access rights to do so.", e);
				} finally {
					if ( !installSucceeded ) try {
						lombokJar.delete();
						agentJar.delete();
					} catch ( Throwable ignore ) {}
				}
			}
		}
	}
}