<filename>chrome/test/mini_installer/variable_expander.py
# Copyright 2013 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.

import base64
import hashlib
import os
import string
import win32api
import win32file
import win32com.client
from win32com.shell import shell, shellcon
import win32security


def _GetFileVersion(file_path):
  """Returns the file version of the given file."""
  return win32com.client.Dispatch('Scripting.FileSystemObject').GetFileVersion(
      file_path)


def _GetFileBitness(file_path):
  """Returns the bitness of the given file."""
  if win32file.GetBinaryType(file_path) == win32file.SCS_32BIT_BINARY:
    return '32'
  return '64'


def _GetProductName(file_path):
  """Returns the product name of the given file.

  Args:
    file_path: The absolute or relative path to the file.

  Returns:
    A string representing the product name of the file, or None if the product
    name was not found.
  """
  language_and_codepage_pairs = win32api.GetFileVersionInfo(
      file_path, '\\VarFileInfo\\Translation')
  if not language_and_codepage_pairs:
    return None
  product_name_entry = ('\\StringFileInfo\\%04x%04x\\ProductName' %
                        language_and_codepage_pairs[0])
  return win32api.GetFileVersionInfo(file_path, product_name_entry)


def _GetUserSpecificRegistrySuffix():
  """Returns '.' plus the unpadded Base32 encoding of the MD5 of the user's SID.

  The result must match the output from the method
  UserSpecificRegistrySuffix::GetSuffix() in
  chrome/installer/util/shell_util.cc. It will always be 27 characters long.
  """
  token_handle = win32security.OpenProcessToken(win32api.GetCurrentProcess(),
                                                win32security.TOKEN_QUERY)
  user_sid, _ = win32security.GetTokenInformation(token_handle,
                                                  win32security.TokenUser)
  user_sid_string = win32security.ConvertSidToStringSid(user_sid)
  md5_digest = hashlib.md5(user_sid_string).digest()
  return '.' + base64.b32encode(md5_digest).rstrip('=')


