demos/org.eclipse.fx.demo.contacts.app/src/org/eclipse/fx/demo/contacts/model/VCardContactsResource.java

/*******************************************************************************
 * Copyright (c) 2009, 2010 Siemens AG and others.
 * 
 * All rights reserved. This program and the accompanying materials 
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html.
 * 
 * Contributors:
 *     Kai T????dter - initial implementation
 ******************************************************************************/

package org.eclipse.fx.demo.contacts.model;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FilenameFilter;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.Enumeration;

import org.eclipse.core.runtime.FileLocator;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.Platform;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.resource.impl.ResourceImpl;
import org.eclipse.fx.demo.contacts.BundleActivatorImpl;
import org.eclipse.fx.demo.contacts.Contact;
import org.eclipse.fx.demo.contacts.ContactsFactory;
import org.eclipse.fx.demo.contacts.Group;
import org.osgi.framework.Bundle;


public class VCardContactsResource extends ResourceImpl {

	public VCardContactsResource() {
		super(URI.createPlatformResourceURI("org.eclipse.fx.demo.contacts/vcards", false));
		
		try {
			Group rootGroup = ContactsFactory.eINSTANCE.createGroup();
			for (File file : getContacts()) {
				Contact contact = readFromVCard(file.getAbsolutePath());
				rootGroup.getContacts().add(contact);
			}
			getContents().add(rootGroup);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	private File[] getContacts() throws Exception {
		File[] localContacts = getLocalContacts();
		if (localContacts.length == 0) {
			IPath path = BundleActivatorImpl.getInstance().getStateLocation();
			byte[] buffer = new byte[8192];
			Bundle bundle = Platform.getBundle("org.eclipse.fx.demo.contacts.app"); //$NON-NLS-1$

			for (Enumeration<?> contacts = getStoredContacts(); contacts.hasMoreElements();) {
				String bundlePath = (String) contacts.nextElement();
				if (!bundlePath.endsWith(".vcf")) { //$NON-NLS-1$
					continue;
				}

				InputStream inputStream = FileLocator.openStream(bundle, new Path(bundlePath), false);
				FileOutputStream outputStream = new FileOutputStream(path.append(
						bundlePath.substring(bundlePath.indexOf('/') + 1)).toFile());

				int read = inputStream.read(buffer);
				while (read != -1) {
					outputStream.write(buffer, 0, read);
					read = inputStream.read(buffer);
				}

				inputStream.close();
				outputStream.close();
			}

			return getLocalContacts();
		}
		return localContacts;
	}

	private File[] getLocalContacts() {
		IPath path = BundleActivatorImpl.getInstance().getStateLocation();
		File directory = path.toFile();
		return directory.listFiles(new FilenameFilter() {
			public boolean accept(File dir, String name) {
				return name.endsWith(".vcf"); //$NON-NLS-1$
			}
		});
	}

	private Enumeration<?> getStoredContacts() throws Exception {
		Bundle bundle = Platform.getBundle("org.eclipse.fx.demo.contacts.app"); //$NON-NLS-1$
		return bundle.getEntryPaths("vcards"); //$NON-NLS-1$
	}

	/**
	 * Reads a Contact from a VCard. This method cannot parse a generic VCard,
	 * but can only parse VCards created with Microsoft Outlook. The intention
	 * is not to provide a generic VCard reader but an easy way to get contact
	 * data (including pictures) in the repository.
	 * 
	 * @param fileName
	 *            the vcard file
	 * @return the created Contact
	 */
	public Contact readFromVCard(String fileName) {
		Contact contact = ContactsFactory.eINSTANCE.createContact();
		contact.setSourceFile(fileName);
		BufferedReader bufferedReader = null;
		String charSet = "Cp1252";

		/*
		 * first try to guess the char set (currently not working under some
		 * JVMs
		 */

		/*
		 * try { bufferedReader = new BufferedReader(new InputStreamReader( new
		 * FileInputStream(fileName))); String line; while ((line =
		 * bufferedReader.readLine()) != null) { int index =
		 * line.indexOf("CHARSET="); if (index != -1) { int endIndex = index +
		 * 8; while (line.charAt(endIndex) != ':' && line.charAt(endIndex) !=
		 * ';') { endIndex += 1; } charSet = line.substring(index + 8,
		 * endIndex); break; } } } catch (FileNotFoundException e) { // TODO
		 * Auto-generated catch block e.printStackTrace();
		 * 
		 * } catch (IOException e) { // TODO Auto-generated catch block
		 * e.printStackTrace(); } finally { try { if (bufferedReader != null) {
		 * bufferedReader.close(); } } catch (IOException e) { // TODO
		 * Auto-generated catch block e.printStackTrace(); } }
		 */

		// Then parse the vCard
		try {
			InputStreamReader inputStreamReader = new InputStreamReader(new FileInputStream(fileName), charSet);
			bufferedReader = new BufferedReader(inputStreamReader);
			String line;
			String value;
			while ((line = bufferedReader.readLine()) != null) {
				value = getVCardValue(line, "N");
				if (value != null) {
					String[] result = value.split(";");

					if (result.length > 0) {
						contact.setLastName(result[0]);
					}
					if (result.length > 1) {
						contact.setFirstName(result[1]);
					}
					if (result.length > 2) {
						contact.setMiddleName(result[2]);
					}
					if (result.length > 3) {
						contact.setTitle(result[3]);
					}
					continue;
				}
				value = getVCardValue(line, "TEL;WORK");
				if (value != null) {
					contact.setPhone(value);
					continue;
				}
				value = getVCardValue(line, "TEL;CELL");
				if (value != null) {
					contact.setMobile(value);
					continue;
				}
				value = getVCardValue(line, "ADR;WORK");
				if (value != null) {
					String[] result = value.split(";");

					if (result.length > 2) {
						contact.setStreet(result[2]);
					}
					if (result.length > 3) {
						contact.setCity(result[3]);
					}
					if (result.length > 4) {
						contact.setState(result[4]);
					}
					if (result.length > 5) {
						contact.setZip(result[5]);
					}
					if (result.length > 6) {
						contact.setCountry(result[6]);
					}
					continue;
				}
				value = getVCardValue(line, "EMAIL;PREF;INTERNET");
				if (value != null) {
					contact.setEmail(value);
					continue;
				}
				value = getVCardValue(line, "URL;WORK");
				if (value != null) {
					contact.setWebPage(value);
					continue;
				}
				value = getVCardValue(line, "ORG");
				if (value != null) {
					contact.setCompany(value);
					continue;
				}
				value = getVCardValue(line, "TITLE");
				if (value != null) {
					contact.setJobTitle(value);
					continue;
				}
				value = getVCardValue(line, "NOTE");
				if (value != null) {
					contact.setNote(value);
					continue;
				}
				value = getVCardValue(line, "PHOTO;TYPE=JPEG;ENCODING=BASE64");
				if (value != null) {
					line = bufferedReader.readLine();
					StringBuilder builder = new StringBuilder();
					while (line != null && line.length() > 0 && line.charAt(0) == ' ') {
						builder.append(line.trim());
						line = bufferedReader.readLine();
					}
					String jpegString = builder.toString();

					contact.setJpegString(jpegString);
					continue;
				}
			}
		} catch (FileNotFoundException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();

		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} finally {
			try {
				if (bufferedReader != null) {
					bufferedReader.close();
				}
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
		}
		return contact;
	}

	private String getVCardValue(String line, String token) {
		if (line.startsWith(token + ":") || line.startsWith(token + ";")) {
			String value = line.substring(line.indexOf(":") + 1);
			return value;
		}
		return null;
	}

}