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

SETTINGS=settings

    14

ZONES=zones

    15

SERIALS=serials

    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

    25

## options

    26

IS_TTY=

    27

    28

LOG=y

    29

LOG_INFO=

    30

LOG_DEBUG=

    31

LOG_CMD=

    32

    33

UPDATE_FORCE=

    34

UPDATE_NOOP=

    35

UPDATE_DIFF=

    36

SERIAL_NOUPDATE=

    37

    38

function help_args {

    39

    local prog=$1

    40

    41

    cat <<END

    42

Usage: $prog [options]

    43

    44

    -h      display this help text

    45

    46

    -q      quiet

    47

    -v      verbose

    48

    -D      debug

    49

    -C      debug commands

    50

    51

    -p      show changes

    52

    53

    -F      force-updates without checking src mtime

    54

    -S      do not update serial

    55

    -n      no-op/mock-update; do not actually change anything; implies -Sp

    56

END

    57

}

    58

    59

function parse_args {

    60

    OPTIND=1

    61

    62

    while getopts 'hqvDCpFSn' opt "$@"; do

    63

        case $opt in

    64

            h)  

    65

                help_args $1

    66

                exit 0

    67

            ;;

    68

    69

            q)  LOG= ;;

    70

            v)  LOG_INFO=y ;;

    71

            D)  LOG_DEBUG=y  ;;

    72

            C)  LOG_CMD=y ;;

    73

            p)  UPDATE_DIFF=y ;;

    74

            F)  UPDATE_FORCE=y ;;

    75

            S)  SERIAL_NOUPDATE=y ;;

    76

    77

            n)  

    78

                UPDATE_NOOP=y 

    79

                # implies -Sp

    80

                UPDATE_DIFF=y

    81

                SERIAL_NOUPDATE=y

    82

                ;;

    83

    84

    85

            ?)  

    86

                die 

    87

            ;;

    88

        esac

    89

    90

    done

    91

}

    92

    93

## lib

    94

function log_msg {

    95

    echo "$*" >&2

    96

}

    97

    98

function log_color {

    99

    local code=$1; shift

   100

   101

    if [ $IS_TTY ]; then

   102

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

   103

    else

   104

        echo "$*" >&2

   105

    fi

   106

}

   107

   108

function log_error {

   109

    log_color 31 "$*"

   110

}

   111

   112

function log {

   113

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

   114

}

   115

   116

function log_info {

   117

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

   118

}

   119

   120

function log_debug {

   121

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

   122

}

   123

   124

function log_cmd {

   125

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

   126

}

   127

   128

# XXX: broken

   129

function log_stack {

   130

    local level=1

   131

   132

    while info=$(caller $level); do

   133

        echo $info | read line sub file

   134

   135

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

   136

   137

        level=$(($level + 1))

   138

    done

   139

}

   140

   141

function fail {

   142

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

   143

   144

    log_error "$func: $*"

   145

   146

    exit 2

   147

}

   148

   149

function die {

   150

    log_error "$*"

   151

    exit 1

   152

}

   153

   154

function cmd {

   155

    log_cmd "$@"

   156

   157

    "$@" || die "Failed"

   158

}

   159

   160

function run_cmd {

   161

    local msg=$1; shift

   162

   163

    log_info "$msg... "

   164

   165

    cmd "$@"

   166

}

   167

   168

function indent () {

   169

    local indent=$1; shift

   170

   171

    "$@" | (

   172

        while read line; do

   173

            echo "$indent$line"

   174

        done

   175

    ) || exit $?

   176

}

   177

   178

## test

   179

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

   180

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

   181

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

   182

   183

## functions

   184

function check_update {

   185

    # target

   186

    local dst=$1; shift

   187

   188

    log_debug "check_update: $dst"

   189

   190

    # need update?

   191

    local update=

   192

   193

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

   194

        log_debug "  update forced"

   195

        update=y

   196

    fi

   197

   198

    # check deps

   199

    for dep in "$@"; do

   200

        # don't bother checking if already figured out

   201

        [ $update ] && continue

   202

   203

        # check

   204

   205

        if [ $dst -ot $dep ]; then

   206

            log_debug "  changed: $dep"

   207

            update=y

   208

        fi

   209

    done

   210

   211

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

   212

   213

    # return

   214

    [ $update ]

   215

}

   216

   217

function do_update {

   218

    local dst=$1; shift

   219

    local tmp=$dst.new

   220

   221

    log_debug "update: $dst"

   222

    cmd "$@" > $tmp

   223

   224

    # compare

   225

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

   226

        log_debug "  changes:"

   227

   228

        # terse

   229

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

   230

    fi

   231

   232

    if [ $UPDATE_NOOP ]; then

   233

        # cleanup

   234

        log_debug "  no-op"

   235

   236

        cmd rm $tmp

   237

    else

   238

        # commit

   239

        log_debug "  update"

   240

   241

        cmd mv $tmp $dst

   242

    fi

   243

}

   244

   245