class VariableExpander:
  """Expands variables in strings."""

  def __init__(self, mini_installer_path, previous_version_mini_installer_path,
               chromedriver_path, quiet, output_dir):
    """Constructor.

    The constructor initializes a variable dictionary that maps variables to
    their values. These are the only acceptable variables:
        * $BRAND: the browser brand (e.g., "Google Chrome" or "Chromium").
        * $CHROME_DIR: the directory of Chrome (or Chromium) from the base
            installation directory.
        * $CHROME_HTML_PROG_ID: 'ChromeHTML' (or 'ChromiumHTM').
        * $CHROME_LONG_NAME: 'Google Chrome' (or 'Chromium').
        * $CHROME_LONG_NAME_BETA: 'Google Chrome Beta' if $BRAND is 'Google
        *   Chrome'.
        * $CHROME_LONG_NAME_DEV: 'Google Chrome Dev' if $BRAND is 'Google
        *   Chrome'.
        * $CHROME_LONG_NAME_SXS: 'Google Chrome SxS' if $BRAND is 'Google
        *   Chrome'.
        * $CHROME_SHORT_NAME: 'Chrome' (or 'Chromium').
        * $CHROME_SHORT_NAME_BETA: 'ChromeBeta' if $BRAND is 'Google Chrome'.
        * $CHROME_SHORT_NAME_DEV: 'ChromeDev' if $BRAND is 'Google Chrome'.
        * $CHROME_SHORT_NAME_SXS: 'ChromeCanary' if $BRAND is 'Google Chrome'.
        * $CHROME_UPDATE_REGISTRY_SUBKEY: the registry key, excluding the root
            key, of Chrome for Google Update.
        * $CHROME_UPDATE_REGISTRY_SUBKEY_DEV: the registry key, excluding the
            root key, of Chrome Dev for Google Update.
        * $CHROME_UPDATE_REGISTRY_SUBKEY_BETA: the registry key, excluding the
            root key, of Chrome Beta for Google Update.
        * $CHROME_UPDATE_REGISTRY_SUBKEY_SXS: the registry key, excluding the
            root key, of Chrome SxS for Google Update.
        * $CHROMEDRIVER_PATH: Path to chromedriver.
        * $QUIET: Supress output.
        * $OUTPUT_DIR: "--output-dir=DIR" or an empty string.
        * $LAUNCHER_UPDATE_REGISTRY_SUBKEY: the registry key, excluding the root
            key, of the app launcher for Google Update if $BRAND is 'Google
        *   Chrome'.
        * $LOCAL_APPDATA: the unquoted path to the Local Application Data
            folder.
        * $LOG_FILE: "--log-file=FILE" or an empty string.
        * $MINI_INSTALLER: the unquoted path to the mini_installer.
        * $MINI_INSTALLER_BITNESS: the bitness of the mini_installer.
        * $MINI_INSTALLER_FILE_VERSION: the file version of $MINI_INSTALLER.
        * $PREVIOUS_VERSION_MINI_INSTALLER: the unquoted path to a
             mini_installer whose version is lower than $MINI_INSTALLER.
        * $PREVIOUS_VERSION_MINI_INSTALLER_FILE_VERSION: the file version of
            $PREVIOUS_VERSION_MINI_INSTALLER.
        * $PROGRAM_FILES: the unquoted path to the Program Files folder.
        * $USER_SPECIFIC_REGISTRY_SUFFIX: the output from the function
            _GetUserSpecificRegistrySuffix().
        * $VERSION_[XP/SERVER_2003/VISTA/WIN7/WIN8/WIN8_1/WIN10]: a 2-tuple
            representing the version of the corresponding OS.
        * $WINDOWS_VERSION: a 2-tuple representing the current Windows version.
        * $CHROME_TOAST_ACTIVATOR_CLSID: NotificationActivator's CLSID for
            Chrome.
        * $CHROME_TOAST_ACTIVATOR_CLSID_BETA: NotificationActivator's CLSID for
            Chrome Beta.
        * $CHROME_TOAST_ACTIVATOR_CLSID_DEV: NotificationActivator's CLSID for
            Chrome Dev.
        * $CHROME_TOAST_ACTIVATOR_CLSID_SXS: NotificationActivator's CLSID for
            Chrome SxS.
        * $CHROME_ELEVATOR_CLSID: Elevator Service CLSID for Chrome.
        * $CHROME_ELEVATOR_CLSID_BETA: Elevator Service CLSID for Chrome Beta.
        * $CHROME_ELEVATOR_CLSID_DEV: Elevator Service CLSID for Chrome Dev.
        * $CHROME_ELEVATOR_CLSID_SXS: Elevator Service CLSID for Chrome SxS.
        * $CHROME_ELEVATOR_IID: IElevator IID for Chrome.
        * $CHROME_ELEVATOR_IID_BETA: IElevator IID for Chrome Beta.
        * $CHROME_ELEVATOR_IID_DEV: IElevator IID for Chrome Dev.
        * $CHROME_ELEVATOR_IID_SXS: IElevator IID for Chrome SxS.
        * $CHROME_ELEVATION_SERVICE_NAME: Elevation Service Name for Chrome.
        * $CHROME_ELEVATION_SERVICE_NAME_BETA: Elevation Service Name for Chrome
            Beta.
        * $CHROME_ELEVATION_SERVICE_NAME_DEV: Elevation Service Name for Chrome
            Dev.
        * $CHROME_ELEVATION_SERVICE_NAME_SXS: Elevation Service Name for Chrome
            SxS.
        * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME: Elevation Service Display Name
            for Chrome.
        * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME_BETA: Elevation Service Display
            Name for Chrome Beta.
        * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME_DEV: Elevation Service Display
            Name for Chrome Dev.
        * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME_SXS: Elevation Service Display
            Name for Chrome SxS.

    Args:
      mini_installer_path: The path to a mini_installer.
      previous_version_mini_installer_path: The path to a mini_installer whose
          version is lower than |mini_installer_path|.
    """
    mini_installer_abspath = os.path.abspath(mini_installer_path)
    previous_version_mini_installer_abspath = os.path.abspath(
        previous_version_mini_installer_path)
    windows_major_ver, windows_minor_ver, _, _, _ = win32api.GetVersionEx()
    self._variable_mapping = {
        'CHROMEDRIVER_PATH':
        chromedriver_path,
        'QUIET':
        '-q' if quiet else '',
        'OUTPUT_DIR':
        '"--output-dir=%s"' % output_dir if output_dir else '',
        'LOCAL_APPDATA':
        shell.SHGetFolderPath(0, shellcon.CSIDL_LOCAL_APPDATA, None, 0),
        'LOG_FILE':
        '',
        'MINI_INSTALLER':
        mini_installer_abspath,
        'MINI_INSTALLER_FILE_VERSION':
        _GetFileVersion(mini_installer_abspath),
        'MINI_INSTALLER_BITNESS':
        _GetFileBitness(mini_installer_abspath),
        'PREVIOUS_VERSION_MINI_INSTALLER':
        previous_version_mini_installer_abspath,
        'PREVIOUS_VERSION_MINI_INSTALLER_FILE_VERSION':
        _GetFileVersion(previous_version_mini_installer_abspath),
        'PROGRAM_FILES':
        shell.SHGetFolderPath(
            0, shellcon.CSIDL_PROGRAM_FILES
            if _GetFileBitness(mini_installer_abspath) == '64' else
            shellcon.CSIDL_PROGRAM_FILESX86, None, 0),
        'USER_SPECIFIC_REGISTRY_SUFFIX':
        _GetUserSpecificRegistrySuffix(),
        'VERSION_SERVER_2003':
        '(5, 2)',
        'VERSION_VISTA':
        '(6, 0)',
        'VERSION_WIN10':
        '(10, 0)',
        'VERSION_WIN7':
        '(6, 1)',
        'VERSION_WIN8':
        '(6, 2)',
        'VERSION_WIN8_1':
        '(6, 3)',
        'VERSION_XP':
        '(5, 1)',
        'WINDOWS_VERSION':
        '(%s, %s)' % (windows_major_ver, windows_minor_ver)
    }

    mini_installer_product_name = _GetProductName(mini_installer_abspath)
    if mini_installer_product_name == 'Google Chrome Installer':
      self._variable_mapping.update({
          'BRAND': 'Google Chrome',
          'BINARIES_UPDATE_REGISTRY_SUBKEY': (
            'Software\\Google\\Update\\Clients\\'
            '{4DC8B4CA-1BDA-483e-B5FA-D3C12E15B62D}'),
          'CHROME_DIR': 'Google\\Chrome',
          'CHROME_HTML_PROG_ID': 'ChromeHTML',
          'CHROME_HTML_PROG_ID_BETA': 'ChromeBHTML',
          'CHROME_HTML_PROG_ID_DEV': 'ChromeDHTML',
          'CHROME_HTML_PROG_ID_SXS': 'ChromeSSHTM',
          'CHROME_LONG_NAME': 'Google Chrome',
          'CHROME_SHORT_NAME': 'Chrome',
          'CHROME_UPDATE_REGISTRY_SUBKEY': (
            'Software\\Google\\Update\\Clients\\'
            '{8A69D345-D564-463c-AFF1-A69D9E530F96}'),
          'CHROME_CLIENT_STATE_KEY': (
            'Software\\Google\\Update\\ClientState\\'
            '{8A69D345-D564-463c-AFF1-A69D9E530F96}'),
          'CHROME_TOAST_ACTIVATOR_CLSID': (
            '{A2C6CB58-C076-425C-ACB7-6D19D64428CD}'),
          'CHROME_DIR_BETA': 'Google\\Chrome Beta',
          'CHROME_DIR_DEV': 'Google\\Chrome Dev',
          'CHROME_DIR_SXS': 'Google\\Chrome SxS',
          'CHROME_LONG_NAME_BETA': 'Google Chrome Beta',
          'CHROME_LONG_NAME_DEV': 'Google Chrome Dev',
          'CHROME_LONG_NAME_SXS': 'Google Chrome SxS',
          'CHROME_SHORT_NAME_BETA': 'ChromeBeta',
          'CHROME_SHORT_NAME_DEV': 'ChromeDev',
          'CHROME_SHORT_NAME_SXS': 'ChromeCanary',
          'CHROME_UPDATE_REGISTRY_SUBKEY_BETA': (
            'Software\\Google\\Update\\Clients\\'
            '{8237E44A-0054-442C-B6B6-EA0509993955}'),
          'CHROME_UPDATE_REGISTRY_SUBKEY_DEV': (
            'Software\\Google\\Update\\Clients\\'
            '{401C381F-E0DE-4B85-8BD8-3F3F14FBDA57}'),
          'CHROME_UPDATE_REGISTRY_SUBKEY_SXS': (
            'Software\\Google\\Update\\Clients\\'
            '{4ea16ac7-fd5a-47c3-875b-dbf4a2008c20}'),
          'LAUNCHER_UPDATE_REGISTRY_SUBKEY': (
            'Software\\Google\\Update\\Clients\\'
            '{FDA71E6F-AC4C-4a00-8B70-9958A68906BF}'),
          'CHROME_TOAST_ACTIVATOR_CLSID_BETA': (
            '{B89B137F-96AA-4AE2-98C4-6373EAA1EA4D}'),
          'CHROME_TOAST_ACTIVATOR_CLSID_DEV': (
            '{F01C03EB-D431-4C83-8D7A-902771E732FA}'),
          'CHROME_TOAST_ACTIVATOR_CLSID_SXS': (
            '{FA372A6E-149F-4E95-832D-8F698D40AD7F}'),
          'CHROME_ELEVATOR_CLSID': ('{708860E0-F641-4611-8895-7D867DD3675B}'),
          'CHROME_ELEVATOR_CLSID_BETA': (
            '{DD2646BA-3707-4BF8-B9A7-038691A68FC2}'),
          'CHROME_ELEVATOR_CLSID_DEV': (
            '{DA7FDCA5-2CAA-4637-AA17-0740584DE7DA}'),
          'CHROME_ELEVATOR_CLSID_SXS': (
            '{704C2872-2049-435E-A469-0A534313C42B}'),
          'CHROME_ELEVATOR_IID': ('{463ABECF-410D-407F-8AF5-0DF35A005CC8}'),
          'CHROME_ELEVATOR_IID_BETA': (
            '{A2721D66-376E-4D2F-9F0F-9070E9A42B5F}'),
          'CHROME_ELEVATOR_IID_DEV': (
            '{BB2AA26B-343A-4072-8B6F-80557B8CE571}'),
          'CHROME_ELEVATOR_IID_SXS': (
            '{4F7CE041-28E9-484F-9DD0-61A8CACEFEE4}'),
          'CHROME_ELEVATION_SERVICE_NAME': (
            'GoogleChromeElevationService'),
          'CHROME_ELEVATION_SERVICE_NAME_BETA': (
            'GoogleChromeBetaElevationService'),
          'CHROME_ELEVATION_SERVICE_NAME_DEV': (
            'GoogleChromeDevElevationService'),
          'CHROME_ELEVATION_SERVICE_NAME_SXS': (
            'GoogleChromeCanaryElevationService'),
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME': (
            'Google Chrome Elevation Service (GoogleChromeElevationService)'),
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME_BETA': (
            'Google Chrome Beta Elevation Service'
            ' (GoogleChromeBetaElevationService)'),
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME_DEV': (
            'Google Chrome Dev Elevation Service'
            ' (GoogleChromeDevElevationService)'),
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME_SXS': (
            'Google Chrome Canary Elevation Service'
            ' (GoogleChromeCanaryElevationService)'),
      })
    elif mini_installer_product_name == 'Chromium Installer':
      self._variable_mapping.update({
          'BRAND': 'Chromium',
          'BINARIES_UPDATE_REGISTRY_SUBKEY': 'Software\\Chromium Binaries',
          'CHROME_DIR': 'Chromium',
          'CHROME_HTML_PROG_ID': 'ChromiumHTM',
          'CHROME_LONG_NAME': 'Chromium',
          'CHROME_SHORT_NAME': 'Chromium',
          'CHROME_UPDATE_REGISTRY_SUBKEY': 'Software\\Chromium',
          'CHROME_CLIENT_STATE_KEY': 'Software\\Chromium',
          'CHROME_TOAST_ACTIVATOR_CLSID': (
            '{635EFA6F-08D6-4EC9-BD14-8A0FDE975159}'),
          'CHROME_ELEVATOR_CLSID': ('{D133B120-6DB4-4D6B-8BFE-83BF8CA1B1B0}'),
          'CHROME_ELEVATOR_IID': ('{B88C45B9-8825-4629-B83E-77CC67D9CEED}'),
          'CHROME_ELEVATION_SERVICE_NAME': 'ChromiumElevationService',
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME': (
            'Chromium Elevation Service (ChromiumElevationService)'),
      })
    else:
      raise KeyError("Unknown mini_installer product name '%s'" %
                     mini_installer_product_name)


  def SetLogFile(self, log_file):
    """Updates the value for the LOG_FILE variable"""
    self._variable_mapping['LOG_FILE'] = (
        '"--log-file=%s"' % log_file if log_file else '')


  def Expand(self, str):
    """Expands variables in the given string.

    This method resolves only variables defined in the constructor. It does not
    resolve environment variables. Any dollar signs that are not part of
    variables must be escaped with $$, otherwise a KeyError or a ValueError will
    be raised.

    Args:
      str: A string.

    Returns:
      A new string created by replacing variables with their values.
    """
    return string.Template(str).substitute(self._variable_mapping)

