pac_manage.sh

#!/bin/env bash
# shellcheck disable=SC2155
# global vars
LIST_FILE="$HOME/package-list"
NO_CONFIRM=""

redrawProgressBar() { # int barsize, int base, int i, int top
    local barsize=$1
    local base=$2
    local current=$3
    local top=$4
    local j=0
    local progress=$(( (barsize * ( current - base )) / (top - base ) ))
    echo -n "["
    for ((j=0; j < progress; j++)); do
        echo -n '='
    done
    echo -n '=>'
    for ((j=progress; j < barsize ; j++)); do
        echo -n ' '
    done
    echo -n "] $current / $top " $'\r'
}
getPackageInfo() {
    local package="$1"
    local description="$(pacman -Qi "$package" | grep Description | cut -d: -f2)"

    local info="$(yaourt -Qe "$package")"
    local repository="$(echo "$info" | cut -d' ' -f1 | cut -d/ -f1)"
    local groups="$(echo "$info" | cut -d' ' -f3-)"
    if [ ${#package} -ge 24 ]; then
        printf "%s # %s\n" "$package" "${repository}${groups}${description}"
    else
        printf "%-24s # %s\n" "$package" "${repository}${groups}${description}"
    fi
}
addSection() {
    local list=($1)
    local section=$2
    local comment=$3
    if [ -z "$comment" ]; then
        echo "${section}=( # {{{" >> "$LIST_FILE"
    else
        echo "${section}=( # {{{ $comment" >> "$LIST_FILE"
    fi
    for i in "${!list[@]}"; do
        local package=${list[i]}
        getPackageInfo "$package" >> "$LIST_FILE"
        redrawProgressBar 50 0 "$i" ${#list[@]}
    done
    echo ") # }}}" >> "$LIST_FILE"
}
filterList() {
    local comments='/^\s*#.*/d'
    local inlinecomments='s/#.*$//'
    local syntax='/\([[:alnum:]]*=(\)\|\(^)\)/d'
    sed -e "$comments" -e "$inlinecomments" -e "$syntax" "$LIST_FILE" \
            | while read -r package; do
        echo "$package"
    done
}
createInitialPackageList() {
    # write package-list
    echo "# package-list for $0" > "$LIST_FILE"
    addSection "" "NEW_PACKAGES" "new packages will be added here"
    addSection "$(pacman -Qqen)" "NATIVE_PACKAGES"
    addSection "$(yaourt -Qema | grep -v ^local | cut -d' ' -f1 | cut -d/ -f2)"\
        "AUR_PACKAGES"
    addSection "$(yaourt -Qema | grep ^local | cut -d' ' -f1 | cut -d/ -f2)"\
        "LOCAL_PACKAGES" "true local packages, not from aur"
    echo "# vim: foldmethod=marker foldmarker={{{,}}}" >> "$LIST_FILE"
    echo ""
    echo "You can now edit your package list!"
}
updatePackageList() {
    for package in $(pacman -Qqe); do
        if ! grep --quiet "^$package" "$LIST_FILE"; then
            echo "adding: $package"
            local line="$(getPackageInfo "$package")"
            if [ -z "$DRY_RUN" ]; then
                sed -i "/NEW_PACKAGES=(/a $line" "$LIST_FILE"
            fi
        fi
    done
}
updatePackageListInfo() {
    local list=($(filterList))
    for i in "${!list[@]}"; do
        local package="${list[i]}"
        local line="$(getPackageInfo "$package")"
        # escape replacement string
        line=$(echo "$line" | sed 's/[\/&]/\\&/g')
        sed -i "s/^$package\( \|$\).*/$line/" "$LIST_FILE"
        redrawProgressBar 50 0 "$i" ${#list[@]}
    done
}
checkDuplicates() {
    local dupes="$(filterList | sort | uniq --repeated)"
    if ! [ -z "$dupes" ]; then
        echo duplicates:
        echo "$dupes"
        echo "Duplicates in list detected, please check this. Aborting.."
        exit 1
    fi
}
cleanMissing() {
    local list=($(filterList))
    for i in "${!list[@]}"; do
        local package="${list[i]}"
        if ! (pacman -Qq "$package" &> /dev/null); then
            echo "removing from list: $package"
            if [ -z "$DRY_RUN" ]; then
                sed -i "/^$package\( \|$\)/d" "$LIST_FILE"
            fi
        fi
        redrawProgressBar 50 0 "$i" ${#list[@]}
    done
}
installMissing() {
    local list=($(filterList))
    for i in "${!list[@]}"; do
        local package="${list[i]}"
        if ! (pacman -Qq "$package" &> /dev/null); then
            echo "installing: $package"
            if [ -z "$DRY_RUN" ]; then
                yaourt -S --needed $NO_CONFIRM "$package"
            fi
        fi
        redrawProgressBar 50 0 "$i" ${#list[@]}
    done
    echo ""
}
removePackages() {
    for package in $(pacman -Qqe); do
        if ! (grep --quiet "^$package" "$LIST_FILE"); then
            echo "removing $package"
            if [ -z "$DRY_RUN" ]; then
                sudo pacman -Rns $NO_CONFIRM "$package"
            fi
        fi
    done
}
function printHelpMessage() {
    cat << EOF
$0 manages your packages in a single file where you can add comments
and group your packages.

    USAGE: $0 [OPTIONS] OPERATION

    Available OPERATIONs:
        remove
            Removes all packages (pacman -Rns) that are installed but not in
            the package-list.

        update
            Update the package list. (New packages are added to the section
            'NEW_PACKAGES')

        install
            Install all missing packages.

        clean
            Removes missing packages from the list.

        update-info
            Update the extra information in the package list.
            [package]  # [repo] [(groups)] [description]
            Notes: Can only update info for installed packages. Replaces the
            whole line. If you want personal comments, add them in a line above
            the package.


    If multiple operations are given they are carried out in the order:
        1. remove
        2. update
        3. install
        4. clean
        5. update-info

    OPTIONS:
    -h --help
        Shows this help message.

    -l --list FILE_NAME
        Set the package-list file name. If the file does not exist it is
        created. (Default: '~/package-list')

    -n --no-confirm
        Don't ask for confirmations. Applys to remove and install operations.

    -d --dry-run
        Just print what would be done.


    USE CASES / EXAMPLES:
        # generate an initial package-list
        >>> $0
        # edit the list to your likings (add comments, reorder entries etc.)
        >>> vim ~/package-list
        # install new packages to your system
        >>> sudo pacman -S gvim colorgcc
        # decide you want to keep them
        # save them to your package-list (1)
        >>> $0 update
        # install some packages
        >>> sudo pacman -S cowsay ponysay
        # test the installed applications and decide you dont want to keep them
        # remove them (restore the state from (1))
        >>> $0 remove
EOF
}
commandLineInterface() {
    while true; do
        case "$1" in
            -h|--help)
                shift
                printHelpMessage "$0"
                exit 0
                ;;
            -l|--list)
                shift
                LIST_FILE="$1"
                shift
                ;;
            -n|--no-confirm)
                shift
                NO_CONFIRM="--noconfirm"
                ;;
            -d|--dry-run)
                shift
                DRY_RUN=1
                ;;
            update)
                shift
                local do_update=1
                ;;
            update-info)
                shift
                local do_update_info=1
                ;;
            clean)
                shift
                local do_clean=1
                ;;
            install)
                shift
                local do_install=1
                ;;
            remove)
                shift
                local do_remove=1
                ;;
            '')
                shift
                break
                ;;
            *)
                echo "unsupported argument: $1"
                exit 1
                shift
        esac
    done
    LIST_FILE="${LIST_FILE/#\~/$HOME}" # expand ~
    if ! [ -f "$LIST_FILE" ]; then
        echo "creating initial package list: '$LIST_FILE'"
        createInitialPackageList
    else
        if [ -z "$do_remove" ] && [ -z "$do_update" ] && [ -z "$do_install" ] \
            && [ -z "$do_clean" ] && [ -z "$do_update_info" ]; then
            printHelpMessage "$0"
            exit 1
        fi
    fi
    checkDuplicates
    if [ -n "$DRY_RUN" ]; then
        echo "NOTE: -d(--dry-run) is set, operations wont do anything"
    fi
    if [ -n "$do_remove" ]; then
        echo "removing packages..."
        removePackages
    fi
    if [ -n "$do_update" ]; then
        echo "updating package-list..."
        updatePackageList
    fi
    if [ -n "$do_install" ]; then
        echo "checking if packages are missing... "
        installMissing
    fi
    if [ -n "$do_clean" ]; then
        echo "cleaning package-list..."
        cleanMissing
    fi
    if [ -n "$do_update_info" ];then
        echo "updating package infos"
        updatePackageListInfo
    fi
    echo "all done!"
}
commandLineInterface "$@"