sbm

#!/usr/bin/env bash
# ==============================================================================
# SlackBuild Manager for The Slackware Linux CN Community
# ==============================================================================
# Copyright (c) 2016 Arondight <shell_way@foxmail.com>,
#                    The Slackware Linux CN Community
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
# ==============================================================================
# This to add, delete or update SlackBuilds
# ==============================================================================
set -e

BASEURL=${BASEURL:-'https://github.com/slackwarecn-slackbuilds'}
RDIR=$(dirname $(readlink -f $0))
LOGFILE="${RDIR}/Changes"
okjobs=( )
# XXX This regex can be improved {
URLREG='^https?://(([\w\d\.-]+\.\w{2,6})|(\d{1,3}(\.\d{1,3}){3}))(:\d{1,4})*(/[\w\d\./-]*)?$'
# }

# ==============================================================================
# Show usage of sbm
# ==============================================================================
function usage ()
{
  ( echo 'sbm - SlackBuild Manager for The Slackware Linux CN Community'
    echo
    echo 'Usage:'
    echo '  ./sbm [add|del|sync|init|undo|build|help] [<repo>|<url>]...'
    echo
    echo '  add:      Add an git repository of SlackBuild as sub module'
    echo '  del:      Delete an sub module'
    echo '  sync:     Update sub modules'
    echo '  undo:     Undo any changes'
    echo '  init:     Init repository, only do this at very begining'
    echo '  build:    Build SlackBuilds'
    echo '  help:     Show this message'
    echo
    echo '  <repo>:   Repository name which will cat with environment $BASEURL'
    echo '  <url>:    Full repository url'
    echo
    echo '  $BASEURL: Default is https://github.com/slackwarecn-slackbuilds'
    echo '            you can specfied this with custom base url'
  ) >&2
}

# ==============================================================================
# Check if this is an url
# ==============================================================================
function isUrl ()
{
  if ( echo $1 | grep -oP $URLREG 2>&1 >/dev/null )
  then
    return 0
  fi

  return 1
}

# ==============================================================================
# Check if url is reachable
# ==============================================================================
function reachableUrl ()
{
  if ( wget --spider -q $1 )
  then
    return 0
  fi

  return 1
}

# ==============================================================================
# Check if sub module is exist
# ==============================================================================
function existMod ()
{
  local cwd=$(pwd)
  local ret=0

  cd $RDIR

  if [[ -n $1 ]] && ( git submodule status | grep $1 2>&1 >/dev/null )
  then
    ret=0
  else
    ret=1
  fi

  cd $cwd

  return $ret
}

# ==============================================================================
# Init this repository
# ==============================================================================
function init ()
{
  local ret=0

  cd $RDIR

  echo -n "II: Try to sync this repository  ... "
  if ( git pull -q origin master \
        && git submodule -q update --init --recursive )
  then
    echo 'ok'
  else
    echo 'failed'
    echo "EE: Failed to sync this repository." >&2
    ret=1
  fi

  return $ret
}

# ==============================================================================
# Add sub modules
# ==============================================================================
function add
{
  local ARGS=( $@ )
  local url=''
  local mod=''
  local dir=''
  local ret=0

  cd $RDIR

  for arg in ${ARGS[@]}
  do
    mod=${mod%/}

    # Get url of repository
    url=$arg
    if ! ( isUrl $arg )
    then
      url="${BASEURL}/${arg}"
    fi

    echo -n "II: Try to add \"${url}\" ... "

    # Skip if url is not reachable
    if ! ( reachableUrl $url )
    then
      echo 'skip'
      echo "EE: Can not get \"${BASEURL}\", skip." >&2
      continue
    fi

    mod=$(basename $url)
    dir="${RDIR}/${mod}"

    # Add url as git sub module
    if ( existMod $mod)
    then
      echo 'skip'
      echo 'WW: Sub module already exist, skip.' >&2
      continue
    fi

    cd $RDIR

    if ( git submodule -q add $url $mod \
          && cd $dir \
          && git checkout -q $(git show --pretty='format:%H' --no-patch HEAD) )
    then
      echo 'ok'
      okjobs=( ${okjobs[@]} $url )
    else
      echo 'failed'
      echo "EE: Failed to add sub module \"${mod}\"." >&2
      ret=1
    fi
  done

  return $ret
}

