blob: 646fd5efb797c563c97ad9cc86589410490a4cad [file] [log] [blame]
#!/bin/bash
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
# This tool contains functions that are to be used to handle/enable funsize
# Author: Mihai Tabara
#
HOOK=
SERVER_URL=
LOCAL_CACHE_DIR=
getchecksum(){
echo "$(openssl dgst -sha1 "${1}" | awk '{print $2}')"
}
print_usage(){
echo "$(basename $0) -A SERVER-URL [-c LOCAL-CACHE-DIR-PATH] [-g] [-u] PATH-FROM-URL PATH-TO-URL PATH-PATCH"
echo "Script that saves/retrieves from cache presumptive patches as args"
echo ""
echo "-A SERVER-URL - host where to send the files"
echo "-c LOCAL-CACHE-DIR-PATH local path to which patches are cached"
echo "-g pre hook - tests whether patch already in cache"
echo "-u post hook - upload patch to cache for future use"
echo ""
echo "PATH-FROM-URL : path on disk for source file"
echo "PATH-TO-URL : path on disk for destination file"
echo "PATH-PATCH : path on disk for patch between source and destination"
}
upload_patch(){
sha_from=`getchecksum "$1"`
sha_to=`getchecksum "$2"`
patch_path="$3"
if [ -z "$sha_from" -o -z "$sha_to" ]; then
echo "Cannot calculate checksums"
return 1
fi
# save to local cache first
if [ -n "$LOCAL_CACHE_DIR" ]; then
local_cmd="mkdir -p "$LOCAL_CACHE_DIR/$sha_from""
if `$local_cmd` >&2; then
cp -avf "$patch_path" "$LOCAL_CACHE_DIR/$sha_from/$sha_to"
echo "$patch_path saved on local cache!"
fi
fi
# The remote cache implementation is not used. The code is for usage
# reference only.
return 0
# send it over to funsize
cmd="curl -sSw %{http_code} -o /dev/null -X POST $SERVER_URL -F sha_from="$sha_from" -F sha_to="$sha_to" -F patch_file="@$patch_path""
ret_code=`$cmd`
if [ $ret_code -eq 200 ]; then
echo "$patch_path Successful uploaded to funsize!"
return 0
fi
echo "$patch_path Failed to be uploaded to funsize!"
return 1
}
get_patch(){
sha_from=`getchecksum "$1"`
sha_to=`getchecksum "$2"`
destination_file="$3"
tmp_file="$destination_file.tmp"
if [ -z "$sha_from" -o -z "$sha_to" ]; then
echo "Cannot calculate checksums"
return 1
fi
# try to retrieve from local cache first
if [ -r "$LOCAL_CACHE_DIR/$sha_from/$sha_to" ]; then
cp -avf "$LOCAL_CACHE_DIR/$sha_from/$sha_to" "$destination_file"
echo "Successful retrieved $destination_file from local cache!"
return 0
else
echo "File is not in the locale cache"
return 1
fi
# The remote cache implementation is not used. The code is for usage
# reference only.
# if unsuccessful, try to retrieve from funsize
cmd="curl -LsSGw %{http_code} $SERVER_URL/$sha_from/$sha_to -o $tmp_file"
ret_code=`$cmd`
if [ $ret_code -eq 200 ]; then
mv "$tmp_file" "$destination_file"
echo "Successful retrieved $destination_file from funsize!"
return 0
fi
rm -f "$tmp_file"
echo "Failed to retrieve $destination_file from funsize!"
return 1
}
OPTIND=1
while getopts ":A:c:gu" option; do
case $option in
A)
SERVER_URL="$OPTARG"
;;
c)
LOCAL_CACHE_DIR="$OPTARG"
;;
g)
HOOK="PRE"
;;
u)
HOOK="POST"
;;
\?)
echo "Invalid option: -$OPTARG" >&2
print_usage
exit 1
;;
:)
echo "Option -$OPTARG requires an argument." >&2
print_usage
exit 1
;;
*)
echo "Unimplemented option: -$OPTARG" >&2
print_usage
exit 1
;;
esac
done
shift $((OPTIND-1))
if [ "$HOOK" == "PRE" ]; then
get_patch "$1" "$2" "$3"
elif [ "$HOOK" == "POST" ]; then
upload_patch "$1" "$2" "$3"
fi