push sheeet

pkgs/development/python-modules/av/update-test-samples.bash

@@ -0,0 +1,44 @@
+#!/usr/bin/env bash
+
+set -o errexit
+set -o nounset
+
+if test "$#" != 1; then
+    printf >&2 'usage: update-test-samples.bash /path/to/PyAV/source\n'
+    exit 2
+fi
+
+pyav_source=$1
+
+exec > "$(dirname "$(readlink -f "$0")")/test-samples.toml"
+
+fetch() {
+    path=$1
+    url=$2
+    prefetch_json=$(nix store prefetch-file --json "${url}")
+    sri_hash=$(jq -r .hash <<< "${prefetch_json}")
+    printf '"%s" = { url = "%s", hash = "%s" }\n' "${path}" "${url}" "${sri_hash}"
+}
+
+fetch_all() {
+    function=$1
+    base_path=$2
+    base_url=$3
+
+    samples=$(
+        rg \
+            --only-matching \
+            --no-filename \
+            "\\b${function}\\([\"']([^\"']+)[\"']\\)" \
+            --replace '$1' \
+            "${pyav_source}"
+    )
+    unique_samples=$(sort -u <<< "${samples}")
+
+    while IFS= read -r sample; do
+    fetch "${base_path}/${sample}" "${base_url}/${sample}"
+    done <<< "${unique_samples}"
+}
+
+fetch_all fate_suite fate-suite "http://fate.ffmpeg.org/fate-suite"
+fetch_all curated pyav-curated "https://pyav.org/datasets"