<filename>chrome/test/mini_installer/variable_expander.py
# Copyright 2013 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.

import base64
import hashlib
import os
import string
import win32api
import win32file
import win32com.client
from win32com.shell import shell, shellcon
import win32security


def _GetFileVersion(file_path):
  """Returns the file version of the given file."""
  return win32com.client.Dispatch('Scripting.FileSystemObject').GetFileVersion(
      file_path)


def _GetFileBitness(file_path):
  """Returns the bitness of the given file."""
  if win32file.GetBinaryType(file_path) == win32file.SCS_32BIT_BINARY:
    return '32'
  return '64'


def _GetProductName(file_path):
  """Returns the product name of the given file.

  Args:
    file_path: The absolute or relative path to the file.

  Returns:
    A string representing the product name of the file, or None if the product
    name was not found.
  """
  language_and_codepage_pairs = win32api.GetFileVersionInfo(
      file_path, '\\VarFileInfo\\Translation')
  if not language_and_codepage_pairs:
    return None
  product_name_entry = ('\\StringFileInfo\\%04x%04x\\ProductName' %
                        language_and_codepage_pairs[0])
  return win32api.GetFileVersionInfo(file_path, product_name_entry)


def _GetUserSpecificRegistrySuffix():
  """Returns '.' plus the unpadded Base32 encoding of the MD5 of the user's SID.

  The result must match the output from the method
  UserSpecificRegistrySuffix::GetSuffix() in
  chrome/installer/util/shell_util.cc. It will always be 27 characters long.
  """
  token_handle = win32security.OpenProcessToken(win32api.GetCurrentProcess(),
                                                win32security.TOKEN_QUERY)
  user_sid, _ = win32security.GetTokenInformation(token_handle,
                                                  win32security.TokenUser)
  user_sid_string = win32security.ConvertSidToStringSid(user_sid)
  md5_digest = hashlib.md5(user_sid_string).digest()
  return '.' + base64.b32encode(md5_digest).rstrip('=')


