Add explicit PrivateKeyWallet save/load/export (#62)

Author: 0xFirekeeper

Thirdweb.Tests/Thirdweb.Wallets/Thirdweb.PrivateKeyWallet.Tests.cs

@@ -364,4 +364,70 @@ public async Task ExecuteTransaction_InvalidOperation()
         };
         _ = await Assert.ThrowsAsync<InvalidOperationException>(() => account.ExecuteTransaction(transaction));
     }
+
+    [Fact(Timeout = 120000)]
+    public async Task LoadOrGenerate_LoadsExistingWallet()
+    {
+        // Generate and save a wallet to simulate an existing wallet file
+        var wallet = await PrivateKeyWallet.Generate(this.Client);
+        await wallet.Save();
+
+        var loadedWallet = await PrivateKeyWallet.LoadOrGenerate(this.Client);
+
+        Assert.NotNull(loadedWallet);
+        Assert.Equal(await wallet.Export(), await loadedWallet.Export());
+        Assert.Equal(await wallet.GetAddress(), await loadedWallet.GetAddress());
+
+        // Clean up
+        var path = PrivateKeyWallet.GetSavePath();
+        if (File.Exists(path))
+        {
+            File.Delete(path);
+        }
+    }
+
+    [Fact(Timeout = 120000)]
+    public async Task LoadOrGenerate_GeneratesNewWalletIfNoExistingWallet()
+    {
+        var path = PrivateKeyWallet.GetSavePath();
+
+        if (File.Exists(path))
+        {
+            File.Delete(path);
+        }
+
+        var wallet = await PrivateKeyWallet.LoadOrGenerate(this.Client);
+
+        Assert.NotNull(wallet);
+        Assert.NotNull(await wallet.Export());
+        Assert.False(File.Exists(path));
+    }
+
+    [Fact(Timeout = 120000)]
+    public async Task Save_SavesPrivateKeyToFile()
+    {
+        var wallet = await PrivateKeyWallet.Generate(this.Client);
+
+        await wallet.Save();
+
+        var path = PrivateKeyWallet.GetSavePath();
+        Assert.True(File.Exists(path));
+
+        var savedPrivateKey = await File.ReadAllTextAsync(path);
+        Assert.Equal(await wallet.Export(), savedPrivateKey);
+
+        // Clean up
+        File.Delete(path);
+    }
+
+    [Fact(Timeout = 120000)]
+    public async Task Export_ReturnsPrivateKey()
+    {
+        var wallet = await PrivateKeyWallet.Generate(this.Client);
+
+        var privateKey = await wallet.Export();
+
+        Assert.NotNull(privateKey);
+        Assert.Equal(privateKey, await wallet.Export());
+    }
 }

Thirdweb/Thirdweb.Wallets/InAppWallet/EmbeddedWallet.Storage/LocalStorage.cs

@@ -12,37 +12,37 @@ internal abstract class LocalStorageBase
 
 internal partial class LocalStorage : LocalStorageBase
 {
-    internal override DataStorage Data => this.storage.Data;
-    private readonly Storage storage;
-    private readonly string filePath;
+    internal override DataStorage Data => this._storage.Data;
+    private readonly Storage _storage;
+    private readonly string _filePath;
 
     internal LocalStorage(string clientId, string storageDirectoryPath = null)
     {
         string directory;
         directory = storageDirectoryPath ?? Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData);
         directory = Path.Combine(directory, "Thirdweb", "InAppWallet");
         _ = Directory.CreateDirectory(directory);
-        this.filePath = Path.Combine(directory, $"{clientId}.txt");
+        this._filePath = Path.Combine(directory, $"{clientId}.txt");
         try
         {
-            var json = File.ReadAllBytes(this.filePath);
+            var json = File.ReadAllBytes(this._filePath);
             DataContractJsonSerializer serializer = new(typeof(Storage));
             MemoryStream fin = new(json);
-            this.storage = (Storage)serializer.ReadObject(fin);
+            this._storage = (Storage)serializer.ReadObject(fin);
         }
         catch (Exception)
         {
-            this.storage = new Storage();
+            this._storage = new Storage();
         }
     }
 
     internal override Task RemoveAuthTokenAsync()
     {
         return this.UpdateDataAsync(() =>
         {
-            if (this.storage.Data?.AuthToken != null)
+            if (this._storage.Data?.AuthToken != null)
             {
-                this.storage.Data.ClearAuthToken();
+                this._storage.Data.ClearAuthToken();
                 return true;
             }
             return false;
@@ -55,8 +55,8 @@ private async Task<bool> UpdateDataAsync(Func<bool> fn)
         {
             DataContractJsonSerializer serializer = new(typeof(Storage));
             MemoryStream fout = new();
-            serializer.WriteObject(fout, this.storage);
-            await File.WriteAllBytesAsync(this.filePath, fout.ToArray()).ConfigureAwait(false);
+            serializer.WriteObject(fout, this._storage);
+            await File.WriteAllBytesAsync(this._filePath, fout.ToArray()).ConfigureAwait(false);
             return true;
         }
         return false;
@@ -66,7 +66,7 @@ internal override Task SaveDataAsync(DataStorage data)
     {
         return this.UpdateDataAsync(() =>
         {
-            this.storage.Data = data;
+            this._storage.Data = data;
             return true;
         });
     }