src/club/wpia/gigi/output/CertificateIterable.java

   1 package club.wpia.gigi.output;
   2
   3 import java.io.IOException;
   4 import java.security.GeneralSecurityException;
   5 import java.security.cert.X509Certificate;
   6 import java.util.Date;
   7 import java.util.Map;
   8
   9 import club.wpia.gigi.dbObjects.Certificate;
  10 import club.wpia.gigi.dbObjects.Certificate.CertificateStatus;
  11 import club.wpia.gigi.localisation.Language;
  12 import club.wpia.gigi.output.template.IterableDataset;
  13 import club.wpia.gigi.util.CalendarUtil;
  14
  15 public class CertificateIterable implements IterableDataset {
  16
  17     private Certificate[] certificates;
  18
  19     public static final int EXPIRING_IN_DAYS = 14;
  20
  21     public CertificateIterable(Certificate[] certificates) {
  22         this.certificates = certificates;
  23     }
  24
  25     private int i = 0;
  26
  27     @Override
  28     public boolean next(Language l, Map<String, Object> vars) {
  29         if (i >= certificates.length) {
  30             return false;
  31         }
  32         Certificate c = certificates[i++];
  33         vars.put("state", c.getStatus());
  34         vars.put("CN", c.getDistinguishedName());
  35         vars.put("serial", c.getSerial());
  36         vars.put("digest", c.getMessageDigest());
  37         vars.put("profile", c.getProfile().getVisibleName());
  38         CertificateStatus st = c.getStatus();
  39         vars.put("revokable", st != CertificateStatus.REVOKED && st == CertificateStatus.ISSUED);
  40         String issuedWarning = "";
  41         String expiredWarning = "";
  42         vars.put("issued", l.getTranslation("N/A"));
  43         vars.put("expire", l.getTranslation("N/A"));
  44         vars.put("classIssued", "");
  45         vars.put("classExpired", "");
  46         vars.put("revoked", l.getTranslation("N/A"));
  47
  48         try {
  49             if (st == CertificateStatus.ISSUED || st == CertificateStatus.REVOKED) {
  50                 X509Certificate cert = c.cert();
  51                 vars.put("issued", cert.getNotBefore());
  52                 vars.put("expire", cert.getNotAfter());
  53
  54                 if (cert.getNotBefore().after(new Date())) {
  55                     issuedWarning = "bg-warning";
  56                 }
  57                 vars.put("classIssued", issuedWarning);
  58
  59                 if (cert.getNotAfter().before(CalendarUtil.timeDifferenceDays(EXPIRING_IN_DAYS))) {
  60                     expiredWarning = "bg-warning";
  61                 }
  62                 if (cert.getNotAfter().before(new Date())) {
  63                     expiredWarning = "bg-danger";
  64                 }
  65                 vars.put("classExpired", expiredWarning);
  66             }
  67
  68             if (st == CertificateStatus.REVOKED) {
  69                 vars.put("revoked", c.getRevocationDate());
  70             }
  71         } catch (IOException e) {
  72             e.printStackTrace();
  73         } catch (GeneralSecurityException e) {
  74             e.printStackTrace();
  75         }
  76         vars.put("login", c.isLoginEnabled());
  77         return true;
  78     }
  79 }