1

#!/bin/bash

     2

# vim: set ft=sh :

     3

     4

set -ue

     5

     6

ROOT=$(pwd)

     7

     8

BIN=bin

     9

PROCESS_ZONE=$BIN/process-zone

    10

EXPAND_ZONE=$BIN/expand-zone

    11

UPDATE_SERIAL=$BIN/update-serial

    12

    13

DATA=settings

    14

ZONES=zones

    15

SERIALS=$DATA

    16

    17

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

    18

    19

FORWARD_MX=mail

    20

REVERSE_ZONE=194.197.235

    21

REVERSE_DOMAIN=paivola.fi

    22

    23

    24

NAMED_CHECKZONE=/usr/sbin/named-checkzone

    25

HG=hg

    26

    27

## options

    28

IS_TTY=

    29

    30

LOG=y

    31

LOG_INFO=

    32

LOG_DEBUG=

    33

LOG_CMD=

    34

    35

UPDATE_FORCE=

    36

UPDATE_NOOP=

    37

UPDATE_DIFF=

    38

SERIAL_NOUPDATE=

    39

COMMIT_SKIP=

    40

COMMIT_FORCE=

    41

COMMIT_MSG=' '

    42

    43

function help_args {

    44

    local prog=$1

    45

    46

    cat <<END

    47

Usage: $prog [options]

    48

    49

General:

    50

    -h      display this help text

    51

    52

Logging:    

    53

    -q      quiet

    54

    -v      verbose

    55

    -D      debug

    56

    -V      debug commands

    57

    58

Updates:

    59

    -p      show changes

    60

    -F      force-updates without checking src mtime

    61

    -S      do not update serial

    62

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

    63

    64

Commit    

    65

    -C      do not commit changes

    66

    -c      commit changes

    67

    -m      commit message

    68

END

    69

}

    70

    71

function parse_args {

    72

    OPTIND=1

    73

    74

    while getopts 'hqvDVpFSnCcm:' opt "$@"; do

    75

        case $opt in

    76

            h)  

    77

                help_args $0

    78

                exit 0

    79

            ;;

    80

    81

            q)  LOG= ;;

    82

            v)  LOG_INFO=y ;;

    83

            D)  

    84

                LOG_DEBUG=y

    85

                LOG_INFO=y

    86

                ;;

    87

            V)  LOG_CMD=y ;;

    88

    89

            p)  UPDATE_DIFF=y ;;

    90

            F)  UPDATE_FORCE=y ;;

    91

            S)  SERIAL_NOUPDATE=y ;;

    92

    93

            n)  

    94

                UPDATE_NOOP=y 

    95

                # implies -Sp

    96

                UPDATE_DIFF=y

    97

                SERIAL_NOUPDATE=y

    98

                COMMIT_SKIP=y

    99

                ;;

   100

   101

            C)  COMMIT_SKIP=y ;;

   102

            c)  COMMIT_FORCE=y ;;

   103

            m)  COMMIT_MSG="$OPTARG" ;;

   104

   105

            ?)  

   106

                die 

   107

            ;;

   108

        esac

   109

   110

    done

   111

}

   112

   113

## lib

   114

function log_msg {

   115

    echo "$*" >&2

   116

}

   117

   118

function log_color {

   119

    local code=$1; shift

   120

   121

    if [ $IS_TTY ]; then

   122

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

   123

    else

   124

        echo "$*" >&2

   125

    fi

   126

}

   127

   128

function log_error {

   129

    log_color 31 "$*"

   130

}

   131

   132

function log {

   133

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

   134

}

   135

   136

function log_info {

   137

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

   138

}

   139

   140

function log_debug {

   141

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

   142

}

   143

   144

function log_cmd {

   145

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

   146

}

   147

   148

# XXX: broken

   149

function log_stack {

   150

    local level=1

   151

   152

    while info=$(caller $level); do

   153

        echo $info | read line sub file

   154

   155

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

   156

   157

        level=$(($level + 1))

   158

    done

   159

}

   160

   161

function fail {

   162

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

   163

   164

    log_error "$func: $*"

   165

   166

    exit 2

   167

}

   168

   169

function die {

   170

    log_error "$*"

   171

    exit 1

   172

}

   173

   174

function cmd {

   175

    log_cmd "$@"

   176

   177

    "$@" || die "Failed"

   178

}

   179

   180

function run_cmd {

   181

    local msg=$1; shift

   182

   183

    log_info "$msg... "

   184

   185

    cmd "$@"

   186

}

   187

   188

function indent () {

   189

    local indent=$1; shift

   190

   191

    log_cmd "$@"

   192

   194

}

   195

   196

function abspath () {

   197

    echo "$ROOT/$1"

   198

}

   199

   200

## test

   201

[ -d $DATA ] || die "Missing settings: $DATA"

   202

[ -d $SERIALS ] || die "Missing serials: $SERIALS"

   203

[ -d $ZONES ] || die "Missing zones: $ZONES"

   204

   205

## functions

   206

function check_update {

   207

    # target

   208

    local dst=$1; shift

   209

   210

    log_debug "check_update: $dst"

   211

   212

    # need update?

   213

    local update=

   214

   215

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

   216

        log_debug "  update forced"

   217

        update=y

   218

    fi

   219

   220

    # check deps

   221

    for dep in "$@"; do

   222

        # don't bother checking if already figured out

   223

        [ $update ] && continue

   224

   225

        # check

   226

   227

        if [ $dst -ot $dep ]; then

   228

            log_debug "  changed: $dep"

   229

            update=y

   230

        fi

   231

    done

   232

   233

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

   234

   235

    # return

   236

    [ $update ]

   237

}

   238

   239

