mop_flush modify
[brisk.git] / sql / builder.sh
1 #! /bin/bash
2
3 #
4 #  all this part is from mopshop and we will use it to construct the brisk database
5 #
6
7 if [ -f $HOME/.db.conf ]; then
8     source $HOME/.db.conf
9 else
10     DBHOST=127.0.0.1
11     DBUSER=brisk
12     DBPORT=5432
13     DBBASE=brisk
14     DBPASS=briskpass
15     PFX="bsk_"
16 fi
17
18 if [ -f $HOME/.brisk_install ]; then
19     source $HOME/.brisk_install
20 fi
21
22 pg_args=""
23 test "$DBHOST" != "" && pg_args="$pg_args -h $DBHOST"
24 test "$DBUSER" != "" && pg_args="$pg_args -U $DBUSER"
25 test "$DBPORT" != "" && pg_args="$pg_args -p $DBPORT"
26 test "$DBBASE" != "" && pg_args="$pg_args $DBBASE"
27         
28 sqlexe () {
29     local sht
30     sht=$1
31     
32     if [ $sht -eq 1 ];  then 
33         sed "s/#PFX#/$PFX/g" | psql -a $pg_args 2>&1 | egrep 'ERROR|^-- MESG' 
34     else
35         sed "s/#PFX#/$PFX/g" | psql -a $pg_args
36     fi
37
38     return 0
39 }
40
41 one_or_all() {
42     if [ "$1" = "" ]; then
43         cat sql.d/*.sql
44     else
45         cat "$1"
46     fi
47 }
48
49 #
50 # MAIN
51 #
52 sht=0
53
54 if [ "$1" = "-s" ]; then
55     shift
56     sht=1
57 fi
58
59 if [ "$1" = "create" ]; then
60     echo "su root" 
61     su root -c "su postgres -c \"echo \\\"DBUser passwd: $DBPASS\\\" ; createuser -S -D -R -P $DBUSER && createdb -E utf8 -O $DBUSER $DBBASE\"" 
62 elif [ "$1" = "destroy" ]; then
63     echo "su root" 
64     su root -c "su postgres -c \"dropdb $DBBASE && dropuser $DBUSER\"" 
65 elif [ "$1" = "clean" ]; then
66     ( echo "-- MESG: clean start" ; one_or_all $2 | grep -i '^drop' | tac ; echo "-- MESG: clean end" ;   ) | sqlexe $sht
67 elif [ "$1" = "build" ]; then
68     ( echo "-- MESG: build start" ; one_or_all $2 | grep -iv '^drop' ; echo "-- MESG: build end" ;   ) | sqlexe $sht
69 elif [ "$1" = "rebuild" ]; then
70     ( echo "-- MESG: clean start" ; one_or_all $2 | grep -i '^drop' | tac ; echo "-- MESG: clean end" ; \
71         echo "-- MESG: build start" ; one_or_all $2 | grep -iv '^drop' ; echo "-- MESG: build end" ;   ) \
72         | sqlexe $sht
73 elif [ "$1" = "psql" ]; then
74    shift
75    psql $pg_args $@
76 elif [ "$1" = "piped" ]; then
77    shift
78    psql $pg_args -t -q -A -F '|' $@
79 elif [ "$1" = "dump" ]; then
80     if [ $# -eq 1 ]; then
81         pg_dump -a --inserts -h $DBHOST -U $DBUSER $DBBASE
82     else
83         pg_dump -a --inserts -h $DBHOST -U $DBUSER $DBBASE > $2
84     fi
85 elif [ "$1" = "dumpall" ]; then
86     if [ $# -eq 1 ]; then
87         pg_dump -h $DBHOST -U $DBUSER $DBBASE
88     else
89         pg_dump -h $DBHOST -U $DBUSER $DBBASE > $2
90     fi
91 elif [ "$1" = "add" ]; then
92     cat "$2" | psql $pg_args
93 else
94     echo " USAGE"
95     echo "   ./builder create"
96     echo "   ./builder destroy"
97     echo "   ./builder clean"
98     echo "   ./builder build"
99     echo "   ./builder rebuild"
100     echo "   ./builder psql"
101     echo "   ./builder piped"
102     echo "   ./builder add <filesql>"
103     echo "   ./builder dump [dumpfile]"
104     echo "   ./builder dumpall [dumpfile]"
105     echo "   ./builder all"
106     echo "   ./builder help"
107 fi