/*
* 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);
}
}