de.inren.frontend.health.backup.BackupRestorePanel.java Source code

Introduction

Here is the source code for de.inren.frontend.health.backup.BackupRestorePanel.java

Source

/**
 * Copyright 2014 the original author or authors.
 *
 * 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 de.inren.frontend.health.backup;

import java.util.Calendar;
import java.util.List;

import org.apache.wicket.Component;
import org.apache.wicket.extensions.ajax.markup.html.form.upload.UploadProgressBar;
import org.apache.wicket.markup.html.form.Form;
import org.apache.wicket.markup.html.form.upload.FileUpload;
import org.apache.wicket.markup.html.form.upload.FileUploadField;
import org.apache.wicket.markup.html.link.ResourceLink;
import org.apache.wicket.request.resource.ByteArrayResource;
import org.apache.wicket.request.resource.IResource;
import org.apache.wicket.request.resource.ResourceReference;
import org.apache.wicket.spring.injection.annot.SpringBean;
import org.apache.wicket.util.file.File;
import org.apache.wicket.util.file.Files;
import org.apache.wicket.util.file.Folder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import de.inren.frontend.common.panel.ABasePanel;
import de.inren.service.health.HealthXmlBackupRestoreService;

/**
 * @author Ingo Renner
 *
 */
public class BackupRestorePanel extends ABasePanel {
    private static final Logger log = LoggerFactory.getLogger(BackupRestorePanel.class);

    @SpringBean
    private HealthXmlBackupRestoreService healthXmlBackupRestoreService;

    public BackupRestorePanel(String id) {
        super(id);
    }

    @Override
    protected void initGui() {
        // Backup User
        add(getMyBackupLink("userLink"));
        // Backup all Users
        add(getAllBackupLink("allLink"));
        // Restore with progress bar
        final FileUploadForm progressUploadForm = new FileUploadForm("restore");
        progressUploadForm
                .add(new UploadProgressBar("progress", progressUploadForm, progressUploadForm.fileUploadField));
        add(progressUploadForm);
    }

    private Component getMyBackupLink(String id) {
        Calendar cal = Calendar.getInstance();

        final String key = "HealthBackup_" + getUser().getUsername() + "_" + cal.getTime().toString() + ".xml";

        ResourceReference rr = new ResourceReference(key) {

            @Override
            public IResource getResource() {
                return new ByteArrayResource("text/xml") {

                    @Override
                    protected byte[] getData(Attributes attributes) {
                        try {
                            String xml = healthXmlBackupRestoreService.dumpDbToXml(getUser().getUsername());
                            return xml.getBytes("UTF-8");
                        } catch (Exception e) {
                            log.error(e.getMessage(), e);
                            throw new RuntimeException(e.getMessage());
                        }
                    }
                };
            }
        };
        return new ResourceLink(id, rr);
    }

    private Component getAllBackupLink(String id) {
        Calendar cal = Calendar.getInstance();

        final String key = "HealthBackup_AllUser" + "_" + cal.getTime().toString() + ".xml";

        ResourceReference rr = new ResourceReference(key) {

            @Override
            public IResource getResource() {
                return new ByteArrayResource("text/xml") {

                    @Override
                    protected byte[] getData(Attributes attributes) {
                        try {
                            String xml = healthXmlBackupRestoreService.dumpDbToXml();
                            return xml.getBytes("UTF-8");
                        } catch (Exception e) {
                            log.error(e.getMessage(), e);
                            throw new RuntimeException(e.getMessage());
                        }
                    }
                };
            }
        };
        return new ResourceLink(id, rr);
    }

    /**
     * Form for uploads.
     */
    private class FileUploadForm extends Form<Void> {
        FileUploadField fileUploadField;

        /**
         * Construct.
         * 
         * @param name
         *            Component name
         */
        public FileUploadForm(String name) {
            super(name);

            // set this form to multipart mode (allways needed for uploads!)
            setMultiPart(true);

            // Add one file input field
            add(fileUploadField = new FileUploadField("fileInput"));
        }

        @Override
        protected void onSubmit() {
            final List<FileUpload> uploads = fileUploadField.getFileUploads();
            if (uploads != null) {
                for (FileUpload upload : uploads) {
                    // Create a new file
                    File newFile = new File(getUploadFolder(), upload.getClientFileName());

                    // Check new file, delete if it already existed
                    checkFileExists(newFile);
                    try {
                        // Save to new file
                        newFile.createNewFile();
                        upload.writeTo(newFile);
                        // kick restore
                        healthXmlBackupRestoreService.restoreFromXmlFile(newFile);
                        getFeedback().info("Restored data from " + upload.getClientFileName() + ".");
                    } catch (Exception e) {
                        throw new IllegalStateException("Unable to restore data.", e);
                    }
                }
            }
        }
    }

    private void checkFileExists(File newFile) {
        if (newFile.exists()) {
            // Try to delete the file
            if (!Files.remove(newFile)) {
                throw new IllegalStateException("Unable to overwrite " + newFile.getAbsolutePath());
            }
        }
    }

    private Folder getUploadFolder() {
        return null; // ((InRenApplication) Application.get()).getUploadFolder();
    }
}