forked from Kloadut/dokku-pg-plugin
-
Notifications
You must be signed in to change notification settings - Fork 0
/
commands
executable file
·232 lines (206 loc) · 6.9 KB
/
commands
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
#!/bin/bash
set -e;
# Check if name is specified
if [[ $1 == postgresql:* ]]; then
if [ -z $2 ] && [ $1 != postgresql:list ]; then
echo "You must specify an app name"
exit 1
else
APP="$2"
# Check if app exists with the same name
if [[ -d "$DOKKU_ROOT/$APP" ]]; then
APP_EXISTS=true
else
APP_EXISTS=false
fi
fi
fi
# Create .postgresql diretory if not exists
if [[ ! -d $DOKKU_ROOT/.postgresql ]]; then
mkdir -p $DOKKU_ROOT/.postgresql
chown -R dokku: $DOKKU_ROOT/.postgresql
fi
function check_postgresql_container() {
if [[ ! -f "$DOKKU_ROOT/.postgresql/pwd_$APP" ]]; then
echo "Unknown (or too old) PostgreSQL container"
exit 1
fi
}
function check_postgresql_tool() {
local tool_name="$1"
if [[ $(whereis $tool_name | awk '{ print NF }') -eq 1 ]]; then
echo "'${tool_name}' not found. Is the package 'postgresql-client' installed?" 1>&2
exit 1
fi
}
function get_postgresql_id() {
ID=$(docker ps | grep postgresql/"$APP":latest | awk {'print $1'})
echo $ID
}
function get_postgresql_port() {
ID=$(get_postgresql_id)
PORT=$(docker port "$ID" 5432 | sed 's/0.0.0.0://')
echo $PORT
}
function connect_to_db() {
export PGPASSWORD=$(cat "$DOKKU_ROOT/.postgresql/pwd_$APP")
PORT=$(get_postgresql_port)
psql -h 172.17.42.1 -p $PORT -U root db
}
case "$1" in
postgresql:console)
check_postgresql_container
check_postgresql_tool psql
connect_to_db
;;
postgresql:create)
DB_IMAGE=postgresql/$APP
# Check if DB container is installed
IMAGE=$(docker images | grep "kloadut/postgresql " | awk '{print $3}')
if [[ -z $IMAGE ]]; then
echo "PostgreSQL image not found... Did you run 'dokku plugins-install' ?"
exit 1
fi
# Check if an existing DB volume exists
if [[ -f "$DOKKU_ROOT/.postgresql/volume_$APP" ]]; then
VOLUME="`cat $DOKKU_ROOT/.postgresql/volume_$APP`:/opt/postgresql"
echo
echo "-----> Reusing postgresql/$APP database"
else
VOLUME="/opt/postgresql"
# Generate a random password for DB user
DB_PASSWORD=$(< /dev/urandom tr -dc A-Za-z0-9 | head -c 16)
echo $DB_PASSWORD > "$DOKKU_ROOT/.postgresql/pwd_$APP"
chown dokku: "$DOKKU_ROOT/.postgresql/pwd_$APP"
chmod 700 "$DOKKU_ROOT/.postgresql/pwd_$APP"
fi
# Fork DB image
ID=$(docker run -d kloadut/postgresql echo '')
docker wait $ID > /dev/null
IMAGE=$(docker commit $ID)
docker tag $IMAGE $DB_IMAGE
# Launch container
ID=$(docker run -v $VOLUME -p 5432 -d $DB_IMAGE /usr/bin/start_pgsql.sh $DB_PASSWORD)
sleep 4
# Rename persistent volume
if [[ ! -f "$DOKKU_ROOT/.postgresql/volume_$APP" ]]; then
VOLUME_PATH=$(docker inspect $ID | grep /var/lib/docker/vfs/dir/ | awk '{print $2}' | sed -e's/"//g')
if [[ -z $VOLUME_PATH ]]; then
echo "Your docker version is too old, please update it"
exit 1
fi
echo $VOLUME_PATH > "$DOKKU_ROOT/.postgresql/volume_$APP"
fi
# Link to a potential existing app
dokku postgresql:link $APP $APP
echo
echo "-----> PostgreSQL container created: $DB_IMAGE"
sleep 1
dokku postgresql:info $APP
;;
postgresql:delete)
DB_IMAGE=postgresql/$APP
ID=$(get_postgresql_id)
# Stop and remove the container
if [[ ! -z $ID ]]; then
docker kill $ID > /dev/null
sleep 1
docker rm -v $ID > /dev/null
sleep 1
fi
# Remove image
IMAGE=$(docker images | grep "$DB_IMAGE " | awk '{print $1}')
if [[ ! -z $IMAGE ]]; then
docker rmi $IMAGE > /dev/null
fi
# Remove container root password
if [[ -f "$DOKKU_ROOT/.postgresql/pwd_$APP" ]]; then
rm -f "$DOKKU_ROOT/.postgresql/pwd_$APP"
fi
# Remove persistent volume
if [[ -f "$DOKKU_ROOT/.postgresql/volume_$APP" ]]; then
rm -f "$DOKKU_ROOT/.postgresql/volume_$APP"
fi
echo
echo "-----> PostgreSQL container deleted: $DB_IMAGE"
;;
postgresql:dump)
check_postgresql_container
check_postgresql_tool pg_dump
export PGPASSWORD=$(cat "$DOKKU_ROOT/.postgresql/pwd_$APP")
PORT=$(get_postgresql_port)
pg_dump -h 172.17.42.1 -p $PORT -U root -c -O db
# echo to stderr, as stdout will probably be redirected to a file
echo 1>&2
echo "-----> $APP database dumped" 1>&2
;;
postgresql:info)
check_postgresql_container
DB_PASSWORD=$(cat "$DOKKU_ROOT/.postgresql/pwd_$APP")
PORT=$(get_postgresql_port)
echo
echo " Host: 172.17.42.1"
echo " Port: $PORT"
echo " User: 'root'"
echo " Password: '$DB_PASSWORD'"
echo " Database: 'db'"
echo
echo " Url: 'postgres://root:[email protected]:$PORT/db'"
echo
;;
postgresql:link)
if $APP_EXISTS; then
# Check argument
if [[ -z $3 ]]; then
echo "You must specify a database name"
exit 1
fi
DB_IMAGE="postgresql/$3"
if [[ ! -f "$DOKKU_ROOT/.postgresql/pwd_$3" ]]; then
echo "Database is not initialized correctly. Did you run 'dokku plugins-install'?"
exit 1
fi
DB_PASSWORD=$(cat "$DOKKU_ROOT/.postgresql/pwd_$3")
PORT=$(get_postgresql_port)
# Link database using dokku command
dokku config:set $APP "DATABASE_URL=postgres://root:[email protected]:$PORT/db"
echo
echo "-----> $APP linked to $DB_IMAGE database"
fi
;;
postgresql:logs)
ID=$(get_postgresql_id)
docker logs $ID | tail -n 100
;;
postgresql:list)
CONTAINERS=$(ls $DOKKU_ROOT/.postgresql/volume* 2> /dev/null | sed -e 's/_/ /' | awk '{print $2}')
if [[ -z $CONTAINERS ]]; then
echo "There are no PostgreSQL containers created."
else
echo "PostgreSQL containers:"
for CONTAINER in $CONTAINERS; do
echo " - $CONTAINER"
done
fi
;;
postgresql:restore)
check_postgresql_container
check_postgresql_tool psql
connect_to_db
echo
echo "-----> $APP database restored"
;;
help)
cat && cat<<EOF
postgresql:console <db> Open a PostgreSQL console
postgresql:create <db> Create a PostgreSQL container
postgresql:delete <db> Delete specified PostgreSQL container
postgresql:dump <db> > dump_file.sql Dump database data
postgresql:info <db> Display database informations
postgresql:link <app> <db> Link an app to a PostgreSQL database
postgresql:list Display list of PostgreSQL containers
postgresql:logs <db> Display last logs from PostgreSQL container
postgresql:restore <db> < dump_file.sql Restore database data from a previous dump
EOF
;;
esac