1

#!/bin/bash

     2

# vim: set ft=sh :

     3

     4

set -ue

     5

     6

ROOT=$(pwd)

     7

     8

# resolve $0

     9

self=$0

    10

while [ -L $self ]; do

    11

    tgt=$(readlink $self)

    12

    13

    if [ "${tgt:0:1}" == "/" ]; then

    14

        self=$tgt

    15

    else

    16

        self=$(dirname $self)/$tgt

    17

    fi

    18

done

    19

    20

# bin dir

    21

BIN=$(dirname $self)

    22

    23

# data files

    24

DATA=settings

    25

ZONES=zones

    26

SERIALS=$DATA

    27

REPO=

    28

    29

# data args

    30

PROCESS_ARGS='--input-charset latin-1'

    31

    32

FORWARD_MX=mail

    33

REVERSE_ZONE=194.197.235

    34

REVERSE_DOMAIN=paivola.fi

    35

    36

# external progs

    37

NAMED_CHECKZONE=/usr/sbin/named-checkzone

    38

HG=/usr/bin/hg

    39

RNDC=/usr/sbin/rndc

    40

    41

## options

    42

IS_TTY=

    43

    44

LOG_WARN=y

    45

LOG=y

    46

LOG_INFO=y

    47

LOG_SKIP=

    48

LOG_DEBUG=

    49

LOG_CMD=

    50

    51

UPDATE_FORCE=

    52

UPDATE_NOOP=

    53

UPDATE_DIFF=

    54

SERIAL_NOUPDATE=

    55

COMMIT_SKIP=

    56

COMMIT_FORCE=

    57

COMMIT_MSG=' '

    58

    59

DEPLOY_SKIP=

    60

    61

function help_args {

    62

    local prog=$1

    63

    64

    cat <<END

    65

Usage: $prog [options]

    66

    67

General:

    68

    -h      display this help text

    69

    -d DIR  datadir

    70

    71

Logging:    

    72

    -q      quiet

    73

    -v      verbose

    74

    -D      debug

    75

    -V      debug commands

    76

    77

Updates:

    78

    -p      show changes

    79

    -F      force-updates without checking src mtime

    80

    -S      do not update serial

    81

    -n      no-op/mock-update; don't actually change anything; implies -SpC

    82

    83

Commit    

    84

    -C      do not commit changes

    85

    -c      commit changes

    86

    -m MSG  commit message

    87

END

    88

}

    89

    90

function parse_args {

    91

    OPTIND=1

    92

    93

    while getopts 'hd:qvDVpFSnCcm:' opt "$@"; do

    94

        case $opt in

    95

            h)  

    96

                help_args $0

    97

                exit 0

    98

            ;;

    99

   100

            d)  ROOT="$OPTARG" ;;

   101

   102

            q)  

   103

                LOG= 

   104

                LOG_WARN=

   105

                ;;

   106

   107

            v)  LOG_SKIP=y ;;

   108

            D)  

   109

                LOG_DEBUG=y

   110

                LOG_INFO=y

   111

                ;;

   112

            V)  LOG_CMD=y ;;

   113

   114

            p)  UPDATE_DIFF=y ;;

   115

            F)  UPDATE_FORCE=y ;;

   116

            S)  SERIAL_NOUPDATE=y ;;

   117

   118

            n)  

   119

                UPDATE_NOOP=y 

   120

                # implies -Sp

   121

                UPDATE_DIFF=y

   122

                SERIAL_NOUPDATE=y

   123

                COMMIT_SKIP=y

   124

                DEPLOY_SKIP=y

   125

                ;;

   126

   127

            C)  COMMIT_SKIP=y ;;

   128

            c)  COMMIT_FORCE=y ;;

   129

            m)  COMMIT_MSG="$OPTARG" ;;

   130

   131

            ?)  

   132

                die 

   133

            ;;

   134

        esac

   135

   136

    done

   137

}

   138

   139

