FileDialog.java

/**
 * Based on Alexander Ponomarev's file dialog class (https://code.google.com/p/android-file-dialog/)
 *
 * @author Alexander Ponomarev
 * @author Peter Brinkmann (peter.brinkmann@gmail.com)
 */

package com.lamerman;

import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.TreeMap;
import java.util.regex.Pattern;

import org.puredata.android.scenes.R;

import android.app.AlertDialog;
import android.app.ListActivity;
import android.content.DialogInterface;
import android.content.Intent;
import android.os.Bundle;
import android.view.View;
import android.widget.ListView;
import android.widget.SimpleAdapter;
import android.widget.TextView;

public class FileDialog extends ListActivity {

	private static final String ITEM_KEY = "key";
	private static final String ITEM_IMAGE = "image";

	public static final String START_PATH = "START_PATH";
	public static final String RESULT_PATH = "RESULT_PATH";
	public static final String SELECT_PATTERN = "SELECT_PATTERN";
	public static final String ACCEPT_FOLDER = "ACCEPT_FOLDER";
	public static final String ACCEPT_FILE = "ACCEPT_FILE";

	private Intent intent;
	private List<String> item = null;
	private List<String> path = null;
	private String root = "/";
	private TextView myPath;
	private ArrayList<HashMap<String, Object>> mList;

	private String parentPath;
	private String currentPath = root;
	private boolean acceptFolder;
	private boolean acceptFile;
	private Pattern pattern;

	private File selectedFile;
	private HashMap<String, Integer> lastPositions = new HashMap<String, Integer>();

	@Override
	public void onCreate(Bundle savedInstanceState) {
		super.onCreate(savedInstanceState);
		intent = getIntent();
		setResult(RESULT_CANCELED, intent);
		setContentView(R.layout.file_dialog_main);
		myPath = (TextView) findViewById(R.id.path);
		String startPath = intent.getStringExtra(START_PATH);
		selectedFile = new File(startPath);
		acceptFolder = intent.getBooleanExtra(ACCEPT_FOLDER, false);
		acceptFile = intent.getBooleanExtra(ACCEPT_FILE, true);
		String regExp = intent.getStringExtra(SELECT_PATTERN);
		pattern = regExp != null ? Pattern.compile(regExp) : null;
		getDir(startPath);
	}

	private void getDir(String dirPath) {
		boolean useAutoSelection = dirPath.length() < currentPath.length();
		Integer position = lastPositions.get(parentPath);
		getDirImpl(dirPath);
		if (position != null && useAutoSelection) {
			getListView().setSelection(position);
		}
	}

	private void getDirImpl(String dirPath) {
		File f = new File(dirPath);
		File[] files = f.listFiles();
		if (files == null) {
			dirPath = "/";
			f = new File(dirPath);
			files = f.listFiles();
		}
		myPath.setText(getText(R.string.location) + ": " + dirPath);
		currentPath = dirPath;
		item = new ArrayList<String>();
		path = new ArrayList<String>();
		mList = new ArrayList<HashMap<String, Object>>();
		if (!dirPath.equals(root)) {
			item.add(root);
			addItem(root, R.drawable.folder);
			path.add(root);
			item.add("../");
			addItem("../", R.drawable.folder);
			path.add(f.getParent());
			parentPath = f.getParent();
		}
		TreeMap<String, String> dirsMap = new TreeMap<String, String>();
		TreeMap<String, String> dirsPathMap = new TreeMap<String, String>();
		TreeMap<String, String> filesMap = new TreeMap<String, String>();
		TreeMap<String, String> filesPathMap = new TreeMap<String, String>();
		for (File file : files) {
			if (file.isDirectory()) {
				String dirName = file.getName();
				dirsMap.put(dirName, dirName);
				dirsPathMap.put(dirName, file.getPath());
			} else if (acceptFile && matches(file)) {
				filesMap.put(file.getName(), file.getName());
				filesPathMap.put(file.getName(), file.getPath());
			}
		}
		item.addAll(dirsMap.tailMap("").values());
		item.addAll(filesMap.tailMap("").values());
		path.addAll(dirsPathMap.tailMap("").values());
		path.addAll(filesPathMap.tailMap("").values());
		SimpleAdapter fileList = new SimpleAdapter(this, mList,
				R.layout.file_dialog_row,
				new String[] { ITEM_KEY, ITEM_IMAGE }, new int[] {
				R.id.fdrowtext, R.id.fdrowimage });
		for (String dir : dirsMap.tailMap("").values()) {
			addItem(dir, R.drawable.folder);
		}
		for (String file : filesMap.tailMap("").values()) {
			addItem(file, R.drawable.file);
		}
		fileList.notifyDataSetChanged();
		setListAdapter(fileList);
	}

	private boolean matches(File file) {
		return pattern == null || pattern.matcher(file.getName()).matches();
	}

	private void addItem(String fileName, int imageId) {
		HashMap<String, Object> item = new HashMap<String, Object>();
		item.put(ITEM_KEY, fileName);
		item.put(ITEM_IMAGE, imageId);
		mList.add(item);
	}

	@Override
	protected void onListItemClick(ListView l, View v, int position, long id) {
		selectedFile = new File(path.get(position));
		if (selectedFile.isDirectory()) {
			if (acceptFolder && pattern != null && matches(selectedFile)) {
				setResult(selectedFile);
				return;
			}
			if (selectedFile.canRead()) {
				lastPositions.put(currentPath, position);
				getDir(path.get(position));
			} else {
				new AlertDialog.Builder(this).setIcon(R.drawable.icon).setTitle(
						"[" + selectedFile.getName() + "] "
						+ getText(R.string.cant_read_folder))
						.setPositiveButton("OK",
								new DialogInterface.OnClickListener() {

							@Override
							public void onClick(DialogInterface dialog,
									int which) {

							}
						}).show();
			}
		} else if (acceptFile && matches(selectedFile)) {
			setResult(selectedFile);
			return;
		}
		v.setSelected(true);
	}

	private void setResult(File file) {
		intent.putExtra(RESULT_PATH, file.getPath());
		setResult(RESULT_OK, getIntent());
		finish();
	}
}