src/club/wpia/gigi/pages/account/certs/CertificateModificationForm.java

   1 package club.wpia.gigi.pages.account.certs;
   2
   3 import java.io.PrintWriter;
   4 import java.util.LinkedList;
   5 import java.util.Map;
   6
   7 import javax.servlet.http.HttpServletRequest;
   8
   9 import club.wpia.gigi.GigiApiException;
  10 import club.wpia.gigi.dbObjects.Certificate;
  11 import club.wpia.gigi.dbObjects.Certificate.RevocationType;
  12 import club.wpia.gigi.dbObjects.CertificateOwner;
  13 import club.wpia.gigi.dbObjects.Job;
  14 import club.wpia.gigi.localisation.Language;
  15 import club.wpia.gigi.output.CertificateIterable;
  16 import club.wpia.gigi.output.template.Form;
  17 import club.wpia.gigi.output.template.Template;
  18 import club.wpia.gigi.pages.LoginPage;
  19
  20 public class CertificateModificationForm extends Form {
  21
  22     private CertificateOwner target;
  23
  24     private final boolean withRevoked;
  25
  26     public CertificateModificationForm(HttpServletRequest hsr, boolean withRevoked) {
  27         super(hsr);
  28         this.withRevoked = withRevoked;
  29         target = LoginPage.getAuthorizationContext(hsr).getTarget();
  30     }
  31
  32     private static final Template certTable = new Template(CertificateIterable.class.getResource("CertificateTable.templ"));
  33
  34     private static final Template myTemplate = new Template(CertificateModificationForm.class.getResource("CertificateModificationForm.templ"));
  35
  36     @Override
  37     public SubmissionResult submit(HttpServletRequest req) throws GigiApiException {
  38         String action = req.getParameter("action");
  39         if ( !"revoke".equals(action)) {
  40             throw new GigiApiException("Incorrect action given.");
  41         }
  42         String[] certs = req.getParameterValues("certs[]");
  43         if (certs == null) {
  44             throw new GigiApiException("No certificates to revoke.");
  45         }
  46         LinkedList<Job> revokes = new LinkedList<Job>();
  47         for (String serial : certs) {
  48             Certificate c = Certificate.getBySerial(serial);
  49             if (c == null || c.getOwner() != target) {
  50                 continue;
  51             }
  52             revokes.add(c.revoke(RevocationType.SUPPORT));
  53         }
  54         long start = System.currentTimeMillis();
  55         for (Job job : revokes) {
  56             int toWait = (int) (60000 + start - System.currentTimeMillis());
  57             if (toWait > 0) {
  58                 job.waitFor(toWait);
  59             } else {
  60                 break; // canceled... waited too log
  61             }
  62         }
  63         return new RedirectResult(req.getPathInfo());
  64     }
  65
  66     @Override
  67     protected void outputContent(PrintWriter out, Language l, Map<String, Object> vars) {
  68         vars.put("certs", new CertificateIterable(target.getCertificates(withRevoked)));
  69         vars.put("certTable", certTable);
  70         if (withRevoked) {
  71             vars.put("all", "btn-primary");
  72             vars.put("current", "btn-info");
  73         } else {
  74             vars.put("all", "btn-info");
  75             vars.put("current", "btn-primary");
  76         }
  77         myTemplate.output(out, l, vars);
  78     }
  79
  80 }