Query addlink-simple on Toolforge

linkrecommendation-toolforge.sh

#!/bin/bash

set -e

WIKI=$1
TITLE=$2
if [ -z "$3" ]; then
  THRESHOLD=0.5
else
  THRESHOLD=$3
fi
URL="https://$WIKI.wikipedia.org/w/api.php?action=query&format=json&prop=revisions&rvprop=ids%7Ccontent&rvslots=main"
RESPONSE=$(curl --data-urlencode titles="$TITLE" -s "$URL")
PAGEID=$(echo "$RESPONSE" | jq -r .query.pages[].pageid )
REVID=$(echo "$RESPONSE" | jq -r .query.pages[].revisions[].revid)
CONTENT=$(echo "$RESPONSE" | jq -r '.query.pages[].revisions[].slots.main."*"')
ENCODED_CONTENT=$(echo "$CONTENT" | jq -sR tojson)

ENDPOINT="https://addlink-simple.toolforge.org/query"

JSON="{\"max_recommendations\": 10,
\"page_title\": \"$TITLE\",
\"revid\": $REVID,
\"pageid\": $PAGEID,
\"wiki_id\": \"$WIKI\",
\"wikitext\": $ENCODED_CONTENT,
\"threshold\": 0.5}"
echo "$JSON" | curl -d @- --header "Content-Type: application/json" -s "$ENDPOINT"