path: root/scripts/repo-add

#!/bin/bash
#
#   repo-add : add a package to a given repo database file
#  
#   Copyright (c) 2006 Aaron Griffin <aaron@archlinux.org>
#  
#   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 2 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, write to the Free Software
#   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, 
#   USA.

myver='3.0.0'

FORCE=0
REPO_DB_FILE=""
TMP_DIR=""

# print usage instructions
usage() {
  echo "repo-add $myver"
  echo
  echo "usage: repo-add <path-to-db> [--force] <package> ..."
  echo
  echo "repo-add will update a package database by reading a package file."
  echo "Multiple packages to add can be specified on the command line."
  echo
  echo "The --force flag will add a 'force' entry to the sync database, which"
  echo "tells pacman to skip its internal version number checking and update"
  echo "the package regardless."
  echo
  echo "Example:"
  echo "  repo-add /path/to/repo.db.tar.gz pacman-3.0.0.pkg.tar.gz"
  echo
}

# return calculated checksum of package
#   arg1 - checksum type
#   arg2 - path to package
get_checksum () {
  case "$(echo "$1" | tr A-Z a-z)" in
    md5) sum=$(md5sum $2); echo ${sum% *} ;;
    sha1) sum=$(sha1sum $2); echo ${sum% *} ;;
    sha256) sum=$(sha256sum $2); echo ${sum% *} ;;
    sha384) sum=$(sha256sum $2); echo ${sum% *} ;;
    sha512) sum=$(sha256sum $2); echo ${sum% *} ;;
  esac
}

# return PKGINFO string for checksum type
#   arg1 - checksum type
checksum_name () {
  case "$(echo "$1" | tr A-Z a-z)" in
    md5) echo "MD5SUM" ;;
    sha1) echo "SHA1SUM" ;;
    sha256) echo "SHA256SUM" ;;
    sha384) echo "SHA384SUM" ;;
    sha512) echo "SHA512SUM" ;;
  esac
}

# test if a file is a repository DB
test_repo_db_file () {
  if [ -f "$REPO_DB_FILE" ]; then
    [ "$(tar tf "$REPO_DB_FILE" | grep -c "/desc")" -gt 0 ] || return 1
  else
    true
  fi
}

# write an entry to the pacman database
#   arg1 - path to package
db_write_entry()
{
  # blank out all variables and set pkgfile
  pkgfile=$(readlink -f $1)
  export pkgname=""
  pkgver=""
  pkgdesc=""
  url=""
  builddate=""
  packager=""
  csize=""
  size="" 
  _groups=""
  _depends=""
  _backups=""
  _licenses=""
  _replaces=""
  _provides=""
  _conflicts=""

  OLDIFS="$IFS"
  # IFS (field seperator) is only the newline character
  IFS="
"

  # read info from the zipped package
  for i in $(tar xOf "$pkgfile" .PKGINFO | grep -v "^#" |sed 's|\(\w*\)\s*=\s*\(.*\)|\1="\2"|'); do
    eval "${i}"
    case "$i" in
      group=*) _groups="$_groups $group" ;;
      depend=*) _depends="$_depends $depend" ;;
      backup=*) _backups="$_backups $backup" ;;
      license=*) _licenses="$_licenses $license" ;;
      replaces=*) _replaces="$_replaces $replaces" ;;
      provides=*) _provides="$_provides $provides" ;;
      conflict=*) _conflicts="$_conflicts $conflicts" ;;
    esac
  done

  IFS=$OLDIFS

  # get compressed size of package
  csize="$(du -b -L $pkgfile | cut -f1)"

  cd $gstmpdir

  # ensure $pkgname and $pkgver variables were found
  if [ -z "$pkgname" -o -z "$pkgver" ]; then
    echo "   error: invalid package file"
    return 1
  fi 

  # remove any other package in the DB with same name
  for existing in *; do
    if [ "${existing%-*-*}" = "$pkgname" ]; then
      echo ":: removing existing package '$existing'"
      rm -rf $existing
    fi
  done

  # create package directory
  mkdir "$pkgname-$pkgver"
  cd "$pkgname-$pkgver"

  # create desc entry
  echo ":: creating 'desc' db entry"
  echo -e "%FILENAME%\n$(basename $1)\n" >>desc
  echo -e "%NAME%\n$pkgname\n" >>desc
  echo -e "%VERSION%\n$pkgver\n" >>desc
  if [ -n "$pkgdesc" ]; then
    echo -e "%DESC%\n$pkgdesc\n" >>desc
  fi
  if [ -n "$_groups" ]; then
    echo "%GROUPS%" >>desc
    echo $_groups | tr -s ' ' '\n' >>desc
    echo "" >>desc
  fi
  [ -n $csize ] && echo -e "%CSIZE%\n$csize\n" >>desc
  [ -n $size ] && echo -e "%ISIZE%\n$size\n" >>desc

  # compute checksums
  for chk in ${DB_CHECKSUMS[@]}; do
    name="$(checksum_name $chk)"
    echo ":: computing $name checksums"
    if [ -n "$name" ]; then
      echo -e "%$name%\n$(get_checksum $chk $pkgfile)\n" >>desc
    fi
  done

  [ -n "$url" ] && echo -e "%URL%\n$url\n" >>desc
  if [ -n "$_licenses" ]; then
    echo "%LICENSE%" >>desc
    echo $_licenses | tr -s ' ' '\n' >>desc
    echo "" >>desc
  fi
  [ -n "$arch" ] && echo -e "%ARCH%\n$arch\n" >>desc
  [ -n "$builddate" ] && echo -e "%BUILDDATE%\n$builddate\n" >>desc
  [ -n "$packager" ] && echo -e "%PACKAGER%\n$packager\n" >>desc

  if [ -n "$_replaces" ]; then
    echo "%REPLACES%" >>desc
    echo $_replaces | tr -s ' ' '\n' >>desc
    echo "" >>desc
  fi
  [ "$FORCE" = "1" ] && echo -e "%FORCE%\n" >>desc

  # create depends entry
  echo ":: creating 'depends' db entry"
  if [ -n "$_depends" ]; then
    echo "%DEPENDS%" >>depends
    echo $_depends | tr -s ' ' '\n' >>depends
    echo "" >>depends
  fi
  if [ -n "$_conflicts" ]; then
    echo "%CONFLICTS%" >>depends
    echo $_conflicts | tr -s ' ' '\n' >>depends
    echo "" >>depends
  fi
  if [ -n "$_provides" ]; then
    echo "%PROVIDES%" >>depends
    echo $_provides | tr -s ' ' '\n' >>depends
    echo "" >>depends
  fi

  # preserve the modification time
  touch -r "$pkgfile" desc depends
} # end db_write_entry

