org.openmrs.module.sync.serialization.ZipPackage.java Source code

Introduction

Here is the source code for org.openmrs.module.sync.serialization.ZipPackage.java

Source

/**
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.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://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Copyright (C) OpenMRS, LLC.  All Rights Reserved.
 */
package org.openmrs.module.sync.serialization;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Date;
import java.util.zip.Adler32;
import java.util.zip.CheckedOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.openmrs.module.sync.SyncConstants;

/**
 * 
 */
public class ZipPackage {
    private static Log log = LogFactory.getLog(FilePackage.class);
    static final int BUFFER = 2048;
    private String rootName;
    private String targetName;
    private FileOutputStream dest;
    private CheckedOutputStream checksum;
    private ZipOutputStream out;
    private byte data[];

    public ZipPackage(File root, String target) {
        try {
            this.rootName = root.getAbsolutePath();
            this.targetName = target;
            File archiveFolder = new File(root, "archive");
            archiveFolder.mkdir();
            File f = new File(archiveFolder, targetName);
            f.mkdir();
            if (!f.exists())
                f.mkdirs();
            File outputStreamDir = new File(f,
                    targetName + "_" + SyncConstants.SYNC_FILENAME_MASK.format(new Date()) + ".zip");
            this.dest = new FileOutputStream(outputStreamDir);
            this.checksum = new CheckedOutputStream(dest, new Adler32());
            this.out = new ZipOutputStream(new BufferedOutputStream(checksum));
            this.data = new byte[BUFFER];
        } catch (Exception e) {
            log.error(e.toString());
        }
    }

    public boolean zip(boolean clearDir) {
        if (this.addFile(this.targetName)) {
            try {
                this.out.close();
                if (clearDir) {
                    File dir = new File(this.rootName + "archive/" + this.targetName);
                    this.clearPath(dir);
                }
            } catch (Exception e) {
                log.error(e.toString());
                return false;
            }
            log.info("ZIP FILE: " + this.rootName + "archive/" + this.targetName + ".zip archived successifully");
            return true;
        } else {
            try {
                this.out.close();
            } catch (Exception e) {
                e.printStackTrace();
                return false;
            }
            log.error("ERROR while archiving folder " + this.rootName + "archive/" + this.targetName);
            return false;
        }
    }

    public void clearPath(File path) throws IOException {
        File[] files = path.listFiles();

        for (int i = 0; i < files.length; ++i) {
            if (files[i].isDirectory()) {
                clearPath(files[i]);
            }
            files[i].delete();
        }
    }

    public boolean addFile(String rootRelativeName) {
        try {

            File currentFile = new File(rootName + "archive/" + rootRelativeName);

            if (currentFile.isFile()) {
                BufferedInputStream origin = new BufferedInputStream(new FileInputStream(currentFile), BUFFER);
                ZipEntry entry = new ZipEntry(rootRelativeName);
                this.out.putNextEntry(entry);
                int count;
                while ((count = origin.read(this.data, 0, BUFFER)) != -1) {
                    this.out.write(this.data, 0, count);
                }
                origin.close();
                return true;
            } else if (currentFile.isDirectory()) {
                File[] fileList = currentFile.listFiles();
                for (int i = 0; i < fileList.length; i++) {
                    if (!addFile(rootRelativeName + "/" + fileList[i].getName())) {
                        return false;
                    }
                }
                return true;
            }
        } catch (Exception e) {
            log.error(e.toString());
            return false;
        }
        return false;
    }
}