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

    self=$(dirname $self)/$(readlink $self)

    12

done

    13

    14

# bin dir

    15

BIN=$(dirname $self)

    16

PROCESS_ZONE=$BIN/process-zone

    17

EXPAND_ZONE=$BIN/expand-zone

    18

UPDATE_SERIAL=$BIN/update-serial

    19

    20

# data files

    21

DATA=settings

    22

ZONES=zones

    23

SERIALS=$DATA

    24

    25

# data args

    26

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

    27

    28

FORWARD_MX=mail

    29

REVERSE_ZONE=194.197.235

    30

REVERSE_DOMAIN=paivola.fi

    31

    32

# external progs

    33

NAMED_CHECKZONE=/usr/sbin/named-checkzone

    34

HG=hg

    35

RNDC=/usr/sbin/rndc

    36

    37

## options

    38

IS_TTY=

    39

    40

LOG=y

    41

LOG_INFO=

    42

LOG_DEBUG=

    43

LOG_CMD=

    44

    45

UPDATE_FORCE=

    46

UPDATE_NOOP=

    47

UPDATE_DIFF=

    48

SERIAL_NOUPDATE=

    49

COMMIT_SKIP=

    50

COMMIT_FORCE=

    51

COMMIT_MSG=' '

    52

    53

DEPLOY_SKIP=

    54

    55

function help_args {

    56

    local prog=$1

    57

    58

    cat <<END

    59

Usage: $prog [options]

    60

    61

General:

    62

    -h      display this help text

    63

    -d DIR  datadir

    64

    65

Logging:    

    66

    -q      quiet

    67

    -v      verbose

    68

    -D      debug

    69

    -V      debug commands

    70

    71

Updates:

    72

    -p      show changes

    73

    -F      force-updates without checking src mtime

    74

    -S      do not update serial

    75

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

    76

    77

Commit    

    78

    -C      do not commit changes

    79

    -c      commit changes

    80

    -m MSG  commit message

    81

END

    82

}

    83

    84

function parse_args {

    85

    OPTIND=1

    86

    87

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

    88

        case $opt in

    89

            h)  

    90

                help_args $0

    91

                exit 0

    92

            ;;

    93

    94

            d)  ROOT="$OPTARG" ;;

    95

    96

            q)  LOG= ;;

    97

            v)  LOG_INFO=y ;;

    98

            D)  

    99

                LOG_DEBUG=y

   100

                LOG_INFO=y

   101

                ;;

   102

            V)  LOG_CMD=y ;;

   103

   104

            p)  UPDATE_DIFF=y ;;

   105

            F)  UPDATE_FORCE=y ;;

   106

            S)  SERIAL_NOUPDATE=y ;;

   107

   108

            n)  

   109

                UPDATE_NOOP=y 

   110

                # implies -Sp

   111

                UPDATE_DIFF=y

   112

                SERIAL_NOUPDATE=y

   113

                COMMIT_SKIP=y

   114

                DEPLOY_SKIP=y

   115

                ;;

   116

   117

            C)  COMMIT_SKIP=y ;;

   118

            c)  COMMIT_FORCE=y ;;

   119

            m)  COMMIT_MSG="$OPTARG" ;;

   120

   121

            ?)  

   122

                die 

   123

            ;;

   124

        esac

   125

   126

    done

   127

}

   128

   129

## lib

   130

function log_msg {

   131

    echo "$*" >&2

   132

}

   133

   134

function log_color {

   135

    local code=$1; shift

   136

   137

    if [ $IS_TTY ]; then

   138

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

   139

    else

   140

        echo "$*" >&2

   141

    fi

   142

}

   143

   144

function log_error {

   145

    log_color 31 "$*"

   146

}

   147

   148

function log {

   149

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

   150

}

   151

   152

function log_info {

   153

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

   154

}

   155

   156

function log_debug {

   157

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

   158

}

   159

   160

function log_cmd {

   161

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

   162

}

   163

   164

# XXX: broken

   165

