Files
mattermost-retention/mattermost-retention.sh
2023-07-27 15:16:57 +00:00

107 lines
3.3 KiB
Bash
Executable File

#!/bin/bash
###
# configure vars
####
# Database user name
DB_USER="mmuser"
# Database name
DB_NAME="mattermost"
# Database password
DB_PASS=""
# Database hostname
DB_HOST="127.0.0.1"
# How many days to keep of messages/files?
RETENTION="30"
# Mattermost data directory
DATA_PATH="/mattermost/data/"
# Database drive (postgres OR mysql)
DB_DRIVE="postgres"
# Set the docker command prefex for accessing DB
DB_DOCKER_CMD=""
#DB_DOCKER_CMD="docker compose exec -it postgres" #for example
# Set the docker command prefex for accessing mattermost
#MM_DOCKER_CMD="docker compose exec -it mattermost" # for example
###
# calculate epoch in milisec
###
delete_before=$(date --date="$RETENTION day ago" "+%s%3N")
echo $(date --date="$RETENTION day ago")
case $DB_DRIVE in
postgres)
echo "Using postgres database."
export PGPASSWORD=$DB_PASS
###
# get list of files to be removed
###
$DB_DOCKER_CMD psql -h "$DB_HOST" -U"$DB_USER" "$DB_NAME" -t -c "select path from fileinfo where createat < $delete_before;" > /tmp/mattermost-paths.list
$DB_DOCKER_CMD psql -h "$DB_HOST" -U"$DB_USER" "$DB_NAME" -t -c "select thumbnailpath from fileinfo where createat < $delete_before;" >> /tmp/mattermost-paths.list
$DB_DOCKER_CMD psql -h "$DB_HOST" -U"$DB_USER" "$DB_NAME" -t -c "select previewpath from fileinfo where createat < $delete_before;" >> /tmp/mattermost-paths.list
###
# cleanup db
###
$DB_DOCKER_CMD psql -h "$DB_HOST" -U"$DB_USER" "$DB_NAME" -t -c "delete from posts where createat < $delete_before;"
$DB_DOCKER_CMD psql -h "$DB_HOST" -U"$DB_USER" "$DB_NAME" -t -c "delete from fileinfo where createat < $delete_before;"
;;
mysql)
echo "Using mysql database."
###
# get list of files to be removed
###
$DB_DOCKER_CMD mysql --password=$DB_PASS --user=$DB_USER --host=$DB_HOST --database=$DB_NAME --execute="select path from FileInfo where createat < $delete_before;" > /tmp/mattermost-paths.list
$DB_DOCKER_CMD mysql --password=$DB_PASS --user=$DB_USER --host=$DB_HOST --database=$DB_NAME --execute="select thumbnailpath from FileInfo where createat < $delete_before;" >> /tmp/mattermost-paths.list
$DB_DOCKER_CMD mysql --password=$DB_PASS --user=$DB_USER --host=$DB_HOST --database=$DB_NAME --execute="select previewpath from FileInfo where createat < $delete_before;" >> /tmp/mattermost-paths.list
###
# cleanup db
###
$DB_DOCKER_CMD mysql --password=$DB_PASS --user=$DB_USER --host=$DB_HOST --database=$DB_NAME --execute="delete from Posts where createat < $delete_before;"
$DB_DOCKER_CMD mysql --password=$DB_PASS --user=$DB_USER --host=$DB_HOST --database=$DB_NAME --execute="delete from FileInfo where createat < $delete_before;"
;;
*)
echo "Unknown DB_DRIVE option. Currently ONLY mysql AND postgres are available."
exit 1
;;
esac
###
# delete files
###
for fp in `cat /tmp/mattermost-paths.list`
do
if [ -n "$fp" ]; then
echo "$DATA_PATH""$fp"
$MM_DOCKER_CMD shred -u "$DATA_PATH""$fp"
fi
done
###
# cleanup after script execution
###
rm /tmp/mattermost-paths.list
###
# cleanup empty data dirs
###
$MM_DOCKER_CMD find $DATA_PATH -type d -empty -delete
exit 0