paul@634 | 1 | #!/bin/sh |
paul@634 | 2 | |
paul@1108 | 3 | DIRNAME=`dirname "$0"` |
paul@1096 | 4 | CONFIG="$DIRNAME/config.sh" |
paul@891 | 5 | |
paul@1096 | 6 | if [ -e "$CONFIG" ]; then |
paul@1089 | 7 | . "$CONFIG" |
paul@891 | 8 | else |
paul@1089 | 9 | CONFIG=/etc/imip-agent/config.sh |
paul@1089 | 10 | . "$CONFIG" |
paul@891 | 11 | fi |
paul@891 | 12 | |
paul@1096 | 13 | SCHEMA="$DIRNAME/../conf/postgresql/schema.sql" |
paul@1096 | 14 | |
paul@1096 | 15 | if [ ! -e "$SCHEMA" ]; then |
paul@1096 | 16 | SCHEMA=/etc/imip-agent/postgresql/schema.sql |
paul@891 | 17 | fi |
paul@891 | 18 | |
paul@1108 | 19 | PROGNAME=`basename "$0"` |
paul@731 | 20 | |
paul@731 | 21 | if [ "$1" = "--help" ]; then |
paul@731 | 22 | cat 1>&2 <<EOF |
paul@1096 | 23 | Usage: $PROGNAME |
paul@1096 | 24 | |
paul@1096 | 25 | Initialise stored and published data directories at... |
paul@731 | 26 | |
paul@1096 | 27 | * $INSTALL_DIR |
paul@1096 | 28 | * $WEB_INSTALL_DIR |
paul@891 | 29 | |
paul@1096 | 30 | ...respectively. |
paul@1096 | 31 | |
paul@1096 | 32 | Set permissions to the user and group respectively given as $IMIP_AGENT_USER |
paul@891 | 33 | and $IMIP_AGENT_GROUP. |
paul@731 | 34 | |
paul@1096 | 35 | Within the stored data directory, the following directories will be created |
paul@1096 | 36 | (with STORE_TYPE currently set as "$STORE_TYPE"): |
paul@731 | 37 | |
paul@1039 | 38 | * $INSTALL_DIR/preferences |
paul@1096 | 39 | EOF |
paul@731 | 40 | |
paul@1096 | 41 | if [ "$STORE_TYPE" = "file" ]; then |
paul@1096 | 42 | cat 1>&2 <<EOF |
paul@1096 | 43 | * $INSTALL_DIR/journal (if STORE_TYPE is "file") |
paul@1096 | 44 | * $INSTALL_DIR/store (if STORE_TYPE is "file") |
paul@1096 | 45 | EOF |
paul@1096 | 46 | fi |
paul@1096 | 47 | |
paul@1096 | 48 | cat 1>&2 <<EOF |
paul@1096 | 49 | |
paul@1096 | 50 | Within the published data directory the following directory will be created: |
paul@731 | 51 | |
paul@891 | 52 | * $WEB_INSTALL_DIR/static |
paul@731 | 53 | EOF |
paul@1089 | 54 | |
paul@1096 | 55 | if [ "$STORE_TYPE" = "postgresql" ]; then |
paul@1096 | 56 | cat 1>&2 <<EOF |
paul@1096 | 57 | |
paul@1096 | 58 | With STORE_TYPE set as "database", a database schema will be initialised for the |
paul@1096 | 59 | following database: |
paul@1096 | 60 | |
paul@1096 | 61 | * $POSTGRESQL_DB |
paul@1096 | 62 | EOF |
paul@1096 | 63 | fi |
paul@1096 | 64 | |
paul@1096 | 65 | cat 1>&2 <<EOF |
paul@1096 | 66 | |
paul@1096 | 67 | See $CONFIG for the settings used as described above. |
paul@731 | 68 | EOF |
paul@731 | 69 | exit 1 |
paul@731 | 70 | fi |
paul@731 | 71 | |
paul@1135 | 72 | # Test for a privileged user. |
paul@1135 | 73 | |
paul@1135 | 74 | if [ `whoami` != 'root' ]; then |
paul@1135 | 75 | cat 1>&2 <<EOF |
paul@1135 | 76 | You will need to become a privileged user using su or sudo to run this program |
paul@1135 | 77 | because it changes file ownership and may also switch users to run database |
paul@1135 | 78 | administration commands. |
paul@1135 | 79 | EOF |
paul@1135 | 80 | exit 1 |
paul@1135 | 81 | fi |
paul@1135 | 82 | |
paul@1135 | 83 | # Create necessary directories regardless of store type. |
paul@1135 | 84 | |
paul@1135 | 85 | echo "Creating preferences and static Web directories..." 1>&2 |
paul@1135 | 86 | |
paul@1089 | 87 | for DIR in "$INSTALL_DIR"/preferences "$WEB_INSTALL_DIR"/static ; do |
paul@634 | 88 | mkdir -p "$DIR" |
paul@1096 | 89 | chown "$IMIP_AGENT_USER" "$DIR" |
paul@1096 | 90 | chgrp "$IMIP_AGENT_GROUP" "$DIR" |
paul@634 | 91 | chmod g+ws "$DIR" |
paul@634 | 92 | done |
paul@1089 | 93 | |
paul@1096 | 94 | # Initialise a file store. |
paul@1096 | 95 | |
paul@1089 | 96 | if [ "$STORE_TYPE" = "file" ]; then |
paul@1135 | 97 | |
paul@1135 | 98 | echo "Creating store and journal directories..." 1>&2 |
paul@1135 | 99 | |
paul@1089 | 100 | for DIR in "$INSTALL_DIR"/store "$INSTALL_DIR"/journal ; do |
paul@1089 | 101 | mkdir -p "$DIR" |
paul@1096 | 102 | chown "$IMIP_AGENT_USER" "$DIR" |
paul@1096 | 103 | chgrp "$IMIP_AGENT_GROUP" "$DIR" |
paul@1089 | 104 | chmod g+ws "$DIR" |
paul@1089 | 105 | done |
paul@1096 | 106 | |
paul@1096 | 107 | # Initialise a PostgreSQL store. |
paul@1096 | 108 | |
paul@1096 | 109 | elif [ "$STORE_TYPE" = "postgresql" ]; then |
paul@1096 | 110 | |
paul@1096 | 111 | # Check for the database. |
paul@1096 | 112 | |
paul@1135 | 113 | echo "Checking for the database ${POSTGRESQL_DB}..." 1>&2 |
paul@1135 | 114 | |
paul@1101 | 115 | if $AS_POSTGRES psql -tA -c 'select datname from pg_database' postgres | grep -q ^"$POSTGRESQL_DB"$ ; then |
paul@1096 | 116 | cat 1>&2 <<EOF |
paul@1096 | 117 | Database $POSTGRESQL_DB already exists. |
paul@1096 | 118 | EOF |
paul@1096 | 119 | exit 1 |
paul@1096 | 120 | fi |
paul@1096 | 121 | |
paul@1096 | 122 | # Attempt to create the database. |
paul@1096 | 123 | |
paul@1135 | 124 | echo "Creating database ${POSTGRESQL_DB}..." 1>&2 |
paul@1135 | 125 | |
paul@1101 | 126 | if ! $AS_POSTGRES createdb "$POSTGRESQL_DB" ; then |
paul@1096 | 127 | cat 1>&2 <<EOF |
paul@1096 | 128 | Could not create database $POSTGRESQL_DB using createdb. |
paul@1096 | 129 | EOF |
paul@1096 | 130 | exit 1 |
paul@1096 | 131 | fi |
paul@1096 | 132 | |
paul@1096 | 133 | # Attempt to initialise the schema. |
paul@1096 | 134 | |
paul@1135 | 135 | echo "Initialising the schema for database ${POSTGRESQL_DB}..." 1>&2 |
paul@1135 | 136 | |
paul@1135 | 137 | if ! $AS_POSTGRES psql -q -f "$SCHEMA" "$POSTGRESQL_DB" ; then |
paul@1096 | 138 | cat 1>&2 <<EOF |
paul@1096 | 139 | Could not initialise schema in database $POSTGRESQL_DB using psql. |
paul@1096 | 140 | EOF |
paul@1096 | 141 | exit 1 |
paul@1096 | 142 | fi |
paul@1096 | 143 | |
paul@1096 | 144 | # For each user needing to connect, attempt to create a role and grant it |
paul@1096 | 145 | # privileges on the tables. |
paul@1096 | 146 | |
paul@1096 | 147 | for USER in $POSTGRESQL_USERS ; do |
paul@1135 | 148 | |
paul@1135 | 149 | echo "Creating a database user for ${USER}..." 1>&2 |
paul@1135 | 150 | |
paul@1101 | 151 | if ! $AS_POSTGRES createuser -D -R -S "$USER" ; then |
paul@1096 | 152 | cat 1>&2 <<EOF |
paul@1096 | 153 | Could not create database user $USER using createuser. |
paul@1096 | 154 | EOF |
paul@1096 | 155 | fi |
paul@1135 | 156 | |
paul@1135 | 157 | echo "Granting privileges to database user for ${USER}..." 1>&2 |
paul@1135 | 158 | |
paul@1101 | 159 | if ! $AS_POSTGRES psql -Atc '\dt' "$POSTGRESQL_DB" \ |
paul@1096 | 160 | | cut -d '|' -f 2 \ |
paul@1135 | 161 | | xargs -I{} $AS_POSTGRES psql -q -c "grant all privileges on table {} to \"$USER\"" "$POSTGRESQL_DB" ; then |
paul@1096 | 162 | |
paul@1096 | 163 | cat 1>&2 <<EOF |
paul@1096 | 164 | Could not grant permissions for schema in database $POSTGRESQL_DB to $USER |
paul@1096 | 165 | using psql. |
paul@1096 | 166 | EOF |
paul@1096 | 167 | fi |
paul@1096 | 168 | done |
paul@1089 | 169 | fi |