add (set|get)Mode and enhance label functionality

append label now stable and will not revert back and set|get_mode method were added to library

Signed-off-by: Mahdi Ardekanian <mahdi@ibm.com>

src/NodeHelper.groovy

@@ -93,6 +93,51 @@ class  NodeHelper {
         return ret;
     } 
 
+    /**
+     * Overwrites the existing mode with the one passed in.
+     * modes could be either normal or exclusive
+     *
+     * @param computerName the computer whose labels need to be updated
+     * @param mode         the net mode for the computer
+     *
+     * @return Done if mode changed and setMode:SLAVE_NOT_FOUND if no node found
+     */
+    public String setMode(String computerName, String mode) {
+        String ret = "setMode:SLAVE_NOT_FOUND";
+
+        def slave = Jenkins.getInstance().getSlave(computerName)
+        if (slave != null) {
+            if(mode == 'NORMAL'){
+                slave.setMode(hudson.model.Node.Mode.NORMAL);
+            }else{
+                slave.setMode(hudson.model.Node.Mode.EXCLUSIVE);
+            }
+            slave.save()
+            return 'Done'
+        }
+
+        return ret;
+    }
+
+
+    /**
+     * Returns mode of the node.
+     *
+     * @param computerName the computer whose labels need to be updated
+     *
+     * @return either NORMAL or EXCLUSIVE
+     */
+    public String getMode(String computerName) {
+        String ret = "setMode:SLAVE_NOT_FOUND";
+
+        def slave = Jenkins.getInstance().getSlave(computerName)
+        if (slave != null) {
+            return slave.getMode() == hudson.model.Node.Mode.NORMAL ? "NORMAL" : "EXCLUSIVE"
+        }
+
+        return ret;
+    }
+
     /**
      * Overwrites the existing labels with the ones passed
      * in. Labels should be seperated by spaces.
@@ -108,6 +153,7 @@ class  NodeHelper {
         Computer computer = getComputer(computerName);
         if (computer != null) {
             computer.getNode().setLabelString(label.toLowerCase());
+            computer.getNode().save()
             ret = getLabels(computer.getName());
         }