From d22fc152197edefc923655d9b15bbf265efea3b4 Mon Sep 17 00:00:00 2001
From: Brandon Berhent <bbedward@gmail.com>
Date: Tue, 9 Jul 2024 10:17:19 -0400
Subject: [PATCH] Add repair adhoc option

---
 apps/cli/main.go | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/apps/cli/main.go b/apps/cli/main.go
index 6047e04..bdb3cd7 100644
--- a/apps/cli/main.go
+++ b/apps/cli/main.go
@@ -160,6 +160,7 @@ func main() {
 	accountIndex := accountCmd.Int("index", 0, "Specify an index to use when creating account (optional, cannot be used with --count)")
 	accountCount := accountCmd.Int("count", 0, "Specify how many accounts to create (optional, cannot be used with --index)")
 	accountKey := accountCmd.String("key", "", "Specify a private key to use when creating account (optional, cannot be used with --index or --count)")
+	repairAdhoc := accountCmd.Bool("repair-adhoc", false, "Repair adhoc accounts")
 
 	if *showHelp {
 		usage()
@@ -459,6 +460,36 @@ func main() {
 				}
 				fmt.Printf("Account %d created: %s\n", *acc.AccountIndex, acc.Address)
 			}
+		} else if *repairAdhoc {
+			// Get all accounts with 64-length private keys
+			accounts, err := nanoWallet.DB.Account.Query().Where(account.PrivateKeyNotNil()).All(ctx)
+			if err != nil {
+				fmt.Printf("Failed to get accounts: %v\n", err)
+				os.Exit(1)
+			}
+			for _, a := range accounts {
+				if len(*a.PrivateKey) != 64 {
+					continue
+				}
+				// Convert to ed25519 key
+				asByte, err := hex.DecodeString(*a.PrivateKey)
+				if err != nil {
+					fmt.Printf("Failed to decode private key: %v\n", err)
+					os.Exit(1)
+				}
+				priv, err := ed25519.NewKeyFromSeed(asByte)
+				if err != nil {
+					fmt.Printf("Failed to create private key: %v\n", err)
+					os.Exit(1)
+				}
+				// Update account
+				_, err = nanoWallet.DB.Account.UpdateOneID(a.ID).SetPrivateKey(hex.EncodeToString(priv)).Save(ctx)
+				if err != nil {
+					fmt.Printf("Failed to update account: %v\n", err)
+					os.Exit(1)
+				}
+				fmt.Printf("Account %s repaired\n", a.Address)
+			}
 		}
 	default:
 		fmt.Println("expected 'foo' or 'bar' subcommands")