#
#
# 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.
#
#
class hdp2-hadoop::hdfs::service_check()
{
  $unique = hdp_unique_id_and_date()
  $dir = '/tmp'
  $tmp_file = "${dir}/${unique}"
  $user_dir='/user'
  $apps_dir='/apps'

  $safemode_command = "dfsadmin -safemode get | grep OFF"

  $create_dir_cmd = "fs -mkdir ${dir} ; hadoop fs -chmod -R 777 ${dir} ; hadoop fs -mkdir ${user_dir} ; hadoop fs -chmod 755 ${user_dir} ; hadoop fs -mkdir ${apps_dir} ; hadoop fs -chmod 755 ${apps_dir} ;"
  $test_dir_exists = "hadoop fs -test -e ${dir}" #TODO: may fix up fact that test needs explicit hadoop while omamnd does not
  $cleanup_cmd = "fs -rm ${tmp_file}"
  #cleanup put below to handle retries; if retrying there wil be a stale file that needs cleanup; exit code is fn of second command
  $create_file_cmd = "${cleanup_cmd}; hadoop fs -put /etc/passwd ${tmp_file}" #TODO: inconsistent that second comamnd needs hadoop
  $test_cmd = "fs -test -e ${tmp_file}"

  anchor { 'hdp2-hadoop::hdfs::service_check::begin':}

  hdp2-hadoop::exec-hadoop { 'hdfs::service_check::check_safemode':
    command   => $safemode_command,
    tries     => 40,
    try_sleep => 15,
    logoutput => true,
    require   => Anchor['hdp2-hadoop::hdfs::service_check::begin']
  }

  hdp2-hadoop::exec-hadoop { 'hdfs::service_check::create_dir':
    command   => $create_dir_cmd,
    unless    => $test_dir_exists,
    tries     => 3,
    try_sleep => 5,
    require   => Hdp2-hadoop::Exec-hadoop['hdfs::service_check::check_safemode']
  }

  hdp2-hadoop::exec-hadoop { 'hdfs::service_check::create_file':
    command   => $create_file_cmd,
    tries     => 3,
    try_sleep => 5,
    require   => Hdp2-hadoop::Exec-hadoop['hdfs::service_check::create_dir'],
    notify    => Hdp2-hadoop::Exec-hadoop['hdfs::service_check::test']
  }

  hdp2-hadoop::exec-hadoop { 'hdfs::service_check::test':
    command     => $test_cmd,
    refreshonly => true,
    require     => Hdp2-hadoop::Exec-hadoop['hdfs::service_check::create_file'],
    #notify      => Hdp2-hadoop::Exec-hadoop['hdfs::service_check::cleanup']  #TODO: put in after testing
    before      => Anchor['hdp2-hadoop::hdfs::service_check::end'] #TODO: remove after testing
  }

   #TODO: put in after testing
 #  hdp2-hadoop::exec-hadoop { 'hdfs::service_check::cleanup':
 #   command     => $cleanup_cmd,
 #   refreshonly => true,
 #   require     => Hdp2-hadoop::Exec-hadoop['hdfs::service_check::test'],
 #   before      => Anchor['hdp2-hadoop::hdfs::service_check::end']
  #}
  anchor{ 'hdp2-hadoop::hdfs::service_check::end':}

  class { 'hdp2-hadoop::hdfs-directories' :
    service_state => running  }
}

class hdp2-hadoop::hdfs-directories($service_state)
{
  $smoke_test_user = $hdp2::params::smokeuser
  hdp2-hadoop::hdfs::directory{ "/user/${smoke_test_user}":
    service_state => $service_state,
    owner => $smoke_test_user,
    mode  => '770',
    recursive_chmod => true
  }
}