class VariableExpander:
  """Expands variables in strings."""

  def __init__(self, mini_installer_path, previous_version_mini_installer_path,
               chromedriver_path, quiet, output_dir):
    """Constructor.

    The constructor initializes a variable dictionary that maps variables to
    their values. These are the only acceptable variables:
        * $BRAND: the browser brand (e.g., "Google Chrome" or "Chromium").
        * $CHROME_DIR: the directory of Chrome (or Chromium) from the base
            installation directory.
        * $CHROME_HTML_PROG_ID: 'ChromeHTML' (or 'ChromiumHTM').
        * $CHROME_LONG_NAME: 'Google Chrome' (or 'Chromium').
        * $CHROME_LONG_NAME_BETA: 'Google Chrome Beta' if $BRAND is 'Google
        *   Chrome'.
        * $CHROME_LONG_NAME_DEV: 'Google Chrome Dev' if $BRAND is 'Google
        *   Chrome'.
        * $CHROME_LONG_NAME_SXS: 'Google Chrome SxS' if $BRAND is 'Google
        *   Chrome'.
        * $CHROME_SHORT_NAME: 'Chrome' (or 'Chromium').
        * $CHROME_SHORT_NAME_BETA: 'ChromeBeta' if $BRAND is 'Google Chrome'.
        * $CHROME_SHORT_NAME_DEV: 'ChromeDev' if $BRAND is 'Google Chrome'.
        * $CHROME_SHORT_NAME_SXS: 'ChromeCanary' if $BRAND is 'Google Chrome'.
        * $CHROME_UPDATE_REGISTRY_SUBKEY: the registry key, excluding the root
            key, of Chrome for Google Update.
        * $CHROME_UPDATE_REGISTRY_SUBKEY_DEV: the registry key, excluding the
            root key, of Chrome Dev for Google Update.
        * $CHROME_UPDATE_REGISTRY_SUBKEY_BETA: the registry key, excluding the
            root key, of Chrome Beta for Google Update.
        * $CHROME_UPDATE_REGISTRY_SUBKEY_SXS: the registry key, excluding the
            root key, of Chrome SxS for Google Update.
        * $CHROMEDRIVER_PATH: Path to chromedriver.
        * $QUIET: Supress output.
        * $OUTPUT_DIR: "--output-dir=DIR" or an empty string.
        * $LAUNCHER_UPDATE_REGISTRY_SUBKEY: the registry key, excluding the root
            key, of the app launcher for Google Update if $BRAND is 'Google
        *   Chrome'.
        * $LOCAL_APPDATA: the unquoted path to the Local Application Data
            folder.
        * $LOG_FILE: "--log-file=FILE" or an empty string.
        * $MINI_INSTALLER: the unquoted path to the mini_installer.
        * $MINI_INSTALLER_BITNESS: the bitness of the mini_installer.
        * $MINI_INSTALLER_FILE_VERSION: the file version of $MINI_INSTALLER.
        * $PREVIOUS_VERSION_MINI_INSTALLER: the unquoted path to a
             mini_installer whose version is lower than $MINI_INSTALLER.
        * $PREVIOUS_VERSION_MINI_INSTALLER_FILE_VERSION: the file version of
            $PREVIOUS_VERSION_MINI_INSTALLER.
        * $PROGRAM_FILES: the unquoted path to the Program Files folder.
        * $USER_SPECIFIC_REGISTRY_SUFFIX: the output from the function
            _GetUserSpecificRegistrySuffix().
        * $VERSION_[XP/SERVER_2003/VISTA/WIN7/WIN8/WIN8_1/WIN10]: a 2-tuple
            representing the version of the corresponding OS.
        * $WINDOWS_VERSION: a 2-tuple representing the current Windows version.
        * $CHROME_TOAST_ACTIVATOR_CLSID: NotificationActivator's CLSID for
            Chrome.
        * $CHROME_TOAST_ACTIVATOR_CLSID_BETA: NotificationActivator's CLSID for
            Chrome Beta.
        * $CHROME_TOAST_ACTIVATOR_CLSID_DEV: NotificationActivator's CLSID for
            Chrome Dev.
        * $CHROME_TOAST_ACTIVATOR_CLSID_SXS: NotificationActivator's CLSID for
            Chrome SxS.
        * $CHROME_ELEVATOR_CLSID: Elevator Service CLSID for Chrome.
        * $CHROME_ELEVATOR_CLSID_BETA: Elevator Service CLSID for Chrome Beta.
        * $CHROME_ELEVATOR_CLSID_DEV: Elevator Service CLSID for Chrome Dev.
        * $CHROME_ELEVATOR_CLSID_SXS: Elevator Service CLSID for Chrome SxS.
        * $CHROME_ELEVATOR_IID: IElevator IID for Chrome.
        * $CHROME_ELEVATOR_IID_BETA: IElevator IID for Chrome Beta.
        * $CHROME_ELEVATOR_IID_DEV: IElevator IID for Chrome Dev.
        * $CHROME_ELEVATOR_IID_SXS: IElevator IID for Chrome SxS.
        * $CHROME_ELEVATION_SERVICE_NAME: Elevation Service Name for Chrome.
        * $CHROME_ELEVATION_SERVICE_NAME_BETA: Elevation Service Name for Chrome
            Beta.
        * $CHROME_ELEVATION_SERVICE_NAME_DEV: Elevation Service Name for Chrome
            Dev.
        * $CHROME_ELEVATION_SERVICE_NAME_SXS: Elevation Service Name for Chrome
            SxS.
        * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME: Elevation Service Display Name
            for Chrome.
        * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME_BETA: Elevation Service Display
            Name for Chrome Beta.
        * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME_DEV: Elevation Service Display
            Name for Chrome Dev.
        * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME_SXS: Elevation Service Display
            Name for Chrome SxS.

    Args:
      mini_installer_path: The path to a mini_installer.
      previous_version_mini_installer_path: The path to a mini_installer whose
          version is lower than |mini_installer_path|.
    """
    mini_installer_abspath = os.path.abspath(mini_installer_path)
    previous_version_mini_installer_abspath = os.path.abspath(
        previous_version_mini_installer_path)
    windows_major_ver, windows_minor_ver, _, _, _ = win32api.GetVersionEx()
    self._variable_mapping = {
        'CHROMEDRIVER_PATH':
        chromedriver_path,
        'QUIET':
        '-q' if quiet else '',
        'OUTPUT_DIR':
        '"--output-dir=%s"' % output_dir if output_dir else '',
        'LOCAL_APPDATA':
        shell.SHGetFolderPath(0, shellcon.CSIDL_LOCAL_APPDATA, None, 0),
        'LOG_FILE':
        '',
        'MINI_INSTALLER':
        mini_installer_abspath,
        'MINI_INSTALLER_FILE_VERSION':
        _GetFileVersion(mini_installer_abspath),
        'MINI_INSTALLER_BITNESS':
        _GetFileBitness(mini_installer_abspath),
        'PREVIOUS_VERSION_MINI_INSTALLER':
        previous_version_mini_installer_abspath,
        'PREVIOUS_VERSION_MINI_INSTALLER_FILE_VERSION':
        _GetFileVersion(previous_version_mini_installer_abspath),
        'PROGRAM_FILES':
        shell.SHGetFolderPath(
            0, shellcon.CSIDL_PROGRAM_FILES
            if _GetFileBitness(mini_installer_abspath) == '64' else
            shellcon.CSIDL_PROGRAM_FILESX86, None, 0),
        'USER_SPECIFIC_REGISTRY_SUFFIX':
        _GetUserSpecificRegistrySuffix(),
        'VERSION_SERVER_2003':
        '(5, 2)',
        'VERSION_VISTA':
        '(6, 0)',
        'VERSION_WIN10':
        '(10, 0)',
        'VERSION_WIN7':
        '(6, 1)',
        'VERSION_WIN8':
        '(6, 2)',
        'VERSION_WIN8_1':
        '(6, 3)',
        'VERSION_XP':
        '(5, 1)',
        'WINDOWS_VERSION':
        '(%s, %s)' % (windows_major_ver, windows_minor_ver)
    }

    mini_installer_product_name = _GetProductName(mini_installer_abspath)
    if mini_installer_product_name == 'Google Chrome Installer':
      self._variable_mapping.update({
          'BRAND': 'Google Chrome',
          'BINARIES_UPDATE_REGISTRY_SUBKEY': (
            'Software\\Google\\Update\\Clients\\'
            '{4DC8B4CA-1BDA-483e-B5FA-D3C12E15B62D}'),
          'CHROME_DIR': 'Google\\Chrome',
          'CHROME_HTML_PROG_ID': 'ChromeHTML',
          'CHROME_HTML_PROG_ID_BETA': 'ChromeBHTML',
          'CHROME_HTML_PROG_ID_DEV': 'ChromeDHTML',
          'CHROME_HTML_PROG_ID_SXS': 'ChromeSSHTM',
          'CHROME_LONG_NAME': 'Google Chrome',
          'CHROME_SHORT_NAME': 'Chrome',
          'CHROME_UPDATE_REGISTRY_SUBKEY': (
            'Software\\Google\\Update\\Clients\\'
            '{8A69D345-D564-463c-AFF1-A69D9E530F96}'),
          'CHROME_CLIENT_STATE_KEY': (
            'Software\\Google\\Update\\ClientState\\'
            '{8A69D345-D564-463c-AFF1-A69D9E530F96}'),
          'CHROME_TOAST_ACTIVATOR_CLSID': (
            '{A2C6CB58-C076-425C-ACB7-6D19D64428CD}'),
          'CHROME_DIR_BETA': 'Google\\Chrome Beta',
          'CHROME_DIR_DEV': 'Google\\Chrome Dev',
          'CHROME_DIR_SXS': 'Google\\Chrome SxS',
          'CHROME_LONG_NAME_BETA': 'Google Chrome Beta',
          'CHROME_LONG_NAME_DEV': 'Google Chrome Dev',
          'CHROME_LONG_NAME_SXS': 'Google Chrome SxS',
          'CHROME_SHORT_NAME_BETA': 'ChromeBeta',
          'CHROME_SHORT_NAME_DEV': 'ChromeDev',
          'CHROME_SHORT_NAME_SXS': 'ChromeCanary',
          'CHROME_UPDATE_REGISTRY_SUBKEY_BETA': (
            'Software\\Google\\Update\\Clients\\'
            '{8237E44A-0054-442C-B6B6-EA0509993955}'),
          'CHROME_UPDATE_REGISTRY_SUBKEY_DEV': (
            'Software\\Google\\Update\\Clients\\'
            '{401C381F-E0DE-4B85-8BD8-3F3F14FBDA57}'),
          'CHROME_UPDATE_REGISTRY_SUBKEY_SXS': (
            'Software\\Google\\Update\\Clients\\'
            '{4ea16ac7-fd5a-47c3-875b-dbf4a2008c20}'),
          'LAUNCHER_UPDATE_REGISTRY_SUBKEY': (
            'Software\\Google\\Update\\Clients\\'
            '{FDA71E6F-AC4C-4a00-8B70-9958A68906BF}'),
          'CHROME_TOAST_ACTIVATOR_CLSID_BETA': (
            '{B89B137F-96AA-4AE2-98C4-6373EAA1EA4D}'),
          'CHROME_TOAST_ACTIVATOR_CLSID_DEV': (
            '{F01C03EB-D431-4C83-8D7A-902771E732FA}'),
          'CHROME_TOAST_ACTIVATOR_CLSID_SXS': (
            '{FA372A6E-149F-4E95-832D-8F698D40AD7F}'),
          'CHROME_ELEVATOR_CLSID': ('{708860E0-F641-4611-8895-7D867DD3675B}'),
          'CHROME_ELEVATOR_CLSID_BETA': (
            '{DD2646BA-3707-4BF8-B9A7-038691A68FC2}'),
          'CHROME_ELEVATOR_CLSID_DEV': (
            '{DA7FDCA5-2CAA-4637-AA17-0740584DE7DA}'),
          'CHROME_ELEVATOR_CLSID_SXS': (
            '{704C2872-2049-435E-A469-0A534313C42B}'),
          'CHROME_ELEVATOR_IID': ('{463ABECF-410D-407F-8AF5-0DF35A005CC8}'),
          'CHROME_ELEVATOR_IID_BETA': (
            '{A2721D66-376E-4D2F-9F0F-9070E9A42B5F}'),
          'CHROME_ELEVATOR_IID_DEV': (
            '{BB2AA26B-343A-4072-8B6F-80557B8CE571}'),
          'CHROME_ELEVATOR_IID_SXS': (
            '{4F7CE041-28E9-484F-9DD0-61A8CACEFEE4}'),
          'CHROME_ELEVATION_SERVICE_NAME': (
            'GoogleChromeElevationService'),
          'CHROME_ELEVATION_SERVICE_NAME_BETA': (
            'GoogleChromeBetaElevationService'),
          'CHROME_ELEVATION_SERVICE_NAME_DEV': (
            'GoogleChromeDevElevationService'),
          'CHROME_ELEVATION_SERVICE_NAME_SXS': (
            'GoogleChromeCanaryElevationService'),
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME': (
            'Google Chrome Elevation Service (GoogleChromeElevationService)'),
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME_BETA': (
            'Google Chrome Beta Elevation Service'
            ' (GoogleChromeBetaElevationService)'),
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME_DEV': (
            'Google Chrome Dev Elevation Service'
            ' (GoogleChromeDevElevationService)'),
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME_SXS': (
            'Google Chrome Canary Elevation Service'
            ' (GoogleChromeCanaryElevationService)'),
      })
    elif mini_installer_product_name == 'Chromium Installer':
      self._variable_mapping.update({
          'BRAND': 'Chromium',
          'BINARIES_UPDATE_REGISTRY_SUBKEY': 'Software\\Chromium Binaries',
          'CHROME_DIR': 'Chromium',
          'CHROME_HTML_PROG_ID': 'ChromiumHTM',
          'CHROME_LONG_NAME': 'Chromium',
          'CHROME_SHORT_NAME': 'Chromium',
          'CHROME_UPDATE_REGISTRY_SUBKEY': 'Software\\Chromium',
          'CHROME_CLIENT_STATE_KEY': 'Software\\Chromium',
          'CHROME_TOAST_ACTIVATOR_CLSID': (
            '{635EFA6F-08D6-4EC9-BD14-8A0FDE975159}'),
          'CHROME_ELEVATOR_CLSID': ('{D133B120-6DB4-4D6B-8BFE-83BF8CA1B1B0}'),
          'CHROME_ELEVATOR_IID': ('{B88C45B9-8825-4629-B83E-77CC67D9CEED}'),
          'CHROME_ELEVATION_SERVICE_NAME': 'ChromiumElevationService',
          'CHROME_ELEVATION_SERVICE_DISPLAY_NAME': (
            'Chromium Elevation Service (ChromiumElevationService)'),
      })
    else:
      raise KeyError("Unknown mini_installer product name '%s'" %
                     mini_installer_product_name)


  def SetLogFile(self, log_file):
    """Updates the value for the LOG_FILE variable"""
    self._variable_mapping['LOG_FILE'] = (
        '"--log-file=%s"' % log_file if log_file else '')


  def Expand(self, str):
    """Expands variables in the given string.

    This method resolves only variables defined in the constructor. It does not
    resolve environment variables. Any dollar signs that are not part of
    variables must be escaped with $$, otherwise a KeyError or a ValueError will
    be raised.

    Args:
      str: A string.

    Returns:
      A new string created by replacing variables with their values.
    """
    return string.Template(str).substitute(self._variable_mapping)