function log_stack {

   166

    local level=1

   167

   168

    while info=$(caller $level); do

   169

        echo $info | read line sub file

   170

   171

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

   172

   173

        level=$(($level + 1))

   174

    done

   175

}

   176

   177

function fail {

   178

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

   179

   180

    log_error "$func: $*"

   181

   182

    exit 2

   183

}

   184

   185

function die {

   186

    log_error "$*"

   187

    exit 1

   188

}

   189

   190

function cmd {

   191

    log_cmd "$@"

   192

   193

    "$@" || die "Failed"

   194

}

   195

   196

function run_cmd {

   197

    local msg=$1; shift

   198

   199

    log_info "$msg... "

   200

   201

    cmd "$@"

   202

}

   203

   204

function indent () {

   205

    local indent=$1; shift

   206

   207

    log_cmd "$@"

   208

   209

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

   210

}

   211

   212

function abspath () {

   213

    echo "$ROOT/$1"

   214

}

   215

   216

## hg

   217

function hg {

   218

    local repo=$r

   219

   220

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

   221

}

   222

   223

function hg_modified {

   224

    local repo=$1

   225

   226

    hg $repo id | grep -q '+'

   227

}

   228

   229

function hg_user {

   230

    if [ ${SUDO_USER:-} ]; then

   231

        echo '-u' "$SUDO_USER"

   232

   233

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

   234

        log_debug "using .hgrc user"

   235

        echo ''

   236

   237

    else

   238

        echo '-u' "$USER"

   239

    fi

   240

}

   241

   242

function hg_diff {

   243

    local repo=$1

   244

   245

    hg $repo diff

   246

}

   247

   248

function hg_commit {

   249

    local repo=$1

   250

    local msg=$2

   251

    local user_opt=$(hg_user)

   252

   253

    log_debug "commit: $user_opt: $msg"

   254

    cmd hg $repo commit $user_opt -m "$msg"

   255

}

   256

   257

   258

## functions

   259

function check_update {

   260

    # target

   261

    local dst=$1; shift

   262

   263

    log_debug "check_update: $dst"

   264

   265

    # need update?

   266

    local update=

   267

   268

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

   269

        log_debug "  update forced"

   270

        update=y

   271

    fi

   272

   273

    # check deps

   274

    for dep in "$@"; do

   275

        # don't bother checking if already figured out

   276

        [ $update ] && continue

   277

   278

        # check

   279

   280

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

   281

            log_debug "  changed: $dep"

   282

            update=y

   283

        fi

   284

    done

   285

   286

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

   287

   288

    # return

   289

    [ $update ]

   290

}

   291

   292

function do_update {

   293

    local dst=$1; shift

   294

    local tmp=$dst.new

   295

   296

    log_debug "update: $dst"

   297

    cmd "$@" > $ROOT/$tmp

   298

   299

    # compare

   300

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

   301

        log_debug "  changes:"

   302

   303

        # terse

   304

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

   305

    fi

   306

   307

    if [ $UPDATE_NOOP ]; then

   308

        # cleanup

   309

        log_debug "  no-op"

   310

   311

        cmd rm $ROOT/$tmp

   312

    else

   313

        # commit

   314

        log_debug "  update"

   315

   316

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

   317

    fi

   318

}

   319

   320

function update {

   321

    local dst=$1; shift;

   322

   323

    local sep=

   324

    local dep=()

   325

    local cmd=()

   326

   327

    for arg in "$@"; do

   328

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

   329

            sep=y

   330

        fi

   331

   332

        if [ $sep ]; then

   333

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

   334

        else

   335

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

   336

        fi

   337

    done

   338

   339

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

   340

   341

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

   342

}

   343

   344

## actions

   345

function update_zone_serial {

   346

    local name=$1; shift

   347

    local file=$SERIALS/$name.serial

   348

   349

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

   350

   351

    log_info "Updating $file..."

   352

   353

    cmd $UPDATE_SERIAL $* $ROOT/$file

   354

   355

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

   356

   357

    log_debug "  $old -> $new"

   358

}

   359

   360

