package acl2s.plugin.dialogs;
	
	import java.io.File;
	import java.io.FilenameFilter;
	import java.lang.reflect.InvocationTargetException;
	import java.util.ArrayList;
	import java.util.Arrays;
	import java.util.LinkedList;
	import java.util.List;
	
	import org.eclipse.core.runtime.IProgressMonitor;
	import org.eclipse.jface.dialogs.ProgressMonitorDialog;
	import org.eclipse.jface.operation.IRunnableWithProgress;
	import org.eclipse.swt.SWT;
	import org.eclipse.swt.events.SelectionEvent;
	import org.eclipse.swt.events.SelectionListener;
	import org.eclipse.swt.graphics.Rectangle;
	import org.eclipse.swt.layout.GridData;
	import org.eclipse.swt.layout.GridLayout;
	import org.eclipse.swt.widgets.Button;
	import org.eclipse.swt.widgets.Display;
	import org.eclipse.swt.widgets.Group;
	import org.eclipse.swt.widgets.Label;
	import org.eclipse.swt.widgets.Shell;
	import org.peterd.util.Misc;
	
	import acl2s.plugin.Acl2sPlugin;
	import acl2s.plugin.prefs.Acl2Prefs;
	
	public class RemoveOldStuffDialog implements SelectionListener {
		protected static class OldStuff {
			public OldStuff(File base, String desc) {
				this.base = base.getAbsoluteFile();
				this.desc = desc;
			}
			
			public final File base;
			public final String desc;
			
			public boolean equals(Object obj) {
				if (!(obj instanceof OldStuff)) return false;
				OldStuff that = (OldStuff) obj;
				return this.base.equals(that.base);
			}
			
			public int hashCode() {
				return base.hashCode();
			}
			
			
			private int fileCount = 0;
			
			public int countFiles() {
				if (fileCount == 0) {
					fileCount = countFiles(base);
				}
				return fileCount;
			}
			
			public static int countFiles(File f) {
				if (f.isDirectory()) {
					int count = 1;
					for (File f2 : f.listFiles()) {
						count += countFiles(f2);
					}
					return count;
				} else {
					return 1;
				}
			}
			
			
			private long totalSize = 0;
			
			public long totalSize() {
				if (totalSize == 0) {
					totalSize = totalSize(base);
				}
				return totalSize;
			}
			
			public static long totalSize(File f) {
				if (f.isDirectory()) {
					long size = 4096; // roughly
					for (File f2 : f.listFiles()) {
						size += totalSize(f2);
					}
					return size;
				} else {
					return f.length();
				}
			}
			
			
			public void remove(IProgressMonitor m) throws InterruptedException {
				removeFiles(base,m);
			}
			
			public static boolean removeFiles(File f, IProgressMonitor m) throws InterruptedException {

				if (Thread.interrupted() || m.isCanceled()) throw new InterruptedException();

				if (f.isDirectory()) {
					for (File f2 : f.listFiles()) {
						removeFiles(f2, m);
					}
				}
				boolean b = f.delete();

				if (m != null) m.worked(1);
				return b;
			}
			
			private String str = null;
			
			public String toString() {
				if (str == null) {
					int count = countFiles();
					StringBuilder b = new StringBuilder(base.getAbsolutePath().length() + 80);
					b.append(base.getAbsolutePath());
					b.append("\n");
					if (desc != null) {
						b.append(desc);
						b.append("\n");
					}
					if (count > 1) {
						b.append(countFiles());
						b.append(" files");
						b.append("\n");
					}
					long size = totalSize();
					if (size < 10000) {
						b.append(size);
						b.append(" bytes");
					} else if (size < 10240000) {
						b.append((size + 512) / 1024);
						b.append(" KB");
					} else {
						b.append((size / 1024 + 512) / 1024);
						b.append(" MB");
					}
					str = b.toString();
				}
				return str;
			}
		}
		
		/* This is really old now.  No sense in checking. */
		/*
		static final String[] oldHomeImageFilenames = {
			"acl2s.exe",
			"acl2s.image",
			"acl2s-sbcl.exe",
			"acl2s-dppccl.exe",
			"acl2s-dx86cl64.exe",
			"hiddencon.exe",
			"sendbreak.exe",
			"sendctrlc.exe"
		};
		*/
		
		static class UnverFilter implements FilenameFilter {
			final String basePlus;
			final int len;
			
			public UnverFilter(String base) {
				this.basePlus = base.toLowerCase() + "_";
				this.len = this.basePlus.length();
			}
			
			public boolean accept(File dir, String name) {
				return name.length() > len &&
					   name.toLowerCase().startsWith(basePlus) &&
				       Character.isDigit(name.charAt(len));
			}
		}
		
		private static final File acl2sDir = Acl2sPlugin.getAcl2sBundleDir();
		private static final File hooksDir = Acl2sPlugin.getAcl2sHooksDir();
		private static final File imageDir = Acl2Prefs.getAcl2ImageDir();
		
		private static void addAllButNewest(File[] files, ArrayList<OldStuff> v, String desc) {
			if (files == null || files.length <= 1) return;
			Arrays.sort(files);
			for (int i = 0; i < files.length - 1; i++) {
				File f = files[i];
				
				if (Misc.equal(f, acl2sDir) ||
					Misc.equal(f, hooksDir) ||
					Misc.equal(f, imageDir)) continue; // don't remove current ones
				
				v.add(new OldStuff(files[i], desc));
			}
		}
		
		protected static OldStuff[] getOldStuff() {
			ArrayList<OldStuff> v = new ArrayList<OldStuff>();
			
			/* This is really old now.  No sense in checking. */
			/*
			File home = new File(System.getProperty("user.home"));
			for (String name : oldHomeImageFilenames) {
				File f = new File(home,name);
				if (f.exists()) {
					v.add(new OldStuff(f,"Likely from old ACL2 image downloaded by old ACL2s"));
				}
			}
			*/
			
			try {
				File pluginsDir = acl2sDir.getParentFile();
				File featuresDir = new File(pluginsDir.getParentFile(), "features");
				
				if (featuresDir.isDirectory()) {
					for (File f : pluginsDir.listFiles(new UnverFilter("a2s"))) {
						v.add(new OldStuff(f, "Old ACL2s plugin"));
					}
					for (File f : featuresDir.listFiles(new UnverFilter("a2s"))) {
						v.add(new OldStuff(f, "Old ACL2s feature"));
					}
	
					addAllButNewest(pluginsDir.listFiles(new UnverFilter("acl2s")),
							v, "Old ACL2s plugin");
					addAllButNewest(featuresDir.listFiles(new UnverFilter("acl2s")),
							v, "Old ACL2s feature");
	
					addAllButNewest(pluginsDir.listFiles(new UnverFilter("acl2s_hooks")),
							v, "Old ACL2s hooks plugin");
	
					addAllButNewest(pluginsDir.listFiles(new UnverFilter("acl2s_modes")),
							v, "Old ACL2s modes plugin");
	
					addAllButNewest(pluginsDir.listFiles(new UnverFilter(Acl2Prefs.acl2BundleName)),
							v, "Old ACL2 image plugin");
					addAllButNewest(featuresDir.listFiles(new UnverFilter(Acl2Prefs.acl2BundleName)),
							v, "Old ACL2 image feature");
				}
			} catch (Exception e) {
				Acl2sPlugin.logWarning("Error while looking for old stuff", e);
			}
	
			return v.toArray(new OldStuff[v.size()]);
		}
		
		public static void go(Shell parent) {
			OldStuff[] oldStuff = getOldStuff();
			if (oldStuff.length > 0) {
				RemoveOldStuffDialog d = new RemoveOldStuffDialog(parent, oldStuff);
				final List<OldStuff> lst = d.open();
				if (lst.size() > 0) {
					try {
						IRunnableWithProgress op = new IRunnableWithProgress() {
							public void run(IProgressMonitor monitor)
							throws InterruptedException {
								int totalWork = 0;
								for (OldStuff o : lst) {
									totalWork += o.countFiles();
								}
								monitor.beginTask("Removing old ACL2s files", totalWork);
	
								for (OldStuff o : lst) {
									o.remove(monitor);
								}
							}
						};
						new ProgressMonitorDialog(parent).run(true, true, op);
					} catch (InvocationTargetException e) {
						// shouldn't happen
					} catch (InterruptedException e) {
						// oh well
					}
				}
			}
		}
		
		protected final Shell parent;
		protected final int style;
		protected final String text;
		protected final Display display;
		
		protected final OldStuff[] actions;
		protected final List<OldStuff> chosenActions = new LinkedList<OldStuff>();
		protected Shell sh;
		
		protected Button[] actionButtons;
	
		protected Button okButton;
		
		public RemoveOldStuffDialog(Shell parent, OldStuff[] actions) {
			this.parent = parent;
			this.actions = actions;
			
			this.display = parent == null ? Acl2sPlugin.getDisplay() : parent.getDisplay();
			this.text = "Select old ACL2s files/versions to remove";
			this.style = SWT.APPLICATION_MODAL | SWT.DIALOG_TRIM;
		}
	
		public List<OldStuff> open() {
			sh = new Shell(parent, style);
			sh.setText(text);
			sh.setLayout(new GridLayout(1,true));
			
			GridData gd;
					
			Label l = new Label(sh, SWT.WRAP);
			l.setText("The following old ACL2s components were found on your system.  If you would like to remove them to free the space, select them below.");
			gd = new GridData();
			gd.horizontalAlignment = SWT.FILL;
			//gd.horizontalSpan = 2;
			l.setLayoutData(gd);
	
			Group modeGroup = new Group(sh, 0);
			//modeGroup.setText("Select a mode for " + src + ":");
			gd = new GridData();
			gd.horizontalAlignment = SWT.FILL;
	
			modeGroup.setLayoutData(gd);
			modeGroup.setLayout(new GridLayout(2,false));
			actionButtons = new Button[actions.length];
			for (int i = 0; i < actionButtons.length; i++) {
				actionButtons[i] = new Button(modeGroup, SWT.CHECK);
				
				l = new Label(modeGroup,0);
				l.setText(actions[i].toString());
				gd = new GridData();
				gd.horizontalAlignment = SWT.FILL;
				l.setLayoutData(gd);
				
				//actionButtons[i].setText();
				//actionButtons[i].setEnabled(true);
				//actionButtons[i].setSelection(chosenActions.contains(actions[i]));
			}
	
			okButton = new Button(sh, SWT.PUSH);
			okButton.setText("OK");
			gd = new GridData(GridData.FILL_HORIZONTAL);
			okButton.setLayoutData(gd);
			okButton.addSelectionListener(this);
			
			/*
			Label l = new Label(sh,0);
			l.setText("Note: Any change will take effect when session is restarted.");
			gd = new GridData(GridData.FILL_HORIZONTAL);
			gd.horizontalSpan = 2;
			l.setLayoutData(gd);
			*/
			
			sh.pack();
	
			Rectangle bounds = parent == null ? display.getBounds() : parent.getBounds();
			Rectangle rect = sh.getBounds ();
			int x = bounds.x + (bounds.width - rect.width) / 2;
			int y = bounds.y + (bounds.height - rect.height) / 2;
			sh.setLocation (x, y);
			
			sh.open();
			
			while (!sh.isDisposed()) {
				if (!display.readAndDispatch()) {
					display.sleep();
				}
			}
			return chosenActions;
		}
	
		public void widgetSelected(SelectionEvent e) {
			if (e.getSource() == okButton) {
				chosenActions.clear();
				for (int i = 0; i < actionButtons.length; i++) {
					if (actionButtons[i].getSelection()) {
						chosenActions.add(actions[i]);
					}
				}
			}
			sh.close();
		}
	
		public void widgetDefaultSelected(SelectionEvent e) { }
	}