cn.edu.zjnu.acm.judge.admin.AdminController.java Source code

Introduction

Here is the source code for cn.edu.zjnu.acm.judge.admin.AdminController.java

Source

/*
 * Copyright 2016 ZJNU ACM.
 *
 * 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 cn.edu.zjnu.acm.judge.admin;

import cn.edu.zjnu.acm.judge.config.JudgeConfiguration;
import cn.edu.zjnu.acm.judge.domain.Contest;
import cn.edu.zjnu.acm.judge.exception.MessageException;
import cn.edu.zjnu.acm.judge.mapper.ContestMapper;
import cn.edu.zjnu.acm.judge.mapper.ProblemMapper;
import cn.edu.zjnu.acm.judge.mapper.UserProblemMapper;
import java.time.Instant;
import java.time.LocalDateTime;
import java.time.ZoneId;
import javax.validation.constraints.Pattern;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.format.annotation.DateTimeFormat;
import org.springframework.http.HttpStatus;
import org.springframework.security.access.annotation.Secured;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.thymeleaf.util.StringUtils;

/**
 *
 * @author zhanhb
 */
@Controller
@Secured("ROLE_ADMIN")
@Validated
public class AdminController {

    @Autowired
    private ProblemMapper problemMapper;
    @Autowired
    private ContestMapper contestMapper;
    @Autowired
    private UserProblemMapper userProblemMapper;
    @Autowired
    private JudgeConfiguration judgeConfiguration;

    @GetMapping("/admin/contests/new")
    public String addcontestpage() {
        return "admin/contests/addpage";
    }

    @GetMapping("/admin")
    public String index() {
        return "admin/index";
    }

    @GetMapping("/admin/problems/{problemId}/disable")
    public String disableProblem(Model model, @PathVariable("problemId") long problemId) {

        problemMapper.setDisabled(problemId, true);

        model.addAttribute("problemId", problemId);
        return "admin/problems/disable";
    }

    // TODO request method
    @GetMapping("/admin/problems/{problemId}/enable")
    public String enableProblem(Model model, @PathVariable("problemId") long problemId) {
        problemMapper.setDisabled(problemId, false);

        model.addAttribute("problemId", problemId);
        return "admin/problems/enable";
    }

    @PostMapping("/admin/contests")
    public String addContest(@RequestParam(value = "title", required = false) String title,
            @RequestParam(value = "description", required = false) String description,
            @RequestParam(value = "start") @DateTimeFormat(pattern = "yyyy-M-d H:m") LocalDateTime start,
            @RequestParam("length") @Pattern(regexp = "\\d+:[0-5]?[0-9]:[0-5]?[0-9]") String length, Model model) {
        Instant startTime = start.atZone(ZoneId.systemDefault()).toInstant();
        String[] split = length.split(":", 3);
        long h = Long.parseLong(split[0]);
        int m = Integer.parseInt(split[1]);
        int s = Integer.parseInt(split[2]);
        Instant endTime = startTime.plusSeconds(h * 3600 + m * 60 + s);

        Contest contest = Contest.builder().title(title).description(description).startTime(startTime)
                .endTime(endTime).build();
        contestMapper.save(contest);

        model.addAttribute("contestId", contest.getId());
        model.addAttribute("startTime", startTime);
        model.addAttribute("endTime", endTime);

        return "admin/contests/add";
    }

    @PostMapping("/admin/tools")
    public String tools(@RequestParam(value = "op", required = false) String op,
            @RequestParam(value = "sinfo", required = false) String sinfo,
            @RequestParam(value = "pureText", required = false) String pureText) {
        if (StringUtils.isEmptyOrWhitespace(op)) {
            throw new MessageException("Nothing I can do for you.", HttpStatus.BAD_REQUEST);
        }

        switch (op.toLowerCase()) {
        case "test":
            userProblemMapper.init();
            userProblemMapper.updateProblems();
            userProblemMapper.updateUsers();
            break;
        case "setsysteminfo":
            if (StringUtils.isEmptyOrWhitespace(sinfo)) {
                judgeConfiguration.setSystemInfo(null);
            } else if (pureText != null) {
                judgeConfiguration.setSystemInfo(StringUtils.escapeXml(sinfo.trim()));
            } else {
                judgeConfiguration.setSystemInfo(sinfo.trim());
            }
            break;
        default:
            throw new IllegalStateException("unknown operation '" + op + "'");
        }

        return "admin/recieved";
    }

}