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

    23

## options

    24

IS_TTY=

    25

    26

LOG=y

    27

LOG_INFO=

    28

LOG_DEBUG=

    29

LOG_CMD=

    30

    31

UPDATE_FORCE=

    32

UPDATE_NOOP=

    33

UPDATE_DIFF=

    34

SERIAL_NOUPDATE=

    35

    36

function help_args {

    37

    local prog=$1

    38

    39

    cat <<END

    40

Usage: $prog [options]

    41

    42

    -h      display this help text

    43

    44

    -q      quiet

    45

    -v      verbose

    46

    -D      debug

    48

    50

    51

    -F      force-updates without checking src mtime

    52

    -S      do not update serial

    54

END

    55

}

    56

    57

function parse_args {

    58

    OPTIND=1

    59

    61

        case $opt in

    62

            h)  

    63

                help_args $1

    64

                exit 0

    65

            ;;

    66

    67

            q)  LOG= ;;

    68

            v)  LOG_INFO=y ;;

    72

            F)  UPDATE_FORCE=y ;;

    73

            S)  SERIAL_NOUPDATE=y ;;

    74

    75

            n)  

    76

                UPDATE_NOOP=y 

    79

                SERIAL_NOUPDATE=y

    80

                ;;

    81

    82

    83

            ?)  

    84

                die 

    85

            ;;

    86

        esac

    87

    88

    done

    89

}

    90

    91

## lib

    92

function log_msg {

    93

    echo "$*" >&2

    94

}

    95

    96

function log_color {

    97

    local code=$1; shift

    98

    99

    if [ $IS_TTY ]; then

   100

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

   101

    else

   102

        echo "$*" >&2

   103

    fi

   104

}

   105

   106

function log_error {

   107

    log_color 31 "$*"

   108

}

   109

   110

function log {

   111

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

   112

}

   113

   114

function log_info {

   115

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

   116

}

   117

   118

function log_debug {

   119

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

   120

}

   121

   122

function log_cmd {

   123

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

   124

}

   125

   126

# XXX: broken

   127

function log_stack {

   128

    local level=1

   129

   130

    while info=$(caller $level); do

   131

        echo $info | read line sub file

   132

   133

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

   134

   135

        level=$(($level + 1))

   136

    done

   137

}

   138

   139

function fail {

   140

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

   141

   142

    log_error "$func: $*"

   143

   144

    exit 2

   145

}

   146

   147

function die {

   148

    log_error "$*"

   149

    exit 1

   150

}

   151

   152

function cmd {

   153

    log_cmd "$@"

   154

   155

    "$@" || die "Failed"

   156

}

   157

   158

function run_cmd {

   159

    local msg=$1; shift

   160

   161

    log_info "$msg... "

   162

   163

    cmd "$@"

   164

}

   165

   166

function indent () {

   167

    local indent=$1; shift

   168

   169

    "$@" | (

   170

        while read line; do

   171

            echo "$indent$line"

   172

        done

   173

    ) || exit $?

   174

}

   175

   176

## test

   177

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

   178

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

   179

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

   180

   181

## functions

   182

function check_update {

   183

    # target

   184

    local dst=$1; shift

   185

   187

   188

    # need update?

   189

    local update=

   190

   191

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

   192

        log_debug "  update forced"

   193

        update=y

   194

    fi

   195

   196

    # check deps

   197

    for dep in "$@"; do

   198

        # don't bother checking if already figured out

   199

        [ $update ] && continue

   200

   201

        # check

   202

   203

        if [ $dst -ot $dep ]; then

   204

            log_debug "  changed: $dep"

   205

            update=y

   206

        fi

   207

    done

   208

   209

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

   210

   211

    # return

   212

    [ $update ]

   213

}

   214

   215

function do_update {

   216

    local dst=$1; shift

   217

    local tmp=$dst.new

   218

   220

    cmd "$@" > $tmp

   221

   222

    # compare

   223

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

   224

        log_debug "  changes:"

   225

   226

        # terse

   227

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

   228

    fi

   229

   230

    if [ $UPDATE_NOOP ]; then

   231

        # cleanup

   233

   234

        cmd rm $tmp

   235

    else

   236

        # commit

   238

   239

        cmd mv $tmp $dst

   240

    fi

   241

}

   242

   243

