systemd/sfa-setup.sh

   1 #!/bin/bash
   2
   3 set -e
   4
   5 ####################
   6 PGDATA=/var/lib/pgsql/data/
   7 PGLOG=/var/log/pgsql
   8
   9 postgresql_conf=$PGDATA/postgresql.conf
  10 pg_hba_conf=$PGDATA/pg_hba.conf
  11
  12 # SFA consolidated (merged) config file
  13 sfa_whole_config=/etc/sfa/sfa_config
  14 # SFA default config (read-only template)
  15 sfa_default_config=/etc/sfa/default_config.xml
  16 # SFA local (site-dependent) file
  17 sfa_local_config=/etc/sfa/configs/site_config
  18 sfa_local_config_xml=/etc/sfa/configs/site_config.xml
  19 sfa_local_config_sh=/etc/sfa/sfa_config.sh
  20
  21 # Regenerate configuration files - almost verbatim from plc.init
  22 function reload () {
  23
  24     # Regenerate the main configuration file from default values
  25     # overlaid with site-specific and current values.
  26     files=( $sfa_default_config $sfa_local_config )
  27     tmp=$(mktemp /tmp/sfa_config.XXXXXX)
  28     sfa-config --python "${files[@]}" > $tmp
  29     if [ $? -eq 0 ] ; then
  30                 mv $tmp $sfa_whole_config
  31                 chmod 444 $sfa_whole_config
  32     else
  33                 echo "SFA: Warning: Invalid configuration file(s) detected"
  34             rm -f $tmp
  35         exit 1
  36     fi
  37
  38     # Convert configuration to various formats
  39     if [ -f $sfa_local_config_xml ] ; then
  40         sfa-config --python $sfa_local_config_xml > $sfa_local_config
  41         rm $sfa_local_config_xml
  42     fi
  43     if [ -n "$force" -o $sfa_local_config -nt $sfa_whole_config ] ; then
  44         sfa-config --python $sfa_default_config $sfa_local_config > $sfa_whole_config
  45     fi
  46     if [ -n "$force" -o $sfa_whole_config -nt /etc/sfa/sfa_config.sh ] ; then
  47         sfa-config --shell $sfa_default_config $sfa_local_config > /etc/sfa/sfa_config.sh
  48     fi
  49
  50     # reload the shell version
  51     source $sfa_local_config_sh
  52
  53 }
  54
  55 function postgresql_setting() {
  56     param="$1"; shift
  57     value="$1"; shift
  58
  59     # is that setting already present in file ?
  60     if grep --extended-regexp -q "#?${param} *=.*" $postgresql_conf; then
  61         sed --regexp-extended --in-place \
  62             --expression="s|#?${param} = .*|${param} = ${value}|" \
  63             $postgresql_conf
  64     else
  65         echo "${param} = ${value}" >> $postgresql_conf
  66     fi
  67 }
  68
  69 function start-db () {
  70
  71     # source shell config if present
  72     # but it might not be present the very first time
  73     [ ! -f $sfa_local_config_sh ] && reload
  74
  75     source $sfa_local_config_sh
  76
  77     # Export so that we do not have to specify -p to psql invocations
  78     export PGPORT=$SFA_DB_PORT
  79
  80     # only if enabled
  81     [ "$SFA_DB_ENABLED" == 1 -o "$SFA_DB_ENABLED" == True ] || return
  82
  83     postgresql_setting port "'$SFA_DB_PORT'"
  84     mkdir -p $PGLOG
  85     chown postgres:postgres $PGLOG
  86     postgresql_setting log_directory "'$PGLOG'"
  87
  88     ######## /var/lib/pgsql/data
  89     # Fix ownership (rpm installation may have changed it)
  90     chown -R -H postgres:postgres $(dirname $PGDATA)
  91
  92     # PostgreSQL must be started at least once to bootstrap
  93     # /var/lib/pgsql/data
  94     if [ ! -f $postgresql_conf ] ; then
  95         /usr/bin/postgresql-setup --initdb --unit postgresql
  96     fi
  97
  98     ######## /var/lib/pgsql/data/postgresql.conf
  99     registry_ip=""
 100     foo=$(python -c "import socket; print socket.gethostbyname('$SFA_REGISTRY_HOST')") && registry_ip="$foo"
 101     db_ip=""
 102     foo=$(python -c "import socket; print socket.gethostbyname('$SFA_DB_HOST')") && db_ip="$foo"
 103     # Enable DB server. drop Postgresql<=7.x
 104     # PostgreSQL >=8.0 defines listen_addresses
 105     # listen on a specific IP + localhost, more robust when run within a vserver
 106     sed -i -e '/^listen_addresses/d' $postgresql_conf
 107     case "$db_ip" in
 108         ""|127.0.0.1|localhost*)
 109             postgresql_setting listen_addresses "'localhost'" ;;
 110         *)
 111             postgresql_setting listen_addresses "'${db_ip},localhost'" ;;
 112     esac
 113     postgresql_setting timezone "'UTC'"
 114     postgresql_setting log_timezone "'UTC'"
 115
 116     ######## /var/lib/pgsql/data/pg_hba.conf
 117     # remove/recreate passwordless localhost entry
 118     sed -i -e "/^local/d" $pg_hba_conf
 119     echo "local all all trust" >> $pg_hba_conf
 120
 121     # Disable access to our DB from all hosts
 122     sed -i -e "/^host ${SFA_DB_NAME}/d" $pg_hba_conf
 123     # grant access
 124     {
 125         echo "host $SFA_DB_NAME $SFA_DB_USER 127.0.0.1/32 password"
 126         [ -n "$registry_ip" ] && echo "host $SFA_DB_NAME $SFA_DB_USER ${registry_ip}/32 password"
 127     } >> $pg_hba_conf
 128
 129     # Fix ownership (sed -i changes it)
 130     chown postgres:postgres $postgresql_conf $pg_hba_conf
 131
 132     ######## compute a password if needed
 133     if [ -z "$SFA_DB_PASSWORD" ] ; then
 134         SFA_DB_PASSWORD=$(uuidgen)
 135         sfa-config --category=sfa_db --variable=password --value="$SFA_DB_PASSWORD" --save=$sfa_local_config $sfa_local_config >& /dev/null
 136         reload
 137     fi
 138
 139     # tell postgresql that settings have changed
 140     # note that changes to listen_addresses do require a full restart
 141     # but it's too intrusive to do this each time, as it would in turn
 142     # require a restart of the plc service
 143     su - postgres bash -c "pg_ctl reload"
 144
 145     ######## make sure we have the user and db created
 146     # user
 147     if ! psql -U $SFA_DB_USER -c "" template1 >/dev/null 2>&1 ; then
 148         psql -U postgres -c "CREATE USER $SFA_DB_USER PASSWORD '$SFA_DB_PASSWORD'" template1 >& /dev/null
 149     else
 150         psql -U postgres -c "ALTER USER $SFA_DB_USER WITH PASSWORD '$SFA_DB_PASSWORD'" template1 >& /dev/null
 151     fi
 152
 153     # db
 154     if ! psql -U $SFA_DB_USER -c "" $SFA_DB_NAME >/dev/null 2>&1 ; then
 155         createdb -U postgres --template=template0 --encoding=UNICODE --owner=$SFA_DB_USER $SFA_DB_NAME
 156     fi
 157
 158     # create schema; sfaadmin.py is safer than just sfaadmin
 159     sfaadmin.py reg sync_db
 160
 161 }
 162
 163 usage="$0 start-db|reload
 164   start-db: configure postgresql database and restart postgresql
 165   reload: recompute miscell configuration files after changes are made in master config
 166 "
 167
 168 func="$1"; shift
 169
 170 case "$func" in
 171     start-db|reload) $func;;
 172     *) echo "$usage"; exit 1;;
 173 esac