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
28use strict;
29
30use File::Basename;
31use File::Spec;
32use File::Temp qw(tempdir);
33use FindBin;
34use Getopt::Long;
35
36use lib $FindBin::Bin;
37use webkitdirs;
38use VCSUtils;
39
40my $macPythonURL = "http://www.python.org/ftp/python/2.6.5/python-2.6.5-macosx10.3-2010-03-24.dmg";
41my $macPythonMD5 = "84489bba813fdbb6041b69d4310a86da";
42my $macPythonInstallerName = "Python.mpkg";
43
44# We could use a consistent download location, like the source or build directory.
45my $tempDirectory = File::Temp::tempdir("WebKitPythonXXXX", TMPDIR => 1, CLEANUP => 1);
46my $downloadDirectory = $tempDirectory;
47my $mountPoint = File::Spec->join($tempDirectory, "mount");
48
49sub 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
56sub 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
64sub 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
74sub 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
81sub 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
89sub 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
98sub 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
109sub 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
119sub 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;
129Usage: $0 [options]
130  --check-only        Check python version only.
131  -h|--help           Show this help message.
132HELP
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
152exit(main());
153