merge with stable
authorMatt Mackall <mpm@selenic.com>
Tue, 19 Jul 2011 14:19:04 -0500
changeset 14896 1a841d2b4bb1
parent 14895 a35d6f822e3e (current diff)
parent 14892 d2d592718e90 (diff)
child 14899 6d1d0b9c4ecc
merge with stable
--- a/mercurial/hook.py	Sun Jul 17 00:36:43 2011 +0200
+++ b/mercurial/hook.py	Tue Jul 19 14:19:04 2011 -0500
@@ -65,6 +65,12 @@
                                '("%s" is not callable)') %
                              (hname, funcname))
     try:
+        # redirect IO descriptors the the ui descriptors so hooks that write
+        # directly to these don't mess the command protocol when running through
+        # the command server
+        old = sys.stdout, sys.stderr, sys.stdin
+        sys.stdout, sys.stderr, sys.stdin = ui.fout, ui.ferr, ui.fin
+
         r = obj(ui=ui, repo=repo, hooktype=name, **args)
     except KeyboardInterrupt:
         raise
@@ -79,6 +85,8 @@
             raise
         ui.traceback()
         return True
+    finally:
+        sys.stdout, sys.stderr, sys.stdin = old
     if r:
         if throw:
             raise util.Abort(_('%s hook failed') % hname)
--- a/mercurial/windows.py	Sun Jul 17 00:36:43 2011 +0200
+++ b/mercurial/windows.py	Tue Jul 19 14:19:04 2011 -0500
@@ -62,7 +62,7 @@
             self.close()
             raise IOError(errno.EPIPE, 'Broken pipe')
 
-sys.stdout = winstdout(sys.stdout)
+sys.__stdout__ = sys.stdout = winstdout(sys.stdout)
 
 def _is_win_9x():
     '''return true if run on windows 95, 98 or me.'''
--- a/tests/test-commandserver.py	Sun Jul 17 00:36:43 2011 +0200
+++ b/tests/test-commandserver.py	Tue Jul 19 14:19:04 2011 -0500
@@ -144,6 +144,16 @@
     runcommand(server, ['-R', 'foo', 'showconfig'])
     shutil.rmtree('foo')
 
+def hook(**args):
+    print 'hook talking'
+    print 'now try to read something: %r' % sys.stdin.read()
+
+def hookoutput(server):
+    readchannel(server)
+    runcommand(server, ['--config',
+                        'hooks.pre-identify=python:test-commandserver.hook', 'id'],
+               input=cStringIO.StringIO('some input'))
+
 if __name__ == '__main__':
     os.system('hg init')
 
@@ -158,3 +168,4 @@
     hgrc.write('[ui]\nfoo=bar\n')
     hgrc.close()
     check(localhgrc)
+    check(hookoutput)
--- a/tests/test-commandserver.py.out	Sun Jul 17 00:36:43 2011 +0200
+++ b/tests/test-commandserver.py.out	Tue Jul 19 14:19:04 2011 -0500
@@ -49,3 +49,6 @@
 defaults.commit=-d "0 0"
 defaults.tag=-d "0 0"
 ui.slash=True
+hook talking
+now try to read something: 'some input'
+eff892de26ec tip