Improve shell scripts (#705)

* Improve init-db.sh

Allow custom database parameters.
Abstract common functionality.

* Improve install.sh

Abstract common functionality.
Ask to automagically reload the project when source files are changed.
This commit is contained in:
Richie Zhang 2020-05-11 10:16:08 -07:00 committed by GitHub
parent fd1ceb0ea9
commit 1eea23bcfe
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 139 additions and 47 deletions

57
install.sh vendored
View file

@ -1,4 +1,4 @@
#!/bin/bash #!/bin/sh
set -e set -e
# Set the database variable to the default first. # Set the database variable to the default first.
@ -10,25 +10,55 @@ export LEMMY_DATABASE_URL=postgres://lemmy:password@localhost:5432/lemmy
export JWT_SECRET=changeme export JWT_SECRET=changeme
export HOSTNAME=rrr export HOSTNAME=rrr
# Optionally initialize the database yes_no_prompt_invalid() {
init_db_valid=0 echo "Invalid input. Please enter either \"y\" or \"n\"." 1>&2
init_db_final=0 }
while [ "$init_db_valid" == 0 ]
do ask_to_init_db() {
init_db_valid=0
init_db_final=0
while [ "$init_db_valid" == 0 ]
do
read -p "Initialize database (y/n)? " init_db read -p "Initialize database (y/n)? " init_db
case "$init_db" in case "$init_db" in
[yY]* ) init_db_valid=1; init_db_final=1;; [yY]* ) init_db_valid=1; init_db_final=1;;
[nN]* ) init_db_valid=1; init_db_final=0;; [nN]* ) init_db_valid=1; init_db_final=0;;
* ) echo "Invalid input. Please enter either \"y\" or \"n\"." 1>&2;; * ) yes_no_prompt_invalid;;
esac esac
echo echo
done done
if [ "$init_db_final" = 1 ] if [ "$init_db_final" = 1 ]
then then
source ./server/db-init.sh source ./server/db-init.sh
read -n 1 -s -r -p "Press ANY KEY to continue execution of this script, press CTRL+C to quit..." read -n 1 -s -r -p "Press ANY KEY to continue execution of this script, press CTRL+C to quit..."
echo echo
fi fi
}
ask_to_auto_reload() {
auto_reload_valid=0
auto_reload_final=0
while [ "$auto_reload_valid" == 0 ]
do
echo "Automagically reload the project when source files are changed?"
echo "ONLY ENABLE THIS FOR DEVELOPMENT!"
read -p "(y/n) " auto_reload
case "$auto_reload" in
[yY]* ) auto_reload_valid=1; auto_reload_final=1;;
[nN]* ) auto_reload_valid=1; auto_reload_final=0;;
* ) yes_no_prompt_invalid;;
esac
echo
done
if [ "$auto_reload_final" = 1 ]
then
cd ui && yarn start
cd server && cargo watch -x run
fi
}
# Optionally initialize the database
ask_to_init_db
# Build the web client # Build the web client
cd ui cd ui
@ -39,6 +69,5 @@ yarn build
cd ../server cd ../server
RUST_LOG=debug cargo run RUST_LOG=debug cargo run
# For live coding, where both the front and back end, automagically reload on any save, do: # For live coding, where both the front and back end, automagically reload on any save
# cd ui && yarn start ask_to_auto_reload
# cd server && cargo watch -x run

79
server/db-init.sh vendored
View file

@ -1,15 +1,73 @@
#!/bin/bash #!/bin/sh
# Default configurations
username=lemmy username=lemmy
dbname=lemmy dbname=lemmy
port=5432 port=5432
password="" yes_no_prompt_invalid() {
password_confirm="" echo "Invalid input. Please enter either \"y\" or \"n\"." 1>&2
password_valid=0 }
while [ "$password_valid" == 0 ] print_config() {
do echo " database name: $dbname"
echo " username: $username"
echo " port: $port"
}
ask_for_db_config() {
echo "The default database configuration is:"
print_config
echo
default_config_final=0
default_config_valid=0
while [ "$default_config_valid" == 0 ]
do
read -p "Use this configuration (y/n)? " default_config
case "$default_config" in
[yY]* ) default_config_valid=1; default_config_final=1;;
[nN]* ) default_config_valid=1; default_config_final=0;;
* ) yes_no_prompt_invalid;;
esac
echo
done
if [ "$default_config_final" == 0 ]
then
config_ok_final=0
while [ "$config_ok_final" == 0 ]
do
read -p "Database name: " dbname
read -p "Username: " username
read -p "Port: " port
#echo
#echo "The database configuration is:"
#print_config
#echo
config_ok_valid=0
while [ "$config_ok_valid" == 0 ]
do
read -p "Use this configuration (y/n)? " config_ok
case "$config_ok" in
[yY]* ) config_ok_valid=1; config_ok_final=1;;
[nN]* ) config_ok_valid=1; config_ok_final=0;;
* ) yes_no_prompt_invalid;;
esac
echo
done
done
fi
}
ask_for_password() {
password=""
password_confirm=""
password_valid=0
while [ "$password_valid" == 0 ]
do
read -p "Enter database password: " -s password read -p "Enter database password: " -s password
echo echo
@ -33,11 +91,16 @@ do
# Set the password_valid variable to break out of the loop # Set the password_valid variable to break out of the loop
password_valid=1 password_valid=1
done done
}
ask_for_db_config
ask_for_password
psql -c "CREATE USER $username WITH PASSWORD '$password' SUPERUSER;" -U postgres psql -c "CREATE USER $username WITH PASSWORD '$password' SUPERUSER;" -U postgres
psql -c "CREATE DATABASE $dbname WITH OWNER $username;" -U postgres psql -c "CREATE DATABASE $dbname WITH OWNER $username;" -U postgres
export LEMMY_DATABASE_URL=postgres://$username:$password@localhost:$port/$dbname export LEMMY_DATABASE_URL=postgres://$username:$password@localhost:$port/$dbname
echo $LEMMY_DATABASE_URL echo "The database URL is $LEMMY_DATABASE_URL"