StandaloneRepositoryHelper.java

/*
 * Copyright (C) 2011 Marius Giepz
 *
 * This program is free software; you can redistribute it and/or modify it 
 * under the terms of the GNU General Public License as published by the Free 
 * Software Foundation; either version 2 of the License, or (at your option) 
 * any later version.
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * 
 * See the GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License along 
 * with this program; if not, write to the Free Software Foundation, Inc., 
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 
 *
 */

package org.saiku.adhoc.service.repository;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FilenameFilter;
import java.io.IOException;
import java.util.Map;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.dom4j.Document;
import org.pentaho.metadata.model.Domain;
import org.pentaho.platform.api.engine.IPentahoSession;
import org.pentaho.platform.api.engine.ISolutionFile;
import org.pentaho.platform.api.repository.ISolutionRepository;
import org.pentaho.platform.engine.core.solution.ActionInfo;
import org.pentaho.platform.engine.core.system.PentahoSessionHolder;
import org.pentaho.platform.engine.core.system.PentahoSystem;
import org.pentaho.platform.util.messages.LocaleHelper;
import org.saiku.adhoc.model.master.ReportTemplate;
import org.saiku.adhoc.model.master.ReportTemplateStandalone;
import org.saiku.adhoc.providers.IPrptProvider;
import org.saiku.adhoc.service.PluginConfig;

public class StandaloneRepositoryHelper implements IRepositoryHelper {

	private static Log log = LogFactory.getLog(StandaloneRepositoryHelper.class);

	private static final String ENCODING = "UTF-8";

	private IPrptProvider prptProvider;
	
    
	public void setPrptProvider(IPrptProvider prptProvider) {
		this.prptProvider = prptProvider;
	}

	@Override
	public boolean writeFile(String solution, String path, String action,
			String contents) {

		try {
			return writeFile(solution, path, action,
					contents.getBytes(ENCODING));
		} catch (Exception e) {
			log.error("Cannot save cda (" + action + ")", e);
			return (false);
		}

	}

	@Override
	public boolean writeFile(String solution, String path, String artifact,
			ByteArrayOutputStream contents) {

		try {
			return writeFile(solution, path, artifact, contents.toByteArray());
		} catch (Exception e) {
			log.error("Cannot save cda (" + artifact + ")", e);
			return (false);
		}

	}

	@Override
	public String loadFile(String solution, String path, String action)
			throws IOException {

		// do we need to load other files too?
		if (!action.endsWith(".adhoc")) {
			action += ".adhoc";
		}

		String fullPath = ActionInfo.buildSolutionPath(solution, path, action);
		IPentahoSession userSession = PentahoSessionHolder.getSession();
		ISolutionRepository repository = PentahoSystem.get(
				ISolutionRepository.class, userSession);

		if (repository.resourceExists(fullPath)) {
			String doc = repository.getResourceAsString(fullPath,
					ISolutionRepository.ACTION_EXECUTE);

			if (doc == null) {
				log.error("Error retrieving document from solution repository");
				throw new NullPointerException(
						"Error retrieving document from solution repository");
			}

			return doc;

		}
		return null;
	}

	private static boolean writeFile(String solution, String path,
			String artifact, byte[] contents)
			throws Exception{

		String fullPath = ActionInfo
				.buildSolutionPath(solution, path, artifact);
		IPentahoSession userSession = PentahoSessionHolder.getSession();
		ISolutionRepository repository = PentahoSystem.get(
				ISolutionRepository.class, userSession);

		if (repository == null) {
			log.error("Access to Repository has failed");
			throw new NullPointerException("Access to Repository has failed");
		}

		String base = PentahoSystem.getApplicationContext()
				.getSolutionRootPath();
		String parentPath = ActionInfo.buildSolutionPath(solution, path, "");
		ISolutionFile parentFile = repository.getSolutionFile(parentPath,
				ISolutionRepository.ACTION_CREATE);
		String filePath = parentPath + ISolutionRepository.SEPARATOR + artifact;
		ISolutionFile fileToSave = repository.getSolutionFile(fullPath,
				ISolutionRepository.ACTION_UPDATE);

		if (fileToSave != null
				|| (!repository.resourceExists(filePath) && parentFile != null)) {
			repository
					.publish(base, '/' + parentPath, artifact, contents, true);
			log.debug(PluginConfig.PLUGIN_NAME + " : Published " + solution
					+ " / " + path + " / " + artifact);
		} else {
			throw new Exception(
					"Error ocurred while saving to solution repository");
		}
		return (true);
	}

	@Override
	/**
	 * Works out what is the most appropriate locale to use given a domain and the user's
	 * current locale
	 * @return
	 */
	public String getLocale(Domain domain) {
		String locale = LocaleHelper.getClosestLocale(LocaleHelper.getLocale()
				.toString(), domain.getLocaleCodes());
		return locale;
	}

	public ReportTemplate[] getTemplateList(final IPentahoSession userSession) {
		log.debug("Getting template list");


		Map<String, ReportTemplate> datasources = prptProvider.getDatasources();
        return datasources.values().toArray(new ReportTemplateStandalone[]{});


	}

	class PrptFilter implements FilenameFilter {
		public boolean accept(File f, String s) {
			return s.toLowerCase().endsWith(".prpt");
		}
	}

	/**
	 * Browse the content of the repository
	 * 
	 * @param dir
	 * @param fileExtensions
	 */
	public Document browse(String dir, final String fileExtensions) {

		return null;

	}

    public void cleanTemp() {
        // TODO Auto-generated method stub
        
    }
    
    
//    public boolean writeLocalFile(String path,
//                String artifact, byte[] contents){
//    FileOutputStream fos =null;
//    try {
//                fos = new FileOutputStream(path+artifact);
//            } catch (FileNotFoundException e) {
//                // TODO Auto-generated catch block
//                e.printStackTrace();
//            }
//
//            try {
//                fos.write(contents);
//            } catch (IOException e) {
//                // TODO Auto-generated catch block
//                e.printStackTrace();
//            }
//            
//            try {
//                fos.close();
//                return true;
//            } catch (IOException e) {
//                // TODO Auto-generated catch block
//                e.printStackTrace();
//                return false;
//            }
//
//
//
//}
}