Attempt to register session with WaaS

0xsequence · Nov 16, 2023 · 0d1bac6 · 0d1bac6
1 parent f70c871
commit 0d1bac6
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 11 deletions.
diff --git a/Assets/SequenceSDK/WaaS/HttpClient.cs b/Assets/SequenceSDK/WaaS/HttpClient.cs
@@ -19,8 +19,9 @@ public class HttpClient
             NullValueHandling = NullValueHandling.Ignore
         };
 
-        public HttpClient()
+        public HttpClient(string url)
         {
+            _url = url;
             this._defaultHeaders = new Dictionary<string, string>();
         }
 

diff --git a/Assets/SequenceSDK/WaaS/WaaSLogin.cs b/Assets/SequenceSDK/WaaS/WaaSLogin.cs
@@ -2,14 +2,16 @@
 using System.Threading.Tasks;
 using Amazon.CognitoIdentity.Model;
 using Sequence.Authentication;
+using Sequence.Extensions;
+using Sequence.Utils;
 using Sequence.Wallet;
 using UnityEngine;
 
 namespace Sequence.WaaS
 {
     public class WaaSLogin : ILogin 
     {
-        public static readonly string WaaSLoginUrl = "https://d14tu8valot5m0.cloudfront.net/rpc/WaaSAuthenticator/";
+        public static readonly string WaaSLoginUrl = "https://d14tu8valot5m0.cloudfront.net/rpc/WaasAuthenticator";
 
         private AWSConfig _awsConfig;
         private string _waasProjectId;
@@ -88,18 +90,26 @@ public async Task ConnectToWaaS(string idToken)
 
             EthWallet sessionWallet = new EthWallet();
 
-            string signedPayled = await PrepareSignedPayload(sessionWallet, idToken, dataKey);
+            string loginPayload = AssembleLoginPayloadJson(idToken, sessionWallet);
+            string payloadCiphertext = await PrepareEncryptedPayload(sessionWallet, idToken, dataKey, loginPayload);
+            string signedPayload = await sessionWallet.SignMessage(loginPayload);
 
-            // Todo register session with waas rpc
-            Debug.LogError("Signed payload: " + signedPayled);
+            try
+            {
+                (string sessionId, string dataWallet) = await RegisterSession(dataKey.Ciphertext.ByteArrayToHexStringWithPrefix(), payloadCiphertext, signedPayload);
+                Debug.LogError($"Session ID: {sessionId} | Data Wallet: {dataWallet}");
+            }
+            catch (Exception e)
+            {
+                OnLoginFailed?.Invoke("Error registering session: " + e.Message);
+                return;
+            }
         }
 
-        private async Task<string> PrepareSignedPayload(Wallet.IWallet sessionWallet, string idToken, DataKey dataKey)
+        private async Task<string> PrepareEncryptedPayload(Wallet.IWallet sessionWallet, string idToken, DataKey dataKey, string loginPayload)
         {
-            string loginPayload = AssembleLoginPayloadJson(idToken, sessionWallet);
             byte[] encryptedPayload = Encryptor.AES256CBCEncryption(dataKey.Plaintext, loginPayload);
-            string signedPayled = await sessionWallet.SignMessage(encryptedPayload);
-            return signedPayled;
+            return encryptedPayload.ByteArrayToHexStringWithPrefix();
         }
 
         private string AssembleLoginPayloadJson(string idToken, Wallet.IWallet sessionWallet)
@@ -108,10 +118,18 @@ private string AssembleLoginPayloadJson(string idToken, Wallet.IWallet sessionWa
                 sessionWallet.GetAddress(), idToken);
             string intentJson = JsonUtility.ToJson(intent);
             WaaSLoginPayload payload = new WaaSLoginPayload(_waasProjectId, idToken, sessionWallet.GetAddress(),
-                "UserWallet", intentJson);
+                "FRIENDLY SESSION WALLET", intentJson);
             string payloadJson = JsonUtility.ToJson(payload);
             return payloadJson;
         }
+
+        private async Task<(string, string)> RegisterSession(string encryptedPayloadKey, string payloadCiphertext, string signedPayload) 
+        {
+            HttpClient client = new HttpClient(WaaSLoginUrl);
+            RegisterSessionPayload payload = new RegisterSessionPayload(encryptedPayloadKey, payloadCiphertext, signedPayload);
+            (string sessionId, string dataWallet) = await client.SendRequest<RegisterSessionPayload, (string, string)>("RegisterSession", payload);
+            return (sessionId, dataWallet);
+        }
     }
 
     [Serializable]
@@ -169,4 +187,19 @@ public WaaSLoginPayload(string projectId, string idToken, string sessionAddress,
             intentJSON = intentJson;
         }
     }
+
+    [Serializable]
+    public class RegisterSessionPayload
+    {
+        public string encryptedPayloadKey;
+        public string payloadCiphertext;
+        public string payloadSig;
+
+        public RegisterSessionPayload(string encryptedPayloadKey, string payloadCiphertext, string payloadSig)
+        {
+            this.encryptedPayloadKey = encryptedPayloadKey;
+            this.payloadCiphertext = payloadCiphertext;
+            this.payloadSig = payloadSig;
+        }
+    }
 }
diff --git a/Assets/SequenceSDK/WaaS/WaaSWallet.cs b/Assets/SequenceSDK/WaaS/WaaSWallet.cs
@@ -10,7 +10,7 @@ public class WaaSWallet : IWallet
 
         public WaaSWallet(string jwt)
         {
-            this._httpClient = new HttpClient();
+            this._httpClient = new HttpClient("https://next-api.sequence.app/rpc/Wallet");
             this._address = JwtHelper.GetWalletAddressFromJwt(jwt);
             this._httpClient.AddDefaultHeader("Authorization", $"Bearer {jwt}");
         }