com.peadargrant.filecheck.web.controllers.CheckController.java Source code

Introduction

Here is the source code for com.peadargrant.filecheck.web.controllers.CheckController.java

Source

/* 
 * Copyright (C) 2014 Peadar Grant
 *
 * This program 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.
 *
 * This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.peadargrant.filecheck.web.controllers;

import com.peadargrant.filecheck.core.assignments.Assignment;
import com.peadargrant.filecheck.core.assignments.Assignments;
import com.peadargrant.filecheck.core.checker.Checker;
import com.peadargrant.filecheck.core.provider.AssignmentsProvider;
import com.peadargrant.filecheck.core.resultmodels.ReportTableModel;
import com.peadargrant.filecheck.core.resultmodels.SummaryTableModel;
import com.peadargrant.filecheck.web.reports.FileCheckWebTableTransformer;
import com.peadargrant.filecheck.web.support.ServerEnvironment;
import java.awt.Color;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.multipart.MultipartFile;

/**
 *
 * @author Peadar Grant
 */
@Controller
@RequestMapping("/check")
public class CheckController {

    @Autowired
    private ServerEnvironment serverEnvironment;

    @Autowired
    private HttpServletRequest request;

    @RequestMapping(method = RequestMethod.POST)
    public String performCheck(@RequestParam(value = "assignment", required = true) String assignmentCode,
            @RequestParam("file") MultipartFile file, ModelMap model) throws Exception {

        String assignmentsUrl = serverEnvironment.getPropertyAsString("assignmentsUrl");
        model.addAttribute("assignmentsUrl", assignmentsUrl);

        // bail out if the file is empty
        if (file.isEmpty()) {
            model.addAttribute("message", "file.was.empty");
            return "error";
        }

        // input stream from file 
        byte[] bytes = file.getBytes();
        String name = file.getOriginalFilename();

        // write to temp dir
        String filePath = System.getProperty("java.io.tmpdir") + "/" + name;
        BufferedOutputStream stream = new BufferedOutputStream(new FileOutputStream(new File(filePath)));
        stream.write(bytes);
        stream.close();

        // load file
        File inputFile = new File(filePath);

        // get assignment
        Assignment assignment = this.getAssignmentForCode(assignmentCode);
        if (assignment == null) {
            return "assignmentNotFound";
        }
        model.addAttribute(assignment);

        // GUI report table model
        SummaryTableModel summaryTableModel = new SummaryTableModel();
        ReportTableModel reportTableModel = new ReportTableModel(summaryTableModel);

        // checker
        Checker checker = new Checker();
        checker.setReport(reportTableModel);
        checker.runChecks(inputFile, assignment);

        // details for output
        model.addAttribute("fileName", name);
        model.addAttribute("startTime", new java.util.Date());
        String ipAddress = request.getHeader("X-FORWARDED-FOR");
        if (ipAddress == null) {
            ipAddress = request.getRemoteAddr();
        }
        model.addAttribute("remoteIP", ipAddress);

        // final outcome
        model.addAttribute("outcome", summaryTableModel.getFinalOutcome());
        Color finalOutcomeColor = summaryTableModel.getFinalOutcome().getSaturatedColor();
        model.addAttribute("colourr", finalOutcomeColor.getRed());
        model.addAttribute("colourg", finalOutcomeColor.getGreen());
        model.addAttribute("colourb", finalOutcomeColor.getBlue());

        // transformer for parsing tables
        FileCheckWebTableTransformer transformer = new FileCheckWebTableTransformer();

        // summary table headings
        List<String> summaryColumns = transformer.getColumnHeaders(summaryTableModel);
        model.addAttribute("summaryColumns", summaryColumns);

        // summary table
        List summaryContents = transformer.getTableContents(summaryTableModel);
        model.addAttribute("summary", summaryContents);

        // detail table headings
        List<String> detailColumns = transformer.getColumnHeaders(reportTableModel);
        model.addAttribute("detailColumns", detailColumns);

        // detail report table
        List detailContents = transformer.getTableContents(reportTableModel);
        model.addAttribute("detail", detailContents);

        // delete the uploaded file
        inputFile.delete();

        // Return results display
        return "check";
    }

    private Assignment getAssignmentForCode(String code) throws Exception {
        String assignmentsUrl = serverEnvironment.getPropertyAsString("assignmentsUrl");

        AssignmentsProvider assignmentsProvider = new AssignmentsProvider();
        Assignments assignments = assignmentsProvider.customLibrary(assignmentsUrl);
        List<Assignment> assignmentList = assignments.getAssignment();
        for (Assignment assignment : assignmentList) {
            if (assignment.getCode().equals(code)) {
                return assignment;
            }
        }
        return null;
    }
}