| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 # A simple native client in python. | 6 # A simple native client in python. |
| 7 # All this client does is echo the text it receives back at the extension. | 7 # All this client does is echo the text it receives back at the extension. |
| 8 | 8 |
| 9 import os | 9 import os |
| 10 import platform | 10 import platform |
| (...skipping 20 matching lines...) Expand all Loading... |
| 31 return 1 | 31 return 1 |
| 32 # TODO(sergeyu): Use argparse module to parse the arguments (not available in | 32 # TODO(sergeyu): Use argparse module to parse the arguments (not available in |
| 33 # Python 2.6). | 33 # Python 2.6). |
| 34 for arg in sys.argv[1:]: | 34 for arg in sys.argv[1:]: |
| 35 if arg.startswith('--'): | 35 if arg.startswith('--'): |
| 36 if arg.startswith('--parent-window='): | 36 if arg.startswith('--parent-window='): |
| 37 parent_window = long(arg[len('--parent-window='):]) | 37 parent_window = long(arg[len('--parent-window='):]) |
| 38 elif caller_url == None: | 38 elif caller_url == None: |
| 39 caller_url = arg | 39 caller_url = arg |
| 40 | 40 |
| 41 # Verify that the process was started in the correct directory. |
| 42 cwd = os.getcwd() |
| 43 script_path = os.path.dirname(os.path.abspath(sys.argv[0])) |
| 44 if cwd.lower() != script_path.lower(): |
| 45 sys.stderr.write('Native messaging host started in a wrong directory.') |
| 46 return 1 |
| 47 |
| 48 # Verify that --parent-window parameter is correct. |
| 41 if platform.system() == 'Windows' and parent_window: | 49 if platform.system() == 'Windows' and parent_window: |
| 42 import win32gui | 50 import win32gui |
| 43 if not win32gui.IsWindow(parent_window): | 51 if not win32gui.IsWindow(parent_window): |
| 44 sys.stderr.write('Invalid --parent-window.\n') | 52 sys.stderr.write('Invalid --parent-window.\n') |
| 45 return 1 | 53 return 1 |
| 46 | 54 |
| 47 while 1: | 55 while 1: |
| 48 # Read the message type (first 4 bytes). | 56 # Read the message type (first 4 bytes). |
| 49 text_length_bytes = sys.stdin.read(4) | 57 text_length_bytes = sys.stdin.read(4) |
| 50 | 58 |
| (...skipping 26 matching lines...) Expand all Loading... |
| 77 | 85 |
| 78 message_number += 1 | 86 message_number += 1 |
| 79 | 87 |
| 80 message = '{{"id": {0}, "echo": {1}, "caller_url": "{2}"}}'.format( | 88 message = '{{"id": {0}, "echo": {1}, "caller_url": "{2}"}}'.format( |
| 81 message_number, text, caller_url).encode('utf-8') | 89 message_number, text, caller_url).encode('utf-8') |
| 82 if not WriteMessage(message): | 90 if not WriteMessage(message): |
| 83 break | 91 break |
| 84 | 92 |
| 85 if __name__ == '__main__': | 93 if __name__ == '__main__': |
| 86 sys.exit(Main()) | 94 sys.exit(Main()) |
| OLD | NEW |