function update {

   246

    local dst=$1; shift;

   247

   248

    local sep=

   249

    local dep=()

   250

    local cmd=()

   251

   252

    for arg in "$@"; do

   253

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

   254

            sep=y

   255

        fi

   256

   257

        if [ $sep ]; then

   258

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

   259

        else

   260

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

   261

        fi

   262

    done

   263

   264

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

   265

   266

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

   267

}

   268

   269

## bin wrappers

   270

function update_serial {

   271

    local serial=$1; shift

   272

    local old=$(cat $serial)

   273

   274

    log_info "Updating serial: $serial"

   275

   276

    cmd $UPDATE_SERIAL $* $serial

   277

   278

    local new=$(cat $serial)

   279

   280

    log_debug "  $old -> $new"

   281

}

   282

   283

function expand_zone {

   284

    local output=$1; shift

   285

    local src=$1; shift

   286

   287

}

   288

   289

function process_zone {

   290

    local output=$1; shift

   291

    local src=$1; shift

   292

   293

    check_update $output $src && update $output $PROCESS_ZONE $PROCESS_ARGS "$@" $src

   294

}

   295

   296

## actions

   297

function copy_zone_part {

   298

    local zone=$1

   299

    local part=$2

   300

   301

    local name=$zone.zone.$part

   302

    local src=$SETTINGS/$name

   303

    local dst=$ZONES/$name

   304

   305

   306

    if check_update $dst $src; then

   307

        log_info "Copying zone $zone.$part..."

   308

   309

        do_update $dst cat $src

   310

    else

   311

        log_info "Copying zone $zone.$part: not changed"

   312

    fi

   313

}

   314

   315

function update_zone {

   316

    local zone=$1

   317

   318

    local name=$zone.zone

   319

   320

    local out=$ZONES/$name

   321

    local in=$SETTINGS/$zone.zone

   322

    local serial=$SERIALS/$zone.serial

   323

   324

    if check_update $out $in $serial; then

   325

        log_info "Generating $zone zone headers..." 

   326

   327

        do_update $out \

   328

            $EXPAND_ZONE $SETTINGS/$zone.zone   \

   329

                --serial $SERIALS/$zone.serial  \

   330

                --expand zones=$ROOT/$ZONES

   331

    else

   332

        log_info "Generating $zone zone headers: not changed" 

   333

    fi

   334

}

   335

   336

function update_zone_view {

   337

    local zone=$1

   338

    local view=$2

   339

   340

    local name=$view/$zone.zone

   341

   342

    local out=$ZONES/$name

   343

    local in=$SETTINGS/$zone.zone

   344

    local serial=$SERIALS/$zone.serial

   345

   346

    if check_update $out $in $serial; then

   347

        log_info "Generating $zone:$view zone headers..."

   348

   349

        do_update $out \

   350

            $EXPAND_ZONE $SETTINGS/$zone.zone   \

   351

                --serial $SERIALS/$zone.serial  \

   352

                --expand zones=$ROOT/$ZONES     \

   353

                --expand view=$view

   354

    else

   355

        log_info "Generating $zone:$view zone headers: not changed"

   356

    fi

   357

}

   358

   359

function update_hosts {

   360

    local dst=$1; shift

   361

    local src=$1; shift

   362

   363

   364

    if check_update $dst $src; then

   365

        log_info "Generating $dst..."

   366

   367

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

   368

    else

   369

        log_info "Generating $dst: not changed"

   370

    fi

   371

}

   372

   387

function main {

   388

    # test tty

   389

    [ -t 1 ] && IS_TTY=y

   390

   391

    parse_args "$@"

   392

   393

    log "Updating serials..."

   394

   395

    if [ $SERIAL_NOUPDATE ]; then

   397

    else

   398

        update_serial   $SERIALS/paivola.serial

   399

        update_serial   $SERIALS/paivola-reverse.serial

   400

    fi

   401

   402

    log "Generating host zones..."

   403

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

   404

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

   405

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

   406

   407

    log "Copying zone parts..."

   408

        copy_zone_part      paivola             auto

   409

        copy_zone_part      paivola             services

   410

        copy_zone_part      paivola             internal

   411

        copy_zone_part      paivola             external

   412

   413

    log "Updating zones headers..."

   414

        update_zone         paivola-reverse

   415

        update_zone_view    paivola             internal

   416

        update_zone_view    paivola             external

   423

}

   424

   425

main "$@"