163 lines
4.3 KiB
Bash
163 lines
4.3 KiB
Bash
#!/usr/bin/env sh
|
|
LOG_DIR=/rpxy/log
|
|
LOG_FILE=${LOG_DIR}/rpxy.log
|
|
LOG_SIZE=10M
|
|
LOG_NUM=10
|
|
|
|
LOGGING=${LOG_TO_FILE:-false}
|
|
USER=${HOST_USER:-rpxy}
|
|
USER_ID=${HOST_UID:-900}
|
|
GROUP_ID=${HOST_GID:-900}
|
|
|
|
CONFIG_FILE=/etc/rpxy.toml
|
|
CONFIG_DIR=/rpxy/config
|
|
CONFIG_FILE_IN_DIR=${CONFIG_FILENAME:-rpxy.toml}
|
|
|
|
#######################################
|
|
# Setup logrotate
|
|
function setup_logrotate () {
|
|
if [ $LOGROTATE_NUM ]; then
|
|
LOG_NUM=${LOGROTATE_NUM}
|
|
fi
|
|
if [ $LOGROTATE_SIZE ]; then
|
|
LOG_SIZE=${LOGROTATE_SIZE}
|
|
fi
|
|
|
|
cat > /etc/logrotate.conf << EOF
|
|
# see "man logrotate" for details
|
|
# rotate log files weekly
|
|
weekly
|
|
# use the adm group by default, since this is the owning group
|
|
# of /var/log/syslog.
|
|
# su root adm
|
|
# keep 4 weeks worth of backlogs
|
|
rotate 4
|
|
# create new (empty) log files after rotating old ones
|
|
create
|
|
# use date as a suffix of the rotated file
|
|
#dateext
|
|
# uncomment this if you want your log files compressed
|
|
#compress
|
|
# packages drop log rotation information into this directory
|
|
include /etc/logrotate.d
|
|
# system-specific logs may be also be configured here.
|
|
EOF
|
|
|
|
cat > /etc/logrotate.d/rpxy.conf << EOF
|
|
${LOG_FILE} {
|
|
dateext
|
|
daily
|
|
missingok
|
|
rotate ${LOG_NUM}
|
|
notifempty
|
|
compress
|
|
delaycompress
|
|
dateformat -%Y-%m-%d-%s
|
|
size ${LOG_SIZE}
|
|
copytruncate
|
|
su ${USER} ${USER}
|
|
}
|
|
EOF
|
|
}
|
|
|
|
#######################################
|
|
function setup_ubuntu () {
|
|
id ${USER} > /dev/null
|
|
# Check the existence of the user, if not exist, create it.
|
|
if [ $? -eq 1 ]; then
|
|
echo "rpxy: Create user ${USER} with ${USER_ID}:${GROUP_ID}"
|
|
groupadd -g ${GROUP_ID} ${USER}
|
|
useradd -u ${USER_ID} -g ${GROUP_ID} ${USER}
|
|
fi
|
|
|
|
# for crontab when logging
|
|
if "${LOGGING}"; then
|
|
# Set up logrotate
|
|
setup_logrotate
|
|
|
|
# Setup cron
|
|
mkdir -p /etc/cron.15min/
|
|
cp -p /etc/cron.daily/logrotate /etc/cron.15min/
|
|
echo "*/15 * * * * root cd / && run-parts --report /etc/cron.15min" >> /etc/crontab
|
|
# cp -p /etc/cron.daily/logrotate /etc/cron.hourly/
|
|
service cron start
|
|
fi
|
|
}
|
|
|
|
#######################################
|
|
function setup_alpine () {
|
|
id ${USER} > /dev/null
|
|
# Check the existence of the user, if not exist, create it.
|
|
if [ $? -eq 1 ]; then
|
|
echo "rpxy: Create user ${USER} with ${USER_ID}:${GROUP_ID}"
|
|
addgroup -g ${GROUP_ID} ${USER}
|
|
adduser -H -D -u ${USER_ID} -G ${USER} ${USER}
|
|
fi
|
|
|
|
# for crontab when logging
|
|
if "${LOGGING}"; then
|
|
# Set up logrotate
|
|
setup_logrotate
|
|
|
|
# Setup cron
|
|
cp -f /etc/periodic/daily/logrotate /etc/periodic/15min
|
|
crond -b -l 8
|
|
fi
|
|
}
|
|
|
|
#######################################
|
|
|
|
if [ $(whoami) != "root" -o $(id -u) -ne 0 -a $(id -g) -ne 0 ]; then
|
|
echo "Do not execute 'docker run' or 'docker-compose up' with a specific user through '-u'."
|
|
echo "If you want to run 'rpxy' with a specific user, use HOST_USER, HOST_UID and HOST_GID environment variables."
|
|
exit 1
|
|
fi
|
|
|
|
# Check gosu or su-exec, determine linux distribution, and set up user
|
|
if [ $(command -v gosu) ]; then
|
|
# Ubuntu Linux
|
|
alias gosu='gosu'
|
|
setup_ubuntu
|
|
LINUX="Ubuntu"
|
|
elif [ $(command -v su-exec) ]; then
|
|
# Alpine Linux
|
|
alias gosu='su-exec'
|
|
setup_alpine
|
|
LINUX="Alpine"
|
|
else
|
|
echo "Unknown distribution!"
|
|
exit 1
|
|
fi
|
|
|
|
# Check the given user and its uid:gid
|
|
if [ $(id -u ${USER}) -ne ${USER_ID} -a $(id -g ${USER}) -ne ${GROUP_ID} ]; then
|
|
echo "${USER} exists or was previously created. However, its uid and gid are inconsistent. Please recreate your container."
|
|
exit 1
|
|
fi
|
|
|
|
# Change permission according to the given user
|
|
# except for the config dir that possibly get mounted with read-only
|
|
find /rpxy -path ${CONFIG_DIR} -prune -o -exec chown ${USER_ID}:${USER_ID} {} +
|
|
|
|
# Check the config file existence
|
|
if [[ ! -f ${CONFIG_FILE} ]]; then
|
|
if [[ ! -f ${CONFIG_DIR}/${CONFIG_FILE_IN_DIR} ]]; then
|
|
echo "No config file is given. Mount a config dir or file."
|
|
exit 1
|
|
fi
|
|
echo "rpxy: config file: ${CONFIG_DIR}/${CONFIG_FILE_IN_DIR}"
|
|
ln -s ${CONFIG_DIR}/${CONFIG_FILE_IN_DIR} ${CONFIG_FILE}
|
|
else
|
|
echo "rpxy: config file: ${CONFIG_FILE}"
|
|
fi
|
|
|
|
# Run rpxy
|
|
cd /rpxy
|
|
echo "rpxy: Start with user: ${USER} (${USER_ID}:${GROUP_ID})"
|
|
if "${LOGGING}"; then
|
|
echo "rpxy: Start with writing log file"
|
|
gosu ${USER} sh -c "/rpxy/run.sh 2>&1 | tee ${LOG_FILE}"
|
|
else
|
|
echo "rpxy: Start without writing log file"
|
|
gosu ${USER} sh -c "/rpxy/run.sh 2>&1"
|
|
fi
|