org.verwandlung.voj.judger.core.Preprocessor.java Source code

Introduction

Here is the source code for org.verwandlung.voj.judger.core.Preprocessor.java

Source

/* Verwandlung Online Judge - A cross-platform judge online system
 * Copyright (C) 2018 Haozhe Xie <cshzxie@gmail.com>
 *
 * 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/>.
 *
 *
 *                              _ooOoo_  
 *                             o8888888o  
 *                             88" . "88  
 *                             (| -_- |)  
 *                             O\  =  /O  
 *                          ____/`---'\____  
 *                        .'  \\|     |//  `.  
 *                       /  \\|||  :  |||//  \  
 *                      /  _||||| -:- |||||-  \  
 *                      |   | \\\  -  /// |   |  
 *                      | \_|  ''\---/''  |   |  
 *                      \  .-\__  `-`  ___/-. /  
 *                    ___`. .'  /--.--\  `. . __  
 *                 ."" '<  `.___\_<|>_/___.'  >'"".  
 *                | | :  `- \`.;`\ _ /`;.`/ - ` : | |  
 *                \  \ `-.   \_ __\ /__ _/   .-` /  /  
 *           ======`-.____`-.___\_____/___.-`____.-'======  
 *                              `=---=' 
 *
 *                          HERE BE BUDDHA
 *
 */
package org.verwandlung.voj.judger.core;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.attribute.PosixFilePermission;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.commons.io.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import org.verwandlung.voj.judger.exception.CreateDirectoryException;
import org.verwandlung.voj.judger.mapper.CheckpointMapper;
import org.verwandlung.voj.judger.model.Checkpoint;
import org.verwandlung.voj.judger.model.Language;
import org.verwandlung.voj.judger.model.Submission;

/**
 * ?, ??.
 * 
 * @author Haozhe Xie
 */
@Component
public class Preprocessor {
    /**
     * ??.
     * 
     * @param submission - 
     * @param workDirectory - 
     * @param baseFileName - ???(???)
     * @throws Exception 
     */
    public void createTestCode(Submission submission, String workDirectory, String baseFileName) throws Exception {
        File workDirFile = new File(workDirectory);
        if (!workDirFile.exists() && !workDirFile.mkdirs()) {
            throw new CreateDirectoryException("Failed to create directory: " + workDirectory);
        }
        setWorkDirectoryPermission(workDirFile);

        Language language = submission.getLanguage();
        String code = replaceClassName(language, submission.getCode(), baseFileName);
        String codeFilePath = String.format("%s/%s.%s",
                new Object[] { workDirectory, baseFileName, getCodeFileSuffix(language) });

        FileOutputStream outputStream = new FileOutputStream(new File(codeFilePath));
        IOUtils.write(code, outputStream);
        IOUtils.closeQuietly(outputStream);
    }

    /**
     * ?????.
     * @param language - 
     * @return ????
     */
    private String getCodeFileSuffix(Language language) {
        String compileCommand = language.getCompileCommand();

        Pattern pattern = Pattern.compile("\\{filename\\}\\.((?!exe| ).)+");
        Matcher matcher = pattern.matcher(compileCommand);

        if (matcher.find()) {
            String sourceFileName = matcher.group();
            return sourceFileName.replaceAll("\\{filename\\}\\.", "");
        }
        return "";
    }

    /**
     * ???(Java), ??.
     * @param language - 
     * @param code - ??
     * @param newClassName - ??
     */
    private String replaceClassName(Language language, String code, String newClassName) {
        if (!language.getLanguageName().equalsIgnoreCase("Java")) {
            return code;
        }
        return code.replaceAll("class[ \n]+Main", "class " + newClassName);
    }

    /**
     * ???.
     * Linux, ?UID=1536?, ?Others???.
     * @param workDirectory 
     */
    private void setWorkDirectoryPermission(File workDirectory) throws IOException {
        if (!System.getProperty("os.name").contains("Windows")) {
            Set<PosixFilePermission> permissions = new HashSet<>();

            permissions.add(PosixFilePermission.OWNER_READ);
            permissions.add(PosixFilePermission.OWNER_WRITE);
            permissions.add(PosixFilePermission.OWNER_EXECUTE);

            permissions.add(PosixFilePermission.GROUP_READ);
            permissions.add(PosixFilePermission.GROUP_WRITE);
            permissions.add(PosixFilePermission.GROUP_EXECUTE);

            permissions.add(PosixFilePermission.OTHERS_READ);
            permissions.add(PosixFilePermission.OTHERS_WRITE);
            permissions.add(PosixFilePermission.OTHERS_EXECUTE);
            Files.setPosixFilePermissions(workDirectory.toPath(), permissions);
        }
    }

    /**
     * ???.
     * @param problemId - 
     * @throws Exception 
     */
    public void fetchTestPoints(long problemId) throws Exception {
        String checkpointsFilePath = String.format("%s/%s", new Object[] { checkpointDirectory, problemId });
        File checkpointsDirFile = new File(checkpointsFilePath);
        if (!checkpointsDirFile.exists() && !checkpointsDirFile.mkdirs()) {
            throw new CreateDirectoryException(
                    "Failed to create the checkpoints directory: " + checkpointsFilePath);
        }

        List<Checkpoint> checkpoints = checkpointMapper.getCheckpointsUsingProblemId(problemId);
        for (Checkpoint checkpoint : checkpoints) {
            long checkpointId = checkpoint.getCheckpointId();
            { // Standard Input File
                String filePath = String.format("%s/input#%s.txt",
                        new Object[] { checkpointsFilePath, checkpointId });
                FileOutputStream outputStream = new FileOutputStream(new File(filePath));
                String input = checkpoint.getInput();
                IOUtils.write(input, outputStream);
                IOUtils.closeQuietly(outputStream);
            }
            { // Standard Output File
                String filePath = String.format("%s/output#%s.txt",
                        new Object[] { checkpointsFilePath, checkpointId });
                FileOutputStream outputStream = new FileOutputStream(new File(filePath));
                String output = checkpoint.getOutput();
                IOUtils.write(output, outputStream);
                IOUtils.closeQuietly(outputStream);
            }
        }
    }

    /**
     * CheckpointMapper.
     * ?.
     */
    @Autowired
    private CheckpointMapper checkpointMapper;

    /**
     * .
     * ?.
     */
    @Value("${judger.checkpointDir}")
    private String checkpointDirectory;
}