Skip to content

Instantly share code, notes, and snippets.

@bviktor bviktor/glavatar.sh
Last active Jun 28, 2018

Embed
What would you like to do?
GitLab Avatar Updater
#!/bin/sh
set -Eeu
########################################################################
############################ START SETTINGS ############################
########################################################################
# your GitLab instance' URL
export GITLAB_URL='https://gitlab.foobar.com'
# directory where you installed glavatar
export GLAVATAR_ROOT='/opt/glavatar'
# your netscape formatted cookie for the site you download the avatars from
# for firefox you can use https://addons.mozilla.org/en-US/firefox/addon/cookies-txt/
export COOKIE_FILE="${GLAVATAR_ROOT}/cookie.txt"
export IMAGE_BASE='https://portal.foobar.com/images/members'
# database server host
export DB_HOST='127.0.0.1'
# database server port
export DB_PORT='5432'
# database name
export DB_NAME='gitlabhq_production'
# database user
export DB_USER='gitlab'
# its password
export DB_PASS='P@ssw0rd'
# needed scopes: api
export GITLAB_TOKEN='T0k3n'
########################################################################
############################# END SETTINGS #############################
########################################################################
# work variables, you shouldn't change anything from this point
export USER_FILE="${GLAVATAR_ROOT}/users.txt"
export AVATAR_FILE="${GLAVATAR_ROOT}/avatar.jpg"
function pg_query ()
{
sudo -u postgres sh -c "cd && PGPASSWORD=\"${DB_PASS}\" psql --host=${DB_HOST} --port=${DB_PORT} --dbname=${DB_NAME} --username=${DB_USER} --tuples-only --no-align --field-separator=',' --command \"${1}\""
}
# define the query, this assumes you use LDAP users
QUERY="SELECT id,username FROM users WHERE password_automatically_set IS TRUE"
# execute the query
pg_query "${QUERY}" > ${USER_FILE}
# loop through user list
test -e ${USER_FILE} || exit
while read -r line
do
ID=$(cut -d',' -f1 <<<${line})
UNAME=$(cut -d',' -f2 <<<${line})
# delete previous user's avatar
rm -f ${AVATAR_FILE}
# download current user's avatar
wget -nv --content-on-error --load-cookies=${COOKIE_FILE} "${IMAGE_BASE}/${UNAME}.jpg" -O ${AVATAR_FILE} || true
# continue if grep doesn't match, but preserve the match state anyway
RET=1
file ${AVATAR_FILE} | grep -i jpeg && RET=$? || true
# only do API request if it's actually an image
if [ ${RET} -eq 0 ]
then
SIZE=$(stat --printf="%s" ${AVATAR_FILE})
# gitlab rejects avatars bigger than 200k
if [ ${SIZE} -gt 200000 ]
then
# extent is not exact, use a slightly lower value
convert ${AVATAR_FILE} -define jpeg:extent=180kb ${AVATAR_FILE}
fi
# push it to GitLab via API
curl --request PUT --header "PRIVATE-TOKEN: ${GITLAB_TOKEN}" --form "avatar=@/${AVATAR_FILE}" "${GITLAB_URL}/api/v4/users/${ID}"
# cosmetics
echo -e '\n'
fi
done < ${USER_FILE}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.