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         vars.put("description", c.getDescription());
  40         CertificateStatus st = c.getStatus();
  41         vars.put("revokable", st != CertificateStatus.REVOKED && st == CertificateStatus.ISSUED);
  42         String issuedWarning = "";
  43         String expiredWarning = "";
  44         vars.put("issued", l.getTranslation("N/A"));
  45         vars.put("expire", l.getTranslation("N/A"));
  46         vars.put("classIssued", "");
  47         vars.put("classExpired", "");
  48         vars.put("revoked", l.getTranslation("N/A"));
  49
  50         try {
  51             if (st == CertificateStatus.ISSUED || st == CertificateStatus.REVOKED) {
  52                 X509Certificate cert = c.cert();
  53                 vars.put("issued", cert.getNotBefore());
  54                 vars.put("expire", cert.getNotAfter());
  55
  56                 if (cert.getNotBefore().after(new Date())) {
  57                     issuedWarning = "bg-warning";
  58                 }
  59                 vars.put("classIssued", issuedWarning);
  60
  61                 if (cert.getNotAfter().before(CalendarUtil.timeDifferenceDays(EXPIRING_IN_DAYS))) {
  62                     expiredWarning = "bg-warning";
  63                 }
  64                 if (cert.getNotAfter().before(new Date())) {
  65                     expiredWarning = "bg-danger";
  66                 }
  67                 vars.put("classExpired", expiredWarning);
  68             }
  69
  70             if (st == CertificateStatus.REVOKED) {
  71                 vars.put("revoked", c.getRevocationDate());
  72             }
  73         } catch (GigiApiException e) {
  74             e.printStackTrace();
  75         } catch (IOException e) {
  76             e.printStackTrace();
  77         } catch (GeneralSecurityException e) {
  78             e.printStackTrace();
  79         }
  80         vars.put("login", c.isLoginEnabled());
  81         return true;
  82     }
  83 }