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     sed --regexp-extended --in-place \
  60       --expression="s|#?${param} = .*|${param} = ${value}|" \
  61       $postgresql_conf
  62 }
  63
  64 function start-db () {
  65
  66     # source shell config if present
  67     # but it might not be present the very first time
  68     [ ! -f $sfa_local_config_sh ] && reload
  69
  70     source $sfa_local_config_sh
  71
  72     # Export so that we do not have to specify -p to psql invocations
  73     export PGPORT=$SFA_DB_PORT
  74
  75     # only if enabled
  76     [ "$SFA_DB_ENABLED" == 1 -o "$SFA_DB_ENABLED" == True ] || return
  77
  78     postgresql_setting port "'$SFA_DB_PORT'"
  79     mkdir -p $PGLOG
  80     chown postgres:postgres $PGLOG
  81     postgresql_setting log_directory "'$PGLOG'"
  82
  83     ######## /var/lib/pgsql/data
  84     # Fix ownership (rpm installation may have changed it)
  85     chown -R -H postgres:postgres $(dirname $PGDATA)
  86
  87     # PostgreSQL must be started at least once to bootstrap
  88     # /var/lib/pgsql/data
  89     if [ ! -f $postgresql_conf ] ; then
  90         /usr/bin/postgresql-setup --initdb --unit postgresql
  91     fi
  92
  93     ######## /var/lib/pgsql/data/postgresql.conf
  94     registry_ip=""
  95     foo=$(python -c "import socket; print socket.gethostbyname('$SFA_REGISTRY_HOST')") && registry_ip="$foo"
  96     # Enable DB server. drop Postgresql<=7.x
  97     # PostgreSQL >=8.0 defines listen_addresses
  98     # listen on a specific IP + localhost, more robust when run within a vserver
  99     sed -i -e '/^listen_addresses/d' $postgresql_conf
 100     if [ -z "$registry_ip" ] ; then
 101         postgresql_setting listen_addresses "'localhost'"
 102     else
 103         postgresql_setting listen_addresses "'${registry_ip},localhost'"
 104     fi
 105     postgresql_setting timezone "'UTC'"
 106     postgresql_setting log_timezone "'UTC'"
 107
 108     ######## /var/lib/pgsql/data/pg_hba.conf
 109     # remove/recreate passwordless localhost entry
 110     sed -i -e "/^local/d" $pg_hba_conf
 111     echo "local all all trust" >> $pg_hba_conf
 112
 113     # Disable access to our DB from all hosts
 114     sed -i -e "/^host ${SFA_DB_NAME}/d" $pg_hba_conf
 115     # grant access
 116     {
 117         echo "host $SFA_DB_NAME $SFA_DB_USER 127.0.0.1/32 password"
 118         [ -n "$registry_ip" ] && echo "host $SFA_DB_NAME $SFA_DB_USER ${registry_ip}/32 password"
 119     } >> $pg_hba_conf
 120
 121     # Fix ownership (sed -i changes it)
 122     chown postgres:postgres $postgresql_conf $pg_hba_conf
 123
 124     ######## compute a password if needed
 125     if [ -z "$SFA_DB_PASSWORD" ] ; then
 126         SFA_DB_PASSWORD=$(uuidgen)
 127         sfa-config --category=sfa_db --variable=password --value="$SFA_DB_PASSWORD" --save=$sfa_local_config $sfa_local_config >& /dev/null
 128         reload
 129     fi
 130
 131     # tell postgresql that settings have changed
 132     su - postgres bash -c "pg_ctl reload"
 133
 134     ######## make sure we have the user and db created
 135     # user
 136     if ! psql -U $SFA_DB_USER -c "" template1 >/dev/null 2>&1 ; then
 137         psql -U postgres -c "CREATE USER $SFA_DB_USER PASSWORD '$SFA_DB_PASSWORD'" template1 >& /dev/null
 138     else
 139         psql -U postgres -c "ALTER USER $SFA_DB_USER WITH PASSWORD '$SFA_DB_PASSWORD'" template1 >& /dev/null
 140     fi
 141
 142     # db
 143     if ! psql -U $SFA_DB_USER -c "" $SFA_DB_NAME >/dev/null 2>&1 ; then
 144         createdb -U postgres --template=template0 --encoding=UNICODE --owner=$SFA_DB_USER $SFA_DB_NAME
 145     fi
 146
 147     # create schema; sfaadmin.py is safer than just sfaadmin
 148     sfaadmin.py reg sync_db
 149
 150 }
 151
 152 usage="$0 start-db|reload
 153   start-db: configure postgresql database and restart postgresql
 154   reload: recompute miscell configuration files after changes are made in master config
 155 "
 156
 157 func="$1"; shift
 158
 159 case "$func" in
 160     start-db|reload) $func;;
 161     *) echo "$usage"; exit 1;;
 162 esac