/*
    * Copyright (c) 2007, James Leigh All rights reserved.
    * 
    * Redistribution and use in source and binary forms, with or without
    * modification, are permitted provided that the following conditions are met:
    * 
    * - Redistributions of source code must retain the above copyright notice, this
    *   list of conditions and the following disclaimer.
    * - Redistributions in binary form must reproduce the above copyright notice,
   *   this list of conditions and the following disclaimer in the documentation
   *   and/or other materials provided with the distribution. 
   * - Neither the name of the openrdf.org nor the names of its contributors may
   *   be used to endorse or promote products derived from this software without
   *   specific prior written permission.
   * 
   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
   * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
   * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
   * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
   * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
   * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
   * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
   * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
   * POSSIBILITY OF SUCH DAMAGE.
   * 
   */
  package org.openrdf.elmo.tapestry;
  
  import info.aduna.platform.Platform;
  import info.aduna.platform.PlatformFactory;
  
  import java.io.File;
  import java.io.IOException;
  import java.net.URL;
  
  import org.apache.hivemind.ApplicationRuntimeException;
  import org.apache.hivemind.Resource;
  import org.apache.hivemind.events.RegistryShutdownListener;
  import org.openrdf.model.ValueFactory;
  import org.openrdf.repository.Repository;
  import org.openrdf.repository.RepositoryConnection;
  import org.openrdf.repository.RepositoryException;
  import org.openrdf.repository.config.RepositoryConfig;
  import org.openrdf.repository.config.RepositoryConfigException;
  import org.openrdf.repository.config.RepositoryConfigUtil;
  import org.openrdf.repository.manager.LocalRepositoryManager;
  import org.openrdf.repository.sail.config.SailRepositoryConfig;
  import org.openrdf.repository.threadproxy.ThreadProxyRepository;
  import org.openrdf.rio.RDFFormat;
  import org.openrdf.rio.RDFParseException;
  import org.openrdf.sail.memory.config.MemoryStoreConfig;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  /**
   * HiveMind aware Repository, the repository is closed when HiveMind shutdowns.
   * 
   * @author James Leigh
   * 
   */
  public class HiveMindSesameRepository implements Repository,
  		RegistryShutdownListener {
  
  	private final Logger logger = LoggerFactory
  			.getLogger(HiveMindSesameRepository.class);
  
  	private String applicationId = "elmo-tapestry";
  
  	private File dataDir;
  
  	private String repositoryId = "default";
  
  	private Resource initData;
  
  	private Repository repository;
  
  	public HiveMindSesameRepository(Repository repository) {
  		this.repository = repository;
  	}
  
  	public void setApplicationId(String applicationId) {
  		this.applicationId = applicationId;
  	}
  
  	public File getDataDir() {
  		if (repository == null)
  			return dataDir;
  		return repository.getDataDir();
  	}
  
  	public void setDataDir(File dataDir) {
  		this.dataDir = dataDir;
  		if (repository != null) {
  			repository.setDataDir(dataDir);
  		}
  	}
  
 	public void setInitData(Resource initData) {
 		this.initData = initData;
 	}
 
 	public void setRepositoryId(String repositoryId) {
 		this.repositoryId = repositoryId;
 	}
 
 	public void initialize() {
 		Platform platform = PlatformFactory.getPlatform();
 		File dir = platform.getApplicationDataDir(applicationId);
 		if (dataDir != null) {
 			dir = dataDir;
 		}
 		logger.info("Using data dir: {}", dir);
 		try {
 			repository = getRepository(dir, repositoryId);
 			if (initData != null) {
 				RepositoryConnection conn = repository.getConnection();
 				try {
 					if (conn.isEmpty()) {
 						URL data = initData.getResourceURL();
 						conn.add(data, "", RDFFormat
 								.forFileName(data.getFile()));
 					}
 				} finally {
 					conn.close();
 				}
 			}
 		} catch (IOException e) {
 			throw new ApplicationRuntimeException(e);
 		} catch (RDFParseException e) {
 			throw new ApplicationRuntimeException(e);
 		} catch (RepositoryException e) {
 			throw new ApplicationRuntimeException(e);
 		} catch (RepositoryConfigException e) {
 			throw new ApplicationRuntimeException(e);
 		}
 	}
 
 	protected Repository getRepository(File dir, String id)
 			throws RepositoryException, RepositoryConfigException {
 		LocalRepositoryManager manager = new LocalRepositoryManager(dir);
 		manager.initialize();
 		Repository repository = manager.getRepository(id);
 		if (repository == null) {
 			logger.warn("Creating repository configuration for: {}", id);
 			MemoryStoreConfig mc = new MemoryStoreConfig();
 			mc.setPersist(true);
 			SailRepositoryConfig sc = new SailRepositoryConfig(mc);
 			RepositoryConfig config = new RepositoryConfig(id, sc);
 			Repository system = manager.getSystemRepository();
 			RepositoryConfigUtil.updateRepositoryConfigs(system, config);
 			repository = manager.getRepository(id);
 		}
 		repository = new ThreadProxyRepository(repository);
 		return repository;
 	}
 
 	public ValueFactory getValueFactory() {
 		return repository.getValueFactory();
 	}
 
 	public boolean isWritable() throws RepositoryException {
 		return repository.isWritable();
 	}
 
 	public RepositoryConnection getConnection() throws RepositoryException {
 		return repository.getConnection();
 	}
 
 	public void registryDidShutdown() {
 		try {
 			repository.shutDown();
 		} catch (RepositoryException e) {
 			throw new ApplicationRuntimeException(e);
 		}
 	}
 
 	public void shutDown() throws RepositoryException {
 		repository.shutDown();
 	}
 
 }