142 lines
		
	
	
		
			6.5 KiB
		
	
	
	
		
			Bash
		
	
	
			
		
		
	
	
			142 lines
		
	
	
		
			6.5 KiB
		
	
	
	
		
			Bash
		
	
	
| #!/bin/bash
 | |
| 
 | |
| # this script will use the api:
 | |
| #    https://github.com/matrix-org/synapse/blob/master/docs/admin_api/purge_history_api.rst
 | |
| # 
 | |
| # It will purge all messages in a list of rooms up to a cetrain event
 | |
| 
 | |
| ###################################################################################################
 | |
| # define your domain and admin user
 | |
| ###################################################################################################
 | |
| # add this user as admin in your home server:
 | |
| DOMAIN=yourserver.tld
 | |
| # add this user as admin in your home server:
 | |
| ADMIN="@you_admin_username:$DOMAIN"
 | |
| 
 | |
| API_URL="$DOMAIN:8008/_matrix/client/r0"
 | |
| 
 | |
| ###################################################################################################
 | |
| #choose the rooms to prune old messages from (add a free comment at the end)
 | |
| ###################################################################################################
 | |
| # the room_id's you can get e.g. from your Riot clients "View Source" button on each message
 | |
| ROOMS_ARRAY=(
 | |
| '!DgvjtOljKujDBrxyHk:matrix.org#riot:matrix.org'
 | |
| '!QtykxKocfZaZOUrTwp:matrix.org#Matrix HQ'
 | |
| )
 | |
| 
 | |
| # ALTERNATIVELY:
 | |
| # you can select all the rooms that are not encrypted and loop over the result:
 | |
| # SELECT room_id FROM rooms WHERE room_id NOT IN (SELECT DISTINCT room_id FROM events WHERE type ='m.room.encrypted')
 | |
| # or
 | |
| # select all rooms with at least 100 members:
 | |
| # SELECT q.room_id FROM (select count(*) as numberofusers, room_id FROM current_state_events WHERE type ='m.room.member'
 | |
| #   GROUP BY room_id) AS q LEFT JOIN room_aliases a ON q.room_id=a.room_id WHERE q.numberofusers > 100 ORDER BY numberofusers desc
 | |
| 
 | |
| ###################################################################################################
 | |
| # evaluate the EVENT_ID before which should be pruned
 | |
| ###################################################################################################
 | |
| # choose a time before which the messages should be pruned:
 | |
| TIME='12 months ago'
 | |
| # ALTERNATIVELY:
 | |
| # a certain time:
 | |
| # TIME='2016-08-31 23:59:59'
 | |
| 
 | |
| # creates a timestamp from the given time string:
 | |
| UNIX_TIMESTAMP=$(date +%s%3N --date='TZ="UTC+2" '"$TIME")
 | |
| 
 | |
| # ALTERNATIVELY:
 | |
| # prune all messages that are older than 1000 messages ago:
 | |
| # LAST_MESSAGES=1000
 | |
| # SQL_GET_EVENT="SELECT event_id from events WHERE type='m.room.message' AND room_id ='$ROOM' ORDER BY received_ts DESC LIMIT 1 offset $(($LAST_MESSAGES - 1))"
 | |
| 
 | |
| # ALTERNATIVELY:
 | |
| # select the EVENT_ID manually:
 | |
| #EVENT_ID='$1471814088343495zpPNI:matrix.org' # an example event from 21st of Aug 2016 by Matthew
 | |
| 
 | |
| ###################################################################################################
 | |
| # make the admin user a server admin in the database with
 | |
| ###################################################################################################
 | |
| # psql -A -t --dbname=synapse -c "UPDATE users SET admin=1 WHERE name LIKE '$ADMIN'"
 | |
| 
 | |
| ###################################################################################################
 | |
| # database function
 | |
| ###################################################################################################
 | |
| sql (){
 | |
|   # for sqlite3:
 | |
|   #sqlite3 homeserver.db "pragma busy_timeout=20000;$1" | awk '{print $2}'
 | |
|   # for postgres:
 | |
|   psql -A -t --dbname=synapse -c "$1" | grep -v 'Pager'
 | |
| }
 | |
