org.wso2.carbon.appfactory.utilities.storage.FileArtifactStorage.java Source code

Introduction

Here is the source code for org.wso2.carbon.appfactory.utilities.storage.FileArtifactStorage.java

Source

/*
 * Copyright 2005-2011 WSO2, Inc. (http://wso2.com)
 *
 *      Licensed under the Apache License, Version 2.0 (the "License");
 *      you may not use this file except in compliance with the License.
 *      You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *      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 License for the specific language governing permissions and
 *      limitations under the License.
 */

package org.wso2.carbon.appfactory.utilities.storage;

import org.apache.commons.io.FileUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.wso2.carbon.appfactory.common.AppFactoryException;
import org.wso2.carbon.appfactory.core.ArtifactStorage;
import org.wso2.carbon.appfactory.core.dao.ApplicationDAO;
import org.wso2.carbon.utils.CarbonUtils;

import javax.activation.DataHandler;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.List;

public class FileArtifactStorage implements ArtifactStorage {
    private static final Log log = LogFactory.getLog(FileArtifactStorage.class);
    private static final String ARTIFACT_STORAGE = "artifactdb";

    /**
     * {@inheritDoc}
     */
    @Override
    public File retrieveArtifact(String applicationId, String version, String revision, String tenantDomain)
            throws AppFactoryException {
        File targetDir = null;
        List<File> artifactFiles = null;
        File workDir = new File(getApplicationStorageDirectoryPath(applicationId, version, revision));
        String applicationType = ApplicationDAO.getInstance().getApplicationType(applicationId);
        String[] fileExtension = { applicationType };

        if (workDir.isDirectory()) {
            artifactFiles = (List<File>) FileUtils.listFiles(workDir, fileExtension, false);
            targetDir = artifactFiles.get(0);
            return targetDir;
        } else {
            return null;
        }
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public File retrieveArtifact(String applicationId, String version, String revision, String buildId,
            String tenantDomain) throws AppFactoryException {
        File targetDir = null;
        List<File> artifactFiles = null;
        File workDir = new File(getStoragePathForArtifact(applicationId, version, buildId));
        String applicationType = ApplicationDAO.getInstance().getApplicationType(applicationId);
        String[] fileExtension = { applicationType };
        artifactFiles = (List<File>) FileUtils.listFiles(workDir, fileExtension, false);
        targetDir = artifactFiles.get(0);
        return targetDir;
    }

    @Override
    public void storeArtifact(String applicationId, String version, String revision, DataHandler data,
            String fileName) throws AppFactoryException {
        String path = getApplicationStorageDirectoryPath(applicationId, version, revision) + File.separator
                + fileName;
        try {
            File destFile = new File(path);

            if (destFile.exists() == true) {
                destFile.delete();
            }

            if (!destFile.getParentFile().exists()) {
                if (!destFile.getParentFile().mkdirs()) {
                    throw new IOException("Failed to create directory structure.");
                }
            }

            FileOutputStream fileOutputStream = new FileOutputStream(destFile);
            data.writeTo(fileOutputStream);
            fileOutputStream.flush();
            fileOutputStream.close();

        } catch (IOException e) {
            log.error("Error storing the artifact file : " + e.getMessage(), e);
            throw new AppFactoryException("Error storing the artifact file : " + e.getMessage(), e);
        }
    }

    @Override
    public void storeArtifact(String applicationId, String version, String revision, String buildId,
            DataHandler data, String fileName) throws AppFactoryException {
        //String path = getStoragePathForArtifact(applicationId, version, buildId) + File.separator + fileName;
        String path = getApplicationStorageDirectoryPath(applicationId, version, revision) + File.separator
                + fileName;
        try {
            File destFile = new File(path);

            if (destFile.exists() == true) {
                if (destFile.isDirectory()) {
                    FileUtils.deleteDirectory(destFile);
                } else {
                    destFile.delete();
                }

            }

            if (!destFile.getParentFile().exists()) {
                if (!destFile.getParentFile().mkdirs()) {
                    throw new IOException("Failed to create directory structure.");
                }
            }

            FileOutputStream fileOutputStream = new FileOutputStream(destFile);
            data.writeTo(fileOutputStream);
            fileOutputStream.flush();
            fileOutputStream.close();

        } catch (IOException e) {
            log.error("Error storing the artifact file : " + e.getMessage(), e);
            throw new AppFactoryException("Error storing the artifact file : " + e.getMessage(), e);
        }
    }

    public String getStoragePathForArtifact(String applicationId, String version, String buildId) {
        return CarbonUtils.getTmpDir() + File.separator + ARTIFACT_STORAGE + File.separator + applicationId
                + File.separator + version;
    }

    public String getApplicationStorageDirectoryPath(String applicationId, String version, String revision) {
        return CarbonUtils.getTmpDir() + File.separator + ARTIFACT_STORAGE + File.separator + applicationId
                + File.separator + version;
    }
}