build/cache.sh

   1 #!/bin/bash
   2 ##############################################################################
   3 # Copyright (c) 2015 Ericsson AB and others.
   4 # stefan.k.berg@ericsson.com
   5 # jonas.bjurel@ericsson.com
   6 # All rights reserved. This program and the accompanying materials
   7 # are made available under the terms of the Apache License, Version 2.0
   8 # which accompanies this distribution, and is available at
   9 # http://www.apache.org/licenses/LICENSE-2.0
  10 ##############################################################################
  11
  12
  13 exit_trap() {
  14     if [ -d "$TMPDIR" ]; then
  15         rm -rf $TMPDIR
  16     fi
  17 }
  18
  19 trap exit_trap EXIT
  20
  21 CACHETRANSPORT=${CACHETRANSPORT:-"curl --silent"}
  22 CACHEBASE=${CACHEBASE:-"file://${HOME}/cache"}
  23 CACHEMAXAGE=${CACHEMAXAGE:-$[14*24*3600]}
  24 CACHEDEBUG=${CACHEDEBUG:-1}
  25
  26 debugmsg () {
  27     if [ "$CACHEDEBUG" -eq 1 ]; then
  28         echo "$@" >&2
  29     fi
  30 }
  31
  32 errorexit () {
  33     echo "$@" >&2
  34     exit 1
  35 }
  36
  37 # Get a SHA1 based on what's piped into the cache command
  38 getid() {
  39     debugmsg "Generating sha1sum"
  40     sha1sum | sed 's/ .*//'
  41 }
  42
  43
  44 # Put in cache
  45 put() {
  46     if check $1; then
  47        debugmsg "SHA1 $1 already in cache, skipping storage"
  48     else
  49         debugmsg "Storing SHA1 $1 in cache"
  50         ${CACHETRANSPORT} -T - ${CACHEBASE}/$1.blob
  51         echo "Expires: $[`date +"%s"` + $CACHEMAXAGE]" | ${CACHETRANSPORT} -T - ${CACHEBASE}/$1.meta
  52     fi
  53     exit 0
  54 }
  55
  56 # Get from cache
  57 get() {
  58     local rc
  59
  60     ${CACHETRANSPORT} -o - ${CACHEBASE}/$1.blob 2>/dev/null
  61     rc=$?
  62
  63     if [ $rc -eq 0 ]; then
  64         echo "Got SHA1 $1 from cache" 2>/dev/null
  65     else
  66         echo "Tried to get SHA1 $1 from cache but failed" 2>/dev/null
  67     fi
  68
  69     return $?
  70 }
  71
  72 # Check if in cache
  73 check() {
  74     local rc
  75
  76     ${CACHETRANSPORT} ${CACHEBASE}/$1.meta &>/dev/null
  77     rc=$?
  78
  79     if [ $rc -eq 0 ]; then
  80         debugmsg "Checking for SHA1 $1 in cache and found it, rc = $rc"
  81     else
  82         debugmsg "Checking for SHA1 $1 in cache and failed, rc = $rc"
  83     fi
  84
  85     return $rc
  86 }
  87
  88 # Verify that SHA1 seems to be a SHA1...
  89 validSHA1() {
  90     if [ $(echo $1 | wc -c) -ne 41 ]; then
  91         return 1
  92     else
  93         return 0
  94     fi
  95 }
  96
  97 # Figure out commit ID from URI and tag/branch/commit ID
  98 getcommitid() {
  99     HEADMATCH=`git ls-remote $1 | grep "refs/heads/$2$" | awk '{ print $1 }'`
 100     TAGMATCH=`git ls-remote $1 | grep "refs/tags/$2$" | awk '{ print $1 }'`
 101
 102     if [ -n "$HEADMATCH" ]; then
 103         echo "$HEADMATCH"
 104     elif [ -n "$TAGMATCH" ]; then
 105         echo "$TAGMATCH"
 106     else
 107         TMPDIR=`mktemp -d /tmp/cacheXXXXX`
 108         cd $TMPDIR
 109         git clone $1 &>/dev/null || errorexit "Could not clone $1"
 110         cd * || errorexit "Could not enter clone of $1"
 111         git show $2 &>/dev/null || errorexit "Could not find commit $2"
 112         git show $2 | head -1 | awk '{ print $2 }'
 113     fi
 114 }
 115
 116 case $1 in
 117     getcommitid)
 118         if [ $# -ne 3 ]; then
 119             errorexit "Arg 1 needs to be URI and arg 2 tag/branch/commit"
 120         fi
 121         shift
 122         getcommitid $@
 123         ;;
 124     getid)
 125         if [ $# -ne 1 ]; then
 126             errorexit "No arguments can be given to getid!"
 127         fi
 128         getid
 129         ;;
 130     get|check|put)
 131         if [ $# -ne 2 ]; then
 132             errorexit "Only one argument, the SHA1 sum, can be given to getid!"
 133         else
 134             if ! validSHA1 $2; then
 135                 errorexit "Invalid SHA1 format!"
 136             fi
 137         fi
 138
 139         $1 $2
 140         exit $rc
 141         ;;
 142     *)
 143         errorexit "I only know about getcommitid, getid, check, get and put!"
 144 esac