sql/builder.sh

   1 #! /bin/bash
   2 # set -x
   3 #
   4 MATCH_DROP='^DROP.*([^-]...|.[^-]..|..[^M].|...[^F])$|^ALTER TABLE.* DROP .*([^-]...|.[^-]..|..[^M].|...[^F])$|^DELETE .*([^-]...|.[^-]..|..[^M].|...[^F])$|--MB$'
   5
   6 DATECUR="$(date +%s)"
   7
   8 #  functions
   9 usage () {
  10     echo " USAGE"
  11     echo "   ./builder <command> [-d|--dryrun] [-a|-p|--allfiles|--devfiles] [-s|--short] ..."
  12     echo "   ./builder <-h|--help|help>"
  13     echo "   commands are:"
  14     echo "       create"
  15     echo "       destroy"
  16     echo "       clean"
  17     echo "       build"
  18     echo "       rebuild"
  19     echo "       psql"
  20     echo "       piped"
  21     echo "       add <filesql> [<filesql2> [..]]"
  22     echo "       del <filesql> [<filesql2> [..]]"
  23     echo "       res <filesql> [<filesql2> [..]]"
  24     echo "       dump [dumpfile]"
  25     echo "       dumpall [dumpfile]"
  26     echo "       all"
  27     echo
  28     echo "The match rule for clean lines is:"
  29     echo "  [$MATCH_DROP]"
  30     echo "NOTE: to invert normal 'del' rules add '--MF' (move forward) suffix to each line"
  31     echo "      to invert normal 'add' rules add '--MB' (move backward) suffix to each line"
  32
  33     exit $1
  34 }
  35
  36 sqlexe () {
  37     local sht
  38     sht=$1
  39
  40     if [ "$SHORT" = "y" ];  then
  41         sed "s/#PFX#/$PFX/g;s/#NOW#/$DATECUR/g" | psql -a $pg_args 2>&1 | egrep 'ERROR|^-- MESG|^-- FILE '
  42     else
  43         sed "s/#PFX#/$PFX/g;s/#NOW#/$DATECUR/g" | psql -a $pg_args
  44     fi
  45
  46     return 0
  47 }
  48
  49 one_or_all() {
  50     local old_ifs
  51
  52     old_ifs="$IFS"
  53     IFS=" "
  54     for fil in $(
  55         if [ "$1" ]; then
  56             echo "$1"
  57         elif [ "$TYPE_FILES" = "a" ]; then
  58             echo sql.d/[0-9]*
  59         elif [ "$TYPE_FILES" = "d" ]; then
  60             echo sql.d/[0-9]*.{sql,devel}
  61         else
  62             echo sql.d/[0-9]*.sql
  63             fi); do
  64         echo "-- FILE BEG: $fil"
  65         cat "$fil"
  66         echo "-- FILE END: $fil"
  67     done
  68     IFS="$old_ifs"
  69 }
  70
  71 #
  72 #  MAIN
  73 #
  74
  75 CMD=$1
  76 shift
  77
  78 while [ $# -gt 0 ]; do
  79     case $1 in
  80         -d|--dryrun)
  81             DRY_RUN=y
  82             psql () {
  83                 echo "MOCKPSQL params: $@"
  84                 cat
  85             }
  86             ;;
  87         -a|--allfiles)
  88             TYPE_FILES=a
  89             ;;
  90         -p|--devfiles)
  91             TYPE_FILES=d
  92             ;;
  93         -s|--short)
  94             SHORT=y
  95             ;;
  96         *)
  97             break
  98             ;;
  99     esac
 100     shift
 101 done
 102
 103 if [ -f $HOME/.brisk-db.conf ]; then
 104     source $HOME/.brisk-db.conf
 105 elif [ -f $HOME/.db.conf ]; then
 106     source $HOME/.db.conf
 107 else
 108     DBHOST=127.0.0.1
 109     DBUSER=brisk
 110     DBPORT=5432
 111     DBBASE=brisk
 112     DBPASS=briskpass
 113     PFX="bsk_"
 114 fi
 115
 116 if [ -f $HOME/.brisk_install ]; then
 117     source $HOME/.brisk_install
 118 fi
 119
 120 pg_args=""
 121 test "$DBHOST" != "" && pg_args="$pg_args -h $DBHOST"
 122 test "$DBUSER" != "" && pg_args="$pg_args -U $DBUSER"
 123 test "$DBPORT" != "" && pg_args="$pg_args -p $DBPORT"
 124 test "$DBBASE" != "" && pg_args="$pg_args $DBBASE"
 125
 126
 127 case $CMD in
 128     "create")
 129         echo "su root"
 130         su root -c "su postgres -c \"echo \\\"DBUser passwd: $DBPASS\\\" ; createuser -S -D -R -P $DBUSER && createdb -E utf8 -O $DBUSER $DBBASE\""
 131         ;;
 132
 133     "destroy")
 134         echo "su root"
 135         su root -c "su postgres -c \"dropdb $DBBASE && dropuser $DBUSER\""
 136         ;;
 137     "clean")
 138         ( echo "-- MESG: clean start" ; one_or_all $2 | egrep "$MATCH_DROP|^-- MESG|^-- FILE " | tac ; echo "-- MESG: clean end" ;   ) | sqlexe
 139         ;;
 140     "build")
 141         ( echo "-- MESG: build start" ; one_or_all $2 | egrep -v "$MATCH_DROP" ; echo "-- MESG: build end" ;   ) | sqlexe
 142         ;;
 143     "rebuild")
 144         ( echo "-- MESG: clean start" ; one_or_all $2 | egrep "$MATCH_DROP|^-- MESG|^-- FILE " | tac ; echo "-- MESG: clean end" ; \
 145             echo "-- MESG: build start" ; one_or_all $2 | egrep -v "$MATCH_DROP" ; echo "-- MESG: build end" ;   ) \
 146             | sqlexe
 147         ;;
 148     "psql")
 149         psql $pg_args $@
 150         ;;
 151
 152     "piped")
 153         psql $pg_args -t -q -A -F '|' $@
 154         ;;
 155     "dump")
 156         if [ $# -eq 1 ]; then
 157             pg_dump -a --inserts -h $DBHOST -U $DBUSER $DBBASE
 158         else
 159             pg_dump -a --inserts -h $DBHOST -U $DBUSER $DBBASE > $1
 160         fi
 161         ;;
 162     "dumpall")
 163         if [ $# -eq 1 ]; then
 164             pg_dump -h $DBHOST -U $DBUSER $DBBASE
 165         else
 166             pg_dump -h $DBHOST -U $DBUSER $DBBASE > $1
 167         fi
 168         ;;
 169     "add")
 170         ( echo "-- MESG: add start" ; cat "$@" | egrep -v "$MATCH_DROP" ; echo "-- MESG: add end" ;   ) | sqlexe
 171         ;;
 172     "del")
 173         ( echo "-- MESG: del start" ; cat "$@" | egrep "$MATCH_DROP|^-- MESG|^-- FILE " | tac ; echo "-- MESG: del end" ;   ) | sqlexe
 174         ;;
 175     "res")
 176         ( echo "-- MESG: res start" ; cat "$@" | egrep "$MATCH_DROP|^-- MESG|^-- FILE " | tac ; cat "$@" | egrep -v "$MATCH_DROP" ; echo "-- MESG: del end" ;   ) | sqlexe
 177         ;;
 178     "help"|"-h"|"--help")
 179         usage 0
 180         ;;
 181     *)
 182         usage 1
 183         ;;
 184 esac
 185
 186 exit 0