155 lines
3.2 KiB
Bash
Executable File
155 lines
3.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
error() {
|
|
log "ERROR" "$1"
|
|
}
|
|
|
|
warn() {
|
|
log "WARNING" "$1"
|
|
}
|
|
|
|
info() {
|
|
log "info" "$1"
|
|
}
|
|
|
|
debug() {
|
|
if [ "${LOG_LEVEL}" = "debug" ]; then
|
|
log "debug" "$1"
|
|
fi
|
|
}
|
|
|
|
log() {
|
|
LEVEL=$1
|
|
MESSAGE=$2
|
|
printf "$(date '+%F %T') %s: %s\n" "${LEVEL}" "${MESSAGE}"
|
|
}
|
|
|
|
init_config_file() {
|
|
FILE=$1
|
|
FULLPATH=${LIBATION_CONFIG_DIR}/${FILE}
|
|
if [ -f ${FULLPATH} ]; then
|
|
info "loading ${FILE}"
|
|
cp ${FULLPATH} ${LIBATION_CONFIG_INTERNAL}/
|
|
return 0
|
|
else
|
|
warn "${FULLPATH} not found, using defaults"
|
|
echo "{}" > ${LIBATION_CONFIG_INTERNAL}/${FILE}
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
update_settings() {
|
|
FILE=$1
|
|
KEY=$2
|
|
VALUE=$3
|
|
info "setting ${KEY} to ${VALUE}"
|
|
echo $(jq --arg k "${KEY}" --arg v "${VALUE}" '.[$k] = $v' ${LIBATION_CONFIG_INTERNAL}/${FILE}) > ${LIBATION_CONFIG_INTERNAL}/${FILE}.tmp
|
|
mv ${LIBATION_CONFIG_INTERNAL}/${FILE}.tmp ${LIBATION_CONFIG_INTERNAL}/${FILE}
|
|
}
|
|
|
|
is_mounted() {
|
|
DIR=$1
|
|
if mount | grep -q "${DIR}";
|
|
then
|
|
return 0
|
|
else
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
create_db() {
|
|
DBFILE=$1
|
|
if [ -f "${DBFILE}" ]; then
|
|
warn "prexisting database found when creating"
|
|
return 0
|
|
else
|
|
warn "database not found, creating one at ${DBFILE}"
|
|
if ! touch "${DBFILE}"; then
|
|
error "unable to create database, check permissions on host"
|
|
exit 1
|
|
fi
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
run() {
|
|
info "scanning accounts"
|
|
/libation/LibationCli scan
|
|
info "liberating books"
|
|
/libation/LibationCli liberate
|
|
}
|
|
|
|
main() {
|
|
# TERM isn't set by default in docker images
|
|
if [[ -z ${TERM} || ${TERM} = "dumb" ]]; then
|
|
TERM=xterm-256color
|
|
fi
|
|
|
|
info ${TERM}
|
|
|
|
info "initializing libation"
|
|
init_config_file AccountsSettings.json
|
|
init_config_file Settings.json
|
|
|
|
info "loading settings"
|
|
update_settings Settings.json Books /data
|
|
update_settings Settings.json InProgress /tmp
|
|
|
|
info "loading database"
|
|
FILE=LibationContext.db
|
|
# If user provides a separate database mount, use that
|
|
if is_mounted ${LIBATION_DB_DIR};
|
|
then
|
|
debug "using database directory ${LIBATION_DB_DIR}"
|
|
if [ -f "${LIBATION_DB_DIR}/${FILE}" ]; then
|
|
info "database found in ${LIBATION_DB_DIR}"
|
|
else
|
|
create_db ${LIBATION_DB_DIR}/${FILE}
|
|
fi
|
|
ln -s /${LIBATION_DB_DIR}/${FILE} ${LIBATION_CONFIG_INTERNAL}/${FILE}
|
|
# Otherwise, use the config directory
|
|
else
|
|
debug "using config directory ${LIBATION_CONFIG_DIR}"
|
|
if [ -f "${LIBATION_CONFIG_DIR}/${FILE}" ]; then
|
|
info "database found in ${LIBATION_CONFIG_DIR}"
|
|
else
|
|
create_db ${LIBATION_CONFIG_DIR}/${FILE}
|
|
fi
|
|
ln -s /${LIBATION_CONFIG_DIR}/${FILE} ${LIBATION_CONFIG_INTERNAL}/${FILE}
|
|
fi
|
|
|
|
# Try to warn if books dir wasn't mounted in
|
|
if ! is_mounted ${LIBATION_BOOKS_DIR};
|
|
then
|
|
warn "${LIBATION_BOOKS_DIR} does not appear to be mounted, books will not be saved"
|
|
fi
|
|
|
|
# Let the user know what the run type will be
|
|
if [[ -z "${SLEEP_TIME}" ]]; then
|
|
SLEEP_TIME=-1
|
|
fi
|
|
|
|
if [ "${SLEEP_TIME}" = -1 ]; then
|
|
info "running once"
|
|
else
|
|
info "running every ${SLEEP_TIME}"
|
|
fi
|
|
|
|
# loop
|
|
while true
|
|
do
|
|
run
|
|
|
|
# Liberate only once if SLEEP_TIME was set to -1
|
|
if [ "${SLEEP_TIME}" = -1 ]; then
|
|
break
|
|
fi
|
|
|
|
sleep "${SLEEP_TIME}"
|
|
done
|
|
|
|
info "exiting"
|
|
}
|
|
|
|
main
|