Add support for getAggregateTorusKey

torusresearch · Mar 12, 2021 · e1642dc · e1642dc
1 parent c83e775
commit e1642dc
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 1 deletion.
diff --git a/torusdirect/build.gradle b/torusdirect/build.gradle
@@ -11,7 +11,7 @@ android {
         minSdkVersion 21
         targetSdkVersion 30
         versionCode 1
-        versionName "1.2.0"
+        versionName "1.3.0"
 
         testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner"
         consumerProguardFiles 'consumer-rules.pro'

diff --git a/torusdirect/src/main/java/org/torusresearch/torusdirect/TorusDirectSdk.java b/torusdirect/src/main/java/org/torusresearch/torusdirect/TorusDirectSdk.java
@@ -20,6 +20,7 @@
 import org.torusresearch.torusdirect.types.TorusKey;
 import org.torusresearch.torusdirect.types.TorusLoginResponse;
 import org.torusresearch.torusdirect.types.TorusNetwork;
+import org.torusresearch.torusdirect.types.TorusSubVerifierInfo;
 import org.torusresearch.torusdirect.types.TorusVerifierResponse;
 import org.torusresearch.torusdirect.types.TorusVerifierUnionResponse;
 import org.torusresearch.torusdirect.utils.Helpers;
@@ -158,4 +159,29 @@ public CompletableFuture<TorusKey> getTorusKey(String verifier, String verifierI
             return new TorusKey(shareResponse.getPrivKey(), shareResponse.getEthAddress());
         });
     }
+
+    public CompletableFuture<TorusKey> getAggregateTorusKey(String verifier, String verifierId, TorusSubVerifierInfo[] subVerifierInfoArray) {
+        AggregateVerifierParams aggregateVerifierParams = new AggregateVerifierParams();
+        aggregateVerifierParams.setVerify_params(new AggregateVerifierParams.VerifierParams[subVerifierInfoArray.length]);
+        aggregateVerifierParams.setSub_verifier_ids(new String[subVerifierInfoArray.length]);
+        List<String> aggregateIdTokenSeeds = new ArrayList<>();
+        String aggregateVerifierId = "";
+        for (int i = 0; i < subVerifierInfoArray.length; i++) {
+            TorusSubVerifierInfo userInfo = subVerifierInfoArray[i];
+            String finalToken = userInfo.getIdToken();
+            aggregateVerifierParams.setVerifyParamItem(new AggregateVerifierParams.VerifierParams(verifierId, finalToken), i);
+            aggregateVerifierParams.setSubVerifierIdItem(userInfo.getVerifier(), i);
+            aggregateIdTokenSeeds.add(finalToken);
+            aggregateVerifierId = verifierId;
+        }
+        Collections.sort(aggregateIdTokenSeeds);
+        String aggregateTokenString = TextUtils.join(Character.toString((char) 29), aggregateIdTokenSeeds);
+        String aggregateIdToken = Hash.sha3String(aggregateTokenString).substring(2);
+        aggregateVerifierParams.setVerifier_id(aggregateVerifierId);
+        HashMap<String, Object> aggregateVerifierParamsHashMap = new HashMap<>();
+        aggregateVerifierParamsHashMap.put("verify_params", aggregateVerifierParams.getVerify_params());
+        aggregateVerifierParamsHashMap.put("sub_verifier_ids", aggregateVerifierParams.getSub_verifier_ids());
+        aggregateVerifierParamsHashMap.put("verifier_id", aggregateVerifierParams.getVerifier_id());
+        return this.getTorusKey(verifier, aggregateVerifierId, aggregateVerifierParamsHashMap, aggregateIdToken);
+    }
 }
diff --git a/torusdirect/src/main/java/org/torusresearch/torusdirect/types/TorusSubVerifierInfo.java b/torusdirect/src/main/java/org/torusresearch/torusdirect/types/TorusSubVerifierInfo.java
@@ -0,0 +1,19 @@
+package org.torusresearch.torusdirect.types;
+
+public class TorusSubVerifierInfo {
+    String verifier;
+    String idToken;
+
+    public TorusSubVerifierInfo(String verifier, String idToken) {
+        this.verifier = verifier;
+        this.idToken = idToken;
+    }
+
+    public String getVerifier() {
+        return verifier;
+    }
+
+    public String getIdToken() {
+        return idToken;
+    }
+}