function update_zone_part {

   361

    local zone=$1

   362

    local part=$2

   363

   364

    local name=$zone.zone.$part

   365

    local src=$DATA/$name

   366

    local dst=$ZONES/$name

   367

   368

   369

    if check_update $dst $src; then

   370

        log_info "Copying zones/$name..."

   371

   372

        do_update $dst cat $ROOT/$src

   373

    else

   374

        log_info "Copying zones/$name: not changed"

   375

    fi

   376

}

   377

   378

function update_zone {

   379

    local zone=$1

   380

   381

    local name=$zone.zone

   382

   383

    local out=$ZONES/$name

   384

    local in=$DATA/$zone.zone

   385

    local serial=$SERIALS/$zone.serial

   386

   387

    if check_update $out $in $serial; then

   388

        log_info "Generating $out..." 

   389

   390

        do_update $out \

   391

            $EXPAND_ZONE $ROOT/$DATA/$zone.zone   \

   392

                --serial $ROOT/$SERIALS/$zone.serial  \

   393

                --expand zones=$(abspath $ZONES)

   394

    else

   395

        log_info "Generating $out: not changed" 

   396

    fi

   397

}

   398

   399

function update_zone_view {

   400

    local zone=$1

   401

    local view=$2

   402

   403

    local name=$view/$zone.zone

   404

   405

    local out=$ZONES/$name

   406

    local in=$DATA/$zone.zone

   407

    local serial=$SERIALS/$zone.serial

   408

   409

    if check_update $out $in $serial; then

   410

        log_info "Generating $out..."

   411

   412

        do_update $out \

   413

            $EXPAND_ZONE $ROOT/$DATA/$zone.zone   \

   414

                --serial $ROOT/$SERIALS/$zone.serial  \

   415

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

   416

                --expand view=$view

   417

    else

   418

        log_info "Generating $out: not changed"

   419

    fi

   420

}

   421

   422

function update_hosts {

   423

    local dst=$1; shift

   424

    local src=$1; shift

   425

   426

   427

    if check_update $dst $src; then

   428

        log_info "Generating $dst..."

   429

   430

        do_update $dst $PROCESS_ZONE $PROCESS_ARGS $ROOT/$src "$@"

   431

    else

   432

        log_info "Generating $dst: not changed"

   433

    fi

   434

}

   435

   436

# feed `hg annotate -qd`'d version of input to process-zone --input-line-date

   437

# only applies commit'd data

   438

function update_hosts_meta {

   439

    local dst=$1; shift

   440

    local src=$1; shift

   441

   442

    if check_update $dst $src; then

   443

        log_info "Generating $dst..."

   444

   445

        # via stdin

   446

        $hg $DATA annotate -qd $ROOT/$src | \

   447

            do_update $dst $PROCESS_ZONE $PROCESS_ARGS --input-line-date --meta-zone "$@"

   448

    else

   449

        log_info "Generating $dst: not changed"

   450

    fi

   451

}

   452

   453

function check_hosts {

   454

    local hosts=$1; shift 1

   455

   456

    local cmd=($PROCESS_ZONE $PROCESS_ARGS $ROOT/$hosts --check-hosts "$@")

   457

   458

    if "${cmd[@]}" -q; then

   459

        log_info "Check $hosts: OK"

   460

    else

   461

        log_error "  Check $hosts: Failed"

   462

   463

        indent "    " "${cmd[@]}"

   464

   465

        exit 1

   466

    fi

   467

}

   468

   469

function check_zone {

   470

    local name=$1

   471

    local file=$2

   472

   473

    local cmd=($NAMED_CHECKZONE $name $ROOT/$file)

   474

   475

    # test

   476

    # XXX: checkzone is very specific about the order of arguments, -q must be first

   477

    if $NAMED_CHECKZONE -q $name $ROOT/$file; then

   478

        log_info "Check $file($name): OK"

   479

    else

   480

        log_error "  Check $file($name): Failed:"

   481

   482

        indent "    " "${cmd[@]}"

   483

   484

        exit 1

   485

    fi

   486

}

   487

   488

# deploy new zone data to bind