#!/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 {

}

function clean_up {

}

# 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