| 
 | |
| ###################################################################################################
 | |
| # get an access token
 | |
| ###################################################################################################
 | |
| # for example externally by watching Riot in your browser's network inspector
 | |
| # or internally on the server locally, use this:
 | |
| TOKEN=$(sql "SELECT token FROM access_tokens WHERE user_id='$ADMIN' ORDER BY id DESC LIMIT 1")
 | |
| AUTH="Authorization: Bearer $TOKEN"
 | |
| 
 | |
| ###################################################################################################
 | |
| # check, if your TOKEN works. For example this works: 
 | |
| ###################################################################################################
 | |
| # $ curl --header "$AUTH" "$API_URL/rooms/$ROOM/state/m.room.power_levels" 
 | |
| 
 | |
| ###################################################################################################
 | |
| # finally start pruning the room:
 | |
| ###################################################################################################
 | |
| POSTDATA='{"delete_local_events":"true"}' # this will really delete local events, so the messages in the room really disappear unless they are restored by remote federation
 | |
| 
 | |
| for ROOM in "${ROOMS_ARRAY[@]}"; do
 | |
|     echo "########################################### $(date) ################# "
 | |
|     echo "pruning room: $ROOM ..."
 | |
|     ROOM=${ROOM%#*}
 | |
|     #set -x
 | |
|     echo "check for alias in db..."
 | |
|     # for postgres:
 | |
|     sql "SELECT * FROM room_aliases WHERE room_id='$ROOM'"
 | |
|     echo "get event..."
 | |
|     # for postgres:
 | |
|     EVENT_ID=$(sql "SELECT event_id FROM events WHERE type='m.room.message' AND received_ts<'$UNIX_TIMESTAMP' AND room_id='$ROOM' ORDER BY received_ts DESC LIMIT 1;")
 | |
|     if [ "$EVENT_ID" == "" ]; then
 | |
|       echo "no event $TIME"
 | |
|     else
 | |
|       echo "event: $EVENT_ID"
 | |
|       SLEEP=2
 | |
|       set -x
 | |
|       # call purge
 | |
|       OUT=$(curl --header "$AUTH" -s -d $POSTDATA POST "$API_URL/admin/purge_history/$ROOM/$EVENT_ID")
 | |
|       PURGE_ID=$(echo "$OUT" |grep purge_id|cut -d'"' -f4 )
 | |
|       if [ "$PURGE_ID" == "" ]; then
 | |
|         # probably the history purge is already in progress for $ROOM
 | |
|         : "continuing with next room"
 | |
|       else
 | |
|         while : ; do
 | |
|           # get status of purge and sleep longer each time if still active
 | |
|           sleep $SLEEP
 | |
|           STATUS=$(curl --header "$AUTH" -s GET "$API_URL/admin/purge_history_status/$PURGE_ID" |grep status|cut -d'"' -f4)
 | |
|           : "$ROOM --> Status: $STATUS"
 | |
|           [[ "$STATUS" == "active" ]] || break 
 | |
|           SLEEP=$((SLEEP + 1))
 | |
|         done 
 | |
|       fi
 | |
|       set +x
 | |
|       sleep 1
 | |
|     fi  
 | |
| done
 | |
| 
 | |
| 
 | |
| ###################################################################################################
 | |
| # additionally
 | |
| ###################################################################################################
 | |
| # to benefit from pruning large amounts of data, you need to call VACUUM to free the unused space.
 | |
| # This can take a very long time (hours) and the client have to be stopped while you do so:
 | |
| # $ synctl stop
 | |
| # $ sqlite3 -line homeserver.db "vacuum;"
 | |
| # $ synctl start
 | |
| 
 | |
| # This could be set, so you don't need to prune every time after deleting some rows:
 | |
| # $ sqlite3 homeserver.db "PRAGMA auto_vacuum = FULL;"
 | |
| # be cautious, it could make the database somewhat slow if there are a lot of deletions
 | |
| 
 | |
| exit
 |