#!/usr/bin/env python
"""
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements.  See the NOTICE file
distributed with this work for additional information
regarding copyright ownership.  The ASF licenses this file
to you under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance
with the License.  You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

"""
import urllib2

from resource_management import *
from ambari_commons.os_family_impl import OsFamilyFuncImpl, OsFamilyImpl
from ambari_commons import OSConst
import time

@OsFamilyFuncImpl(os_family=OSConst.WINSRV_FAMILY)
def webhcat_service_check():
  Logger.info("Webhcat smoke test - service status")

  import params
  # AMBARI-11633 [WinTP2] Webhcat service check fails
  # Hive doesn't pass the environment variables correctly to child processes, which fails the smoke test.
  # Reducing the amount of URLs checked to the minimum required.
  #smoke_cmd = os.path.join(params.stack_root,"Run-SmokeTests.cmd")
  #service = "WEBHCAT"
  #Execute(format("cmd /C {smoke_cmd} {service}"), user=params.hcat_user, logoutput=True)

  url_tests = [
    "status",
    #These are the failing ones:
    #"ddl/database?user.name=hadoop",
    #"ddl/database/default/table?user.name=hadoop"
  ]


  import socket

  url_host = socket.getfqdn()
  url_port = params.config["configurations"]["webhcat-site"]["templeton.port"]

  for url_test in url_tests:
    url_request = "http://{0}:{1}/templeton/v1/{2}".format(url_host, url_port, url_test)
    url_response = None

    try:
      # execute the query for the JSON that includes WebHCat status
      url_response = urllib2.urlopen(url_request, timeout=30)

      status = url_response.getcode()
      response = url_response.read()

      if status != 200:
        Logger.warning("Webhcat service check status: {0}".format(status))
      Logger.info("Webhcat service check response: {0}".format(response))
    except urllib2.HTTPError as he:
      raise Fail("Webhcat check {0} failed: {1}".format(url_request, he.msg))
    finally:
      if url_response is not None:
        try:
          url_response.close()
        except:
          pass


@OsFamilyFuncImpl(os_family=OsFamilyImpl.DEFAULT)
def webhcat_service_check():
  import params
  File(format("{tmp_dir}/templetonSmoke.sh"),
       content= StaticFile('templetonSmoke.sh'),
       mode=0755
  )

  if params.security_enabled:
    smokeuser_keytab=params.smoke_user_keytab
    smoke_user_principal=params.smokeuser_principal
  else:
    smokeuser_keytab= "no_keytab"
    smoke_user_principal="no_principal"
    
  unique_name = format("{smokeuser}.{timestamp}", timestamp = time.time())
  templeton_test_script = format("idtest.{unique_name}.pig")
  templeton_test_input = format("/tmp/idtest.{unique_name}.in")
  templeton_test_output = format("/tmp/idtest.{unique_name}.out")

  File(format("{tmp_dir}/{templeton_test_script}"),
       content = Template("templeton_smoke.pig.j2", templeton_test_input=templeton_test_input, templeton_test_output=templeton_test_output),
       owner=params.hdfs_user
  )
  
  params.HdfsResource(format("/tmp/{templeton_test_script}"),
                      action = "create_on_execute",
                      type = "file",
                      source = format("{tmp_dir}/{templeton_test_script}"),
                      owner = params.smokeuser
  )
  
  params.HdfsResource(templeton_test_input,
                      action = "create_on_execute",
                      type = "file",
                      source = "/etc/passwd",
                      owner = params.smokeuser
  )
  
  params.HdfsResource(None, action = "execute")

  cmd = format("{tmp_dir}/templetonSmoke.sh {webhcat_server_host[0]} {smokeuser} {templeton_port} {templeton_test_script} {smokeuser_keytab}"
               " {security_param} {kinit_path_local} {smoke_user_principal}"
               " {tmp_dir}")

  Execute(cmd,
          tries=3,
          try_sleep=5,
          path='/usr/sbin:/sbin:/usr/local/bin:/bin:/usr/bin',
          logoutput=True)