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=$DATA

    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 {

   236

    local repo=$1; shift

   237

   238

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

   239

}

   240

   241

function hg_modified {

   242

    local repo=$1

   243

   244

    hg $repo id | grep -q '+'

   245

}

   246

   247

function hg_user {

   248

    if [ ${SUDO_USER:-} ]; then

   249

        echo '-u' "$SUDO_USER"

   250

   251

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

   252

        log_debug "using .hgrc user"

   253

        echo ''

   254

   255

    else

   256

        echo '-u' "$USER"

   257

    fi

   258

}

   259

   260

function hg_diff {

   261

    local repo=$1

   262

   263

    hg $repo diff

   264

}

   265

   266

function hg_commit {

   267

    local repo=$1

   268

    local msg=$2

   269

    local user_opt=$(hg_user)

   270

   271

    log_debug "commit: $user_opt: $msg"

   272

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

   273

}

   274

   275

   276

## functions

   277

function check_update {

   278

    # target

   279

    local dst=$1; shift

   280

   281

    log_debug "check_update: $dst"

   282

   283

    # need update?

   284

    local update=

   285

   286

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

   287

        log_debug "  update forced"

   288

        update=y

   289

    fi

   290

   291

    # check deps

   292

    for dep in "$@"; do

   293

        # don't bother checking if already figured out

   294

        [ $update ] && continue

   295

   296

        # check

   297

   298

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

   299

            log_debug "  changed: $dep"

   300

            update=y

   301

        fi

   302

    done

   303

   304

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

   305

   306

    # return

   307

    [ $update ]

   308

}

   309

   310

function do_update {

   311

    local dst=$1; shift

   312

    local tmp=$dst.new

   313

   314

    log_debug "update: $dst"

   315

    cmd "$@" > $ROOT/$tmp

   316

   317

    # compare

   318

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

   319

        log_debug "  changes:"

   320

   321

        # terse

   322

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

   323

    fi

   324

   325

    if [ $UPDATE_NOOP ]; then

   326

        # cleanup

   327

        log_debug "  no-op"

   328

   329

        cmd rm $ROOT/$tmp

   330

    else

   331

        # commit

   332

        log_debug "  update"

   333

   334

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

   335

    fi

   336

}

   337

   338

function update {

   339

    local dst=$1; shift;

   340

   341

    local sep=

   342

    local dep=()

   343

    local cmd=()

   344

   345

    for arg in "$@"; do

   346

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

   347

            sep=y

   348

        fi

   349

   350

        if [ $sep ]; then

   351

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

   352

        else

   353

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

   354

        fi

   355

    done

   356

   357

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

   358

   359

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

   360

}

   361

   362

## actions

   363

function update_zone_serial {

   364

    local name=$1; shift

   365

    local file=$SERIALS/$name.serial

   366

   367

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

   368

   369

    log_info "Updating $file..."

   370

   371

    cmd $UPDATE_SERIAL $* $ROOT/$file

   372

   373

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

   374

   375

    log_debug "  $old -> $new"

   376

}

   377

   378

function link_zone_serial {

   379

    local name=$1

   380

    local base=$2

   381

   382

    local lnk=$SERIALS/$name.serial

   383

    local tgt=$SERIALS/$base.serial

   384

   385

    if [ -e $ROOT/$lnk ] && [ $(readlink $ROOT/$lnk) == $ROOT/$tgt ]; then

   386

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

   387

   388

    else

   389

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

   390

   391

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

   392

    fi

   393

}

   394

   395

function update_zone_part {

   396

    local zone=$1

   397

    local part=$2

   398

   399

    local name=$zone.zone.$part

   400

    local src=$DATA/$name

   401

    local dst=$ZONES/$name

   402

   403

   404

    if check_update $dst $src; then

   405

        log_info "Copying zones/$name..."

   406

   407

        do_update $dst cat $ROOT/$src

   408

    else

   409

        log_info "Copying zones/$name: not changed"

   410

    fi

   411

}

   412

   413

function update_zone {

   414

    local zone=$1

   415

   416

    local out=$ZONES/$zone

   417

    local in=$DATA/$zone.zone

   418

    local serial=$SERIALS/$zone.serial

   419

   420

    if check_update $out $in $serial; then

   421

        log_info "Generating $out..." 

   422

   423

        do_update $out \

   424

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

   425

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

   426

                --expand zones=$(abspath $ZONES)

   427

    else

   428

        log_info "Generating $out: not changed" 

   429

    fi

   430

}

   431

   432

function update_zone_view {

   433

    local zone=$1

   434

    local view=$2

   435

   436

    local out=$ZONES/$view/$zone

   437

    local in=$DATA/$zone.zone

   438

    local serial=$SERIALS/$zone.serial

   439

   440

    if check_update $out $in $serial; then

   441

        log_info "Generating $out..."

   442

   443

        do_update $out \

   444

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

   445

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

   446

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

   447

                --expand view=$view

   448

    else

   449

        log_info "Generating $out: not changed"

   450

    fi

   451

}

   452

   453

function link_zone_view {

   454

    local zone=$1

   455

    local view=$2

   456

    local base=$3

   457

   458

    local lnk=$ZONES/$view/$zone

   459

    local tgt=

   460

   461

    # find tgt

   462

    for path in "$ZONES/$view/$base" "$ZONES/$base"; do

   463

        if [ -e "$ROOT/$path" ]; then

   464

            log_debug "$view/$zone: base $base from $path"

   465

            tgt="$path"

   466

   467

            break

   468

        fi

   469

    done

   470

   471

    if [ ! $tgt ]; then

   472

        die "$view/$zone: base $base not found!"

   473

    fi

   474

   475

    # link

   476

    if [ -e $ROOT/$lnk ] && [ $(readlink $ROOT/$lnk) == $ROOT/$tgt ]; then

   477

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

   478

   479

    else

   480

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

   481

   482

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

   483

    fi

   484

}

   485

   486

function update_hosts {

   487

    local dst=$1; shift

   488