1 files changed, 76 insertions, 0 deletions
diff --git a/buildbot/buildbot/steps/master.py b/buildbot/buildbot/steps/master.py
new file mode 100644
index 0000000..da8a664
--- /dev/null
+++ b/buildbot/buildbot/steps/master.py
@@ -0,0 +1,76 @@
+import os, types
+from twisted.python import log, failure, runtime
+from twisted.internet import reactor, defer, task
+from buildbot.process.buildstep import RemoteCommand, BuildStep
+from buildbot.process.buildstep import SUCCESS, FAILURE
+from twisted.internet.protocol import ProcessProtocol
+
+class MasterShellCommand(BuildStep):
+    """
+    Run a shell command locally - on the buildmaster.  The shell command
+    COMMAND is specified just as for a RemoteShellCommand.  Note that extra
+    logfiles are not sopported.
+    """
+    name='MasterShellCommand'
+    description='Running'
+    descriptionDone='Ran'
+
+    def __init__(self, command, **kwargs):
+        BuildStep.__init__(self, **kwargs)
+        self.addFactoryArguments(command=command)
+        self.command=command
+
+    class LocalPP(ProcessProtocol):
+        def __init__(self, step):
+            self.step = step
+
+        def outReceived(self, data):
+            self.step.stdio_log.addStdout(data)
+
+        def errReceived(self, data):
+            self.step.stdio_log.addStderr(data)
+
+        def processEnded(self, status_object):
+            self.step.stdio_log.addHeader("exit status %d\n" % status_object.value.exitCode)
+            self.step.processEnded(status_object)
+
+    def start(self):
+        # set up argv
+        if type(self.command) in types.StringTypes:
+            if runtime.platformType  == 'win32':
+                argv = os.environ['COMSPEC'].split() # allow %COMSPEC% to have args
+                if '/c' not in argv: argv += ['/c'] 
+                argv += [self.command]
+            else:
+                # for posix, use /bin/sh. for other non-posix, well, doesn't
+                # hurt to try
+                argv = ['/bin/sh', '-c', self.command]
+        else:
+            if runtime.platformType  == 'win32':
+                argv = os.environ['COMSPEC'].split() # allow %COMSPEC% to have args
+                if '/c' not in argv: argv += ['/c'] 
+                argv += list(self.command)
+            else:
+                argv = self.command
+
+        self.stdio_log = stdio_log = self.addLog("stdio")
+
+        if type(self.command) in types.StringTypes:
+            stdio_log.addHeader(self.command.strip() + "\n\n")
+        else:
+            stdio_log.addHeader(" ".join(self.command) + "\n\n")
+        stdio_log.addHeader("** RUNNING ON BUILDMASTER **\n")
+        stdio_log.addHeader(" in dir %s\n" % os.getcwd())
+        stdio_log.addHeader(" argv: %s\n" % (argv,))
+
+        # TODO add a timeout?
+        proc = reactor.spawnProcess(self.LocalPP(self), argv[0], argv)
+        # (the LocalPP object will call processEnded for us)
+
+    def processEnded(self, status_object):
+        if status_object.value.exitCode != 0:
+            self.step_status.setText(["failed (%d)" % status_object.value.exitCode])
+            self.finished(FAILURE)
+        else:
+            self.step_status.setText(["succeeded"])
+            self.finished(SUCCESS)