## lib

   140

function log_msg {

   141

    echo "$*" >&2

   142

}

   143

   144

function log_color {

   145

    local code=$1; shift

   146

   147

    if [ $IS_TTY ]; then

   148

        echo $'\e['${code}'m'"$*"$'\e[00m' >&2

   149

    else

   150

        echo "$*" >&2

   151

    fi

   152

}

   153

   154

function log_error {

   155

    log_color '31' "$*"

   156

}

   157

   158

function log_warn {

   159

    [ $LOG_WARN ] && log_color '33' "$*" || true

   160

}

   161

   162

function log {

   163

    [ $LOG ] && log_msg "$*" || true

   164

}

   165

   166

function log_info {

   167

    [ $LOG_INFO ] && log_color '36' "  $*" || true

   168

}

   169

   170

function log_skip {

   171

    [ $LOG_SKIP ] && log_color '1;34' "  $*" || true

   172

}

   173

   174

function log_debug {

   175

    [ $LOG_DEBUG ] && log_color 32 "    $*" || true

   176

}

   177

   178

function log_cmd {

   179

    [ $LOG_CMD ] && log_color 35 "        \$ $*" || true

   180

}

   181

   182

# XXX: broken

   183

function log_stack {

   184

    local level=1

   185

   186

    while info=$(caller $level); do

   187

        echo $info | read line sub file

   188

   189

        log_msg "$file:$lineno $sub()"

   190

   191

        level=$(($level + 1))

   192

    done

   193

}

   194

   205

function fail {

   207

   208

    exit 2

   209

}

   210

   211

function die {

   212

    log_error "$*"

   213

    exit 1

   214

}

   215

   216

function cmd {

   217

    log_cmd "$@"

   218

   219

    "$@" || die "Failed"

   220

}

   221

   222

function run_cmd {

   223

    local msg=$1; shift

   224

   225

    log_info "$msg... "

   226

   227

    cmd "$@"

   228

}

   229

   230

function indent () {

   231

    local indent=$1; shift

   232

   233

    log_cmd "$@"

   234

   235

    "$@" | sed "s/^/$indent/"

   236

   237

    return ${PIPESTATUS[0]}

   238

}

   239

   240

function ensure_dir {

   241

    local dir=$1

   242

   243

    if [ ! -d $ROOT/$dir ]; then

   244

        log_warn "Creating output dir: $dir"

   245

        cmd mkdir $ROOT/$dir

   246

    fi

   247

}

   248

   249

function abspath () {

   250

    echo "$ROOT/$1"

   251

}

   252

   253

## hg

   254

function hg {

   255

    local repo=$REPO; shift

   256

   257

    cmd $HG -R $ROOT/$repo "$@"

   258

}

   259

   260

function hg_modified {

   261

    hg id | grep -q '+'

   262

}

   263

   264

function hg_user {

   265

    if [ ${SUDO_USER:-} ]; then

   266

        echo '-u' "$SUDO_USER"

   267

   268

    elif [ $HOME ] && [ -e $HOME/.hgrc ]; then

   270

        echo ''

   271

   272

    else

   273

        echo '-u' "$USER"

   274

    fi

   275

}

   276

   277

function hg_diff {

   278

    hg diff

   279

}

   280

   281

function hg_commit {

   282

    local msg=$2

   283

    local user_opt=$(hg_user)

   284

   286

    hg commit $user_opt -m "$msg"

   287

}

   288

   289

   290

## functions

   291

function check_update {

   292

    # target

   293

    local dst=$1; shift

   294

   296

   297

    # need update?

   298

    local update=

   299

   310

        update=y

   311

    fi

   312

   313

    # check deps

   314

    for dep in "$@"; do

   315

        # don't bother checking if already figured out

   316

        [ $update ] && continue

   317

   318

        # check

   319

        if [ ! -e $ROOT/$dep ]; then

   320

            fail "$dst: Missing source: $dep"

   321

   322

        elif [ $ROOT/$dst -ot $ROOT/$dep ]; then

   324

            update=y

   327

        fi

   328

    done

   329

   331

   332

    # return

   333

    [ $update ]

   334

}

   335

   336