# PROGRAM START

# check for help flags
if [ "$1" = "-h" -o "$1" = "--help" ]; then
  usage
  exit 0
fi

# check for correct number of args
if [ $# -lt 2 ]; then
  usage
  exit 1
fi

# source system and user makepkg.conf
if [ -r /etc/makepkg.conf ]; then
  source /etc/makepkg.conf
else
  echo "ERROR: /etc/makepkg.conf not found. Can not continue." >&2
  exit 1 # $E_CONFIG_ERROR # TODO: error codes
fi

if [ -r ~/.makepkg.conf ]; then
  source ~/.makepkg.conf
fi

# main routine
if [ $# -gt 1 ]; then
  gstmpdir=$(mktemp -d /tmp/gensync.XXXXXXXXXX) || (\
  echo "cannot create temp directory for database building"; \
  exit 1)

  success=0
  # parse arguements
  for arg in $@; do
    if [ "$arg" == "--force" -o "$arg" == "-f" ]; then
      FORCE=1
    elif [ -z "$REPO_DB_FILE" ]; then
      REPO_DB_FILE="$(readlink -f $arg)"
      if ! test_repo_db_file; then
        echo "error: repository file '$REPO_DB_FILE' is not a proper pacman db"
        exit 1
      elif [ -f "$REPO_DB_FILE" ]; then
        echo ":: extracting database to a temporary location"
        tar xf "$REPO_DB_FILE" -C "$gstmpdir"
      fi
    else
      if [ -f "$arg" ]; then
        if ! tar tf "$arg" .PKGINFO 2>&1 >/dev/null; then
          echo "error: '$arg' is not a package file, skipping"
        else
          echo ":: adding package '$arg'"

          this_dir="$(pwd)"
          if db_write_entry "$arg"; then
            success=1
          fi
          cd $this_dir
        fi
      else
        echo "error: package '$arg' not found"
      fi
    fi
  done

  # if all operations were a success, rezip database
  if [ "$success" = "1" ]; then
    echo ":: creating updated database file ${REPO_DB_FILE}"
    cd $gstmpdir
    if [ -n "$(ls)" ]; then
      [ -f "${REPO_DB_FILE}.old" ] && rm "${REPO_DB_FILE}.old"
      [ -f "$REPO_DB_FILE" ] && mv "$REPO_DB_FILE" "${REPO_DB_FILE}.old"
      case "$DB_COMPRESSION" in
        gz) tar c * | gzip -9 >$REPO_DB_FILE ;;
        bz2) tar c * | bzip2 -9 >$REPO_DB_FILE ;;
        *) echo "warning: no compression set"
        tar c * >$REPO_DB_FILE;;
      esac
    fi
  else
    echo ":: no packages modified, nothing to do"
  fi
fi

# remove the temp directory used to unzip
[ -d "$gstmpdir" ] && rm -rf $gstmpdir

# vim: set ts=2 sw=2 noet: