# Copyright (c) 2012 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Provides an interface for installing Chrome.

At present the only platform it supports is Windows.
"""

import _winreg
import ctypes
from ctypes import wintypes, windll
import httplib
import logging
import os
import shutil
import socket
import subprocess
import tempfile
import urllib


class InstallationType(object):
  """Defines the Chrome installation types."""
  SYSTEM = 1
  USER = 2


def IsVersionNewer(cur_version, new_version):
  """Determines if new Chrome version is higher than the installed one.

  Args:
    cur_version: Current version of Chrome.
    new_version: New version that will be installed.

  Returns:
    True, if new version is higher, otherwise False.
  """
  if cur_version == new_version:
    return False
  cur = cur_version.split('.')
  new = new_version.split('.')
  if len(cur) != 4 or len(new) != 4:
    raise RuntimeError('One or both of the versions are invalid.')
  for x in range(len(cur)):
    if int(cur[x]) > int(new[x]):
      return False
  return True


def Install(installer_path, install_type, build, options):
  """Installs the specified Chrome version.

  Args:
    installer_path: Path to the Chrome installer.
    install_type: Type of installation (i.e., system or user).
    build: Chrome build number.
    options: Additional installation options.

  Returns:
    An instance of ChromeInstallation.
  """
  current = ChromeInstallation.GetCurrent()
  if current:
    # Make sure new build can be installed over existing Chrome build.
    if (current.GetType() == InstallationType.SYSTEM and
        install_type == InstallationType.USER):
      raise RuntimeError('System level Chrome exists, aborting user level '
                         'installation.')
    # Confirm the new Chrome build is higher than the installed build.
    if not IsVersionNewer(current.GetVersion(), build):
      raise RuntimeError('Installation failed because a newer version is '
                         'already installed.')
  options.append('--install')
  options.append('--do-not-launch-chrome')
  logging.info('Installing Chrome build %s...' % build)
  args = [installer_path]
  args.extend(options)
  if subprocess.Popen(args).wait() != 0:
    raise RuntimeError('Chrome installation for build %s failed.' % build)
  logging.info('Installation complete.')
  return ChromeInstallation.GetCurrent()


class ChromeRegistryValues(object):
  """Defines the Chrome registry key values."""
  PRODUCT_VERSION = 'pv'
  UNINSTALL_STRING = 'UninstallString'
  UNINSTALL_ARGUMENTS = 'UninstallArguments'


class ChromeRegistryKeys(object):
  """An interface for accessing and manipulating Chrome registry keys."""

  _HKEY_LOCAL = r'SOFTWARE\Wow6432Node\Google\Update'
  _HKEY_USER = _HKEY_LOCAL.replace(r'\Wow6432Node', '')
  _chrome_version = r'Clients\{8A69D345-D564-463C-AFF1-A69D9E530F96}'
  _chrome_args = r'ClientState\{8A69D345-D564-463C-AFF1-A69D9E530F96}'

  def _GetKeyName(self, install_type, value):
    """Gets the registry key name for the specified value.

    Args:
      install_type: Type of installation, must be InstallationType type.
      value: ChromeRegistryValues type for which the key name is required.

    Returns:
      A string representing the full key name of the specified key value.
    """
    key_name = None
    if install_type == InstallationType.USER:
      key_name = self._HKEY_USER
    elif install_type == InstallationType.SYSTEM:
      key_name = self._HKEY_LOCAL
    if value == ChromeRegistryValues.PRODUCT_VERSION:
      return r'%s\%s' % (key_name, self._chrome_version)
    elif value == ChromeRegistryValues.UNINSTALL_ARGUMENTS:
      return r'%s\%s' % (key_name, self._chrome_args)
    elif value == ChromeRegistryValues.UNINSTALL_STRING:
      return r'%s\%s' % (key_name, self._chrome_args)
    raise RuntimeError('Invalid registry value.')

  def _GetRegistryType(self, install_type):
    """Determines the registry key to use based on installation type.

    Args:
      install_type: Type of installation, must be InstallationType type.

    Returns:
      A long representing HKLM or HKCU, depending on installation type.
    """
    if install_type == InstallationType.SYSTEM:
      return _winreg.HKEY_LOCAL_MACHINE
    elif install_type == InstallationType.USER:
      return _winreg.HKEY_CURRENT_USER
    raise RuntimeError('Invalid installation type.')

  def DoesKeyExist(self, install_type, subkey):
    """Determines if a particular key exists in the registry.

    Args:
      install_type: Type of installation, must be InstallationType type.
      subkey: Subkey to look up. It must be a ChromeRegistryValues type.

    Returns:
      True if the key exists, otherwise False.
    """
    key = self._GetRegistryType(install_type)
    key_name = self._GetKeyName(install_type, subkey)
    try:
      hkey = _winreg.OpenKey(key, key_name)
    except _winreg.error:
      return False
    if not hkey.handle:
      return False
    hkey.Close()
    return True

  def GetKeyValue(self, install_type, subkey):
    """Gets value of the specified subkey from the registry.

    Args:
      install_type: Type of installation, must be InstallationType type.
      subkey: ChromeRegistryValue type representing the value to be returned.

    Returns:
      A string representing the subkey value.
    """
    key = self._GetRegistryType(install_type)
    key_name = self._GetKeyName(install_type, subkey)
    hkey = _winreg.OpenKey(key, key_name)
    reg_value = str(_winreg.QueryValueEx(hkey, subkey)[0])
    hkey.Close()
    return reg_value

  def DeleteRegistryEntries(self, install_type):
    """Deletes chrome registry settings.

    Args:
      install_type: Type of installation, must be InstallationType type.
    """
    reg_type = self._GetRegistryType(install_type)
    key_name = self._GetKeyName(install_type,
                                ChromeRegistryValues.UNINSTALL_ARGUMENTS)
    root = key_name[:key_name.rfind('\\')]
    child = key_name[key_name.rfind('\\') + 1:]
    key = _winreg.OpenKey(reg_type, root, 0, _winreg.KEY_ALL_ACCESS)
    _winreg.DeleteKey(key, child)
    key.Close()


class ChromeInstallation(object):
  """Provides pertinent information about the installed Chrome version.

  The type of Chrome version must be passed as an argument to the constructor,
  (i.e. - user or system level).
  """

  _CSIDL_COMMON_APPDATA = 0x1C
  _CSIDL_PROGRAM_FILESX86 = 0x2A

  def __init__(self, install_type):
    assert(install_type == InstallationType.SYSTEM or
           install_type == InstallationType.USER)
    self._type = install_type
    self._regedit = ChromeRegistryKeys()

  def GetType(self):
    """Returns the current installation type."""
    return self._type

  def GetVersion(self):
    """Returns the installed version of Chrome."""
    return self._regedit.GetKeyValue(self._type,
                                     ChromeRegistryValues.PRODUCT_VERSION)

  def _GetWinLocalFolder(self, ftype=_CSIDL_COMMON_APPDATA):
    """Returns full path of the 'Local' folder on Windows.

    Args:
      ftype: Location to look up, which could vary based on installation type.

    Returns:
      A String representing the folder path if successful, otherwise an empty
      string.
    """
    SHGetFolderPathW = windll.shell32.SHGetFolderPathW
    SHGetFolderPathW.argtypes = [wintypes.HWND,
                                 ctypes.c_int,
                                 wintypes.HANDLE,
                                 wintypes.DWORD,
                                 wintypes.LPCWSTR]
    path_buf = wintypes.create_unicode_buffer(wintypes.MAX_PATH)
    result = SHGetFolderPathW(0, ftype, 0, 0, path_buf)
    return str(path_buf.value)

  def _GetUninstallString(self):
    """Returns the Chrome uninstall string from the registry."""
    return self._regedit.GetKeyValue(self._type,
                                     ChromeRegistryValues.UNINSTALL_STRING)

  def _GetUninstallArguments(self):
    """Returns the Chrome uninstall arguments from the registry."""
    return self._regedit.GetKeyValue(self._type,
                                     ChromeRegistryValues.UNINSTALL_ARGUMENTS)

  def GetExePath(self):
    """Returns Chrome binary location based on installation type.

    Currently this method only returns the location of the Chrome binary.
    It does not support Chromium.
    """
    if self._type == InstallationType.USER:
      folder_id = self._CSIDL_COMMON_APPDATA
    elif self._type == InstallationType.SYSTEM:
      folder_id = self._CSIDL_PROGRAM_FILESX86
    chrome_path = os.path.join(self._GetWinLocalFolder(folder_id), 'Google',
                               'Chrome', 'Application', 'chrome.exe')
    return (chrome_path if os.path.exists(chrome_path) else '')

  @staticmethod
  def GetCurrent():
    """Determines Chrome installation type.

    Returns:
      ChromeInstallation object if Chrome is present, otherwise None.
    """
    registry = ChromeRegistryKeys()
    if registry.DoesKeyExist(InstallationType.SYSTEM,
                             ChromeRegistryValues.PRODUCT_VERSION):
      return ChromeInstallation(InstallationType.SYSTEM)
    elif registry.DoesKeyExist(InstallationType.USER,
                               ChromeRegistryValues.PRODUCT_VERSION):
      return ChromeInstallation(InstallationType.USER)
    return None

  def Uninstall(self):
    """Uninstalls Chrome."""
    chrome_path = self.GetExePath()
    reg_opts = self._GetUninstallArguments()
    uninstall_str = self._GetUninstallString()
    options = '%s --force-uninstall' % (reg_opts)
    if self._type == InstallationType.SYSTEM:
      options += ' --system-level'
    if not os.path.exists(chrome_path):
      raise RuntimeError('Could not find chrome, aborting uninstall.')
    logging.info('Launching Chrome installer...')
    cmd = '"%s" %s' % (uninstall_str, options)
    subprocess.call(cmd)
    if not os.path.exists(chrome_path):
      logging.info('Chrome was uninstalled successfully...')
      logging.info('Deleting registry entries...')
      self._regedit.DeleteRegistryEntries(self._type)
      logging.info('Uninstall complete.')
    else:
      raise RuntimeError('Uninstall failed.')