104 lines
3 KiB
Bash
104 lines
3 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
# shellcheck source=/dev/null
|
||
|
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/amduat_v2_client.sh"
|
||
|
|
|
||
|
|
app_init() {
|
||
|
|
amduat_client_init
|
||
|
|
}
|
||
|
|
|
||
|
|
app_startup_checks() {
|
||
|
|
amduat_api_call GET "/v2/readyz" || return 1
|
||
|
|
printf '%s\n' "${AMDUAT_LAST_BODY}"
|
||
|
|
|
||
|
|
amduat_api_call GET "/v2/graph/capabilities" || return 1
|
||
|
|
printf '%s\n' "${AMDUAT_LAST_BODY}"
|
||
|
|
}
|
||
|
|
|
||
|
|
app_ingest_batch() {
|
||
|
|
local payload="$1"
|
||
|
|
amduat_api_call POST "/v2/graph/batch" "${payload}"
|
||
|
|
printf '%s\n' "${AMDUAT_LAST_BODY}"
|
||
|
|
}
|
||
|
|
|
||
|
|
app_sync_once() {
|
||
|
|
local path="/v2/graph/changes?limit=${SYNC_LIMIT}&wait_ms=${SYNC_WAIT_MS}"
|
||
|
|
local cursor=""
|
||
|
|
|
||
|
|
if [[ -f "${CURSOR_FILE}" ]]; then
|
||
|
|
cursor="$(cat "${CURSOR_FILE}")"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ -n "${cursor}" ]]; then
|
||
|
|
path+="&since_cursor=${cursor}"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if ! amduat_api_call GET "${path}"; then
|
||
|
|
if [[ "${AMDUAT_LAST_STATUS}" == "410" ]]; then
|
||
|
|
rm -f "${CURSOR_FILE}"
|
||
|
|
echo "changes cursor expired (410); cleared ${CURSOR_FILE}" >&2
|
||
|
|
return 0
|
||
|
|
fi
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
printf '%s\n' "${AMDUAT_LAST_BODY}"
|
||
|
|
|
||
|
|
local next
|
||
|
|
next="$(printf '%s\n' "${AMDUAT_LAST_BODY}" | sed -n 's/.*"next_cursor"[[:space:]]*:[[:space:]]*"\([^"]*\)".*/\1/p')"
|
||
|
|
if [[ -n "${next}" ]]; then
|
||
|
|
printf '%s' "${next}" > "${CURSOR_FILE}"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
app_sync_loop() {
|
||
|
|
while true; do
|
||
|
|
app_sync_once
|
||
|
|
sleep 1
|
||
|
|
done
|
||
|
|
}
|
||
|
|
|
||
|
|
app_retrieve_with_fallback() {
|
||
|
|
local roots_csv="$1"
|
||
|
|
local goals_csv="${2:-}"
|
||
|
|
|
||
|
|
local roots_json
|
||
|
|
roots_json="$(printf '%s' "${roots_csv}" | awk -F',' 'BEGIN{printf "["} {for(i=1;i<=NF;i++){gsub(/^ +| +$/, "", $i); if (length($i)>0){if (printed) printf ","; printf "\"%s\"", $i; printed=1}}} END{printf "]"}')"
|
||
|
|
|
||
|
|
local goals_json="[]"
|
||
|
|
if [[ -n "${goals_csv}" ]]; then
|
||
|
|
goals_json="$(printf '%s' "${goals_csv}" | awk -F',' 'BEGIN{printf "["} {for(i=1;i<=NF;i++){gsub(/^ +| +$/, "", $i); if (length($i)>0){if (printed) printf ","; printf "\"%s\"", $i; printed=1}}} END{printf "]"}')"
|
||
|
|
fi
|
||
|
|
|
||
|
|
local payload
|
||
|
|
payload="{\"roots\":${roots_json},\"goal_predicates\":${goals_json},\"max_depth\":2,\"max_fanout\":256,\"limit_nodes\":200,\"limit_edges\":400,\"max_result_bytes\":1048576}"
|
||
|
|
|
||
|
|
if amduat_api_call POST "/v2/graph/retrieve" "${payload}"; then
|
||
|
|
printf '%s\n' "${AMDUAT_LAST_BODY}"
|
||
|
|
return 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
local first_root
|
||
|
|
first_root="$(printf '%s' "${roots_csv}" | awk -F',' '{gsub(/^ +| +$/, "", $1); printf "%s", $1}')"
|
||
|
|
local fallback_path="/v2/graph/subgraph?roots[]=${first_root}&max_depth=2&dir=outgoing&limit_nodes=200&limit_edges=400&max_result_bytes=1048576"
|
||
|
|
if [[ -n "${goals_csv}" ]]; then
|
||
|
|
local first_goal
|
||
|
|
first_goal="$(printf '%s' "${goals_csv}" | awk -F',' '{gsub(/^ +| +$/, "", $1); printf "%s", $1}')"
|
||
|
|
if [[ -n "${first_goal}" ]]; then
|
||
|
|
fallback_path+="&predicates[]=${first_goal}"
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
amduat_api_call GET "${fallback_path}"
|
||
|
|
printf '%s\n' "${AMDUAT_LAST_BODY}"
|
||
|
|
}
|
||
|
|
|
||
|
|
app_tombstone_edge() {
|
||
|
|
local edge_ref="$1"
|
||
|
|
local payload
|
||
|
|
payload="{\"edge_ref\":\"${edge_ref}\"}"
|
||
|
|
amduat_api_call POST "/v2/graph/edges/tombstone" "${payload}"
|
||
|
|
printf '%s\n' "${AMDUAT_LAST_BODY}"
|
||
|
|
}
|