com.digitalgeneralists.assurance.model.merge.SourceMergeEngine.java Source code

Introduction

Here is the source code for com.digitalgeneralists.assurance.model.merge.SourceMergeEngine.java

Source

/*
 * Assurance
 * 
 * Created by Mark Johnson
 * 
 * Copyright (c) 2015 Digital Generalists, LLC.
 * 
 */
/*
 * Copyright 2015 Digital Generalists, LLC.
 *
 * 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 com.digitalgeneralists.assurance.model.merge;

import java.io.File;
import java.io.IOException;

import org.apache.commons.io.FileUtils;
import org.apache.log4j.Logger;
import org.springframework.stereotype.Component;

import com.digitalgeneralists.assurance.exceptions.AssuranceNullFileReferenceException;
import com.digitalgeneralists.assurance.model.entities.ComparisonResult;
import com.digitalgeneralists.assurance.model.enums.AssuranceResultResolution;
import com.digitalgeneralists.assurance.notification.IProgressMonitor;

@Component("SourceMergeEngine")
public class SourceMergeEngine extends MergeEngine implements IMergeEngine {
    private Logger logger = Logger.getLogger(SourceMergeEngine.class);

    @Override
    public void mergeResult(ComparisonResult result, IProgressMonitor monitor)
            throws AssuranceNullFileReferenceException {
        File sourceFile = result.getSource().getFile();
        File targetFile = result.getTarget().getFile();

        if ((sourceFile == null) || (targetFile == null)) {
            throw new AssuranceNullFileReferenceException("The source or target file is null.");
        }

        if (monitor != null) {
            StringBuilder message = new StringBuilder(512);
            monitor.publish(message.append("Merging ").append(sourceFile.toString()).append(" to ")
                    .append(targetFile.toString()).toString());
            message.setLength(0);
            message = null;
        }

        if (sourceFile.exists()) {
            try {
                if (sourceFile.isDirectory()) {
                    FileUtils.copyDirectory(sourceFile, targetFile);
                } else {
                    FileUtils.copyFile(sourceFile, targetFile);
                }
                result.setResolution(AssuranceResultResolution.REPLACE_TARGET);
            } catch (IOException e) {
                logger.error("An error occurred when replacing the target with the source.");
                result.setResolution(AssuranceResultResolution.PROCESSING_ERROR_ENCOUNTERED);
                result.setResolutionError(e.getMessage());
            }
        } else {
            if (targetFile.exists()) {
                File scanDeletedItemsLocation = result
                        .getTargetDeletedItemLocation(getApplicationDeletedItemsLocation());
                if (scanDeletedItemsLocation == null) {
                    StringBuilder deletedItemPath = new StringBuilder(512);
                    scanDeletedItemsLocation = new File(
                            deletedItemPath.append(this.getDefaultDeletedItemsLocation()).append(File.separator)
                                    .append(targetFile.getName()).toString());
                    deletedItemPath.setLength(0);
                    deletedItemPath = null;
                }
                try {
                    if (targetFile.isDirectory()) {
                        FileUtils.moveDirectory(targetFile, scanDeletedItemsLocation);
                    } else {
                        FileUtils.moveFile(targetFile, scanDeletedItemsLocation);
                    }
                    result.setResolution(AssuranceResultResolution.DELETE_TARGET);
                } catch (IOException e) {
                    StringBuffer message = new StringBuffer(512);
                    logger.warn(message.append("Could not move item to deleted items location ")
                            .append(sourceFile.getPath()));
                    message.setLength(0);
                    message = null;
                    result.setResolution(AssuranceResultResolution.PROCESSING_ERROR_ENCOUNTERED);
                    result.setResolutionError(e.getMessage());
                }
                scanDeletedItemsLocation = null;
            }
        }

        sourceFile = null;
        targetFile = null;
    }
}