#!/usr/bin/env bash
# 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.

# This script automates the process of creating and packaging 
# a new vagrant base_box. For use locally (not aws).

base_dir=`dirname $0`/..
cd $base_dir

backup_vagrantfile=backup_Vagrantfile.local
local_vagrantfile=Vagrantfile.local

# Restore original Vagrantfile.local, if it exists
function revert_vagrantfile {
    rm -f $local_vagrantfile
    if [ -e $backup_vagrantfile ]; then
        mv $backup_vagrantfile $local_vagrantfile
    fi
}

function clean_up {
    echo "Cleaning up..."
    vagrant destroy -f
    rm -f package.box
    revert_vagrantfile
}

# Name of the new base box
base_box="kafkatest-worker"

# vagrant VM name
worker_name="worker1"

echo "Destroying vagrant machines..."
vagrant destroy -f

echo "Removing $base_box from vagrant..."
vagrant box remove $base_box

echo "Bringing up a single vagrant machine from scratch..."
if [ -e $local_vagrantfile ]; then
    mv $local_vagrantfile $backup_vagrantfile
fi
echo "num_workers = 1" > $local_vagrantfile
echo "num_brokers = 0" >> $local_vagrantfile
echo "num_zookeepers = 0" >> $local_vagrantfile
vagrant up
up_status=$?
if [ $up_status != 0 ]; then
    echo "Failed to bring up a template vm, please try running again."
    clean_up
    exit $up_status
fi

echo "Packaging $worker_name..."
vagrant package $worker_name

echo "Adding new base box $base_box to vagrant..."
vagrant box add $base_box package.box

clean_up