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

    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

   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

   149

    else

   150

        echo "$*" >&2

   151

    fi

   152

}

   153

   154

function log_error {

   156

}

   157

   158

function log_warn {

   160

}

   161

   162

function log {

   163

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

   164

}

   165

   166

function log_info {

   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

   195

function fail {

   196

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

   197

   198

    log_error "$func: $*"

   199

   200

    exit 2

   201

}

   202

   203

function die {

   204

    log_error "$*"

   205

    exit 1

   206

}

   207

   208

function cmd {

   209

    log_cmd "$@"

   210

   211

    "$@" || die "Failed"

   212

}

   213

   214

function run_cmd {

   215

    local msg=$1; shift

   216

   217

    log_info "$msg... "

   218

   219

    cmd "$@"

   220

}

   221

   222

function indent () {

   223

    local indent=$1; shift

   224

   225

    log_cmd "$@"

   226

   227

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

   228

   229

    return ${PIPESTATUS[0]}

   230

}

   231

   241

function abspath () {

   242

    echo "$ROOT/$1"

   243

}

   244

   245

## hg

   246

function hg {

   247

    local repo=$REPO; shift

   248

   249

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

   250

}

   251

   252

function hg_modified {

   253

    hg id | grep -q '+'

   254

}

   255

   256

function hg_user {

   257

    if [ ${SUDO_USER:-} ]; then

   258

        echo '-u' "$SUDO_USER"

   259

   260

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

   261

        log_debug "using .hgrc user"

   262

        echo ''

   263

   264

    else

   265

        echo '-u' "$USER"

   266

    fi

   267

}

   268

   269

function hg_diff {

   270

    hg diff

   271

}

   272

   273

function hg_commit {

   274

    local msg=$2

   275

    local user_opt=$(hg_user)

   276

   277

    log_debug "commit: $user_opt: $msg"

   278

    hg commit $user_opt -m "$msg"

   279

}

   280

   281

   282

## functions

   283

function check_update {

   284

    # target

   285

    local dst=$1; shift

   286

   287

    log_debug "check_update: $dst"

   288

   289

    # need update?

   290

    local update=

   291

   292

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

   293

        log_debug "  update forced"

   294

        update=y

   295

    fi

   296

   297

    # check deps

   298

    for dep in "$@"; do

   299

        # don't bother checking if already figured out

   300

        [ $update ] && continue

   301

   302

        # check

   305

   307

            log_debug "  changed: $dep"

   308

            update=y

   309

        fi

   310

    done

   311

   312

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

   313

   314

    # return

   315

    [ $update ]

   316

}

   317

   318

function do_update {

   319

    local dst=$1; shift

   320

    local tmp=$dst.new

   321

   322

    log_debug "update: $dst"

   323

    cmd "$@" > $ROOT/$tmp

   324

   325

    # compare

   326

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

   327

        log_debug "  changes:"

   328

   329

        # terse

   330

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

   331

    fi

   332

   333

    if [ $UPDATE_NOOP ]; then

   334

        # cleanup

   335

        log_debug "  no-op"

   336

   337

        cmd rm $ROOT/$tmp

   338

    else

   339

        # commit

   340

        log_debug "  update"

   341

   342

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

   343

    fi

   344

}

   345

   346

function update {

   347

    local dst=$1; shift;

   349

   350

    local sep=

   351

    local dep=()

   352

    local cmd=()

   353

   354

    for arg in "$@"; do

   355

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

   356

            sep=y

   357

        fi

   358

   359

        if [ $sep ]; then

   360

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

   361

        else

   362

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

   363

        fi

   364

    done

   365

   366

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

   367

   376

}

   377

   378

function check_link {

   379

    local lnk=$1

   380

    local tgt=$2

   381

   382

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

   383

}

   384

   385

function do_link {

   386

    local lnk=$1

   387

    local tgt=$2

   388

   389

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

   390

}

   391

   392

## hosts

   393

# copy hosts input zone verbatim

   394

function copy_hosts {

   395

    local zone=$1

   396

    local base=$2

   397

   398

    if check_update $zone $base; then

   399

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

   400

   401

        do_update $zone \

   402

            cat $ROOT/$base

   403

    else

   405

    fi

   406

}

   407

   408

# generate hosts zone from input zone

   409

function update_hosts {

   410

    local zone=$1; shift

   411

    local base=$1; shift

   412

   413

    if check_update $zone $base; then

   414

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

   415

   416

        do_update $zone \

   417

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

   418

    else

   420

    fi

   421

}

   422

   423

## actions

   424

# serial

   425

function update_serial {

   426

    local zone=$1; shift

   427

   428

    local serial=$SERIALS/$zone.serial

   431

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

   432

   433

    log_info "Updating $serial..."

   434

   436

   438

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

   439

   440

    log_debug "  $old -> $new"

   441

}

   442

   443

function link_serial {

   444

    local zone=$1

   445

    local base=$2

   446

   447

    local lnk=$SERIALS/$zone.serial

   448

    local tgt=$SERIALS/$base.serial

   449

   450

    if check_link $lnk $tgt; then

   451

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

   452

   453

        do_link $lnk $tgt

   454

   455

    else

   457

    fi

   458

}

   459

   460

# zone

   461

function copy_zone {

   462

    local view=$1

   463

    local zone=$2

   464

    local base=${3:-$zone}

   465

   466

    local out=$ZONES/$view/$zone

   467

    local src=$DATA/$base

   468

   469

    if check_update $out $src; then

   470

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

   471

   474

    else

   476

    fi

   477

}

   478

   479

function update_zone {

   480

    local view=$1

   481

    local zone=$2

   482

    local base=${3:-$zone}

   483

   484

    local out=$ZONES/$view/$zone

   485

    local src=$DATA/$base.zone

   486

    local lnk=$ZONES/$base