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=

    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

   193

    "$@" | (

   194

        while read line; do

   195

            echo "$indent$line"

   196

        done

   197

    ) || return $?

   198

}

   199

   200

function abspath () {

   201

    echo "$ROOT/$1"

   202

}

   203

   204

## test

   205

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

   206

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

   207

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

   208

   209

## functions

   210

function check_update {

   211

    # target

   212

    local dst=$1; shift

   213

   214

    log_debug "check_update: $dst"

   215

   216

    # need update?

   217

    local update=

   218

   219

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

   220

        log_debug "  update forced"

   221

        update=y

   222

    fi

   223

   224

    # check deps

   225

    for dep in "$@"; do

   226

        # don't bother checking if already figured out

   227

        [ $update ] && continue

   228

   229

        # check

   230

   231

        if [ $dst -ot $dep ]; then

   232

            log_debug "  changed: $dep"

   233

            update=y

   234

        fi

   235

    done

   236

   237

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

   238

   239

    # return

   240

    [ $update ]

   241

}

   242

   243

function do_update {

   244

    local dst=$1; shift

   245

    local tmp=$dst.new

   246

   247

    log_debug "update: $dst"

   248

    cmd "$@" > $tmp

   249

   250

    # compare

   251

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

   252

        log_debug "  changes:"

   253

   254

        # terse

   255

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

   256

    fi

   257

   258

    if [ $UPDATE_NOOP ]; then

   259

        # cleanup

   260

        log_debug "  no-op"

   261

   262

        cmd rm $tmp

   263

    else

   264

        # commit

   265

        log_debug "  update"

   266

   267

        cmd mv $tmp $dst

   268

    fi

   269

}

   270

   271

function update {

   272

    local dst=$1; shift;

   273

   274

    local sep=

   275

    local dep=()

   276

    local cmd=()

   277

   278

    for arg in "$@"; do

   279

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

   280

            sep=y

   281

        fi

   282

   283

        if [ $sep ]; then

   284

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

   285

        else

   286

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

   287

        fi

   288

    done

   289

   290

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

   291

   292

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

   293

}

   294

   295

## actions

   296

function update_zone_serial {

   297

    local name=$1; shift

   298

    local file=$SERIALS/$name.serial

   299

   300

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

   301

   302

    log_info "Updating $file..."

   303

   304

    cmd $UPDATE_SERIAL $* $file

   305

   306

    local new=$(cat $file)

   307

   308

    log_debug "  $old -> $new"

   309

}

   310

   311

function update_zone_part {

   312

    local zone=$1

   313

    local part=$2

   314

   315

    local name=$zone.zone.$part

   316

    local src=$DATA/$name

   317

    local dst=$ZONES/$name

   318

   319

   320

    if check_update $dst $src; then

   321

        log_info "Copying zones/$name..."

   322

   323

        do_update $dst cat $src

   324

    else

   325

        log_info "Copying zones/$name: not changed"

   326

    fi

   327

}

   328

   329

function update_zone {

   330

    local zone=$1

   331

   332

    local name=$zone.zone

   333

   334

    local out=$ZONES/$name

   335

    local in=$DATA/$zone.zone

   336

    local serial=$SERIALS/$zone.serial

   337

   338

    if check_update $out $in $serial; then

   339

        log_info "Generating $out..." 

   340

   341

        do_update $out \

   342

            $EXPAND_ZONE $DATA/$zone.zone   \

   343

                --serial $SERIALS/$zone.serial  \

   344

                --expand zones=$(abspath $ZONES)

   345

    else

   346

        log_info "Generating $out: not changed" 

   347

    fi

   348

}

   349

   350

function update_zone_view {

   351

    local zone=$1

   352

    local view=$2

   353

   354

    local name=$view/$zone.zone

   355

   356

    local out=$ZONES/$name

   357

    local in=$DATA/$zone.zone

   358

    local serial=$SERIALS/$zone.serial

   359

   360

    if check_update $out $in $serial; then

   361

        log_info "Generating $out..."

   362

   363

        do_update $out \

   364

            $EXPAND_ZONE $DATA/$zone.zone   \

   365

                --serial $SERIALS/$zone.serial  \

   366

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

   367

                --expand view=$view

   368

    else

   369

        log_info "Generating $out: not changed"

   370

    fi

   371

}

   372

   373

function update_hosts {

   374

    local dst=$1; shift

   375

    local src=$1; shift

   376

   377

   378

    if check_update $dst $src; then

   379

        log_info "Generating $dst..."

   380

   381

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

   382

    else

   383

        log_info "Generating $dst: not changed"

   384

    fi

   385

}

   386

   387

function check_hosts {

   388

    local hosts=$1; shift 1

   389

   390

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

   391

   392

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

   393

        log_info "Check $hosts: OK"

   394

    else

   395

        log_error "  Check $hosts: Failed"

   396

   397

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

   398

   399

        exit 1

   400

    fi

   401

}

   402

   403

function check_zone {

   404

    local name=$1

   405

    local file=$2

   406

   407

    local cmd=($NAMED_CHECKZONE $name $file)

   408

   409

    # test

   410

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

   411

    if $NAMED_CHECKZONE -q $name $file; then

   412

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

   413

    else

   414

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

   415

   416

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

   417

   418

        exit 1

   419

    fi

   420

}

   421

   422

# commit data changes

   423

function hg_modified {

   424

    local repo=$1

   425

   426

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

   427

}

   428

   429

function hg_user {

   430

    if [ ${SUDO_USER:-} ]; then

   431

        echo '-u' "$SUDO_USER"

   432

   433

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

   434

        log_debug "using .hgrc user"

   435

        echo ''

   436

   437

    else

   438

        echo '-u' "$USER"

   439

    fi

   440

}

   441

   442

function hg_diff {

   443

    local repo=$1

   444

   445

    $HG -R $repo diff

   446

}

   447

   448

function hg_commit {

   449

    local repo=$1

   450

    local msg=$2

   451

    local user_opt=$(hg_user)

   452

   453

    log_debug "commit: $user_opt: $msg"

   454

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

   455

}

   456

   457

function commit_data {

   458

    local data=$DATA

   459

   460

    if hg_modified $data; then

   461

        log_info "Commit changes in $data:"

   462

   463

        indent "    " hg_diff $data

   464

   465

        hg_commit $data "$COMMIT_MSG"

   466

    else

   467

        log_info "Commit changes in $data: no changes"

   468

    fi

   469

}

   470

   471

function main {

   472

    # test tty

   473

    [ -t 1 ] && IS_TTY=y

   474

   475

    parse_args "$@"

   476

   477

    log "Testing hosts..."

   478

        check_hosts     $DATA/paivola.txt --check-exempt ufc

   479

   480

    log "Generating host zones..."

   481

        update_hosts    $ZONES/external/paivola.zone.hosts  $DATA/paivola.txt --forward-zone

   482

        update_hosts    $ZONES/internal/paivola.zone.hosts  $DATA/paivola.txt --forward-zone --forward-txt --forward-mx $FORWARD_MX

   483

        update_hosts    $ZONES/paivola-reverse.zone.hosts   $DATA/paivola.txt --reverse-zone $REVERSE_ZONE --reverse-domain $REVERSE_DOMAIN

   484

   485

    log "Copying zone parts..."

   486

        update_zone_part    paivola         auto

   487

        update_zone_part    paivola         services

   488

        update_zone_part    paivola         internal

   489

        update_zone_part    paivola         external