install script now knows how to kill MSBuild and devenv

visual_studio/NativeClientVSAddIn/InstallerResources/install.py

 #!/usr/bin/env python
 # 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.
 
 """Copies necessary add-in files into place to install the add-in.
 
 This script will copy the necessary files for the Visual Studio add-in
 to where Visual Studio can find them. It assumes the current directory
 contains the necessary files to copy.
 """
 
 import create_ppapi_platform
 import ctypes
 import os
 import optparse
 import platform
 import shutil
 import sys
+import subprocess
+import time
 
 NACL_PLATFORM_NAME = 'NaCl'
 PEPPER_PLATFORM_NAME = 'PPAPI'
 
 DEFAULT_VS_USER_DIRECTORY = os.path.expandvars(
     '%USERPROFILE%\\My Documents\\Visual Studio 2010')
 
 DEFAULT_MS_BUILD_DIRECTORY = os.path.expandvars('%ProgramFiles(x86)%\\MSBuild')
 
 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ skipping 20 matching lines @@
     print 'Failed to remove non-existant file: %s' % (file_path)
 
 
 def Uninstall(nacl_directory, pepper_directory, addin_directory):
   UninstallDirectory(nacl_directory)
   UninstallDirectory(pepper_directory)
   for file_name in ADDIN_FILES:
     UninstallFile(os.path.join(addin_directory, file_name))
 
 
+def CheckForRunningProgams():
+  tasklist = os.popen('tasklist.exe').readlines()

[binji, 2012/10/04 18:23:40]

why not subprocess? I suppose if popen works, fine. (though the docs say it is
deprecated since 2.6)
But remove import above in that case.

+  tasklist = [l for l in tasklist if 'MSBuild' in l or 'devenv' in l]
+  return tasklist
+
+
+def Ask(question):
+  while True:
+    print '\n'
+    print question
+    print "Continue? ((Yes))/((No))"
+    answer = raw_input().strip().lower()
+    if answer in ('y', 'yes'):
+      return True
+    if answer in ('n', 'no'):
+      return False
+
+
 def main():
   parser = optparse.OptionParser(usage='Usage: %prog [options]')
   parser.add_option('-b', '--msbuild-path', dest='msbuild_path',
       default=DEFAULT_MS_BUILD_DIRECTORY, metavar='PATH',
       help='Provide the path to the MSBuild directory')
   parser.add_option('-a', '--vsuser-path', dest='vsuser_path',
       default=DEFAULT_VS_USER_DIRECTORY, metavar='PATH',
       help='Provide the path to the Visual Studio user directory')
