|
| 1 | +#!/usr/bin/env bash |
| 2 | +# |
| 3 | +# Re-pin base images in Retrack Dockerfiles to their current SHA256 manifest-list digests. |
| 4 | +# Usage: ./dev/scripts/docker-pin-digests.sh |
| 5 | +# |
| 6 | +set -euo pipefail |
| 7 | + |
| 8 | +RETRACK_ROOT="$(cd "$(dirname "$0")/../.." && pwd)" |
| 9 | + |
| 10 | +DOCKERFILES=( |
| 11 | + "$RETRACK_ROOT/Dockerfile" |
| 12 | + "$RETRACK_ROOT/Dockerfile.web-scraper" |
| 13 | + "$RETRACK_ROOT/Dockerfile.web-scraper-camoufox" |
| 14 | +) |
| 15 | + |
| 16 | +# File-based cache so subshells can share resolved digests. |
| 17 | +CACHE_FILE="$(mktemp)" |
| 18 | +trap 'rm -f "$CACHE_FILE"' EXIT |
| 19 | + |
| 20 | +get_digest() { |
| 21 | + local image="$1" |
| 22 | + |
| 23 | + # Check cache. |
| 24 | + local cached |
| 25 | + cached="$(grep -F "$image " "$CACHE_FILE" 2>/dev/null | head -1 | awk '{print $2}')" || true |
| 26 | + if [[ -n "$cached" ]]; then |
| 27 | + echo "$cached" |
| 28 | + return |
| 29 | + fi |
| 30 | + |
| 31 | + echo " Fetching digest for $image ..." >&2 |
| 32 | + local digest |
| 33 | + digest="$(docker buildx imagetools inspect "$image" 2>/dev/null \ |
| 34 | + | grep -m1 '^Digest:' | awk '{print $2}')" |
| 35 | + |
| 36 | + if [[ -z "$digest" || ! "$digest" =~ ^sha256: ]]; then |
| 37 | + echo "ERROR: failed to fetch digest for $image" >&2 |
| 38 | + return 1 |
| 39 | + fi |
| 40 | + |
| 41 | + # Strip the sha256: prefix we add it back when rewriting. |
| 42 | + digest="${digest#sha256:}" |
| 43 | + echo "$image $digest" >> "$CACHE_FILE" |
| 44 | + echo "$digest" |
| 45 | +} |
| 46 | + |
| 47 | +for dockerfile in "${DOCKERFILES[@]}"; do |
| 48 | + [[ -f "$dockerfile" ]] || { echo "SKIP: $dockerfile not found"; continue; } |
| 49 | + |
| 50 | + tmp="$(mktemp)" |
| 51 | + changed=false |
| 52 | + |
| 53 | + while IFS= read -r line; do |
| 54 | + if [[ "$line" =~ ^FROM[[:space:]] ]]; then |
| 55 | + # Strip any existing @sha256:... from the image reference. |
| 56 | + stripped="$(echo "$line" | sed -E 's/@sha256:[0-9a-f]+//')" |
| 57 | + |
| 58 | + # Extract the image:tag - it's the token after FROM (and optional --platform=...). |
| 59 | + image_tag="$(echo "$stripped" | sed -E 's/^FROM[[:space:]]+(--platform=[^ ]+[[:space:]]+)?([^ ]+).*/\2/')" |
| 60 | + |
| 61 | + digest="$(get_digest "$image_tag")" |
| 62 | + # Insert @sha256:digest right after the image:tag in the stripped line. |
| 63 | + line="$(echo "$stripped" | sed -E "s|${image_tag}|${image_tag}@sha256:${digest}|")" |
| 64 | + changed=true |
| 65 | + fi |
| 66 | + echo "$line" |
| 67 | + done < "$dockerfile" > "$tmp" |
| 68 | + |
| 69 | + if $changed; then |
| 70 | + mv "$tmp" "$dockerfile" |
| 71 | + echo "Pinned: $dockerfile" |
| 72 | + else |
| 73 | + rm "$tmp" |
| 74 | + echo "SKIP: no FROM lines in $dockerfile" |
| 75 | + fi |
| 76 | +done |
0 commit comments