db-helper

#! /usr/bin/env bash


###################################################################################
#                                                                                 #
#                   Copyright 2010-2011 Ning, Inc.                                #
#                                                                                 #
#      Ning 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.                                                         #
#                                                                                 #
###################################################################################

#set -x 

HERE=`cd \`dirname $0\`; pwd`
TOP=$HERE/..

POM="$TOP/pom.xml"

ACTION=
DATABASE="killbill"
USER="root"
PWD="root"
TEST_ALSO=

DDL_FILE=
CLEAN_FILE=

# Egrep like for skipping some modules until they are ready
SKIP="(usage|server)"

function usage() {
    echo -n "./db_helper "
    echo -n " -a <create|clean|dump>"
    echo -n " -d database_name (default = killbill)"    
    echo -n " -u user_name (default = root)"
    echo -n " -p password (default = root)"
    echo -n " -t (also include test ddl)"
    echo -n "-h this message"
    echo
    exit 1
}

function get_modules() {
    local modules=`grep module $POM  | grep -v modules | cut -d '>' -f 2 | cut -d '<' -f 1 | egrep -v "$SKIP"`
    echo $modules
}

function find_test_ddl() {
    local modules=`get_modules`
    local ddl_test=
    
    local cur_ddl=
    for m in $modules; do
        cur_ddl=`find $m/src/test/resources/ -name ddl_test.sql 2>/dev/null`
        ddl_test="$ddl_test $cur_ddl"
    done
    echo "$ddl_test"
    
}
function find_src_ddl() {
    
    local modules=`get_modules`
    local ddl_src=
    
    local cur_ddl=
    for m in $modules; do
        cur_ddl=`find $m/src/main/resources/ -name ddl.sql 2>/dev/null`
        ddl_src="$ddl_src $cur_ddl"
    done
    echo "$ddl_src"
}


function create_clean_file() {
    local ddl_file=$1
    local tables=`cat $ddl_file | grep -i "create table" | awk ' { print $3 } '` 

    local tmp="/tmp/clean-$DATABASE.$$" 
    echo "use $DATABASE;" >> $tmp
    echo "" >> $tmp
    for t in $tables; do
        echo "truncate $t;" >> $tmp
    done
    echo $tmp
}

function create_ddl_file() {
    local ddls=`find_src_ddl`
    local test_ddls=
    if [ ! -z $TEST_ALSO ]; then
        test_ddls=`find_test_ddl`
        ddls="$ddls $test_ddls"
    fi

    local tmp="/tmp/ddl-$DATABASE.$$"
    touch $tmp
    echo "use $DATABASE;" >> $tmp
    echo "" >> $tmp
    for d in $ddls; do
        cat $d >> $tmp
        echo "" >> $tmp
    done
    echo $tmp
}

function cleanup() {
    rm -f "/tmp/*.$$"
}


while getopts ":a:d:u:pt" options; do
  case $options in
    a ) ACTION=$OPTARG;;
	d ) DATABASE=$OPTARG;;
	u ) USER=$OPTARG;;
	p ) PWD=$OPTARG;;
	t ) TEST_ALSO=1;;
    h ) usage;;
    * ) usage;;
  esac
done



if [ -z $ACTION ]; then
    echo "Need to specify an action <CREATE|CLEAN>"
    usage
fi


if [ $ACTION == "dump" ]; then
     DDL_FILE=`create_ddl_file`
    cat $DDL_FILE
fi

if [ $ACTION == "create" ]; then
    DDL_FILE=`create_ddl_file`
    echo "Applying new schema $tmp to database $DATABASE"    
    mysql -u $USER --password=$PWD < $DDL_FILE
fi

if [ $ACTION == "clean" ]; then
    DDL_FILE=`create_ddl_file` 
    CLEAN_FILE=`create_clean_file $DDL_FILE`   
    echo "Cleaning db tables on database $DATABASE"
    mysql -u $USER --password=$PWD < $DDL_FILE
fi

cleanup