Make Json store use Jackson instead of Gson (as it's already linked)

[signal-cli] / src / main / java / cli / JsonAxolotlStore.java

diff --git a/src/main/java/cli/JsonAxolotlStore.java b/src/main/java/cli/JsonAxolotlStore.java
index 02b9cdf1bc3ce152922556257c90781f81bc17fc..57282f129d1bc5a87f11cc1957b28f534dd49769 100644 (file)
--- a/src/main/java/cli/JsonAxolotlStore.java
+++ b/src/main/java/cli/JsonAxolotlStore.java
@@ -1,5 +1,8 @@
 package cli;
 
 package cli;
 

+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.fasterxml.jackson.databind.annotation.JsonSerialize;

 import org.json.JSONObject;
 import org.whispersystems.libaxolotl.*;
 import org.whispersystems.libaxolotl.state.AxolotlStore;
 import org.json.JSONObject;
 import org.whispersystems.libaxolotl.*;
 import org.whispersystems.libaxolotl.state.AxolotlStore;


@@ -11,17 +14,35 @@ import java.io.IOException;
 import java.util.List;
 
 class JsonAxolotlStore implements AxolotlStore {
 import java.util.List;
 
 class JsonAxolotlStore implements AxolotlStore {

-    private final JsonPreKeyStore preKeyStore;
-    private final JsonSessionStore sessionStore;
-    private final JsonSignedPreKeyStore signedPreKeyStore;

 
 

-    private final JsonIdentityKeyStore identityKeyStore;
+    @JsonProperty("preKeys")
+    @JsonDeserialize(using = JsonPreKeyStore.JsonPreKeyStoreDeserializer.class)
+    @JsonSerialize(using = JsonPreKeyStore.JsonPreKeyStoreSerializer.class)
+    protected JsonPreKeyStore preKeyStore;

 
 

-    public JsonAxolotlStore(JSONObject jsonAxolotl) throws IOException, InvalidKeyException {
-        this.preKeyStore = new JsonPreKeyStore(jsonAxolotl.getJSONArray("preKeys"));
-        this.sessionStore = new JsonSessionStore(jsonAxolotl.getJSONArray("sessionStore"));
-        this.signedPreKeyStore = new JsonSignedPreKeyStore(jsonAxolotl.getJSONArray("signedPreKeyStore"));
-        this.identityKeyStore = new JsonIdentityKeyStore(jsonAxolotl.getJSONObject("identityKeyStore"));
+    @JsonProperty("sessionStore")
+    @JsonDeserialize(using = JsonSessionStore.JsonSessionStoreDeserializer.class)
+    @JsonSerialize(using = JsonSessionStore.JsonPreKeyStoreSerializer.class)
+    protected JsonSessionStore sessionStore;
+
+    @JsonProperty("signedPreKeyStore")
+    @JsonDeserialize(using = JsonSignedPreKeyStore.JsonSignedPreKeyStoreDeserializer.class)
+    @JsonSerialize(using = JsonSignedPreKeyStore.JsonSignedPreKeyStoreSerializer.class)
+    protected JsonSignedPreKeyStore signedPreKeyStore;
+
+    @JsonProperty("identityKeyStore")
+    @JsonDeserialize(using = JsonIdentityKeyStore.JsonIdentityKeyStoreDeserializer.class)
+    @JsonSerialize(using = JsonIdentityKeyStore.JsonIdentityKeyStoreSerializer.class)
+    protected JsonIdentityKeyStore identityKeyStore;
+
+    public JsonAxolotlStore() {
+    }
+
+    public JsonAxolotlStore(JsonPreKeyStore preKeyStore, JsonSessionStore sessionStore, JsonSignedPreKeyStore signedPreKeyStore, JsonIdentityKeyStore identityKeyStore) {
+        this.preKeyStore = preKeyStore;
+        this.sessionStore = sessionStore;
+        this.signedPreKeyStore = signedPreKeyStore;
+        this.identityKeyStore = identityKeyStore;

     }
 
     public JsonAxolotlStore(IdentityKeyPair identityKeyPair, int registrationId) {
     }
 
     public JsonAxolotlStore(IdentityKeyPair identityKeyPair, int registrationId) {


@@ -31,13 +52,6 @@ class JsonAxolotlStore implements AxolotlStore {
         this.identityKeyStore = new JsonIdentityKeyStore(identityKeyPair, registrationId);
     }
 
         this.identityKeyStore = new JsonIdentityKeyStore(identityKeyPair, registrationId);
     }
 

-    public JSONObject getJson() {
-        return new JSONObject().put("preKeys", preKeyStore.getJson())
-                .put("sessionStore", sessionStore.getJson())
-                .put("signedPreKeyStore", signedPreKeyStore.getJson())
-                .put("identityKeyStore", identityKeyStore.getJson());
-    }
-

     @Override
     public IdentityKeyPair getIdentityKeyPair() {
         return identityKeyStore.getIdentityKeyPair();
     @Override
     public IdentityKeyPair getIdentityKeyPair() {
         return identityKeyStore.getIdentityKeyPair();