# Copyright 2013 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. Returns the file version of the given file. Returns the bitness of the given file. Returns the product name of the given file.    Args:     file_path: The absolute or relative path to the file.    Returns:     A string representing the product name of the file, or None if the product     name was not found. Returns '.' plus the unpadded Base32 encoding of the MD5 of the user's SID.    The result must match the output from the method   UserSpecificRegistrySuffix::GetSuffix() in   chrome/installer/util/shell_util.cc. It will always be 27 characters long. Expands variables in strings. Constructor.      The constructor initializes a variable dictionary that maps variables to     their values. These are the only acceptable variables:         * $BRAND: the browser brand (e.g., "Google Chrome" or "Chromium").         * $CHROME_DIR: the directory of Chrome (or Chromium) from the base             installation directory.         * $CHROME_HTML_PROG_ID: 'ChromeHTML' (or 'ChromiumHTM').         * $CHROME_LONG_NAME: 'Google Chrome' (or 'Chromium').         * $CHROME_LONG_NAME_BETA: 'Google Chrome Beta' if $BRAND is 'Google         *   Chrome'.         * $CHROME_LONG_NAME_DEV: 'Google Chrome Dev' if $BRAND is 'Google         *   Chrome'.         * $CHROME_LONG_NAME_SXS: 'Google Chrome SxS' if $BRAND is 'Google         *   Chrome'.         * $CHROME_SHORT_NAME: 'Chrome' (or 'Chromium').         * $CHROME_SHORT_NAME_BETA: 'ChromeBeta' if $BRAND is 'Google Chrome'.         * $CHROME_SHORT_NAME_DEV: 'ChromeDev' if $BRAND is 'Google Chrome'.         * $CHROME_SHORT_NAME_SXS: 'ChromeCanary' if $BRAND is 'Google Chrome'.         * $CHROME_UPDATE_REGISTRY_SUBKEY: the registry key, excluding the root             key, of Chrome for Google Update.         * $CHROME_UPDATE_REGISTRY_SUBKEY_DEV: the registry key, excluding the             root key, of Chrome Dev for Google Update.         * $CHROME_UPDATE_REGISTRY_SUBKEY_BETA: the registry key, excluding the             root key, of Chrome Beta for Google Update.         * $CHROME_UPDATE_REGISTRY_SUBKEY_SXS: the registry key, excluding the             root key, of Chrome SxS for Google Update.         * $CHROMEDRIVER_PATH: Path to chromedriver.         * $QUIET: Supress output.         * $OUTPUT_DIR: "--output-dir=DIR" or an empty string.         * $LAUNCHER_UPDATE_REGISTRY_SUBKEY: the registry key, excluding the root             key, of the app launcher for Google Update if $BRAND is 'Google         *   Chrome'.         * $LOCAL_APPDATA: the unquoted path to the Local Application Data             folder.         * $LOG_FILE: "--log-file=FILE" or an empty string.         * $MINI_INSTALLER: the unquoted path to the mini_installer.         * $MINI_INSTALLER_BITNESS: the bitness of the mini_installer.         * $MINI_INSTALLER_FILE_VERSION: the file version of $MINI_INSTALLER.         * $PREVIOUS_VERSION_MINI_INSTALLER: the unquoted path to a              mini_installer whose version is lower than $MINI_INSTALLER.         * $PREVIOUS_VERSION_MINI_INSTALLER_FILE_VERSION: the file version of             $PREVIOUS_VERSION_MINI_INSTALLER.         * $PROGRAM_FILES: the unquoted path to the Program Files folder.         * $USER_SPECIFIC_REGISTRY_SUFFIX: the output from the function             _GetUserSpecificRegistrySuffix().         * $VERSION_[XP/SERVER_2003/VISTA/WIN7/WIN8/WIN8_1/WIN10]: a 2-tuple             representing the version of the corresponding OS.         * $WINDOWS_VERSION: a 2-tuple representing the current Windows version.         * $CHROME_TOAST_ACTIVATOR_CLSID: NotificationActivator's CLSID for             Chrome.         * $CHROME_TOAST_ACTIVATOR_CLSID_BETA: NotificationActivator's CLSID for             Chrome Beta.         * $CHROME_TOAST_ACTIVATOR_CLSID_DEV: NotificationActivator's CLSID for             Chrome Dev.         * $CHROME_TOAST_ACTIVATOR_CLSID_SXS: NotificationActivator's CLSID for             Chrome SxS.         * $CHROME_ELEVATOR_CLSID: Elevator Service CLSID for Chrome.         * $CHROME_ELEVATOR_CLSID_BETA: Elevator Service CLSID for Chrome Beta.         * $CHROME_ELEVATOR_CLSID_DEV: Elevator Service CLSID for Chrome Dev.         * $CHROME_ELEVATOR_CLSID_SXS: Elevator Service CLSID for Chrome SxS.         * $CHROME_ELEVATOR_IID: IElevator IID for Chrome.         * $CHROME_ELEVATOR_IID_BETA: IElevator IID for Chrome Beta.         * $CHROME_ELEVATOR_IID_DEV: IElevator IID for Chrome Dev.         * $CHROME_ELEVATOR_IID_SXS: IElevator IID for Chrome SxS.         * $CHROME_ELEVATION_SERVICE_NAME: Elevation Service Name for Chrome.         * $CHROME_ELEVATION_SERVICE_NAME_BETA: Elevation Service Name for Chrome             Beta.         * $CHROME_ELEVATION_SERVICE_NAME_DEV: Elevation Service Name for Chrome             Dev.         * $CHROME_ELEVATION_SERVICE_NAME_SXS: Elevation Service Name for Chrome             SxS.         * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME: Elevation Service Display Name             for Chrome.         * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME_BETA: Elevation Service Display             Name for Chrome Beta.         * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME_DEV: Elevation Service Display             Name for Chrome Dev.         * $CHROME_ELEVATION_SERVICE_DISPLAY_NAME_SXS: Elevation Service Display             Name for Chrome SxS.      Args:       mini_installer_path: The path to a mini_installer.       previous_version_mini_installer_path: The path to a mini_installer whose           version is lower than |mini_installer_path|. Updates the value for the LOG_FILE variable Expands variables in the given string.      This method resolves only variables defined in the constructor. It does not     resolve environment variables. Any dollar signs that are not part of     variables must be escaped with $$, otherwise a KeyError or a ValueError will     be raised.      Args:       str: A string.      Returns:       A new string created by replacing variables with their values.