{#
# 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.
#}
#!/bin/bash

kafka_home=/usr/hdp/current/kafka-broker
kafka_zk_endpoint=tsys1.field.hortonworks.com:2181,tsys2.field.hortonworks.com:2181,tsys3.field.hortonworks.com:2181
default_replication_factor=1
metadata_user=atlas
rangertagsync_user=rangertagsync

create_topic() {
    topic_name=$1
    topics=`${kafka_home}/bin/kafka-topics.sh --zookeeper ${kafka_zk_endpoint} --topic $topic_name --list`
    if [ -z $topics ]; then
      ${kafka_home}/bin/kafka-topics.sh --zookeeper ${kafka_zk_endpoint} --topic $topic_name --create --partitions 1 --replication-factor ${default_replication_factor}
      echo "Created topic $topic_name with replication factor ${default_replication_factor}"
    else
      echo "Topic $topic_name already exists"
    fi
}

create_topic ATLAS_HOOK
create_topic ATLAS_ENTITIES

${kafka_home}/bin/kafka-acls.sh --authorizer-properties zookeeper.connect=${kafka_zk_endpoint} --add  --topic ATLAS_HOOK --allow-principal User:* --producer
${kafka_home}/bin/kafka-acls.sh --authorizer-properties zookeeper.connect=${kafka_zk_endpoint} --add  --topic ATLAS_HOOK --allow-principal User:${metadata_user} --consumer --group atlas
${kafka_home}/bin/kafka-acls.sh --authorizer-properties zookeeper.connect=${kafka_zk_endpoint} --add  --topic ATLAS_ENTITIES --allow-principal User:${metadata_user} --producer

${kafka_home}/bin/kafka-acls.sh --authorizer-properties zookeeper.connect=${kafka_zk_endpoint} --add  --topic ATLAS_ENTITIES --allow-principal User:${rangertagsync_user} --consumer --group ranger_entities_consumer