#!/bin/bash

set -ex

bottlenecks_env_prepare()
{
   if [ -d $BOTTLENECKS_REPO_DIR ]; then
       rm -rf ${BOTTLENECKS_REPO_DIR}
   fi

   mkdir -p ${BOTTLENECKS_REPO_DIR}
   git config --global http.sslVerify false
   git clone ${BOTTLENECKS_REPO} ${BOTTLENECKS_REPO_DIR}

   source $BOTTLENECKS_REPO_DIR/rubbos/rubbos_scripts/1-1-1/scripts/env_preparation.sh
}

bottlenecks_create_instance()
{
   echo "create bottlenecks instance using heat template"

   echo "upload keypair"
   nova keypair-add --pub_key $KEY_PATH/bottlenecks_key.pub $KEY_NAME
   #need FIX, only upload the public key? should be keypair

   echo "use heat template to create stack"
   cd $HOT_PATH
   heat stack-create bottlenecks -f ${TEMPLATE_NAME} -P "image=$IMAGE_NAME;key=$KEY_NAME;public_network=$PUBLIC_NET_NAME"
   sleep 60
   heat stack-list
   heat stack-show bottlenecks
   nova list
   heat stack-delete bottlenecks

   #need FIX, use stack to create 9 VMs
}

bottlenecks_cleanup()
{
   echo "clean up bottlenecks images and keys"

   if glance image-list; then
       for image in $(glance image-list | grep -e $IMAGE_NAME | awk '{print $2}'); do
           echo "clean up image $image"
           glance image-delete $image || true
       done
   fi

   if nova keypair-list; then
       for key in $(nova keypair-list | grep -e $KEY_NAME | awk '{print $2}'); do
           echo "clean up key $key"
           nova keypair-delete || true
       done
   fi
}

bottlenecks_build_image()
{
   echo "build bottlenecks image"

   #need FIX
}

bottlenecks_load_cirros_image()
{
   echo "load bottlenecks cirros image"

   wget http://download.cirros-cloud.net/0.3.3/cirros-0.3.3-x86_64-disk.img -O /tmp/cirros.img

   result=$(glance image-create \
       --name cirros-0.3.3 \
       --disk-format qcow2 \
       --container-format bare \
       --file /tmp/cirros.img)
   echo "$result"

   rm -rf /tmp/cirros.img

   IMAGE_ID_CIRROS=$(echo "$result" | grep " id " | awk '{print $(NF-1)}')
   if [ -z "$IMAGE_ID_CIRROS" ]; then
        echo 'failed to upload cirros image to openstack'
        exit 1
   fi

   echo "cirros image id: $IMAGE_ID_CIRROS"
}

bottlenecks_load_image()
{
   echo "load bottlenecks image"

   result=$(glance --os-image-api-version 1 image-create \
      --name $IMAGE_NAME \
      --is-public true --disk-format qcow2 \
      --container-format bare \
      --file $IMAGE_FILE_NAME)
   echo "$result"

   GLANCE_IMAGE_ID=$(echo "$result" | grep " id " | awk '{print $(NF-1)}')

   if [ -z "$GLANCE_IMAGE_ID" ]; then
       echo 'add image to glance failed'
       exit 1
   fi

   sudo rm -f $IMAGE_FILE_NAME

   echo "add glance image completed: $GLANCE_IMAGE_ID"
}

main()
{
   echo "create instances with heat template"

   BOTTLENECKS_REPO=https://gerrit.opnfv.org/gerrit/bottlenecks
   BOTTLENECKS_REPO_DIR=/tmp/opnfvrepo/bottlenecks
   #IMAGE_URL=http://205.177.226.235:9999
   IMAGE_NAME=cirros-0.3.3
   #need FIX, need a script to transfer the image from the url to be the installer images
   KEY_PATH=$BOTTLENECKS_REPO_DIR/utils/infra_setup/bottlenecks_key
   HOT_PATH=$BOTTLENECKS_REPO_DIR/utils/infra_setup/heat_template
   KEY_NAME=bottlenecks_key
   TEMPLATE_NAME=bottlenecks_template1.yaml
   PUBLIC_NET_NAME=net04_ext
   #need FIX
   #IMAGE_FILE_NAME=""

   bottlenecks_env_prepare
   bottlenecks_cleanup
   #bottlenecks_build_image
   bottlenecks_load_cirros_image
   bottlenecks_create_instance
}

main
set +ex