Prolific Programmer on the Internets!

package com.prolificprogrammer.dirutils;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.channels.FileChannel;
import java.util.ArrayList;
import java.util.List;
import java.util.ListIterator;
import java.util.zip.GZIPOutputStream;

import javax.swing.JFileChooser;
import javax.swing.JMenu;
import javax.swing.JMenuItem;

public class DirUtils {
	private static final int ZIP = 0;
	private static final int ALPHABETISE = 1;
	private static final int EXTENSIONISE = 2;
	
	public DirUtils() {
		// TODO Auto-generated constructor stub
	}
	File folder;

	public void sort(int alphabetise2) {
		File sortedFilesFolder = new File(folder.getAbsolutePath()+File.pathSeparator+"sorted");
		sortedFilesFolder.mkdir();
		List  files = this.getFiles(folder, null);
		ListIterator fileobj = files.listIterator();
		while (fileobj.hasNext()) {
			if (alphabetise2 == ALPHABETISE) {
				File currentFile = fileobj.next();
				String storageBucket = new String(new Character(currentFile.getName().charAt(1)).toString());
				new File(sortedFilesFolder.getAbsolutePath()+storageBucket).mkdir();
				FileChannel ic = null;
				try {
					ic = new FileInputStream(currentFile).getChannel();
				} catch (FileNotFoundException e4) {
					e4.printStackTrace();
				}
				FileChannel oc = null;
				try {
					oc = new FileOutputStream(sortedFilesFolder.getAbsolutePath()+File.pathSeparator+storageBucket+File.pathSeparator+currentFile.getName()).getChannel();
				} catch (FileNotFoundException e3) {
					e3.printStackTrace();
				}
				try {
					ic.transferTo(0, ic.size(), oc);
				} catch (IOException e2) {
					e2.printStackTrace();
				}
				try {
					oc.close();
				} catch (IOException e1) {
					e1.printStackTrace();
				}
				try {
					ic.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			} else { // alphabetise2 == extensionise
				File currentFile = fileobj.next();
				String storageBucket = new String(new Character(currentFile.getName().substring(currentFile.getName().lastIndexOf(".")).charAt(1)).toString());
				new File(sortedFilesFolder.getAbsolutePath()+storageBucket).mkdir();
				FileChannel ic = null;
				try {
					ic = new FileInputStream(currentFile).getChannel();
				} catch (FileNotFoundException e4) {
					e4.printStackTrace();
				}
				FileChannel oc = null;
				try {
					oc = new FileOutputStream(sortedFilesFolder.getAbsolutePath()+File.pathSeparator+storageBucket+File.pathSeparator+currentFile.getName()).getChannel();
				} catch (FileNotFoundException e3) {
					e3.printStackTrace();
				}
				try {
					ic.transferTo(0, ic.size(), oc);
				} catch (IOException e2) {
					e2.printStackTrace();
				}
				try {
					oc.close();
				} catch (IOException e1) {
					e1.printStackTrace();
				}
				try {
					ic.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

		public void zip() {
			chooseFolder();
			GZIPOutputStream gzip = null;
			try {
				gzip = new GZIPOutputStream(this.makeZipFileName(folder));
			} catch (IOException e1) {
				e1.printStackTrace();
			}
			List  files = this.getFiles(folder, null); 
			ListIterator fileobj = files.listIterator();
			while (fileobj.hasNext()) {
				File file = (File) fileobj.next();
				byte[] data = new byte[(int) file.length()];
				try {
					gzip.write(data);
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
			try {
				gzip.finish();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}

		private void chooseFolder() {
			JFileChooser fc = new JFileChooser(new File("/"));
			fc.showOpenDialog(null);
                        // TODO ensure it's a folder
			folder = fc.getSelectedFile();
		}

		private List getFiles(File folder2, List  helperList) {
			List  ret = new ArrayList();
			if (helperList != null)
				ret.addAll(helperList);
			File [] folder2files = folder2.listFiles();
			for (File f : folder2files) {
				if (f.isFile()) {
					ret.add(f);
				} else { // f.isDirectory() so recurse
					this.getFiles(f, ret);
				}
			}
			return ret;
		}

		private OutputStream makeZipFileName(File directoryName) {
			FileOutputStream ret = null;

			String rootName = directoryName.getAbsolutePath().substring(directoryName.getAbsolutePath().lastIndexOf(File.separator))+".zip";
			try {
				ret = new FileOutputStream(rootName);
			} catch (FileNotFoundException e) { 
				e.printStackTrace();
			}
			return ret;
		}
		
		public static void main (String[] args) {
			DirUtils dirutils = new DirUtils();
			// TODO JMenu with possible options 
		}
	}