mirror of
https://github.com/forcedotcom/afv-library.git
synced 2026-08-06 15:59:59 +08:00
84ee08cd47
* @W-21860796 Adds media-management skill with download script * chore: retrigger CLA check * chore: retrigger CLA validation * fix: use relative path for download script * fix: add metadata block and quote description to pass skill validation --------- Co-authored-by: Shruthi Paramesh <281795489+sparamesh-personal@users.noreply.github.com> Co-authored-by: Hemant Singh Bisht <hsinghbisht@salesforce.com>
85 lines
2.1 KiB
Bash
Executable File
85 lines
2.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
usage() {
|
|
cat <<EOF
|
|
Usage: $(basename "$0") --url <image_url> --id <response_id> --format <output_format> [--output-dir <dir>] [--preview]
|
|
|
|
Required:
|
|
--url URL of the generated image to download
|
|
--id Response ID used as the output filename
|
|
--format Output format (webp, jpeg, png)
|
|
|
|
Optional:
|
|
--output-dir Directory to save the image (default: generatedimages)
|
|
--preview Open the image in VS Code after download
|
|
EOF
|
|
exit 1
|
|
}
|
|
|
|
OUTPUT_DIR="generatedimages"
|
|
PREVIEW=false
|
|
URL=""
|
|
RESPONSE_ID=""
|
|
FORMAT=""
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--url) URL="$2"; shift 2 ;;
|
|
--id) RESPONSE_ID="$2"; shift 2 ;;
|
|
--format) FORMAT="$2"; shift 2 ;;
|
|
--output-dir) OUTPUT_DIR="$2"; shift 2 ;;
|
|
--preview) PREVIEW=true; shift ;;
|
|
*) usage ;;
|
|
esac
|
|
done
|
|
|
|
if [[ -z "$URL" || -z "$RESPONSE_ID" || -z "$FORMAT" ]]; then
|
|
usage
|
|
fi
|
|
|
|
case "$FORMAT" in
|
|
webp|jpeg|png) ;;
|
|
*) echo "Error: invalid format '$FORMAT'. Must be webp, jpeg, or png." >&2; exit 1 ;;
|
|
esac
|
|
|
|
if [[ "$RESPONSE_ID" == */* || "$RESPONSE_ID" == *..* ]]; then
|
|
echo "Error: invalid response ID '$RESPONSE_ID'. Must not contain '/' or '..'." >&2
|
|
exit 1
|
|
fi
|
|
|
|
if [[ ! "$URL" =~ ^https:// ]]; then
|
|
echo "Error: URL must start with https://" >&2
|
|
exit 1
|
|
fi
|
|
|
|
TARGET_ORG=$(sf config get target-org --json | jq -r '.result[0].value')
|
|
|
|
if [[ -z "$TARGET_ORG" || "$TARGET_ORG" == "null" ]]; then
|
|
echo "Error: no target-org configured. Run 'sf config set target-org <org>'" >&2
|
|
exit 1
|
|
fi
|
|
|
|
ACCESS_TOKEN=$(sf org display --target-org "$TARGET_ORG" --json | jq -r '.result.accessToken')
|
|
|
|
if [[ -z "$ACCESS_TOKEN" || "$ACCESS_TOKEN" == "null" ]]; then
|
|
echo "Error: failed to retrieve access token for org '$TARGET_ORG'" >&2
|
|
exit 1
|
|
fi
|
|
|
|
mkdir -p "$OUTPUT_DIR"
|
|
|
|
OUTPUT_FILE="${OUTPUT_DIR}/${RESPONSE_ID}.${FORMAT}"
|
|
|
|
if ! curl -fsS -H "Authorization: Bearer $ACCESS_TOKEN" -o "$OUTPUT_FILE" -- "$URL"; then
|
|
rm -f "$OUTPUT_FILE"
|
|
echo "Error: failed to download image from $URL" >&2
|
|
exit 1
|
|
fi
|
|
|
|
echo "$OUTPUT_FILE"
|
|
|
|
if [[ "$PREVIEW" == true ]]; then
|
|
code "$OUTPUT_FILE"
|
|
fi
|