text: """
  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.
  
  Ambari Agent
  
  """
  
  from resource_management import *
  from resource_management.libraries.functions import conf_select
  from resource_management.libraries.functions import stack_select
  from resource_management.libraries.functions.version import compare_versions, format_stack_version
  from resource_management.libraries.functions.security_commons import build_expectations, \
    cached_kinit_executor, get_params_from_filesystem, validate_security_config_properties,\
    FILE_TYPE_XML
  from resource_management.libraries.functions.format import format
  
  from yarn import yarn
  from service import service
  
  class ApplicationTimelineServer(Script):
  
    def get_component_name(self):
      return "hadoop-yarn-timelineserver"
  
    def install(self, env):
      self.install_packages(env)
      #self.configure(env)
  
    def configure(self, env):
      import params
      env.set_params(params)
      yarn(name='apptimelineserver')
  
    def pre_upgrade_restart(self, env, upgrade_type=None):
      Logger.info("Executing Stack Upgrade pre-restart")
      import params
      env.set_params(params)
  
      if params.version and compare_versions(format_stack_version(params.version), '4.0.0.0') >= 0:
        conf_select.select(params.stack_name, "hadoop", params.version)
        stack_select.select("hadoop-yarn-timelineserver", params.version)
        #Execute(format("iop-select set hadoop-yarn-timelineserver {version}"))
  
    def start(self, env, upgrade_type=None):
      import params
      env.set_params(params)
      self.configure(env) # FOR SECURITY
      service('timelineserver', action='start')
  
    def stop(self, env, upgrade_type=None):
      import params
      env.set_params(params)
      service('timelineserver', action='stop')
  
    def status(self, env):
      import status_params
      env.set_params(status_params)
      Execute(format("mv {yarn_historyserver_pid_file_old} {yarn_historyserver_pid_file}"),
              only_if = format("test -e {yarn_historyserver_pid_file_old}", user=status_params.yarn_user))
      functions.check_process_status(status_params.yarn_historyserver_pid_file)
  
    def security_status(self, env):
      import status_params
      env.set_params(status_params)
      if status_params.security_enabled:
        props_value_check = {"yarn.timeline-service.enabled": "true",
                             "yarn.timeline-service.http-authentication.type": "kerberos",
                             "yarn.acl.enable": "true"}
        props_empty_check = ["yarn.timeline-service.principal",
                             "yarn.timeline-service.keytab",
                             "yarn.timeline-service.http-authentication.kerberos.principal",
                             "yarn.timeline-service.http-authentication.kerberos.keytab"]
  
        props_read_check = ["yarn.timeline-service.keytab",
                            "yarn.timeline-service.http-authentication.kerberos.keytab"]
        yarn_site_props = build_expectations('yarn-site', props_value_check, props_empty_check,
                                                    props_read_check)
  
        yarn_expectations ={}
        yarn_expectations.update(yarn_site_props)
  
        security_params = get_params_from_filesystem(status_params.hadoop_conf_dir,
                                                     {'yarn-site.xml': FILE_TYPE_XML})
        result_issues = validate_security_config_properties(security_params, yarn_expectations)
        if not result_issues: # If all validations passed successfully
          try:
            # Double check the dict before calling execute
            if ( 'yarn-site' not in security_params
                 or 'yarn.timeline-service.keytab' not in security_params['yarn-site']
                 or 'yarn.timeline-service.principal' not in security_params['yarn-site']) \
              or 'yarn.timeline-service.http-authentication.kerberos.keytab' not in security_params['yarn-site'] \
              or 'yarn.timeline-service.http-authentication.kerberos.principal' not in security_params['yarn-site']:
              self.put_structured_out({"securityState": "UNSECURED"})
              self.put_structured_out(
                {"securityIssuesFound": "Keytab file or principal are not set property."})
              return
  
            cached_kinit_executor(status_params.kinit_path_local,
                                  status_params.yarn_user,
                                  security_params['yarn-site']['yarn.timeline-service.keytab'],
                                  security_params['yarn-site']['yarn.timeline-service.principal'],
                                  status_params.hostname,
                                  status_params.tmp_dir)
            cached_kinit_executor(status_params.kinit_path_local,
                                  status_params.yarn_user,
                                  security_params['yarn-site']['yarn.timeline-service.http-authentication.kerberos.keytab'],
                                  security_params['yarn-site']['yarn.timeline-service.http-authentication.kerberos.principal'],
                                  status_params.hostname,
                                  status_params.tmp_dir)
            self.put_structured_out({"securityState": "SECURED_KERBEROS"})
          except Exception as e:
            self.put_structured_out({"securityState": "ERROR"})
            self.put_structured_out({"securityStateErrorInfo": str(e)})
        else:
          issues = []
          for cf in result_issues:
            issues.append("Configuration file %s did not pass the validation. Reason: %s" % (cf, result_issues[cf]))
          self.put_structured_out({"securityIssuesFound": ". ".join(issues)})
          self.put_structured_out({"securityState": "UNSECURED"})
      else:
        self.put_structured_out({"securityState": "UNSECURED"})
  
  
  if __name__ == "__main__":
    ApplicationTimelineServer().execute()
  