org.schoellerfamily.gedbrowser.api.controller.SaveController Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of gedbrowserng Show documentation
There is a newer version: 1.3.0-RC2
package org.schoellerfamily.gedbrowser.api.controller;

import java.util.Collections;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.schoellerfamily.gedbrowser.api.controller.exception.DataSetNotFoundException;
import org.schoellerfamily.gedbrowser.datamodel.Root;
import org.schoellerfamily.gedbrowser.persistence.domain.RootDocument;
import org.schoellerfamily.gedbrowser.persistence.mongo.loader.GedDocumentFileLoader;
import org.schoellerfamily.gedbrowser.writer.GedWriter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.ResponseBody;

/**
 * Emits GEDCOM to the HTTP connection to download the GEDCOM state.
 *
 * @author Dick Schoeller
 */
@CrossOrigin(origins = {
        "http://largo.schoellerfamily.org:4200", "http://localhost:4200" })
@Controller
public class SaveController {
    /** Logger. */
    private final transient Log logger = LogFactory.getLog(getClass());

    /** */
    @Autowired
    private transient GedDocumentFileLoader loader;

    /**
     * Connects HTML template file with data for saving the GEDCOM file.
     *
     * @param db name of database for the lookup
     * @param response the servlet response object, needed for tweaking headers
     * @return a string identifying which HTML template to use.
     */
    @GetMapping(value = "/v1/dbs/{db}/save",
            produces = MediaType.TEXT_PLAIN_VALUE)
    @ResponseBody
    public final ResponseEntity save(@PathVariable final String db,
            final HttpServletResponse response) {
        logger.info("Starting save");
        // check here whether authorized.
        try {
            final Root root = fetchRoot(db);

            final String contents = new GedWriter(root).writeString();
            final String filename = db + ".ged";
            logger.info("filename: " + filename);
            final HttpHeaders headers = new HttpHeaders();
            headers.setAccessControlExposeHeaders(
                    Collections.singletonList("Content-Disposition"));
            headers.set(
                    "Content-Disposition", "attachment; filename=" + filename);
            headers.setContentType(MediaType.TEXT_PLAIN);
            logger.info("Exiting save");
            return new ResponseEntity<>(contents, headers, HttpStatus.OK);
        } catch (DataSetNotFoundException e) {
            final HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.TEXT_PLAIN);
            logger.info("Exiting save");
            return new ResponseEntity<>(
                    e.getMessage(), headers, HttpStatus.NOT_FOUND);
        }
    }

    /**
     * @param dbName the name of the database
     * @return the root object
     */
    protected final Root fetchRoot(final String dbName) {
        final RootDocument rootDocument = loader.loadDocument(dbName);
        if (rootDocument == null) {
            throw new DataSetNotFoundException(
                    "Data set " + dbName + " not found", dbName);
        }
        return rootDocument.getGedObject();
    }
}