ExportProblemsAction.java

/*
 * Copyright 2007 Zhang, Zheng <oldbig@gmail.com>
 * 
 * This file is part of ZOJ.
 * 
 * ZOJ 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 revision 3 of the License, or (at your option) any later revision.
 * 
 * ZOJ 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 ZOJ. if not, see
 * <http://www.gnu.org/licenses/>.
 */

package cn.edu.zju.acm.onlinejudge.action;

import java.io.IOException;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import javax.servlet.http.HttpServletResponse;

import org.apache.struts.action.ActionForm;
import org.apache.struts.action.ActionForward;
import org.apache.struts.action.ActionMapping;

import cn.edu.zju.acm.onlinejudge.bean.AbstractContest;
import cn.edu.zju.acm.onlinejudge.bean.Problem;
import cn.edu.zju.acm.onlinejudge.bean.Reference;
import cn.edu.zju.acm.onlinejudge.bean.enumeration.ReferenceType;
import cn.edu.zju.acm.onlinejudge.persistence.ReferencePersistence;
import cn.edu.zju.acm.onlinejudge.util.ContestManager;
import cn.edu.zju.acm.onlinejudge.util.PersistenceManager;

/**
 * <p>
 * ExportProblemsAction
 * </p>
 * 
 * 
 * @author Zhang, Zheng
 * @version 2.0
 */
public class ExportProblemsAction extends BaseAction {

    /**
     * <p>
     * Default constructor.
     * </p>
     */
    public ExportProblemsAction() {
    // empty
    }

    /**
     * ExportProblemsAction.
     * 
     * @param mapping
     *            action mapping
     * @param form
     *            action form
     * @param request
     *            http servlet request
     * @param response
     *            http servlet response
     * 
     * @return action forward instance
     * 
     * @throws Exception
     *             any errors happened
     */
    @Override
    public ActionForward execute(ActionMapping mapping, ActionForm form, ContextAdapter context) throws Exception {

        // check contest
        boolean isProblemset = context.getRequest().getRequestURI().endsWith("exportProblems.do");

        ActionForward forward = this.checkContestAdminPermission(mapping, context, isProblemset, false);
        if (forward != null) {
            return forward;
        }

        AbstractContest contest = context.getContest();
        List<Problem> problems = ContestManager.getInstance().getContestProblems(contest.getId());

        HttpServletResponse response = context.getResponse();
        response.setContentType("application/zip");
        response.setHeader("Content-disposition", "attachment; filename=" + this.convertFilename(contest.getTitle()) +
            ".zip");

        ZipOutputStream out = null;
        StringBuilder sb = new StringBuilder();
        try {
            out = new ZipOutputStream(response.getOutputStream());
            for (Object obj : problems) {
                Problem p = (Problem) obj;
                this.zipReference(p, "text", ReferenceType.DESCRIPTION, out);
                this.zipReference(p, "input", ReferenceType.INPUT, out);
                this.zipReference(p, "output", ReferenceType.OUTPUT, out);
                this.zipReference(p, "solution", ReferenceType.JUDGE_SOLUTION, out);
                this.zipReference(p, "checker", ReferenceType.HEADER, out);
                this.zipReference(p, "checker", ReferenceType.CHECKER_SOURCE, out);

                
                sb.append(toCsvString(p.getCode())).append(",");
                sb.append(toCsvString(p.getTitle())).append(",");
                sb.append(p.isChecker()).append(",");
                sb.append(p.getLimit().getTimeLimit()).append(",");
                sb.append(p.getLimit().getMemoryLimit()).append(",");
                sb.append(p.getLimit().getOutputLimit()).append(",");
                sb.append(p.getLimit().getSubmissionLimit()).append(",");
                sb.append(toCsvString(p.getAuthor())).append(",");
                sb.append(toCsvString(p.getSource())).append(",");
                sb.append(toCsvString(p.getContest())).append("\n");
            }
            out.putNextEntry(new ZipEntry("problems.csv"));
            out.write(sb.toString().getBytes());

        } catch (IOException e) {
            throw e;
        } finally {
            if (out != null) {
                out.close();
            }
        }

        return null;

    }
    private String toCsvString(String s) {
    	if (s == null) {
    		return "";
    	} else if (s.contains(",")) {
    		return "\"" + s.replace("\"", "\"\"") + "\"";
    	} else {
    		return s;
    	}
    	
    }

    private void zipReference(Problem p, String fileName, ReferenceType type, ZipOutputStream out) throws Exception {
        ReferencePersistence referencePersistence = PersistenceManager.getInstance().getReferencePersistence();
        List<Reference> refs = referencePersistence.getProblemReferences(p.getId(), type);

        if (refs.size() == 0) {
            return;
        }
        Reference ref = refs.get(0);
        if (type == ReferenceType.CHECKER_SOURCE || type == ReferenceType.JUDGE_SOLUTION) {
            String contentType = ref.getContentType();
            if (contentType == null) {
                contentType = "cc";
            }
            fileName += "." + contentType;
        }
        out.putNextEntry(new ZipEntry(p.getCode() + "/" + fileName));

        byte[] data = ref.getContent();
        out.write(data);
        out.closeEntry();
    }

    private String convertFilename(String name) {
        char[] s = name.toCharArray();
        for (int i = 0; i < s.length; i++) {
            if (Character.isSpaceChar(s[i])) {
                s[i] = '_';
            }
        }
        return new String(s);
    }
}