# Copyright (c) 2012 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import time

from appengine_wrappers import taskqueue
from cron_servlet import CronServlet
from instance_servlet import InstanceServlet
from patch_servlet import PatchServlet
from refresh_servlet import RefreshServlet
from servlet import Servlet, Request, Response
from test_servlet import TestServlet


_DEFAULT_SERVLET = InstanceServlet.GetConstructor()


_FORCE_CRON_TARGET = 'force_cron'


_SERVLETS = {
  'cron': CronServlet,
  'patch': PatchServlet,
  'refresh': RefreshServlet,
  'test': TestServlet,
}


class Handler(Servlet):
  def Get(self):
    path = self._request.path

    if path.startswith('_'):
      servlet_path = path[1:]
      if not '/' in servlet_path:
        servlet_path += '/'
      servlet_name, servlet_path = servlet_path.split('/', 1)
      if servlet_name == _FORCE_CRON_TARGET:
        queue = taskqueue.Queue()
        queue.purge()
        time.sleep(2)
        queue.add(taskqueue.Task(url='/_cron'))
        return Response.Ok('Cron job started.')
      if servlet_name == 'enqueue':
        queue = taskqueue.Queue()
        queue.add(taskqueue.Task(url='/%s'%servlet_path))
        return Response.Ok('Task enqueued.')
      servlet = _SERVLETS.get(servlet_name)
      if servlet is None:
        return Response.NotFound('"%s" servlet not found' %  servlet_path)
    else:
      servlet_path = path
      servlet = _DEFAULT_SERVLET

    return servlet(Request(servlet_path,
                           self._request.host,
                           self._request.headers,
                           self._request.arguments)).Get()