Home | History | Annotate | Download | only in Scripts 
      1 #!/usr/bin/perl -w
      2 # Copyright (C) 2010 Google Inc. All rights reserved.
      3 #
      4 # Redistribution and use in source and binary forms, with or without
      5 # modification, are permitted provided that the following conditions
      6 # are met:
      7 #
      8 # 1.  Redistributions of source code must retain the above copyright
      9 #     notice, this list of conditions and the following disclaimer. 
     10 # 2.  Redistributions in binary form must reproduce the above copyright
     11 #     notice, this list of conditions and the following disclaimer in the
     12 #     documentation and/or other materials provided with the distribution. 
     13 # 3.  Neither the name of Apple Inc. ("Apple") nor the names of
     14 #     its contributors may be used to endorse or promote products derived
     15 #     from this software without specific prior written permission. 
     16 #
     17 # THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
     18 # EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     19 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
     20 # DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
     21 # DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
     22 # (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     23 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
     24 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     25 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     26 # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     27 
     28 use strict;
     29 
     30 use File::Basename;
     31 use File::Spec;
     32 use File::Temp qw(tempdir);
     33 use FindBin;
     34 use Getopt::Long;
     35 
     36 use lib $FindBin::Bin;
     37 use webkitdirs;
     38 use VCSUtils;
     39 
     40 my $macPythonURL = "http://www.python.org/ftp/python/2.6.5/python-2.6.5-macosx10.3-2010-03-24.dmg";
     41 my $macPythonMD5 = "84489bba813fdbb6041b69d4310a86da";
     42 my $macPythonInstallerName = "Python.mpkg";
     43 
     44 # We could use a consistent download location, like the source or build directory.
     45 my $tempDirectory = File::Temp::tempdir("WebKitPythonXXXX", TMPDIR => 1, CLEANUP => 1);
     46 my $downloadDirectory = $tempDirectory;
     47 my $mountPoint = File::Spec->join($tempDirectory, "mount");
     48 
     49 sub checkPythonVersion()
     50 {
     51     # Will exit 0 if Python is 2.5 or greater, non-zero otherwise.
     52     `python -c "import sys;sys.exit(sys.version_info[:2] < (2,5))"`;
     53     return exitStatus($?) == 0;
     54 }
     55 
     56 sub downloadFileToPath($$)
     57 {
     58     my ($remoteURL, $localPath) = @_;
     59     print "Downloading $remoteURL to $localPath\n";
     60     my $exitCode = system("curl", "-o", $localPath, $remoteURL);
     61     return exitStatus($exitCode) == 0;
     62 }
     63 
     64 sub checkMD5($$)
     65 {
     66     my ($path, $expectedMD5) = @_;
     67     my $md5Output = `md5 -q "$path"`;
     68     chomp($md5Output);
     69     my $isValid = $md5Output eq $expectedMD5;
     70     print "'$md5Output' does not match expected: '$expectedMD5'\n" unless $isValid;
     71     return $isValid;
     72 }
     73 
     74 sub mountDMG($$)
     75 {
     76     my ($dmgPath, $mountPoint) = @_;
     77     print "Mounting $dmgPath at $mountPoint\n";
     78     return system("hdiutil", "attach", "-mountpoint", $mountPoint, "-nobrowse", $dmgPath) == 0;
     79 }
     80 
     81 sub unmountDMG($)
     82 {
     83     my ($mountPoint) = @_;
     84     print "Unmounting disk image from $mountPoint\n";
     85     my $exitCode = system("hdiutil", "detach", $mountPoint);
     86     return exitStatus($exitCode) == 0;
     87 }
     88 
     89 sub runInstaller($)
     90 {
     91     my ($installerPackage) = @_;
     92     print "sudo will now ask for your password to run the Python installer.\n";
     93     print "The installer will install Python in /Library/Frameworks/Python.framework\n";
     94     print "and add symlinks from /usr/local/bin.\n";
     95     return system("sudo", "installer", "-verbose", "-pkg", $installerPackage, "-target", "/") == 0;
     96 }
     97 
     98 sub downloadAndMountMacPythonDMG($$)
     99 {
    100     my ($pythonURL, $pythonMD5) = @_;
    101     my $localFilename = basename($pythonURL);
    102     my $localPath = File::Spec->join($downloadDirectory, $localFilename);
    103     
    104     downloadFileToPath($pythonURL, $localPath) or die "Failed to download $pythonURL";
    105     checkMD5($localPath, $pythonMD5) or die "MD5 check failed on $localPath";
    106     return mountDMG($localPath, $mountPoint);
    107 }
    108 
    109 sub installMacPython()
    110 {
    111     downloadAndMountMacPythonDMG($macPythonURL, $macPythonMD5) or die "Failed to download and mount disk image.";
    112     print "Mounted python install image at: $mountPoint\n";
    113     my $installerPackage = File::Spec->join($mountPoint, $macPythonInstallerName);
    114     my $installSuccess = runInstaller($installerPackage);
    115     unmountDMG($mountPoint) or die "Failed to unmount disk image from $mountPoint";
    116     return $installSuccess;
    117 }
    118 
    119 sub main()
    120 {
    121     my $checkOnly = 0;
    122     my $showHelp = 0;
    123     my $getOptionsResult = GetOptions(
    124         'check-only!' => \$checkOnly,
    125         'help|h' => \$showHelp,
    126     );
    127     if (!$getOptionsResult || $showHelp) {
    128         print STDERR <<HELP;
    129 Usage: $0 [options]
    130   --check-only        Check python version only.
    131   -h|--help           Show this help message.
    132 HELP
    133         return 1;
    134     }
    135     # Congrats, your Python is fine.
    136     return 0 if checkPythonVersion();
    137 
    138     return 1 if $checkOnly;
    139 
    140     if (!isTiger()) {
    141         print "Your Python version is insufficient to run WebKit's Python code.  Please update.\n";
    142         print "See http://trac.webkit.org/wiki/PythonGuidelines for more info.\n";
    143         return 1;
    144     }
    145 
    146     installMacPython() or die "Failed to install Python.";
    147 
    148     checkPythonVersion() or die "Final version check failed, must have failed to update Python";
    149     print "Successfully updated python.\n";
    150 }
    151 
    152 exit(main());
    153