1

## library functions

     2

     3

LOG_DEBUG=

     4

LOG_CMD=y

     5

LOG_INFO=y

     6

LOG_WARN=y

     7

LOG_ERROR=y

     8

     9

function log_debug () {

    10

    [ $LOG_DEBUG ] && echo "... $@" >&2

    11

    12

    return 0

    13

}

    14

    15

function log_debugf () {

    16

    local fmt=$1; shift 1

    17

    18

    [ $LOG_DEBUG ] && printf "... $fmt\n"  "$@"  >&2

    19

    20

    return 0

    21

}

    22

    23

function log_info () {

    24

    [ $LOG_INFO ] && echo "--- $@" >&2

    25

    26

    return 0

    27

}

    28

    29

function log_cmd () {

    30

    [ $LOG_CMD ] && echo ">>> $@" >&2

    31

    32

    return 0

    33

}

    34

    35

function log_warn () {

    36

    [ $LOG_WARN ] && echo "XXX $@" >&2

    37

    38

    return 0

    39

}

    40

    41

function log_error () {

    42

    [ $LOG_ERROR ] && echo "!!! $@" >&2

    43

    44

    return 0

    45

}

    46

    47

function die () {

    48

    log_error "$@"

    49

    50

    exit 1

    51

}

    52

    53

# Execute command verbosely, and exit on failure

    54

CMD_MOCK=

    55

CMD_PROMPT=y

    56

    57

function cmd () {

    58

    log_cmd "$@"

    59

    60

    [ $CMD_MOCK ] && return 0

    61

    62

    "$@" # return $?

    63

}

    64

    65

function cmd_confirm () {

    66

    [ $CMD_PROMPT ] && read -p "Confirm: $* ? "

    67

    68

    cmd "$@"

    69

}

    70

    78

function expand_MB () {

    79

    local size=${1^}

    80

    81

    case ${size: -1} in 

    82

        G)

    83

            size=$(( ${size%G} * 1024))

    84

    85

            ;;

    86

    esac

    87

    88

    echo $size

    89

}

    90

    91

function extract_iso () {

    92

    iso=$1

    93

    dst=$2

    94

    95

    [ ! -r "$iso" ] && die "Given .iso is not readable: $iso"

    96

    [ -z "$dst" ] && die "Must give destination: $dst"

    97

    [ -e "$dst" ] && die "Given destination already exists: $dst"

    98

    99

    # temporary mount

   100

    mnt=$(mktemp -d mnt.XXXX)

   101

   102

    # clean on exit

   103

    function cleanup () {

   104

        if mountpoint -q $mnt; then

   105

           sudo umount $mnt

   106

        fi

   107

   108

        [ -d $mnt ] && rmdir $mnt

   109

   110

        return $1

   111

    }

   112

   113

    # loop-mount

   115

   116

    # copy

   117

    cmd cp -rd $mnt $dst || cleanup 1

   118

   119

    # done, cleanup

   120

    cleanup 0

   121

}

   122

   123

function my_md5sum () {

   124

    /usr/bin/md5sum $1 | (

   125

        read md5sum path

   126

        echo $md5sum

   127

    )

   128

}

   129

   130

function expand_line () {

   131

    local line=$1

   132

   133

    # evaluate {...} expressions

   134

    # a slight hack, but it works \o/

   135

    # http://stackoverflow.com/questions/415677/how-to-replace-placeholders-in-a-text-file/7633579#7633579

   136

    line="${line//\\/\\\\}"

   137

    line="${line//\"/\\\"}"

   138

    line="${line//\`/\\\`}"

   139

    line="${line//\$/\\\$}"

   140

    line="${line//{/\${}"   # This is just for vim: } "

   141

   142

    # log_debug "($line)" >&2

   143

    eval "echo \"$line\""

   144

}

   145

   146

function expand_template () {

   147

    local linecount=0

   148

   149

    # read in each line at a time

   150

    while IFS='' read -r line; do

   151

        linecount=$((linecount + 1))

   152

   153

        if [ "${line:0:1}" == "#" ]; then

   154

            # ignore comments; pass through as-is

   155

            echo "$line"

   156

   157

        else

   158

            expand_line "$line" || die "Error at $tpl:$linecount: $line"

   159

        fi

   160

   161

    done

   162

}

   163

   164

function expand_file () {

   165

    local src=$1

   166

    local dst=$2

   167

   168

    log_info "expand: file $src -> $dst"

   169

   170

    expand_template < $src > $dst

   171

}

   172

   173

# Recursive expand_template files from src -> dst

   174

# XXX: not used

   175

function expand_tree () {

   176

    local src=$1

   177

    local dst=$2

   178

    local filter=${3:-'*'}

   179

   180

    log_info "expand: tree $src/$filter -> $dst"

   181

   182

    [ ! -d $dst ] && cmd mkdir $dst

   183

   184

    for path in ${src}/${filter}; do

   185

        local name=$(basename $path)

   186

        local target=$dst/$name

   187

   188

        log_debug "expand_tree: $src: path=$path, name=$name, target=$target"

   189

   190

        if [ -d $path ]; then

   191

            log_debug "expand_tree: $src: tree $name -> $target"

   192

            expand_tree $path $target "$filter"

   193

   194

        elif [ -f $path ]; then

   195

            log_debug "expand_tree: $src: file $name -> $target"

   196

            expand_file $path $target

   197

   198

        else

   199

            log_warn "expand_tree: $src: ignore $name"

   200

   201

        fi

   202

    done

   203

}