/windmill/browser/chrome.py

#   Copyright (c) 2009 Mikeal Rogers <mikeal.rogers@gmail.com>
#   Copyright (c) 2009 Domen Kozar <domen@dev.si>
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.

import commands
import tempfile
import logging
import signal
import subprocess
import sys, os
if not sys.version.startswith('2.4'):
    from urlparse import urlparse
else:
    # python 2.4
    from windmill.tools.urlparse_25 import urlparse

import windmill    

logger = logging.getLogger(__name__)

import safari

class Chrome(safari.Safari):
    
    def __init__(self):
        self.chrome_binary = windmill.settings['CHROME_BINARY']
        self.test_url = windmill.settings['TEST_URL']
    
    # def create_redirect(self):
    #     self.redirection_page = tempfile.mktemp(suffix='.html')
    #     f = open(self.redirection_page, 'w') 
    #     test_url = windmill.get_test_url(windmill.settings['TEST_URL']) 
    #     f.write( html_redirection.replace('{replace}', test_url) )
    #     f.flush() ; f.close()
        
    # def set_proxy_mac(self):
    #     """Set local Proxy"""
    #     self.netsetup_binary = windmill.settings['NETWORKSETUP_BINARY']
    #     interface_name = find_default_interface_name()
    #     uri = urlparse.urlparse(self.test_url)
    #     set_proxy_command = ' '.join([ self.netsetup_binary, 
    #                                    '-setwebproxy', 
    #                                    '"'+interface_name+'"', 
    #                                    '127.0.0.1', 
    #                                    str(windmill.settings['SERVER_HTTP_PORT'])
    #                                  ])
    #     commands.getoutput(set_proxy_command)
    #     enable_proxy_command = ' '.join([ self.netsetup_binary,
    #                                       '-setwebproxystate',
    #                                       '"'+interface_name+'"',
    #                                       'on'
    #                                     ])
    #     commands.getoutput(enable_proxy_command)      
    #     self.create_redirect()
    #     self.interface_name = interface_name
    # 
    # def unset_proxy_mac(self):
    #     commands.getoutput(' '.join([self.netsetup_binary, '-setwebproxystate', '"'+self.interface_name+'"', 'off']))
    
    def get_chrome_command(self):
        #default options, what was used for windows chrome build
        chrome_options = ['--homepage', self.test_url+'/windmill-serv/start.html','--disable-popup-blocking']
        if sys.platform in ('cygwin', 'win32','linux2'):
            # options for running in windows and linux (same options format)
            # Run Proxy using option
            chrome_options = ['--homepage', self.test_url+'/windmill-serv/start.html','--disable-popup-blocking','--proxy-server='+'127.0.0.1:'+str(windmill.settings['SERVER_HTTP_PORT'])]
        return [self.chrome_binary]+chrome_options
    
    #def set_proxy_windows(self):
    #    import ie
    #    self.ie_obj = ie.InternetExplorer()
    #    self.ie_obj.set_proxy()
    # 
    # def unset_proxy_windows(self):
    #     self.ie_obj.unset_proxy()
        
    def start(self):
        """Start Chrome"""
        # if sys.platform == 'darwin':
        #     self.set_proxy_mac()
        #if sys.platform in ('cygwin', 'win32'):
        #    self.set_proxy_windows()
        
        # Workaround for bug in nose
        if hasattr(sys.stdout, 'fileno'):
            kwargs = {'stdout':sys.stdout ,'stderr':sys.stderr, 'stdin':sys.stdin}
        else:
            kwargs = {'stdout':sys.__stdout__ ,'stderr':sys.__stderr__, 'stdin':sys.stdin}
        
        command = self.get_chrome_command()
        self.p_handle = subprocess.Popen(command, **kwargs)
        logger.info(command)

    def kill(self, kill_signal=None):
        """Stop Chrome"""
        if not sys.version.startswith('2.6'):
            raise Exception("Kill doesn't work for Chrome on Python version pre-2.6")
        
        #if sys.platform in ('cygwin', 'win32'):
        #    self.unset_proxy_windows()
        #    
        try:
            self.p_handle.kill()
        except:
            logger.exception('Cannot kill Chrome')
            
    # def stop(self):
    #     self.kill(signal.SIGTERM)
    #         
    # def is_alive(self):
    #     if self.p_handle.poll() is None:
    #         return False
    #     return True