This commit is contained in:
parent
7d1910d6bd
commit
79b153819a
151
temp/sos
151
temp/sos
@ -1,151 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
set -euo pipefail
|
|
||||||
|
|
||||||
|
|
||||||
# get dropshell
|
|
||||||
TEMP_DIR=$(mktemp -d)
|
|
||||||
ARCH=$(uname -m)
|
|
||||||
curl -L -s -o "${TEMP_DIR}/dropshell" "https://getbin.xyz/dropshell.${ARCH}" || die "Failed to download dropshell"
|
|
||||||
chmod +x "${TEMP_DIR}/dropshell"
|
|
||||||
trap 'rm -rf "${TEMP_DIR}"' EXIT
|
|
||||||
DROPSHELL="${TEMP_DIR}/dropshell"
|
|
||||||
|
|
||||||
|
|
||||||
function show_help() {
|
|
||||||
cat << EOF
|
|
||||||
|
|
||||||
sos is a script to upload files to a simple object storage server.
|
|
||||||
|
|
||||||
Usage:
|
|
||||||
sos upload <server> <file> <label:tag> [label:tag ...]
|
|
||||||
|
|
||||||
Example:
|
|
||||||
sos upload tools.dropshell.app ./file.txt file:latest
|
|
||||||
|
|
||||||
This will upload the file to the server, and return the download URL.
|
|
||||||
|
|
||||||
Environment variables:
|
|
||||||
SOS_WRITE_TOKEN: The write token to use for the upload. If not set,
|
|
||||||
the script will look in $HOME/.config/sos/write_token.txt.
|
|
||||||
|
|
||||||
EOF
|
|
||||||
exit 0
|
|
||||||
}
|
|
||||||
|
|
||||||
function die() {
|
|
||||||
echo "FATAL:"
|
|
||||||
echo "$@"
|
|
||||||
exit 1
|
|
||||||
}
|
|
||||||
|
|
||||||
function datetime() {
|
|
||||||
date -u +"%Y.%m%d.%H%M"
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
function upload() {
|
|
||||||
if [ "$#" -lt 3 ]; then
|
|
||||||
echo "Usage: sos upload <server> <file> <label:tag> [label:tag ...]"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
server=$1
|
|
||||||
file=$2
|
|
||||||
first_label=$3
|
|
||||||
LABELTAGS=("$first_label")
|
|
||||||
[[ "$first_label" =~ : ]] || die "Label $first_label does not contain a tag!"
|
|
||||||
|
|
||||||
shift 3
|
|
||||||
for label in "$@"; do
|
|
||||||
[[ "$label" =~ : ]] || die "Label $label does not contain a tag!"
|
|
||||||
LABELTAGS+=("$label")
|
|
||||||
done
|
|
||||||
|
|
||||||
# check if file contains :
|
|
||||||
[[ ! "$file" =~ : ]] || die "File contains : - this is not allowed!"
|
|
||||||
[ -f "$file" ] || die "File not found: $file"
|
|
||||||
|
|
||||||
|
|
||||||
# upload the file
|
|
||||||
TARGET_URL="https://$server/upload"
|
|
||||||
echo "Uploading $file to $TARGET_URL"
|
|
||||||
|
|
||||||
DATETIME=$(datetime)
|
|
||||||
|
|
||||||
# deduplicate the labeltags
|
|
||||||
mapfile -t LABELTAGS < <(printf "%s\n" "${LABELTAGS[@]}" | sort -u)
|
|
||||||
LABELTAGS_JSON=$(printf '"%s",' "${LABELTAGS[@]}")
|
|
||||||
LABELTAGS_JSON="[${LABELTAGS_JSON%,}]"
|
|
||||||
|
|
||||||
METADATA_JSON=$(cat <<EOF
|
|
||||||
{
|
|
||||||
"labeltags": $LABELTAGS_JSON,
|
|
||||||
"description": "Uploaded by sos",
|
|
||||||
"version": "$DATETIME"
|
|
||||||
}
|
|
||||||
EOF
|
|
||||||
)
|
|
||||||
if [ -n "${SOS_WRITE_TOKEN:-}" ]; then
|
|
||||||
WRITE_TOKEN="$SOS_WRITE_TOKEN"
|
|
||||||
else
|
|
||||||
TOKENPATH="$HOME/.config/sos/write_token.txt"
|
|
||||||
if [ ! -f "$TOKENPATH" ]; then
|
|
||||||
die "Token file not found in environment variable SOS_WRITE_TOKEN or $TOKENPATH. Please create it and put your write token in it."
|
|
||||||
fi
|
|
||||||
WRITE_TOKEN=$(cat "$TOKENPATH")
|
|
||||||
fi
|
|
||||||
|
|
||||||
# trip whitespace from the file path
|
|
||||||
LOCALHASH=$("${DROPSHELL}" hash "${file}" | tr -d '[:space:]')
|
|
||||||
echo "Local hash: $LOCALHASH"
|
|
||||||
|
|
||||||
EXISTSJSON=$(eval "curl -s \"https://$server/exists/$LOCALHASH\"") || die "Failed to check if file exists"
|
|
||||||
DOESEXIT=$(echo "$EXISTSJSON" | jq -r '.exists')
|
|
||||||
|
|
||||||
HASH=""
|
|
||||||
|
|
||||||
if [ "$DOESEXIT" == "true" ]; then
|
|
||||||
echo "File already exists, skipping upload"
|
|
||||||
|
|
||||||
curl -X PUT -H "Authorization: Bearer ${WRITE_TOKEN}" \
|
|
||||||
-F "metadata=${METADATA_JSON}" \
|
|
||||||
"$TARGET_URL" \
|
|
||||||
|| die "Failed to upload $file to $TARGET_URL"
|
|
||||||
else
|
|
||||||
curl -X PUT -H "Authorization: Bearer ${WRITE_TOKEN}" \
|
|
||||||
-F "metadata=${METADATA_JSON}" \
|
|
||||||
-F "file=@${file}" \
|
|
||||||
"$TARGET_URL" \
|
|
||||||
|| die "Failed to upload $file to $TARGET_URL"
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo " "
|
|
||||||
echo " "
|
|
||||||
|
|
||||||
JSON1=$(eval "curl -s \"https://$server/hash/$first_label\"")
|
|
||||||
HASH=$(echo "$JSON1" | jq -r '.hash')
|
|
||||||
|
|
||||||
JSON2=$(eval "curl -s \"https://$server/meta/$HASH\"") || die "Failed to get meta for $HASH"
|
|
||||||
FILENAME=$(echo "$JSON2" | jq -r '.metadata.filename')
|
|
||||||
|
|
||||||
echo "Metadata:"
|
|
||||||
echo "$JSON2" | jq
|
|
||||||
|
|
||||||
echo " "
|
|
||||||
|
|
||||||
echo "Download URL: https://$server/$first_label > $FILENAME"
|
|
||||||
echo "Alternative: https://$server/$HASH > $FILENAME"
|
|
||||||
}
|
|
||||||
|
|
||||||
# if no arguments, show help
|
|
||||||
[ "$#" -gt 0 ] || show_help
|
|
||||||
|
|
||||||
CMD="$1"
|
|
||||||
shift
|
|
||||||
|
|
||||||
if [ "$CMD" == "upload" ]; then
|
|
||||||
upload "$@"
|
|
||||||
exit $?
|
|
||||||
fi
|
|
||||||
|
|
||||||
die "Unknown command: $CMD"
|
|
Loading…
x
Reference in New Issue
Block a user