# ==============================================================================
# Delete sub modules
# ==============================================================================
function del ()
{
  local MODS=( $@ )
  local ret=0

  cd $RDIR

  for mod in ${MODS[@]}
  do
    mod=${mod%/}

    # TODO May cause error here in some case {
    if ( isUrl $mod )
    then
      mod=$(basename $mod)
    fi
    # }

    echo -n "II: Try to delete \"${mod}\" ... "

    if ! ( existMod $mod )
    then
      echo 'skip'
      echo "WW: Sub module \"${mod}\" not exist, skip." >&2
      continue
    fi

    if ( git submodule -q deinit -f $mod \
          && git config -f .gitmodules --remove-section "submodule.${mod}" \
          && git add .gitmodules \
          && git rm -q --cached $mod \
          && rm -rf $mod \
          && rm -rf .git/modules/${mod} )
    then
      echo 'ok'
      okjobs=( ${okjobs[@]} $mod )
    else
      echo 'failed'
      echo "EE: Failed to delete \"${mod}\"." >&2
      ret=1
    fi
  done

  return $ret
}

# ==============================================================================
# Add sub modules
# ==============================================================================
function sync ()
{
  local mods=( $@ )
  local dir=''
  local pcid=''
  local ccid=''
  local ret=0

  cd $RDIR

  if [[ 0 -eq ${#mods[@]} ]]
  then
    mods=( $(git submodule -q foreach 'basename $(pwd)') )
  fi

  init

  for mod in ${mods[@]}
  do
    mod=${mod%/}
    dir="${RDIR}/${mod}"

    echo -n "II: Try to sync \"${mod}\" ... "

    cd $dir

    if [[ ! -d $dir ]]
    then
      echo 'skip'
      echo "WW: Sub module's directory \"${dir}\" not exist, skip." >&2
      continue
    fi

    pcid=$(git show --pretty='format:%H' --no-patch HEAD)
    if ( git clean -q -dxf \
          && git checkout -q master \
          && git pull -q origin master \
          && git checkout -q $(git show --pretty='format:%H' --no-patch HEAD) )
    then
      ccid=$(git show --pretty='format:%H' --no-patch HEAD)
      if [[ $pcid == $ccid ]]
      then
        echo 'unchange'
      else
        echo 'ok'
        okjobs=( ${okjobs[@]} $mod )
      fi
    else
      echo 'failed'
      echo "EE: Failed to sync \"${mod}\"." >&2
      ret=1
    fi
  done

  return $ret
}

# ==============================================================================
# Undo changes
# ==============================================================================
function undo ()
{
  local ret=0

  cd $RDIR

  echo -n "II: Try to undo changes ... "
  # FIXME Here need to clean direcotry of unregister sub modules
  #       git-clean won't do this, I need to hack this.
  if ( git reset -q --hard HEAD \
        && git clean -q -dxf )
  then
    echo 'ok'
  else
    echo 'failed'
    ( echo 'EE: Failed to undo changes ... '
      echo 'EE: You may need to run:'
      echo 'EE:   rm -rf *'
      echo 'EE:   git checkout .'
      echo 'EE:   ./sbm init'
    ) >&2
    ret=1
  fi

  return $ret
}

# ==============================================================================
# Build SlackBuilds
# ==============================================================================
function build ()
{
  local builds=( $@ )
  local arch=''
  local dir=''
  local slackbuild=''
  local info=''
  local file=''
  local url=( )
  local md5=( )
  local index=0

  if [[ -z $arch ]]
  then
    case "$(uname -m)" in
      i?86) arch='i486' ;;
      arm*) arch='arm' ;;
      *)    arch=$(uname -m) ;;
    esac
  fi

  cd $RDIR

  for build in ${builds[@]}
  do
    build=${build%/}
    dir="${RDIR}/${build}"
    slackbuild="${dir}/${build}.SlackBuild"
    info="${dir}/${build}.info"

    if [[ ! -d $dir ]]
    then
      echo "EE: Can not found \"${build}\", skip." >&2
      continue
    fi

    cd $dir

    echo -n "II: Source \"${build}.info\" ... "
    if [[ ! -r $info ]]
    then
      echo 'failed'
      echo "EE: Can not read \"${build}.info\", skip." >&2
      continue
    fi
    source $info

    if [[ '' == $DOWNLOAD_x86_64 ]]
    then
      url=( $DOWNLOAD )
      md5=( $MD5SUM )
    else
      case $arch in
        *86)
          url=( $DOWNLOAD )
          md5=( $MD5SUM )
          ;;
        *64)
          url=( $DOWNLOAD_x86_64 )
          md5=( $MD5SUM_x86_64 )
          ;;
      esac
    fi

    if [[ 0 -eq ${#url[@]} ]]
    then
      echo 'failed'
      echo "EE: Miss DOWNLOAD or DOWNLOAD_x86_64 in \"${build}.info\", skip." >&2
      continue
    fi

    if [[ ${#url[@]} != ${#md5[@]} ]]
    then
      echo 'failed'
      echo "EE: In \"${build}\" url and source not matched, skip." >&2
      continue
    fi

    echo 'ok'

    for ((index = 0; index < ${#url[@]}; ++index))
    do
      echo -n "II:: Download ${url[$index]} ..."
      if ( wget -qN ${url[$index]} )
      then
        echo 'ok'
      else
        echo 'failed'
        echo "EE: Failed to download ${url[$index]}, skip." >&2
        continue
      fi
    done

    for ((index = 0; index < ${#md5[@]}; ++index))
    do
      expectmd5=${md5[$index]}
      file=$(basename ${url[$index]})
      curmd5=$(md5sum $file | awk '{print $1}')

      echo -n "Verify file \"${file}\" ... "
      if [[ $expectmd5 != $curmd5 ]]
      then
        echo "failed"
        ( echo "Error \"${file}\":"
          echo -e "\texpect md5 \"${expectmd5}\""
          echo -e "\tcurrent md5 \"${curmd5}\""
        ) >&2
      else
        echo "ok"
      fi
    done

    echo "II: Try to build \"${build}\" ..."
    command $slackbuild
  done
}

# ==============================================================================
# Write log
# ==============================================================================
function log ()
{
  local legalActionReg='^(add)|(del(ete)?)|(sync)$'
  local curtime=$(date +'%Y-%m-%d %T %Z')
  local action=$1
  shift

  if [[ 0 -gt $# ]] || ! ( echo $action | grep -oP $legalActionReg 2>&1 >/dev/null )
  then
    return 0
  fi

  if [[ 0 -eq ${#okjobs[@]} ]]
  then
    return 0
  fi

  if [[ 'del' == $action ]]
  then
    action='delete'
  fi

  ( echo "$curtime"
    echo
    for job in ${okjobs[@]}
    do
      echo "  - ${action} ${job}"
    done
    echo
  ) >>$LOGFILE

  return $?
}

# ==============================================================================
# MAIN
# ==============================================================================
# function main ()
{
  doAdd=0
  doDel=0
  doSync=0
  doInit=0
  action=''
  gstat=''
  ret=0

  if [[ 0 -eq $# ]]
  then
    action='help'
  else
    action=$1
    shift
  fi

  cd $RDIR

  case $action in
    add)
      doAdd=1
      ;;
    del|delete)
      doDel=1
      ;;
    sync)
      doSync=1
      ;;
    init)
      doInit=1
      ;;
    undo)
      undo
      exit $?
      ;;
    build)
      build $@
      exit $?
      ;;
    *)
      usage
      exit 1
      ;;
  esac

  cd $RDIR

  gstat=$(git status --porcelain --ignore-submodules=dirty 2>/dev/null | tail -n 1)
  if [[ -n $gstat ]]
  then
    echo "WW: Repo is dirty, quit." >&2
    exit 1
  fi

  if [[ 1 -eq $doAdd ]]
  then
    add $@
    ret=$?
  elif [[ 1 -eq $doDel ]]
  then
    del $@
    ret=$?
  elif [[ 1 -eq $doSync ]]
  then
    sync $@
    ret=$?
  elif [[ 1 -eq $doInit ]]
  then
    init
    ret=$?
  fi

  cd $RDIR

  log $action $@

  exit $ret
}