From 81598083e7c822964815bdb1e26d6c62760e52b8 Mon Sep 17 00:00:00 2001
From: Matt Corallo <matt@bluematt.me>
Date: Wed, 29 Jun 2011 01:08:48 +0200
Subject: [PATCH] Dynamically remove/insert the Options for encryption in the menus.

---
 share/uiproject.fbp |    4 ++--
 src/ui.cpp          |    8 ++++++++
 src/uibase.cpp      |    2 --
 src/uibase.h        |    2 ++
 4 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/share/uiproject.fbp b/share/uiproject.fbp
index cc735c6..d5e6b1e 100644
--- a/share/uiproject.fbp
+++ b/share/uiproject.fbp
@@ -171,7 +171,7 @@
                         <property name="kind">wxITEM_NORMAL</property>
                         <property name="label">&amp;Encrypt Wallet...</property>
                         <property name="name">m_menuOptionsEncryptWallet</property>
-                        <property name="permission">none</property>
+                        <property name="permission">public</property>
                         <property name="shortcut"></property>
                         <property name="unchecked_bitmap"></property>
                         <event name="OnMenuSelection">OnMenuOptionsEncryptWallet</event>
@@ -186,7 +186,7 @@
                         <property name="kind">wxITEM_NORMAL</property>
                         <property name="label">&amp;Change Wallet Encryption Passphrase...</property>
                         <property name="name">m_menuOptionsChangeWalletPassphrase</property>
-                        <property name="permission">none</property>
+                        <property name="permission">public</property>
                         <property name="shortcut"></property>
                         <property name="unchecked_bitmap"></property>
                         <event name="OnMenuSelection">OnMenuOptionsChangeWalletPassphrase</event>
diff --git a/src/ui.cpp b/src/ui.cpp
index 52315e1..9e099ae 100644
--- a/src/ui.cpp
+++ b/src/ui.cpp
@@ -368,6 +368,11 @@ CMainFrame::CMainFrame(wxWindow* parent) : CMainFrameBase(parent)
     if (CWalletDB(pwalletMain->strWalletFile,"r").ReadDefaultKey(vchPubKey))
         m_textCtrlAddress->SetValue(PubKeyToAddress(vchPubKey));
 
+    if (pwalletMain->IsCrypted())
+        m_menuOptions->Remove(m_menuOptionsEncryptWallet);
+    else
+        m_menuOptions->Remove(m_menuOptionsChangeWalletPassphrase);
+
     // Fill listctrl with wallet transactions
     RefreshListCtrl();
 }
@@ -1217,6 +1222,9 @@ void CMainFrame::OnMenuOptionsEncryptWallet(wxCommandEvent& event)
     munlock(&strWalletPass[0], strWalletPass.capacity());
     munlock(&strWalletPassTest[0], strWalletPassTest.capacity());
     wxMessageBox(_("Wallet Encrypted.\nRemember that encrypting your wallet cannot fully protect your bitcoins from being stolen by malware infecting your computer."), "Bitcoin");
+
+    m_menuOptions->Remove(m_menuOptionsEncryptWallet);
+    m_menuOptions->Insert(m_menuOptions->GetMenuItemCount() - 1, m_menuOptionsChangeWalletPassphrase);
 }
 
 void CMainFrame::OnMenuOptionsChangeWalletPassphrase(wxCommandEvent& event)
diff --git a/src/uibase.cpp b/src/uibase.cpp
index 9ac4ec8..18eec44 100644
--- a/src/uibase.cpp
+++ b/src/uibase.cpp
@@ -32,11 +32,9 @@ CMainFrameBase::CMainFrameBase( wxWindow* parent, wxWindowID id, const wxString&
 	m_menuOptionsChangeYourAddress = new wxMenuItem( m_menuOptions, wxID_ANY, wxString( _("&Your Receiving Addresses...") ) , wxEmptyString, wxITEM_NORMAL );
 	m_menuOptions->Append( m_menuOptionsChangeYourAddress );
 	
-	wxMenuItem* m_menuOptionsEncryptWallet;
 	m_menuOptionsEncryptWallet = new wxMenuItem( m_menuOptions, wxID_ANY, wxString( _("&Encrypt Wallet...") ) , wxEmptyString, wxITEM_NORMAL );
 	m_menuOptions->Append( m_menuOptionsEncryptWallet );
 	
-	wxMenuItem* m_menuOptionsChangeWalletPassphrase;
 	m_menuOptionsChangeWalletPassphrase = new wxMenuItem( m_menuOptions, wxID_ANY, wxString( _("&Change Wallet Encryption Passphrase...") ) , wxEmptyString, wxITEM_NORMAL );
 	m_menuOptions->Append( m_menuOptionsChangeWalletPassphrase );
 	
diff --git a/src/uibase.h b/src/uibase.h
index e2dbb0b..ca0730b 100644
--- a/src/uibase.h
+++ b/src/uibase.h
@@ -117,6 +117,8 @@ class CMainFrameBase : public wxFrame
 	
 	public:
 		wxMenu* m_menuOptions;
+		wxMenuItem* m_menuOptionsEncryptWallet;
+		wxMenuItem* m_menuOptionsChangeWalletPassphrase;
 		wxStatusBar* m_statusBar;
 		wxTextCtrl* m_textCtrlAddress;
 		wxListCtrl* m_listCtrlAll;
-- 
1.7.1