de.uzk.hki.da.repository.FakeRepositoryFacade.java Source code

Introduction

Here is the source code for de.uzk.hki.da.repository.FakeRepositoryFacade.java

Source

/*
  DA-NRW Software Suite | ContentBroker
  Copyright (C) 2013 Historisch-Kulturwissenschaftliche Informationsverarbeitung
  Universitt zu Kln
    
  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 3 of the License, or
  (at your option) any later version.
    
  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  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, see <http://www.gnu.org/licenses/>.
*/

package de.uzk.hki.da.repository;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import org.apache.commons.io.FileUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import de.uzk.hki.da.utils.FolderUtils;

/**
 * Implements a simple file system based repository
 * for acceptance testing on developer machines.
 * @author Sebastian Cuy
 */
public class FakeRepositoryFacade implements RepositoryFacade {

    static final Logger logger = LoggerFactory.getLogger(FakeRepositoryFacade.class);

    private String workAreaRootPath;

    @Override
    public boolean purgeObjectIfExists(String objectId, String collection) throws RepositoryException {
        try {
            if (objectExists(objectId, collection)) {
                FolderUtils.deleteDirectorySafe(getFile(objectId, collection, null));
                return true;
            } else {
                return false;
            }
        } catch (IOException e) {
            throw new RepositoryException("Unable to purge object " + objectId, e);
        }
    }

    @Override
    public void createObject(String objectId, String collection, String ownerId) throws RepositoryException {
        if (!getFile(objectId, collection, null).mkdirs()) {
            throw new RepositoryException("Unable to create folder for object " + collection + "/" + objectId);
        }
    }

    @Override
    public void ingestFile(String objectId, String collection, String fileId, File file, String label,
            String mimeType) throws RepositoryException, IOException {
        try {
            File destFile = getFile(objectId, collection, fileId);
            FileUtils.copyFile(file, destFile);
        } catch (IOException e) {
            throw new RepositoryException("Unable to create file " + collection + "/" + objectId + "/" + fileId, e);
        }
    }

    @Override
    public void retrieveTo(OutputStream outputStream, String objectId, String collection, String fileId)
            throws RepositoryException {
        logger.debug("retrieveFile");
        try {
            InputStream inputStream = new FileInputStream(getFile(objectId, collection, fileId));
            org.apache.commons.io.IOUtils.copy(inputStream, outputStream);
        } catch (IOException e) {
            throw new RepositoryException("Unable to copy file " + collection + "/" + objectId + "/" + fileId, e);
        }
    }

    @Override
    public boolean fileExists(String objectId, String collection, String fileId) throws RepositoryException {
        return getFile(objectId, collection, fileId).exists();
    }

    @Override
    public boolean objectExists(String objectId, String collection) throws RepositoryException {
        return getFile(objectId, collection, null).exists();
    }

    @Override
    public void addRelationship(String objectId, String collection, String predicate, String object) {
        // stub, fake repository does not handle relationships
    }

    @Override
    public String generateFileId(String path) {
        return new File(path).getName();
    }

    public String getWorkAreaRootPath() {
        return workAreaRootPath;
    }

    public void setWorkAreaRootPath(String workAreaRootPath) {
        this.workAreaRootPath = workAreaRootPath + "/work";
    }

    private File getFile(String objectId, String collection, String file) {
        String path = workAreaRootPath + File.separator + "_data" + File.separator + collection + File.separator
                + objectId;
        if (file != null && !file.isEmpty()) {
            path += File.separator + file;
        }
        logger.debug("getFile:" + path);
        return new File(path);
    }
}