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