|
#!/bin/sh |
|
|
|
|
|
if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then |
|
echo "Starting without backup functionality - missing WEBDAV_URL, WEBDAV_USERNAME, or WEBDAV_PASSWORD" |
|
exec java -jar /app/bin/reader.jar |
|
exit 0 |
|
fi |
|
|
|
|
|
WEBDAV_BACKUP_PATH=${WEBDAV_BACKUP_PATH:-""} |
|
FULL_WEBDAV_URL="${WEBDAV_URL}" |
|
if [ -n "$WEBDAV_BACKUP_PATH" ]; then |
|
FULL_WEBDAV_URL="${WEBDAV_URL}/${WEBDAV_BACKUP_PATH}" |
|
fi |
|
|
|
|
|
restore_backup() { |
|
python3 -c " |
|
import sys |
|
import os |
|
import tarfile |
|
import requests |
|
from webdav3.client import Client |
|
options = { |
|
'webdav_hostname': '$FULL_WEBDAV_URL', |
|
'webdav_login': '$WEBDAV_USERNAME', |
|
'webdav_password': '$WEBDAV_PASSWORD' |
|
} |
|
client = Client(options) |
|
backups = [file for file in client.list() if file.endswith('.tar.gz') and file.startswith('reader_backup_')] |
|
if not backups: |
|
print('No backup files found') |
|
sys.exit() |
|
latest_backup = sorted(backups)[-1] |
|
with requests.get(f'$FULL_WEBDAV_URL/{latest_backup}', auth=('$WEBDAV_USERNAME', '$WEBDAV_PASSWORD'), stream=True) as r: |
|
if r.status_code == 200: |
|
with open(f'/tmp/{latest_backup}', 'wb') as f: |
|
for chunk in r.iter_content(chunk_size=8192): |
|
f.write(chunk) |
|
|
|
if os.path.exists(f'/tmp/{latest_backup}'): |
|
with tarfile.open(f'/tmp/{latest_backup}', 'r:gz') as tar: |
|
tar.extractall('/storage') |
|
print(f'Successfully restored backup from {latest_backup}') |
|
else: |
|
print('Failed to download backup file') |
|
else: |
|
print(f'Failed to download backup: {r.status_code}') |
|
" |
|
} |
|
|
|
|
|
echo "Downloading latest backup from WebDAV..." |
|
restore_backup |
|
|
|
|
|
sync_data() { |
|
while true; do |
|
echo "Starting sync process at $(date)" |
|
|
|
if [ -d /storage ]; then |
|
timestamp=$(date +%Y%m%d_%H%M%S) |
|
backup_file="reader_backup_${timestamp}.tar.gz" |
|
|
|
|
|
tar -czf "/tmp/${backup_file}" -C /storage . |
|
|
|
|
|
curl -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "/tmp/${backup_file}" "$FULL_WEBDAV_URL/${backup_file}" |
|
if [ $? -eq 0 ]; then |
|
echo "Successfully uploaded ${backup_file} to WebDAV" |
|
else |
|
echo "Failed to upload ${backup_file} to WebDAV" |
|
fi |
|
|
|
|
|
python3 -c " |
|
import sys |
|
from webdav3.client import Client |
|
options = { |
|
'webdav_hostname': '$FULL_WEBDAV_URL', |
|
'webdav_login': '$WEBDAV_USERNAME', |
|
'webdav_password': '$WEBDAV_PASSWORD' |
|
} |
|
client = Client(options) |
|
backups = [file for file in client.list() if file.endswith('.tar.gz') and file.startswith('reader_backup_')] |
|
backups.sort() |
|
if len(backups) > 2: |
|
to_delete = len(backups) - 2 |
|
for file in backups[:to_delete]: |
|
client.clean(file) |
|
print(f'Successfully deleted {file}.') |
|
else: |
|
print('Only {} backups found, no need to clean.'.format(len(backups))) |
|
" 2>&1 |
|
|
|
rm -f "/tmp/${backup_file}" |
|
else |
|
echo "/storage directory does not exist, waiting for next sync..." |
|
fi |
|
|
|
SYNC_INTERVAL=${SYNC_INTERVAL:-600} |
|
sleep $SYNC_INTERVAL |
|
done |
|
} |
|
|
|
|
|
sync_data & |
|
|
|
|
|
exec java -jar /app/bin/reader.jar |