Rename WebKitTools to Tools
[WebKit-https.git] / Tools / Scripts / ensure-valid-python
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());