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

PROCESS_ZONE=$BIN/process-zone

    23

EXPAND_ZONE=$BIN/expand-zone

    24

UPDATE_SERIAL=$BIN/update-serial

    25

    26

# data files

    27

DATA=settings

    28

ZONES=zones

    29

SERIALS=$DATA

    30

REPO=

    31

    32

# data args

    33

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

    34

    35

FORWARD_MX=mail

    36

REVERSE_ZONE=194.197.235

    37

REVERSE_DOMAIN=paivola.fi

    38

    39

# external progs

    40

NAMED_CHECKZONE=/usr/sbin/named-checkzone

    41

HG=/usr/bin/hg

    42

RNDC=/usr/sbin/rndc

    43

    44

## options

    45

IS_TTY=

    46

    47

LOG_WARN=y

    48

LOG=y

    49

LOG_INFO=

    50

LOG_DEBUG=

    51

LOG_CMD=

    52

    53

UPDATE_FORCE=

    54

UPDATE_NOOP=

    55

UPDATE_DIFF=

    56

SERIAL_NOUPDATE=

    57

COMMIT_SKIP=

    58

COMMIT_FORCE=

    59

COMMIT_MSG=' '

    60

    61

DEPLOY_SKIP=

    62

    63

function help_args {

    64

    local prog=$1

    65

    66

    cat <<END

    67

Usage: $prog [options]

    68

    69

General:

    70

    -h      display this help text

    71

    -d DIR  datadir

    72

    73

Logging:    

    74

    -q      quiet

    75

    -v      verbose

    76

    -D      debug

    77

    -V      debug commands

    78

    79

Updates:

    80

    -p      show changes

    81

    -F      force-updates without checking src mtime

    82

    -S      do not update serial

    83

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

    84

    85

Commit    

    86

    -C      do not commit changes

    87

    -c      commit changes

    88

    -m MSG  commit message

    89

END

    90

}

    91

    92

function parse_args {

    93

    OPTIND=1

    94

    95

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

    96

        case $opt in

    97

            h)  

    98

                help_args $0

    99

                exit 0

   100

            ;;

   101

   102

            d)  ROOT="$OPTARG" ;;

   103

   104

            q)  

   105

                LOG= 

   106

                LOG_WARN=

   107

                ;;

   108

   109

            v)  LOG_INFO=y ;;

   110

            D)  

   111

                LOG_DEBUG=y

   112

                LOG_INFO=y

   113

                ;;

   114

            V)  LOG_CMD=y ;;

   115

   116

            p)  UPDATE_DIFF=y ;;

   117

            F)  UPDATE_FORCE=y ;;

   118

            S)  SERIAL_NOUPDATE=y ;;

   119

   120

            n)  

   121

                UPDATE_NOOP=y 

   122

                # implies -Sp

   123

                UPDATE_DIFF=y

   124

                SERIAL_NOUPDATE=y

   125

                COMMIT_SKIP=y

   126

                DEPLOY_SKIP=y

   127

                ;;

   128

   129

            C)  COMMIT_SKIP=y ;;

   130

            c)  COMMIT_FORCE=y ;;

   131

            m)  COMMIT_MSG="$OPTARG" ;;

   132

   133

            ?)  

   134

                die 

   135

            ;;

   136

        esac

   137

   138

    done

   139

}

   140

   141

## lib

   142

function log_msg {

   143

    echo "$*" >&2

   144

}

   145

   146

function log_color {

   147

    local code=$1; shift

   148

   149

    if [ $IS_TTY ]; then

   150

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

   151

    else

   152

        echo "$*" >&2

   153

    fi

   154

}

   155

   156

function log_error {

   157

    log_color 31 "$*"

   158

}

   159

   160

function log_warn {

   161

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

   162

}

   163

   164

function log {

   165

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

   166

}

   167

   168

function log_info {

   169

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

   170

}

   171

   172

function log_debug {

   173

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

   174

}

   175

   176

function log_cmd {

   177

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

   178

}

   179

   180

# XXX: broken

   181

function log_stack {

   182

    local level=1

   183

   184

    while info=$(caller $level); do

   185

        echo $info | read line sub file

   186

   187

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

   188

   189

        level=$(($level + 1))

   190

    done

   191

}

   192

   193

function fail {

   194

    func=$(caller 1 | cut -d ' ' -f 2)

   195

   196

    log_error "$func: $*"

   197

   198

    exit 2

   199

}

   200

   201

function die {

   202

    log_error "$*"

   203

    exit 1

   204

}

   205

   206

function cmd {

   207

    log_cmd "$@"

   208

   209

    "$@" || die "Failed"

   210

}

   211

   212

function run_cmd {

   213

    local msg=$1; shift

   214

   215

    log_info "$msg... "

   216

   217

    cmd "$@"

   218

}

   219

   220

function indent () {

   221

    local indent=$1; shift

   222

   223

    log_cmd "$@"

   224

   225

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

   226

   227

    return ${PIPESTATUS[0]}

   228

}

   229

   230

function abspath () {

   231

    echo "$ROOT/$1"

   232

}

   233

   234

## hg

   235

function hg {

   237

   238

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

   239

}

   240

   241

function hg_modified {

   243

}

   244

   245

function hg_user {

   246

    if [ ${SUDO_USER:-} ]; then

   247

        echo '-u' "$SUDO_USER"

   248

   249

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

   250

        log_debug "using .hgrc user"

   251

        echo ''

   252

   253

    else

   254

        echo '-u' "$USER"

   255

    fi

   256

}

   257

   258

function hg_diff {

   260

}

   261

   262

