server.py

#!/usr/bin/env python
#
# Copyright (C) 2011 Instructure, Inc.
#
# This file is part of StraitJacket.
#
# StraitJacket is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# StraitJacket is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.
#

import web, json
from lib import straitjacket

__author__ = "JT Olds"
__copyright__ = "Copyright 2011 Instructure, Inc."
__license__ = "AGPLv3"
__email__ = "jt@instructure.com"

DEFAULT_CONFIG_DIR = "config"

INDEX_HTML = """
<h1>welcome to straitjacket</h1>
<form method="post" action="/execute">
<dl>
<dt>source:</dt><dd><textarea name="source" rows=8 cols=50></textarea></dd>
<dt>stdin:</dt><dd><textarea name="stdin" rows=8 cols=50></textarea></dd>
<dt>language:</dt><dd><select name="language">
%(languages)s
</select></dd></dl>
<input type="submit"/>
</form>
"""

def webapp(wrapper=None, config_dir=DEFAULT_CONFIG_DIR,
    skip_language_checks=False):
  if not wrapper:
    wrapper = straitjacket.StraitJacket(config_dir,
        skip_language_checks=skip_language_checks)

  index_html = INDEX_HTML % {"languages": "\n".join(
      ('<option value="%s">%s - %s</option>' % (lang,
       wrapper.enabled_languages[lang]["visible_name"],
       wrapper.enabled_languages[lang]["version"])
       for lang in sorted(wrapper.enabled_languages)))}

  class index:
    def GET(self):
      web.header('Content-Type', 'text/html')
      return index_html

  class execute:
    def POST(self):
      web.header('Content-Type', 'text/plain')
      f = web.input()
      timelimit = None
      if f.has_key("timelimit") and len(f.timelimit) > 0:
        try: timelimit = float(f.timelimit)
        except: pass
      try:
        stdout, stderr, exitstatus, runtime, error = wrapper.run(f.language,
            f.source, f.stdin, custom_timelimit=timelimit)
        return json.dumps({"stdout": stdout, "stderr": stderr,
            "exitstatus": exitstatus, "time": runtime, "error": error})
      except straitjacket.InputError: raise web.badrequest()

  class info:
    def GET(self):
      web.header('Content-Type', 'text/json')
      languages = {}
      for lang in wrapper.enabled_languages:
        languages[lang] = {
            "visible_name": wrapper.enabled_languages[lang]["visible_name"],
            "version": wrapper.enabled_languages[lang]["version"],
            "execution_profile": wrapper.enabled_languages[lang][
                "execution_profile"].__class__.__name__
          }
      return json.dumps({"languages": languages})

  app = web.application((
      '/', 'index',
      '/execute', 'execute',
      '/info', 'info',
    ), locals())

  return app

if __name__ == "__main__": webapp().run()