#
#
# 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 hdp-zookeeper(
  $type = server,
  $service_state = $hdp::params::cluster_service_state,
  $myid = 1,
  $opts = {}
) inherits hdp-zookeeper::params 
{

 if ($service_state == 'no_op') {
   if ($type == 'server') {
     $hdp::params::service_exists['hdp-zookeeper'] = true
  }
 } elsif ($service_state in ['running','stopped','installed_and_configured','uninstalled']) { 
   $zk_user = $hdp-zookeeper::params::zk_user
   $zk_config_dir = $hdp-zookeeper::params::conf_dir
 
   anchor{'hdp-zookeeper::begin':}
   anchor{'hdp-zookeeper::end':}

   if ($service_state == 'uninstalled') {
     if ($type == 'server') {
       $hdp::params::service_exists['hdp-zookeeper'] = true
    }
     hdp::package { 'zookeeper':
       ensure => 'uninstalled'
     }
     hdp::directory_recursive_create { $zk_config_dir:
       service_state => $service_state,
       force => true
     }

     if ($type == 'server') {
        class { 'hdp-zookeeper::service':
          ensure => $service_state,
          myid   => $myid
        }
       }

     if ($type == 'server') {
       Anchor['hdp-zookeeper::begin'] -> Hdp::Package['zookeeper'] -> Hdp::Directory_recursive_create[$zk_config_dir] -> Class['hdp-zookeeper::service']  -> Anchor['hdp-zookeeper::end']
     } else {
       Anchor['hdp-zookeeper::begin'] -> Hdp::Package['zookeeper'] -> Hdp::Directory_recursive_create[$zk_config_dir] -> Anchor['hdp-zookeeper::end']
     }
   } else {
     hdp::package { 'zookeeper':}

     hdp::user{ $zk_user:}

     hdp::directory_recursive_create { $zk_config_dir: 
      service_state => $service_state,
      force => true
     }

     hdp-zookeeper::configfile { ['zoo.cfg','zookeeper-env.sh','configuration.xsl','log4j.properties']: }
 
     if ($hdp::params::update_zk_shell_files == true) {
       hdp-zookeeper::shell_file{ ['zkServer.sh','zkEnv.sh']: }
     }

     if ($type == 'server') {
       $hdp::params::service_exists['hdp-zookeeper'] = true
       class { 'hdp-zookeeper::service': 
         ensure => $service_state,
         myid   => $myid
       }
      }

      Anchor['hdp-zookeeper::begin'] -> Hdp::Package['zookeeper'] -> Hdp::User[$zk_user] -> 
        Hdp::Directory_recursive_create[$zk_config_dir] -> Hdp-zookeeper::Configfile<||> -> Anchor['hdp-zookeeper::end']
      if ($type == 'server') {
        Hdp::Directory_recursive_create[$zk_config_dir] -> Hdp-zookeeper::Configfile<||> -> Class['hdp-zookeeper::service'] -> Anchor['hdp-zookeeper::end']
      }
      if ($hdp::params::update_zk_shell_files == true) {
        Hdp::Package['zookeeper'] -> Hdp-zookeeper::Shell_file<||> -> Anchor['hdp-zookeeper::end']
      }
    }
  } else {
    hdp_fail("TODO not implemented yet: service_state = ${service_state}")
  }
}

### config files
define hdp-zookeeper::configfile(
  $mode = undef
) 
{
  hdp::configfile { "${hdp-zookeeper::params::conf_dir}/${name}":
    component       => 'zookeeper',
    owner           => $hdp-zookeeper::params::zk_user,
    mode            => $mode
  }
}

### 
define hdp-zookeeper::shell_file()
{
  file { "${hdp::params::zk_bin}/${name}":
    source => "puppet:///modules/hdp-zookeeper/${name}", 
    mode => '0755'
  }
}