update/org.eclipse.update.ui/src/org/eclipse/update/internal/ui/model/MyComputer.java

package org.eclipse.update.internal.ui.model;
/*
 * (c) Copyright IBM Corp. 2000, 2001.
 * All Rights Reserved.
 */

import java.net.URL;
import org.eclipse.update.core.*;
import org.eclipse.core.runtime.*;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.ui.views.properties.*;
import org.eclipse.ui.model.*;
import java.util.*;
import org.eclipse.update.internal.ui.*;
import org.eclipse.update.internal.ui.search.*;

import java.io.*;
import org.eclipse.update.internal.ui.UpdateUIPlugin;
import org.eclipse.swt.custom.BusyIndicator;
import org.eclipse.update.configuration.LocalSystemInfo;

public class MyComputer extends UIModelObject implements IWorkbenchAdapter {
	private static final String KEY_LABEL = "MyComputer.label";
	private Object[] children;
	public MyComputer() {
	}

	public Object getAdapter(Class adapter) {
		if (adapter.equals(IWorkbenchAdapter.class)) {
			return this;
		}
		return super.getAdapter(adapter);
	}

	public String getName() {
		return UpdateUIPlugin.getResourceString(KEY_LABEL);
	}

	public String toString() {
		return getName();
	}

	/**
	 * @see IWorkbenchAdapter#getChildren(Object)
	 */
	public Object[] getChildren(Object parent) {
		BusyIndicator.showWhile(UpdateUIPlugin.getActiveWorkbenchShell().getDisplay(), new Runnable() {
			public void run() {
				File [] roots = getRoots();
				if (roots.length > 0) {
					children = new MyComputerDirectory[roots.length];
					for (int i = 0; i < children.length; i++) {
						children[i] = new MyComputerDirectory(MyComputer.this, roots[i], true);
					}
				} else
					children = new Object[0];
			}
		});
		return children;
	}
	
	public static File [] getRoots() {
		File [] roots;
		String [] rootPaths = LocalSystemInfo.listMountPoints();
		if (rootPaths!=null) {
			roots = new File[rootPaths.length];
			for (int i=0; i<roots.length; i++) {
				roots[i] = new File(rootPaths[i]);
			}
		}
		else {
			// fallback
			roots = File.listRoots();
			if (roots.length == 1) {
				// just one root - skip it
				File root = roots[0];
				roots = root.listFiles();
			}
		}
		return roots;
	}

	/**
	 * @see IWorkbenchAdapter#getImageDescriptor(Object)
	 */
	public ImageDescriptor getImageDescriptor(Object obj) {
		return UpdateUIPluginImages.DESC_COMPUTER_OBJ;
	}

	/**
	 * @see IWorkbenchAdapter#getLabel(Object)
	 */
	public String getLabel(Object obj) {
		return getName();
	}

	/**
	 * @see IWorkbenchAdapter#getParent(Object)
	 */
	public Object getParent(Object arg0) {
		return null;
	}
	
	public void collectSites(Vector sites, MyComputerSearchSettings settings, IProgressMonitor monitor) {
		File [] drives = MyComputer.getRoots();
		for (int i=0; i<drives.length; i++) {
			File drive = drives[i];
			if (monitor.isCanceled()) return;
			DriveSearchSettings ds = settings.getDriveSettings(drive.getPath());
			if (ds.isChecked()) {
				collectSites(drive, sites, ds, monitor);
			}
		}
	}
	
	private void collectSites(File dir, Vector sites, DriveSearchSettings driveSettings, IProgressMonitor monitor) {
		File [] children = dir.listFiles();
		if (children==null) return;
		
		for (int i=0; i<children.length; i++) {
			File child = children[i];
			if (monitor.isCanceled()) return;
			if (child.isDirectory()) {
				monitor.subTask(child.getPath());				
				SiteBookmark bookmark = MyComputerDirectory.createSite(child);
				if (bookmark!=null) {
					ISite site = bookmark.getSite(false);
					if (site!=null) sites.add(bookmark);
				}
				else 
					collectSites(child, sites, driveSettings, monitor);
			}
		}
	}
}