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=

    47

LOG_DEBUG=

    48

LOG_CMD=

    49

    50

UPDATE_FORCE=

    51

UPDATE_NOOP=

    52

UPDATE_DIFF=

    53

SERIAL_NOUPDATE=

    54

COMMIT_SKIP=

    55

COMMIT_FORCE=

    56

COMMIT_MSG=' '

    57

    58

DEPLOY_SKIP=

    59

    60

function help_args {

    61

    local prog=$1

    62

    63

    cat <<END

    64

Usage: $prog [options]

    65

    66

General:

    67

    -h      display this help text

    68

    -d DIR  datadir

    69

    70

Logging:    

    71

    -q      quiet

    72

    -v      verbose

    73

    -D      debug

    74

    -V      debug commands

    75

    76

Updates:

    77

    -p      show changes

    78

    -F      force-updates without checking src mtime

    79

    -S      do not update serial

    80

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

    81

    82

Commit    

    83

    -C      do not commit changes

    84

    -c      commit changes

    85

    -m MSG  commit message

    86

END

    87

}

    88

    89

function parse_args {

    90

    OPTIND=1

    91

    92

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

    93

        case $opt in

    94

            h)  

    95

                help_args $0

    96

                exit 0

    97

            ;;

    98

    99

            d)  ROOT="$OPTARG" ;;

   100

   101

            q)  

   102

                LOG= 

   103

                LOG_WARN=

   104

                ;;

   105

   106

            v)  LOG_INFO=y ;;

   107

            D)  

   108

                LOG_DEBUG=y

   109

                LOG_INFO=y

   110

                ;;

   111

            V)  LOG_CMD=y ;;

   112

   113

            p)  UPDATE_DIFF=y ;;

   114

            F)  UPDATE_FORCE=y ;;

   115

            S)  SERIAL_NOUPDATE=y ;;

   116

   117

            n)  

   118

                UPDATE_NOOP=y 

   119

                # implies -Sp

   120

                UPDATE_DIFF=y

   121

                SERIAL_NOUPDATE=y

   122

                COMMIT_SKIP=y

   123

                DEPLOY_SKIP=y

   124

                ;;

   125

   126

            C)  COMMIT_SKIP=y ;;

   127

            c)  COMMIT_FORCE=y ;;

   128

            m)  COMMIT_MSG="$OPTARG" ;;

   129

   130

            ?)  

   131

                die 

   132

            ;;

   133

        esac

   134

   135

    done

   136

}

   137

   138

## lib

   139

function log_msg {

   140

    echo "$*" >&2

   141

}

   142

   143

function log_color {

   144

    local code=$1; shift

   145

   146

    if [ $IS_TTY ]; then

   147

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

   148

    else

   149

        echo "$*" >&2

   150

    fi

   151

}

   152

   153

function log_error {

   154

    log_color 31 "$*"

   155

}

   156

   157

function log_warn {

   158

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

   159

}

   160

   161

function log {

   162

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

   163

}

   164

   165

function log_info {

   166

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

   167

}

   168

   169

function log_debug {

   170

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

   171

}

   172

   173

function log_cmd {

   174

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

   175

}

   176

   177

# XXX: broken

   178

function log_stack {

   179

    local level=1

   180

   181

    while info=$(caller $level); do

   182

        echo $info | read line sub file

   183

   184

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

   185

   186

        level=$(($level + 1))

   187

    done

   188

}

   189

   190

function fail {

   191

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

   192

   193

    log_error "$func: $*"

   194

   195

    exit 2

   196

}

   197

   198

function die {

   199

    log_error "$*"

   200

    exit 1

   201

}

   202

   203

function cmd {

   204

    log_cmd "$@"

   205

   206

    "$@" || die "Failed"

   207

}

   208

   209

function run_cmd {

   210

    local msg=$1; shift

   211

   212

    log_info "$msg... "

   213

   214

    cmd "$@"

   215

}

   216

   217

function indent () {

   218

    local indent=$1; shift

   219

   220

    log_cmd "$@"

   221

   222

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

   223

   224

    return ${PIPESTATUS[0]}

   225

}

   226

   227

function abspath () {

   228

    echo "$ROOT/$1"

   229

}

   230

   231

## hg

   232

function hg {

   233

    local repo=$REPO; shift

   234

   235

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

   236

}

   237

   238

function hg_modified {

   239

    hg id | grep -q '+'

   240

}

   241

   242

function hg_user {

   243

    if [ ${SUDO_USER:-} ]; then

   244

        echo '-u' "$SUDO_USER"

   245

   246

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

   247

        log_debug "using .hgrc user"

   248

        echo ''

   249

   250

    else

   251

        echo '-u' "$USER"

   252

    fi

   253

}

   254

   255

function hg_diff {

   256

    hg diff

   257

}

   258

   259

