- implement for backward compatibility

diff --git a/PLC/Methods/BootUpdateNode.py b/PLC/Methods/BootUpdateNode.py
index 48e0219..e612bdf 100644 (file)
--- a/PLC/Methods/BootUpdateNode.py
+++ b/PLC/Methods/BootUpdateNode.py
@@ -1,6 +1,13 @@
+from PLC.Faults import *
 from PLC.Method import Method
 from PLC.Parameter import Parameter, Mixed
 from PLC.Auth import Auth, BootAuth
+from PLC.Nodes import Node, Nodes
+from PLC.NodeNetworks import NodeNetwork, NodeNetworks
+
+can_update = lambda (field, value): field in \
+             ['method', 'mac', 'gateway', 'network',
+              'broadcast', 'netmask', 'dns1', 'dns2']
 
 class BootUpdateNode(Method):
     """
@@ -10,9 +17,44 @@ class BootUpdateNode(Method):
     Returns 1 if updated successfully.
     """
 
-    accepts = [BootAuth(), dict]
+    nodenetwork_fields = dict(filter(can_update, NodeNetwork.fields.items()))
+
+    accepts = [
+        BootAuth(),
+        {'boot_state': Node.fields['boot_state'],
+         'primary_network': nodenetwork_fields,
+         'ssh_host_key': Node.fields['ssh_rsa_key']}
+        ]
     returns = Parameter(int, '1 if successful')
 
-    def call(self, auth, update_fields):
-        # XXX
+    def call(self, auth, node_fields):
+        # Update node state
+        if node_fields.has_key('boot_state'):
+            self.caller['boot_state'] = node_fields['boot_state']
+        if node_fields.has_key('ssh_host_key'):
+            self.caller['ssh_rsa_key'] = node_fields['ssh_host_key']
+
+        # Update primary node network state
+        if node_fields.has_key('primary_network'):
+            primary_network = node_fields['primary_network'] 
+
+            if 'nodenetwork_id' not in primary_network:
+                raise PLCInvalidArgument, "Node network not specified"
+            if primary_network['nodenetwork_id'] not in self.caller['nodenetwork_ids']:
+                raise PLCInvalidArgument, "Node network not associated with calling node"
+
+            nodenetworks = NodeNetworks(self.api, [primary_network['nodenetwork_id']]).values()
+            if not nodenetworks:
+                raise PLCInvalidArgument, "No such node network"
+            nodenetwork = nodenetworks[0]
+
+            if not nodenetwork['is_primary']:
+                raise PLCInvalidArgument, "Not the primary node network on record"
+
+            nodenetwork_fields = dict(filter(can_update, primary_network.items()))
+            nodenetwork.update(nodenetwork_fields)
+            nodenetwork.sync(commit = False)
+
+        self.caller.sync(commit = True)
+
         return 1