-  parser.add_option('-f', '--force', action="store_true", dest='overwrite',
+  parser.add_option('-f', '--force', action="store_true",
       default=False, help='Force an overwrite of existing files')
   parser.add_option('-p', '--ppapi', action="store_true", dest='install_ppapi',
       help='Install PPAPI template without asking.')
   parser.add_option('-n', '--no-ppapi', action="store_false",
       dest='install_ppapi', help='Do not install PPAPI template and do not ask')
   parser.add_option('-u', '--uninstall', action="store_true",
       dest='uninstall', help='Remove the add-in.')
   (options, args) = parser.parse_args()
 
   print "*************************************************"
   print "Native-Client Visual Studio 2010 Add-in Installer"
   print "*************************************************\n"
-  print "Please ensure Visual Studio and MSBuild are closed " \
-        "during installation.\n"
 
   if platform.system() != 'Windows':
     raise InstallError('Must install to Windows system')
 
+  if CheckForRunningProgams():
+    if not options.force:
+      print "Visual Studio and MSBuild must be closed during installation"
+      if not Ask("Kill all instances now?"):
+        raise InstallError('Please close all Visual Studio or MSBuild '
+                           'instances before installing')
+    os.system("taskkill.exe /IM MSBuild.exe /f")
+    os.system("taskkill.exe /IM devenv.exe")
+    if CheckForRunningProgams():
+      for i in xrange(10):
+        if not CheckForRunningProgams():
+          break
+        time.sleep(1)
+        if not CheckForRunningProgams():
+          break
+        if i == 5:
+          print "Trying taskkill with /f"
+          os.system("taskkill.exe /IM devenv.exe /f")
+      if CheckForRunningProgams():
+        raise InstallError('Failed to kill Visual Studio and MSBuild instances')
+
   if sys.version_info < (2, 6, 2):
     print "\n\nWARNING: Only python version 2.6.2 or greater is supported. " \
           "Current version is %s\n\n" % (sys.version_info[:3],)
 
   # Admin is needed to write to the default platform directory.
   if ctypes.windll.shell32.IsUserAnAdmin() != 1:
     raise InstallError("Not running as administrator. The install script needs "
                        "write access to protected Visual Studio directories.")
 
   # Ensure install directories exist.
   if not os.path.exists(options.vsuser_path):
     raise InstallError("Could not find user Visual Studio directory: %s" % (
         options.vsuser_path))
   if not os.path.exists(options.msbuild_path):
     raise InstallError("Could not find MS Build directory: %s" % (
         options.msbuild_path))
 
   addin_directory = os.path.join(options.vsuser_path, 'Addins')
   platform_directory = os.path.join(
       options.msbuild_path, 'Microsoft.Cpp\\v4.0\\Platforms')
   nacl_directory = os.path.join(platform_directory, NACL_PLATFORM_NAME)
   pepper_directory = os.path.join(platform_directory, PEPPER_PLATFORM_NAME)
 
   # If uninstalling then redirect to uninstall program.
   if options.uninstall:
     Uninstall(nacl_directory, pepper_directory, addin_directory)
     print "\nUninstall complete!\n"
-    exit(0)
+    sys.exit(0)
 
   if not os.path.exists(platform_directory):
     raise InstallError("Could not find path: %s" % platform_directory)
   if not os.path.exists(addin_directory):
     os.mkdir(addin_directory)
 
   # Ensure environment variables are set.
   nacl_sdk_root = os.getenv('NACL_SDK_ROOT', None)
   chrome_path = os.getenv('CHROME_PATH', None)
   if nacl_sdk_root is None:
     raise InstallError('Environment Variable NACL_SDK_ROOT is not set')
   if chrome_path is None:
     raise InstallError('Environment Variable CHROME_PATH is not set')
 
+
   # Remove existing installation.
   if os.path.exists(nacl_directory) or os.path.exists(pepper_directory):
     # If not forced then ask user permission.
-    if not options.overwrite:
-      print "\nWarning: Pre-existing add-in installation will be overwritten."
-      print "Continue? ((Yes))/((No))"
-      remove_answer = raw_input().strip()
-      if not (remove_answer.lower() == "yes" or remove_answer.lower() == "y"):
+    if not options.force:
+      if not Ask("Warning: Pre-existing add-in installation "
+                 "will be overwritten.")
         raise InstallError('User did not allow overwrite of existing install.')
     print "Removing existing install..."
     Uninstall(nacl_directory, pepper_directory, addin_directory)
 
+
   # Ask user before installing PPAPI template.
   if options.install_ppapi is None:
-    print "\n"
-    print "Set up configuration to enable Pepper development " \
-          "with Visual Studio?"
-    print "((Yes)) - I want to create and copy relevant files into a " \
-           "Pepper subdirectory"
-    print "((No)) - I am not interested or will set up the configuration later"
-    ppapi_answer = raw_input().strip()
-    if ppapi_answer.lower() == "yes" or ppapi_answer.lower() == "y":
+    ppapi_answer = Ask("Set up configuration to enable Pepper development " \

[binji, 2012/10/04 18:23:40]

nit: line continuations aren't necessary

+          "with Visual Studio?\n" \
+          "((Yes)) - I want to create and copy relevant files into a " \
+          "Pepper subdirectory\n" \
+          "((No)) - I am not interested or will set up the configuration later")
+    if ppapi_answer:
       options.install_ppapi = True
       print "Confirmed installer will include PPAPI platform."
     else:
       options.install_ppapi = False
       print "Will not install PPAPI platform during installation."
 
   print "\nBegin installing components..."
 
   try:
     # Copy the necessary files into place.
@@ skipping 23 matching lines @@
   except shutil.Error as e:
     print "Error while copying file. Please ensure file is not in use."
     print e
   except WindowsError as e:
     if e.winerror == 5:
       print "Access denied error.  Please ensure Visual Studio and MSBuild"
       print "processes are closed."
     else:
       raise