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