function update {

   244

    local dst=$1; shift;

   245

   246

    local sep=

   247

    local dep=()

   248

    local cmd=()

   249

   250

    for arg in "$@"; do

   251

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

   252

            sep=y

   253

        fi

   254

   255

        if [ $sep ]; then

   256

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

   257

        else

   258

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

   259

        fi

   260

    done

   261

   262

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

   263

   264

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

   265

}

   266

   267

## bin wrappers

   268

function update_serial {

   269

    local serial=$1; shift

   270

    local old=$(cat $serial)

   271

   272

    log_info "Updating serial: $serial"

   273

   274

    cmd $UPDATE_SERIAL $* $serial

   275

   276

    local new=$(cat $serial)

   277

   278

    log_debug "  $old -> $new"

   279

}

   280

   281

function expand_zone {

   282

    local output=$1; shift

   283

    local src=$1; shift

   284

   285

}

   286

   287

function process_zone {

   288

    local output=$1; shift

   289

    local src=$1; shift

   290

   291

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

   292

}

   293

   294

## actions

   295

function copy_zone_part {

   296

    local zone=$1

   297

    local part=$2

   298

   299

    local name=$zone.zone.$part

   300

    local src=$SETTINGS/$name

   301

    local dst=$ZONES/$name

   302

   303

   304

    if check_update $dst $src; then

   305

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

   306

   308

    else

   309

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

   310

    fi

   311

}

   312

   313

function update_zone {

   314

    local zone=$1

   315

   316

    local name=$zone.zone

   317

   318

    local out=$ZONES/$name

   319

    local in=$SETTINGS/$zone.zone

   320

    local serial=$SERIALS/$zone.serial

   321

   322

    if check_update $out $in $serial; then

   323

        log_info "Generating $zone zone headers..." 

   324

   325

        do_update $out \

   326

            $EXPAND_ZONE $SETTINGS/$zone.zone   \

   327

                --serial $SERIALS/$zone.serial  \

   328

                --expand zones=$ROOT/$ZONES

   329

    else

   330

        log_info "Generating $zone zone headers: not changed" 

   331

    fi

   332

}

   333

   334

function update_zone_view {

   335

    local zone=$1

   336

    local view=$2

   337

   338

    local name=$view/$zone.zone

   339

   340

    local out=$ZONES/$name

   341

    local in=$SETTINGS/$zone.zone

   342

    local serial=$SERIALS/$zone.serial

   343

   344

    if check_update $out $in $serial; then

   345

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

   346

   347

        do_update $out \

   348

            $EXPAND_ZONE $SETTINGS/$zone.zone   \

   349

                --serial $SERIALS/$zone.serial  \

   350

                --expand zones=$ROOT/$ZONES     \

   351

                --expand view=$view

   352

    else

   353

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

   354

    fi

   355

}

   356

   357

function update_hosts {

   358

    local dst=$1; shift

   359

    local src=$1; shift

   360

   361

   362

    if check_update $dst $src; then

   363

        log_info "Generating $dst..."

   364

   365

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

   366

    else

   367

        log_info "Generating $dst: not changed"

   368

    fi

   369

}

   370

   371

function main {

   372

    # test tty

   373

    [ -t 1 ] && IS_TTY=y

   374

   375

    parse_args "$@"

   376

   377

    log "Updating serials..."

   378

   379

    if [ $SERIAL_NOUPDATE ]; then

   380

        log_debug "skipping"

   381

    else

   382

        update_serial   $SERIALS/paivola.serial

   383

        update_serial   $SERIALS/paivola-reverse.serial

   384

    fi

   385

   386

    log "Generating host zones..."

   387

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

   388

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

   389

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

   390

   391

    log "Copying zone parts..."

   392

        copy_zone_part      paivola             auto

   393

        copy_zone_part      paivola             services

   394

        copy_zone_part      paivola             internal

   395

        copy_zone_part      paivola             external

   396

   397

    log "Updating zones headers..."

   398

        update_zone         paivola-reverse

   399

        update_zone_view    paivola             internal

   400

        update_zone_view    paivola             external

   401

}

   402

   403

main "$@"