function hg_commit {

   260

    local msg=$2

   261

    local user_opt=$(hg_user)

   262

   263

    log_debug "commit: $user_opt: $msg"

   264

    hg commit $user_opt -m "$msg"

   265

}

   266

   267

   268

## functions

   269

function check_update {

   270

    # target

   271

    local dst=$1; shift

   272

   273

    log_debug "check_update: $dst"

   274

   275

    # need update?

   276

    local update=

   277

   278

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

   279

        log_debug "  update forced"

   280

        update=y

   281

    fi

   282

   283

    # check deps

   284

    for dep in "$@"; do

   285

        # don't bother checking if already figured out

   286

        [ $update ] && continue

   287

   288

        # check

   289

   290

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

   291

            log_debug "  changed: $dep"

   292

            update=y

   293

        fi

   294

    done

   295

   296

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

   297

   298

    # return

   299

    [ $update ]

   300

}

   301

   302

function do_update {

   303

    local dst=$1; shift

   304

    local tmp=$dst.new

   305

   306

    log_debug "update: $dst"

   307

    cmd "$@" > $ROOT/$tmp

   308

   309

    # compare

   310

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

   311

        log_debug "  changes:"

   312

   313

        # terse

   314

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

   315

    fi

   316

   317

    if [ $UPDATE_NOOP ]; then

   318

        # cleanup

   319

        log_debug "  no-op"

   320

   321

        cmd rm $ROOT/$tmp

   322

    else

   323

        # commit

   324

        log_debug "  update"

   325

   326

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

   327

    fi

   328

}

   329

   330

function update {

   331

    local dst=$1; shift;

   332

   333

    local sep=

   334

    local dep=()

   335

    local cmd=()

   336

   337

    for arg in "$@"; do

   338

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

   339

            sep=y

   340

        fi

   341

   342

        if [ $sep ]; then

   343

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

   344

        else

   345

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

   346

        fi

   347

    done

   348

   349

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

   350

   351

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

   352

}

   353

   354

function check_link {

   355

    local lnk=$1

   356

    local tgt=$2

   357

   358

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

   359

}

   360

   361

function do_link {

   362

    local lnk=$1

   363

    local tgt=$2

   364

   365

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

   366

}

   367

   368

## hosts

   369

# copy hosts input zone verbatim

   370

function copy_hosts {

   371

    local zone=$1

   372

    local base=$2

   373

   374

    log_debug "base: $base"

   375

   376

    if check_update $zone $base; then

   377

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

   378

   379

        do_update $zone \

   380

            cat $ROOT/$base

   381

    else

   382

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

   383

    fi

   384

}

   385

   386

# generate hosts zone from input zone

   387

function update_hosts {

   388

    local zone=$1; shift

   389

    local base=$1; shift

   390

   391

    if check_update $zone $base; then

   392

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

   393

   394

        do_update $zone \

   396

    else

   397

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

   398

    fi

   399

}

   400

   401

## actions

   402

# serial

   403

function update_serial {

   404

    local zone=$1; shift

   405

   406

    local serial=$SERIALS/$zone.serial

   407

   408

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

   409

   410

    log_info "Updating $serial..."

   411

   413

   414

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

   415

   416

    log_debug "  $old -> $new"

   417

}

   418

   419

function link_serial {

   420

    local zone=$1

   421

    local base=$2

   422

   423

    local lnk=$SERIALS/$zone.serial

   424

    local tgt=$SERIALS/$base.serial

   425

   426

    if check_link $lnk $tgt; then

   427

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

   428

   429

        do_link $lnk $tgt

   430

   431

    else

   432

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

   433

    fi

   434

}

   435

   436

# zone

   437

function copy_zone {

   438

    local view=$1

   439

    local zone=$2

   440

    local base=${3:-$zone}

   441

   442

    local out=$ZONES/$view/$zone

   443

    local src=$DATA/$base

   444

   445

    if check_update $out $src; then

   446

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

   447

   448

        do_update $out cat $ROOT/$src

   449

    else

   450

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

   451

    fi

   452

}

   453

   454

function update_zone {

   455

    local view=$1

   456

    local zone=$2

   457

    local base=${3:-$zone}

   458

   459

    local out=$ZONES/$view/$zone

   460

    local src=$DATA/$base.zone

   461

    local lnk=$ZONES/$base

   462

    local serial=$SERIALS/$base.serial

   463

   464

    log_debug "$out: from src $src"

   465

   466

    if [ ! -e $src ]; then

   467

        fail "Missing source: $src"

   468

   469

    elif check_update $out $src $serial; then

   470

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

   471

   472

        do_update $out \

   474

                --serial $ROOT/$serial              \

   475

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

   476

                --expand view=$view

   477

    else

   478

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

   479

    fi

   480

}

   481

   482

function link_zone {

   483

    local view=$1

   484

    local zone=$2

   485

    local base=${3:-$zone}

   486

   487

    local out=$ZONES/$view/$zone