Merge pull request #18 from AugustNagro/gus-entry-setters

Added Expiration Time Setters
jorabin · Jan 24, 2018 · 8b469ca · 8b469ca
2 parents dcc3dab + b9232a4
commit 8b469ca
Show file tree

Hide file tree

Showing 5 changed files with 51 additions and 2 deletions.
diff --git a/database/src/main/java/org/linguafranca/pwdb/Entry.java b/database/src/main/java/org/linguafranca/pwdb/Entry.java
@@ -297,7 +297,15 @@ interface Matcher {
 
     boolean getExpires();
 
+    void setExpires(boolean expires);
+
     Date getExpiryTime();
 
+    /**
+     * Sets the expiration date of this element. See {@link org.linguafranca.pwdb.Entry#setExpires(boolean)}
+     * @throws IllegalArgumentException if expiryTime is null.
+     */
+    void setExpiryTime(Date expiryTime) throws IllegalArgumentException;
+
     Date getLastModificationTime();
 }
diff --git a/dom/src/main/java/org/linguafranca/pwdb/kdbx/dom/DomEntryWrapper.java b/dom/src/main/java/org/linguafranca/pwdb/kdbx/dom/DomEntryWrapper.java
@@ -180,6 +180,11 @@ public boolean getExpires() {
         return content != null && content.equalsIgnoreCase("true");
     }
 
+    @Override
+    public void setExpires(boolean expires) {
+        DomHelper.setElementContent(DomHelper.EXPIRES_ELEMENT_NAME, element, expires ? "True" : "False");
+    }
+
     @Override
     public Date getExpiryTime() {
         try {
@@ -189,6 +194,13 @@ public Date getExpiryTime() {
         }
     }
 
+    @Override
+    public void setExpiryTime(Date expiryTime) throws IllegalArgumentException {
+        if (expiryTime == null) throw new IllegalArgumentException("expiryTime may not be null");
+        String formatted = DomHelper.dateFormatter.format(expiryTime);
+        DomHelper.setElementContent(DomHelper.EXPIRY_TIME_ELEMENT_NAME, element, formatted);
+    }
+
     @Override
     public Date getLastModificationTime() {
         try {

diff --git a/jaxb/src/main/java/org/linguafranca/pwdb/kdbx/jaxb/JaxbEntry.java b/jaxb/src/main/java/org/linguafranca/pwdb/kdbx/jaxb/JaxbEntry.java
@@ -219,11 +219,22 @@ public boolean getExpires() {
         return delegate.getTimes().getExpires();
     }
 
+    @Override
+    public void setExpires(boolean expires) {
+        delegate.getTimes().setExpires(true);
+    }
+
     @Override
     public Date getExpiryTime() {
         return delegate.getTimes().getExpiryTime();
     }
 
+    @Override
+    public void setExpiryTime(Date expiryTime) throws IllegalArgumentException {
+        if (expiryTime == null) throw new IllegalArgumentException("expiryTime may not be null");
+        delegate.getTimes().setExpiryTime(expiryTime);
+    }
+
     @Override
     public Date getLastModificationTime() {
         return delegate.getTimes().getLastModificationTime();

diff --git a/kdb/src/main/java/org/linguafranca/pwdb/kdb/KdbEntry.java b/kdb/src/main/java/org/linguafranca/pwdb/kdb/KdbEntry.java
@@ -43,6 +43,7 @@ public class KdbEntry extends AbstractEntry<KdbDatabase, KdbGroup, KdbEntry, Kdb
     private Date creationTime = new Date();
     private Date lastModificationTime = new Date();
     private Date lastAccessTime = new Date(Long.MIN_VALUE);
+    private boolean expires = false;
     private Date expiryTime = new Date(Long.MAX_VALUE);
     private String binaryDescription = "";
     private byte[] binaryData = new byte[0];
@@ -174,7 +175,8 @@ public Date getLastAccessTime() {
         return lastAccessTime;
     }
 
-    void setExpiryTime(Date expiryTime) {
+    public void setExpiryTime(Date expiryTime) {
+        if (expiryTime == null) throw new IllegalArgumentException("expiryTime may not be null");
         this.expiryTime = expiryTime;
     }
 
@@ -223,9 +225,14 @@ public List<String> getBinaryPropertyNames() {
         throw new UnsupportedOperationException();
     }
 
+    @Override
+    public void setExpires(boolean expires) {
+        this.expires = expires;
+    }
+
     @Override
     public boolean getExpires() {
-        return false;
+        return expires;
     }
 
     @Override

diff --git a/simple/src/main/java/org/linguafranca/pwdb/kdbx/simple/SimpleEntry.java b/simple/src/main/java/org/linguafranca/pwdb/kdbx/simple/SimpleEntry.java
@@ -219,11 +219,22 @@ public boolean getExpires() {
         return times.getExpires();
     }
 
+    @Override
+    public void setExpires(boolean expires) {
+        times.setExpires(expires);
+    }
+
     @Override
     public Date getExpiryTime() {
         return times.getExpiryTime();
     }
 
+    @Override
+    public void setExpiryTime(Date expiryTime) throws IllegalArgumentException {
+        if (expiryTime == null) throw new IllegalArgumentException("expiryTime may not be null");
+        times.setExpiryTime(expiryTime);
+    }
+
     @Override
     public Date getLastModificationTime() {
         return times.getLastModificationTime();