com.github.viktornar.controller.composer.ComposeController.java Source code

Introduction

Here is the source code for com.github.viktornar.controller.composer.ComposeController.java

Source

/*
 This file is part of Composer.
 Composer is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.
 Composer is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.
 You should have received a copy of the GNU General Public License
 along with Composer.  If not, see <http://www.gnu.org/licenses/>.
 Copyright 2016 (C) Viktor Nareiko
 */
package com.github.viktornar.controller.composer;

import com.github.viktornar.model.Atlas;
import com.github.viktornar.model.Extent;
import com.github.viktornar.service.SettingsService;
import com.github.viktornar.service.repository.Repository;
import com.github.viktornar.task.PrintTask;
import org.apache.tomcat.util.http.fileupload.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.concurrent.ExecutorService;

import static com.github.viktornar.utils.Helper.*;
import static java.lang.String.format;

/**
 * Controller that is responsible for main atlas printing and displaying functionality.
 *
 * @author v.nareiko
 */
@Controller
@RequestMapping("/")
public class ComposeController {
    @Value("${atlas.folder}")
    private String atlasFolder;
    @Value("${atlas.name.prefix}")
    private String atlasNamePrefix;
    private Repository repository;
    private SettingsService settingsService;

    @Autowired
    public ComposeController(Repository _repository, SettingsService _settingsService) {
        repository = _repository;
        settingsService = _settingsService;
    }

    @RequestMapping(value = "/compose", method = RequestMethod.POST)
    public String postCompose(@ModelAttribute("atlas") Atlas atlas, final RedirectAttributes redirectAttributes) {
        String id = getRandomlyNames(8, 1)[0];

        atlas.setAtlasFolder(format("%s/%s", atlasFolder, id));
        atlas.setAtlasName(atlasNamePrefix + id);

        atlas.setId(id);
        repository.createAtlas(atlas);

        final int POOL_SIZE = atlas.getColumns() * atlas.getRows() + 2;
        ExecutorService executorService = getExecutorService(POOL_SIZE);
        runPrintTask(executorService, atlas, repository, settingsService);

        redirectAttributes.addAttribute("id", id);

        return "redirect:/status/{id}?timeout=30";
    }

    @RequestMapping(value = "/status/{id}", method = RequestMethod.GET)
    public String getStatusById(@PathVariable("id") String id, ModelMap model,
            @RequestParam(value = "timeout", required = true) int timeout) {
        Atlas atlas = repository.getAtlasById(id);

        if (timeout > 0) {
            timeout -= 1;
            model.addAttribute("errorExist", false);
        } else {
            model.addAttribute("errorExist", true);
        }

        model.addAttribute("timeout", timeout);
        model.addAttribute("fileExist", isFileExist(id, atlasFolder, atlasNamePrefix));
        model.addAttribute("atlasId", id);
        model.addAttribute("atlasExecutionProgress", atlas.getProgress());
        model.addAttribute("atlasExecutionTotal", atlas.getRows() * atlas.getColumns());

        return "status_by_id";
    }

    @RequestMapping(value = "/status", method = RequestMethod.GET)
    public String getAllStatus(ModelMap model) {
        List<Atlas> atlases = repository.getAllAtlases();
        model.addAttribute("allAtlases", atlases);
        return "status_all";
    }

    @RequestMapping(value = "/print", method = RequestMethod.GET)
    public String getMap(@RequestParam(value = "orientation") String orientation,
            @RequestParam(value = "size") String size, @RequestParam(value = "xmin") Double xmin,
            @RequestParam(value = "ymin") Double ymin, @RequestParam(value = "xmax") Double xmax,
            @RequestParam(value = "ymax") Double ymax, ModelMap model) {

        Extent extent = new Extent();
        extent.setXmin(xmin);
        extent.setYmin(ymin);
        extent.setXmax(xmax);
        extent.setYmax(ymax);

        Atlas atlas = new Atlas();
        atlas.setExtent(extent);
        atlas.setOrientation(orientation);
        atlas.setSize(size);

        model.addAttribute("atlas", atlas);
        return "map";
    }

    @RequestMapping(value = "/download/{id}", method = RequestMethod.GET)
    public void getFile(@PathVariable("id") String id, HttpServletResponse response) {
        try {
            String _atlasFolder = format("%s/%s", atlasFolder, id);
            String _atlasName = format("%s%s.pdf", atlasNamePrefix, id);

            String filePathToBeServed = format("%s/%s", _atlasFolder, _atlasName);
            File fileToDownload = new File(filePathToBeServed);
            InputStream inputStream = new FileInputStream(fileToDownload);

            response.setContentType("application/pdf");
            response.setHeader("Content-Disposition", format("attachment; filename=%s", _atlasName));
            IOUtils.copy(inputStream, response.getOutputStream());
            response.flushBuffer();
        } catch (IOException ex) {
            throw new RuntimeException("IOError writing file to output stream");
        }
    }

    private void runPrintTask(ExecutorService executorService, Atlas atlas, Repository repository,
            SettingsService settingsService) {
        executorService.submit(new PrintTask(executorService, atlas, repository, settingsService));
    }
}