From 5e55b430b771757fc89c6c21d0450f3e707eb446 Mon Sep 17 00:00:00 2001
From: Marek Marczykowski <marmarek@invisiblethingslab.com>
Date: Wed, 13 Feb 2013 11:14:28 +0100
Subject: [PATCH] anaconda: add option to lock root account

Not only default setting, so one could leave account locked if entered
password setting spoke.
---
 .../pyanaconda/ui/gui/spokes/password.glade   | 18 ++++++++++++-
 anaconda/pyanaconda/ui/gui/spokes/password.py | 27 ++++++++++++++++---
 2 files changed, 40 insertions(+), 5 deletions(-)

diff --git a/anaconda/pyanaconda/ui/gui/spokes/password.glade b/anaconda/pyanaconda/ui/gui/spokes/password.glade
index 043f0ad..913f950 100644
--- a/anaconda/pyanaconda/ui/gui/spokes/password.glade
+++ b/anaconda/pyanaconda/ui/gui/spokes/password.glade
@@ -40,6 +40,22 @@
               <object class="GtkBox" id="AnacondaSpokeWindow-action_area1">
                 <property name="can_focus">False</property>
                 <property name="orientation">vertical</property>
+                <child>
+                  <object class="GtkCheckButton" id="lock">
+                    <property name="label" translatable="yes">Lock root account</property>
+                    <property name="visible">True</property>
+                    <property name="can_focus">True</property>
+                    <property name="receives_default">False</property>
+                    <property name="xalign">0</property>
+                    <property name="draw_indicator">True</property>
+                    <signal name="clicked" handler="on_lock_clicked" swapped="no"/>
+                  </object>
+                  <packing>
+                    <property name="expand">False</property>
+                    <property name="fill">True</property>
+                    <property name="position">0</property>
+                  </packing>
+                </child>
                 <child>
                   <object class="GtkGrid" id="pwgrid">
                     <property name="visible">True</property>
@@ -137,7 +153,7 @@
                   <packing>
                     <property name="expand">False</property>
                     <property name="fill">True</property>
-                    <property name="position">0</property>
+                    <property name="position">1</property>
                   </packing>
                 </child>
               </object>
diff --git a/anaconda/pyanaconda/ui/gui/spokes/password.py b/anaconda/pyanaconda/ui/gui/spokes/password.py
index ac113af..64a2f0a 100644
--- a/anaconda/pyanaconda/ui/gui/spokes/password.py
+++ b/anaconda/pyanaconda/ui/gui/spokes/password.py
@@ -52,16 +52,25 @@ class PasswordSpoke(NormalSpoke):
         self._password = None
         self._error = False
         self._oldweak = None
+        self._lock = self.data.rootpw.lock
 
     def initialize(self):
         NormalSpoke.initialize(self)
         # place holders for the text boxes
         self.pw = self.builder.get_object("pw")
         self.confirm = self.builder.get_object("confirm")
+        self.lock = self.builder.get_object("lock")
 
     def refresh(self):
 #        self.setCapsLockLabel()
-        self.pw.grab_focus()
+        self.lock.set_active(self._lock)
+        self.on_lock_clicked(self.lock)
+
+    def on_lock_clicked(self, lock):
+        self.pw.set_sensitive(not lock.get_active())
+        self.confirm.set_sensitive(not lock.get_active())
+        if not lock.get_active():
+            self.pw.grab_focus()
 
 # Caps lock detection isn't hooked up right now
 #    def setCapsLockLabel(self):
@@ -83,9 +92,10 @@ class PasswordSpoke(NormalSpoke):
             return _("Root password is not set")
 
     def apply(self):
-        self.data.rootpw.password = cryptPassword(self._password)
-        self.data.rootpw.isCrypted = True
-        self.data.rootpw.lock = False
+        if self._password:
+            self.data.rootpw.password = cryptPassword(self._password)
+            self.data.rootpw.isCrypted = True
+        self.data.rootpw.lock = self._lock
 
     @property
     def completed(self):
@@ -98,6 +108,14 @@ class PasswordSpoke(NormalSpoke):
         self._error = False
         pw = self.pw.get_text()
         confirm = self.confirm.get_text()
+        lock = self.lock.get_active()
+
+        if lock:
+            self._lock = True
+            self._password = None
+            self.clear_info()
+            self._error = False
+            return True
 
         if not pw and not confirm:
             self._error = _("You must provide and confirm a password.")
@@ -120,6 +138,7 @@ class PasswordSpoke(NormalSpoke):
 
         # if no errors, clear the info for next time we go into the spoke
         self._password = pw
+        self._lock = False
         self.clear_info()
         self._error = False
         return True