function hg_commit {

   263

    local msg=$2

   264

    local user_opt=$(hg_user)

   265

   266

    log_debug "commit: $user_opt: $msg"

   268

}

   269

   270

   271

## functions

   272

function check_update {

   273

    # target

   274

    local dst=$1; shift

   275

   276

    log_debug "check_update: $dst"

   277

   278

    # need update?

   279

    local update=

   280

   281

    if [ ! -e $dst ] || [ $UPDATE_FORCE ]; then

   282

        log_debug "  update forced"

   283

        update=y

   284

    fi

   285

   286

    # check deps

   287

    for dep in "$@"; do

   288

        # don't bother checking if already figured out

   289

        [ $update ] && continue

   290

   291

        # check

   292

   293

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

   294

            log_debug "  changed: $dep"

   295

            update=y

   296

        fi

   297

    done

   298

   299

    [ ! $update ] && log_debug "  up-to-date"

   300

   301

    # return

   302

    [ $update ]

   303

}

   304

   305

function do_update {

   306

    local dst=$1; shift

   307

    local tmp=$dst.new

   308

   309

    log_debug "update: $dst"

   310

    cmd "$@" > $ROOT/$tmp

   311

   312

    # compare

   313

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

   314

        log_debug "  changes:"

   315

   316

        # terse

   317

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

   318

    fi

   319

   320

    if [ $UPDATE_NOOP ]; then

   321

        # cleanup

   322

        log_debug "  no-op"

   323

   324

        cmd rm $ROOT/$tmp

   325

    else

   326

        # commit

   327

        log_debug "  update"

   328

   329

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

   330

    fi

   331

}

   332

   333

function update {

   334

    local dst=$1; shift;

   335

   336

    local sep=

   337

    local dep=()

   338

    local cmd=()

   339

   340

    for arg in "$@"; do

   341

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

   342

            sep=y

   343

        fi

   344

   345

        if [ $sep ]; then

   346

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

   347

        else

   348

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

   349

        fi

   350

    done

   351

   352

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

   353

   354

    check_update $dst "${dep[@]}" && do_update $dst "${cmd[@]}" || true

   355

}

   356

   357

function check_link {

   358

    local lnk=$1

   359

    local tgt=$2

   360

   361

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

   362

}

   363

   364

function do_link {

   365

    local lnk=$1

   366

    local tgt=$2

   367

   368

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

   369

}

   370

   371

## hosts

   372

# copy hosts input zone verbatim

   373

function copy_hosts {

   374

    local zone=$1

   375

    local base=$2

   376

   377

    log_debug "base: $base"

   378

   379

    if check_update $zone $base; then

   380

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

   381

   382

        do_update $zone \

   383

            cat $ROOT/$base

   384

    else

   385

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

   386

    fi

   387

}

   388

   389

# generate hosts zone from input zone

   390

function update_hosts {

   391

    local zone=$1; shift

   392

    local base=$1; shift

   393

   394

    if check_update $zone $base; then

   395

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

   396

   397

        do_update $zone \

   398

            $PROCESS_ZONE $PROCESS_ARGS $ROOT/$base "$@"

   399

    else

   400

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

   401

    fi

   402

}

   403

   404

## actions

   405

# serial

   406

function update_serial {

   407

    local zone=$1; shift

   408

   409

    local serial=$SERIALS/$zone.serial

   410

   411

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

   412

   413

    log_info "Updating $serial..."

   414

   415

    cmd $UPDATE_SERIAL $* $ROOT/$serial

   416

   417

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

   418

   419

    log_debug "  $old -> $new"

   420

}

   421

   422

function link_serial {

   423

    local zone=$1

   424

    local base=$2

   425

   426

    local lnk=$SERIALS/$zone.serial

   427

    local tgt=$SERIALS/$base.serial

   428

   429

    if check_link $lnk $tgt; then

   430

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

   431

   432

        do_link $lnk $tgt

   433

   434

    else

   435

        log_info "Linking $lnk -> $tgt: not changed"

   436

    fi

   437

}

   438

   439

# zone

   440

function copy_zone {

   441

    local view=$1

   442

    local zone=$2

   443

    local base=${3:-$zone}

   444

   445

    local out=$ZONES/$view/$zone

   446

    local src=$DATA/$base

   447

   448

    if check_update $out $src; then

   449

        log_info "Copying $out <- $src..."

   450

   451

        do_update $out cat $ROOT/$src

   452

    else

   453

        log_info "Copying $out <- $src: not changed"

   454

    fi

   455

}

   456

   457

function update_zone {

   458

    local view=$1

   459

    local zone=$2

   460

    local base=${3:-$zone}

   461

   462

    local out=$ZONES/$view/$zone

   463

    local src=$DATA/$base.zone

   464

    local lnk=$ZONES/$base

   465

    local serial=$SERIALS/$base.serial

   466

   467

    log_debug "$out: from src $src"

   468

   469

    if [ ! -e $src ]; then

   470

        fail "Missing source: $src"

   471

   472

    elif check_update $out $src $serial; then

   473

        log_info "Generating $out <- $src..." 

   474

   475

        do_update $out \

   476

            $EXPAND_ZONE $ROOT/$src                 \

   477

                --serial $ROOT/$serial              \

   478

                --expand zones=$(abspath $ZONES)    \

   479

                --expand view=$view

   480

    else

   481

        log_info "Generating $out <- $src: not changed" 

   482

    fi

   483

}

   484

   485

function link_zone {

   486

    local view=$1

   487

    local zone=$2

   488

    local base=${3:-$zone}