function do_update {

   240

    local dst=$1; shift

   241

    local tmp=$dst.new

   242

   243

    log_debug "update: $dst"

   244

    cmd "$@" > $tmp

   245

   246

    # compare

   247

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

   248

        log_debug "  changes:"

   249

   250

        # terse

   251

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

   252

    fi

   253

   254

    if [ $UPDATE_NOOP ]; then

   255

        # cleanup

   256

        log_debug "  no-op"

   257

   258

        cmd rm $tmp

   259

    else

   260

        # commit

   261

        log_debug "  update"

   262

   263

        cmd mv $tmp $dst

   264

    fi

   265

}

   266

   267

function update {

   268

    local dst=$1; shift;

   269

   270

    local sep=

   271

    local dep=()

   272

    local cmd=()

   273

   274

    for arg in "$@"; do

   275

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

   276

            sep=y

   277

        fi

   278

   279

        if [ $sep ]; then

   280

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

   281

        else

   282

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

   283

        fi

   284

    done

   285

   286

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

   287

   288

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

   289

}

   290

   291

## actions

   292

function update_zone_serial {

   293

    local name=$1; shift

   294

    local file=$SERIALS/$name.serial

   295

   296

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

   297

   298

    log_info "Updating $file..."

   299

   300

    cmd $UPDATE_SERIAL $* $file

   301

   302

    local new=$(cat $file)

   303

   304

    log_debug "  $old -> $new"

   305

}

   306

   307

function update_zone_part {

   308

    local zone=$1

   309

    local part=$2

   310

   311

    local name=$zone.zone.$part

   312

    local src=$DATA/$name

   313

    local dst=$ZONES/$name

   314

   315

   316

    if check_update $dst $src; then

   317

        log_info "Copying zones/$name..."

   318

   319

        do_update $dst cat $src

   320

    else

   321

        log_info "Copying zones/$name: not changed"

   322

    fi

   323

}

   324

   325

function update_zone {

   326

    local zone=$1

   327

   328

    local name=$zone.zone

   329

   330

    local out=$ZONES/$name

   331

    local in=$DATA/$zone.zone

   332

    local serial=$SERIALS/$zone.serial

   333

   334

    if check_update $out $in $serial; then

   335

        log_info "Generating $out..." 

   336

   337

        do_update $out \

   338

            $EXPAND_ZONE $DATA/$zone.zone   \

   339

                --serial $SERIALS/$zone.serial  \

   340

                --expand zones=$(abspath $ZONES)

   341

    else

   342

        log_info "Generating $out: not changed" 

   343

    fi

   344

}

   345

   346

function update_zone_view {

   347

    local zone=$1

   348

    local view=$2

   349

   350

    local name=$view/$zone.zone

   351

   352

    local out=$ZONES/$name

   353

    local in=$DATA/$zone.zone

   354

    local serial=$SERIALS/$zone.serial

   355

   356

    if check_update $out $in $serial; then

   357

        log_info "Generating $out..."

   358

   359

        do_update $out \

   360

            $EXPAND_ZONE $DATA/$zone.zone   \

   361

                --serial $SERIALS/$zone.serial  \

   362

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

   363

                --expand view=$view

   364

    else

   365

        log_info "Generating $out: not changed"

   366

    fi

   367

}

   368

   369

function update_hosts {

   370

    local dst=$1; shift

   371

    local src=$1; shift

   372

   373

   374

    if check_update $dst $src; then

   375

        log_info "Generating $dst..."

   376

   377

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

   378

    else

   379

        log_info "Generating $dst: not changed"

   380

    fi

   381

}

   382

   400

function check_hosts {

   401

    local hosts=$1; shift 1

   402

   403

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

   404

   405

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

   406

        log_info "Check $hosts: OK"

   407

    else

   408

        log_error "  Check $hosts: Failed"

   409

   410

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

   411

   412

        exit 1

   413

    fi

   414

}

   415

   416

function check_zone {

   417

    local name=$1

   418

    local file=$2

   419

   420

    local cmd=($NAMED_CHECKZONE $name $file)

   421

   422

    # test

   423

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

   424

    if $NAMED_CHECKZONE -q $name $file; then

   425

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

   426

    else

   427

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

   428

   429

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

   430

   431

        exit 1

   432

    fi

   433

}

   434

   435

# commit data changes

   436

function hg_modified {

   437

    local repo=$1

   438

   439

    $HG -R $repo id | grep -q '+'

   440

}

   441

   442

function hg_user {

   443

    if [ ${SUDO_USER:-} ]; then

   444

        echo '-u' "$SUDO_USER"

   445

   446

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

   447

        log_debug "using .hgrc user"

   448

        echo ''

   449

   450

    else

   451

        echo '-u' "$USER"

   452

    fi

   453

}

   454

   455

function hg_diff {

   456

    local repo=$1

   457

   458

    $HG -R $repo diff

   459

}

   460

   461

function hg_commit {

   462

    local repo=$1

   463

    local msg=$2

   464

    local user_opt=$(hg_user)

   465

   466

    log_debug "commit: $user_opt: $msg"

   467

    cmd $HG -R $repo commit $user_opt -m "$msg"

   468

}

   469

   470

function commit_data {

   471

    local data=$DATA

   472

   473

    if hg_modified $data; then

   474

        log_info "Commit changes in $data:"

   475

   476

        indent "    " hg_diff $data

   477

   478

        hg_commit $data "$COMMIT_MSG"

   479

    else

   480

        log_info "Commit changes in $data: no changes"

   481

    fi

   482

}

   483

   484

function main {

   485

    # test tty

   486

    [ -t 1 ] && IS_TTY=y

   487

   488

    parse_args "$@"

   489

   490