596dbb923d
deleted: .app.py.swp modified: .gitignore new file: README.rst modified: app.py new file: initialize_db.py new file: test_agent.sh
75 lines
2.1 KiB
Bash
75 lines
2.1 KiB
Bash
#!/usr/bin/env bash
|
|
#
|
|
# Usage:
|
|
# MAX_COOKIE_AGE=1800 ./agent_upload.sh /path/to/somefile.jpg
|
|
#
|
|
# Description:
|
|
# Upload a file to the pyrafiles server as an agent, making it public.
|
|
# - If cookies.txt is newer than MAX_COOKIE_AGE seconds (default 31536000, ~1 year),
|
|
# we skip re-login.
|
|
# - Otherwise, we prompt for OTP again.
|
|
#
|
|
# Notes:
|
|
# 1. If not set, MAX_COOKIE_AGE defaults to 31536000 (one year).
|
|
# 2. We assume the server runs on http://localhost:6544. Adjust BASE_URL if needed.
|
|
|
|
BASE_URL="${BASE_URL:-http://localhost:6544}"
|
|
EMAIL="${EMAIL:-agent@example.com}"
|
|
MEDIA_FILE="$1"
|
|
# Default to 1 year in seconds if not provided:
|
|
MAX_COOKIE_AGE="${MAX_COOKIE_AGE:-31536000}"
|
|
|
|
if [[ -z "$MEDIA_FILE" ]]; then
|
|
echo "Usage: MAX_COOKIE_AGE=<seconds> $0 /path/to/mediafile"
|
|
exit 1
|
|
fi
|
|
|
|
# Check if cookies.txt is 'fresh enough':
|
|
COOKIE_FRESH=0
|
|
if [[ -f cookies.txt ]]; then
|
|
# 'stat -c %Y' returns the file's mod time on Linux; 'stat -f %m' on macOS/BSD
|
|
MOD_TIME=$(stat -c %Y cookies.txt 2>/dev/null || stat -f %m cookies.txt 2>/dev/null)
|
|
NOW=$(date +%s)
|
|
AGE=$((NOW - MOD_TIME))
|
|
|
|
if (( AGE < MAX_COOKIE_AGE )); then
|
|
COOKIE_FRESH=1
|
|
fi
|
|
fi
|
|
|
|
if [[ "$COOKIE_FRESH" -eq 1 ]]; then
|
|
echo "Reusing existing cookies (file age < $MAX_COOKIE_AGE seconds)."
|
|
echo "Skipping OTP prompt."
|
|
else
|
|
echo "Starting new session (cookie missing or stale)..."
|
|
curl -s -c cookies.txt -b cookies.txt "$BASE_URL/" >/dev/null
|
|
|
|
echo "Logging in with email: $EMAIL"
|
|
curl -s -c cookies.txt -b cookies.txt \
|
|
-X POST \
|
|
-F "email=$EMAIL" \
|
|
"$BASE_URL/auth/login"
|
|
|
|
echo ""
|
|
echo "Check your console or email for the 6-digit verification code."
|
|
read -p "Enter the 6-digit code: " VERIFICATION_CODE
|
|
|
|
echo "Verifying code..."
|
|
curl -s -c cookies.txt -b cookies.txt \
|
|
-X POST \
|
|
-F "code=$VERIFICATION_CODE" \
|
|
"$BASE_URL/auth/verify"
|
|
|
|
echo "Cookies refreshed."
|
|
fi
|
|
|
|
echo ""
|
|
echo "Uploading media: $MEDIA_FILE (public)"
|
|
curl -s -c cookies.txt -b cookies.txt \
|
|
-X POST \
|
|
-F "media_file=@$MEDIA_FILE" \
|
|
-F "is_public=on" \
|
|
"$BASE_URL/media/upload"
|
|
|
|
echo ""
|
|
echo "Upload complete. File is public."
|