function do_update {

   337

    local dst=$1; shift

   338

    local tmp=$dst.new

   339

   341

    cmd "$@" > $ROOT/$tmp

   342

   343

    # compare

   344

    if [ -e $ROOT/$dst ] && [ $UPDATE_DIFF ]; then

   346

   347

        # terse

   348

        indent "        " diff --unified=1 $ROOT/$dst $ROOT/$tmp || true

   349

    fi

   350

   351

    if [ $UPDATE_NOOP ]; then

   352

        # cleanup

   354

   355

        cmd rm $ROOT/$tmp

   356

    else

   357

        # commit

   359

   360

        cmd mv $ROOT/$tmp $ROOT/$dst

   361

    fi

   362

}

   363

   364

function update {

   365

    local dst=$1; shift;

   366

    local msg=$1; shift

   367

   368

    local sep=

   369

    local dep=()

   370

    local cmd=()

   371

   372

    for arg in "$@"; do

   373

        if [ $arg == '--' ]; then

   374

            sep=y

   375

        fi

   376

   377

        if [ $sep ]; then

   378

            cmd=("${cmd[@]:-}" "$arg")

   379

        else

   380

            dep=("${dep[@]:-}" "$arg")

   381

        fi

   382

    done

   383

   384

    [ ! $sep ] && fail "Invalid args given: $@"

   385

   386

    if check_update $dst "${dep[@]}"; then

   387

        log_info "$msg..."

   388

   389

        do_update $dst "${cmd[@]}"

   390

    else

   392

   393

    fi

   394

}

   395

   396

function check_link {

   397

    local lnk=$1

   398

    local tgt=$2

   399

   400

    [ ! -e $ROOT/$lnk ] || [ $(readlink $ROOT/$lnk) != $ROOT/$tgt ]

   401

}

   402

   403

function do_link {

   404

    local lnk=$1

   405

    local tgt=$2

   406

   407

    cmd ln -sf $ROOT/$tgt $ROOT/$lnk

   408

}

   409

   410

## hosts

   411

# copy hosts input zone verbatim

   412

function copy_hosts {

   413

    local zone=$1

   414

    local base=$2

   415

   416

    if check_update $zone $base; then

   417

        log_info "Copying hosts $zone <- $base..."

   418

   419

        do_update $zone \

   420

            cat $ROOT/$base

   421

    else

   422

        log_skip "Copying hosts $zone <- $base: not changed"

   423

    fi

   424

}

   425

   426

# generate hosts zone from input zone

   427

function update_hosts {

   428

    local zone=$1; shift

   429

    local base=$1; shift

   430

   431

    if check_update $zone $base; then

   432

        log_info "Generating hosts $zone <- $base..."

   433

   434

        do_update $zone \

   435

            $BIN/process-zone $PROCESS_ARGS $ROOT/$base "$@"

   436

    else

   437

        log_skip "Generating hosts $zone <- $base: not changed"

   438

    fi

   439

}

   440

   441

## actions

   442

# serial

   446

    # read

   447

    local old=$(test -e $ROOT/$serial && cat $ROOT/$serial || echo '')

   448

   449

   450

    cmd $BIN/update-serial $ROOT/$serial

   451

   452

    # read

   453

    local new=$(cat $ROOT/$serial)

   454

   475

}

   476

   477

function link_serial {

   478

    local zone=$1

   479

    local base=$2

   480

   481

    local lnk=$SERIALS/$zone.serial

   482

    local tgt=$SERIALS/$base.serial

   483

   484

    if check_link $lnk $tgt; then

   485

        log_info "Linking $lnk -> $tgt..."

   486