build/pgo/profileserver.py.in

   1 #literal #!/usr/bin/python
   2 #
   3 # ***** BEGIN LICENSE BLOCK *****
   4 # Version: MPL 1.1/GPL 2.0/LGPL 2.1
   5 #
   6 # The contents of this file are subject to the Mozilla Public License Version
   7 # 1.1 (the "License"); you may not use this file except in compliance with
   8 # the License. You may obtain a copy of the License at
   9 # http://www.mozilla.org/MPL/
  10 #
  11 # Software distributed under the License is distributed on an "AS IS" basis,
  12 # WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13 # for the specific language governing rights and limitations under the
  14 # License.
  15 #
  16 # The Original Code is mozilla.org code.
  17 #
  18 # The Initial Developer of the Original Code is
  19 # Mozilla Foundation.
  20 # Portions created by the Initial Developer are Copyright (C) 1998
  21 # the Initial Developer. All Rights Reserved.
  22 #
  23 # Contributor(s):
  24 #   Robert Sayre <sayrer@gmail.com>
  25 #   Jeff Walden <jwalden+bmo@mit.edu>
  26 #
  27 # Alternatively, the contents of this file may be used under the terms of
  28 # either the GNU General Public License Version 2 or later (the "GPL"), or
  29 # the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  30 # in which case the provisions of the GPL or the LGPL are applicable instead
  31 # of those above. If you wish to allow use of your version of this file only
  32 # under the terms of either the GPL or the LGPL, and not to allow others to
  33 # use your version of this file under the terms of the MPL, indicate your
  34 # decision by deleting the provisions above and replace them with the notice
  35 # and other provisions required by the GPL or the LGPL. If you do not delete
  36 # the provisions above, a recipient may use your version of this file under
  37 # the terms of any one of the MPL, the GPL or the LGPL.
  38 #
  39 # ***** END LICENSE BLOCK *****
  40
  41 import SimpleHTTPServer
  42 import SocketServer
  43 import socket
  44 import threading
  45 import os
  46 import sys
  47 import shutil
  48 from datetime import datetime
  49 from automation import Automation
  50
  51 PORT = 8888
  52 SCRIPT_DIR = os.path.abspath(os.path.realpath(os.path.dirname(sys.argv[0])))
  53 PROFILE_DIRECTORY = os.path.abspath(os.path.join(SCRIPT_DIR, "./pgoprofile"))
  54 os.chdir(SCRIPT_DIR)
  55
  56 class EasyServer(SocketServer.TCPServer):
  57   allow_reuse_address = True
  58
  59 if __name__ == '__main__':
  60   automation = Automation()
  61   httpd = EasyServer(("", PORT), SimpleHTTPServer.SimpleHTTPRequestHandler)
  62   t = threading.Thread(target=httpd.serve_forever)
  63   t.setDaemon(True) # don't hang on exit
  64   t.start()
  65
  66   automation.initializeProfile(PROFILE_DIRECTORY)
  67   browserEnv = automation.environment()
  68   browserEnv["XPCOM_DEBUG_BREAK"] = "warn"
  69
  70   url = "http://localhost:%d/index.html" % PORT
  71   appPath = os.path.join(SCRIPT_DIR, automation.DEFAULT_APP)
  72   status = automation.runApp(url, browserEnv, appPath, PROFILE_DIRECTORY, {},
  73                              # the profiling HTML doesn't output anything,
  74                              # so let's just run this without a timeout
  75                              timeout = None)
  76   sys.exit(status)