From fe3428039395231ec7e1ba67ec3cefa45b20ed8e Mon Sep 17 00:00:00 2001
From: diginc <adam@diginc.us>
Date: Wed, 18 Oct 2017 21:41:23 -0500
Subject: [PATCH] new customizable Dockerfile.py and travis matrix

---
 .travis.yml   | 20 +++++++++++---
 Dockerfile.py | 73 +++++++++++++++++++++++++++++++++++++--------------
 2 files changed, 69 insertions(+), 24 deletions(-)
 mode change 100644 => 100755 Dockerfile.py

diff --git a/.travis.yml b/.travis.yml
index a3aff05..707324f 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -2,13 +2,25 @@ sudo: required
 services:
   - docker
 language: python
+env:
+    global:
+        - QEMU_VER=v2.9.1
+        - DOCKER_REPO=shaf/test
+    matrix:
+        - OS=debian ARCH=amd64
+        - OS=debian ARCH=armhf
+        - OS=debian ARCH=aarch64
+        - OS=alpine ARCH=amd64
+        - OS=alpine ARCH=armhf
+        - OS=alpine ARCH=aarch64
 python:
   - "2.7"
 install:
-  - pip install tox
-
+  - pip install -r requirements.txt
 script:
   # prepare qemu
   - docker run --rm --privileged multiarch/qemu-user-static:register --reset
-  # requirements, build docker, and tests
-  - tox
+  # generate and build dockerfile
+  - ./Dockerfile.py --os=${OS} --arch=${ARCH} -v
+  # run docker build & tests
+  - py.test -vv -k ${OS}_${ARCH} test/
diff --git a/Dockerfile.py b/Dockerfile.py
old mode 100644
new mode 100755
index 5e78607..89f9055
--- a/Dockerfile.py
+++ b/Dockerfile.py
@@ -1,6 +1,23 @@
-""" Generates Dockerfiles from template and builds them locally """
+#!/usr/bin/env python
 
+""" Dockerfile.py - generates and build dockerfiles
+
+Usage:
+  Dockerfile.py [--os=<os> ...] [--arch=<arch> ...] [-v] [--no-build | --no-generate]
+
+Options:
+    --no-build      Skip building the docker images
+    --no-generate   Skip generating Dockerfiles from template
+    --os=<os>       What OS(s) to build             [default: alpine debian]
+    --arch=<arch>   What Architecture(s) to build   [default: amd64 armhf aarch64]
+    -v              Print docker's command output   [default: False]
+
+Examples:
+"""
+
+from docopt import docopt
 from jinja2 import Environment, FileSystemLoader
+from docopt import docopt
 import os
 import testinfra
 
@@ -54,9 +71,15 @@ images = {
     ]
 }
 
-def generate_dockerfiles():
+def generate_dockerfiles(args):
+    if args['--no-generate']:
+        print " ::: Skipping Dockerfile generation"
+        return
+
     for os, archs in images.iteritems():
         for image in archs:
+            if os not in args['--os'] and image['arch'] not in args['--arch']:
+                return
             merged_data = dict(
                 { 'os': os }.items() +
                 base_vars.items() +
@@ -72,34 +95,44 @@ def generate_dockerfiles():
                 f.write(template.render(pihole=merged_data))
 
 
-def build_everything():
-    for os in ['debian', 'alpine']:
-        for image, archs in {
-            'pi-hole': ['amd64'], 
-            'pi-hole-multiarch': ['armhf', 'aarch64'],
-        }.iteritems():
-            for arch in archs: 
-                build(image, os, arch)
+def build_dockerfiles(args):
+    if args['--no-build']:
+        print " ::: Skipping Dockerfile generation"
+        return
+
+    for os in args['--os']:
+        for arch in args['--arch']:
+            docker_repo = 'pi-hole-multiarch'
+            if arch == 'amd64':
+                docker_repo = 'pi-hole'
+
+            build(docker_repo, os, arch, args)
 
 
-def build(image, os, arch):
+def build(docker_repo, os, arch, args):
     run_local = testinfra.get_backend(
         "local://"
     ).get_module("Command").run
 
     dockerfile = 'Dockerfile_{}_{}'.format(os, arch)
-    image_tag = '{}:{}_{}'.format(image, os, arch)
-    print " ::: Pulling {} to reuse layers".format(dockerfile, image_tag)
-    pull_cmd = run_local('docker pull {}/{}'.format('diginc', image_tag))
-    print pull_cmd.stdout
-    print " ::: Building {} into {}".format(dockerfile, image_tag)
-    build_cmd = run_local('docker build --pull -f {} -t {} .'.format(dockerfile, image_tag))
-    print build_cmd.stdout
+    repo_tag = '{}:{}_{}'.format(docker_repo, os, arch)
+    print " ::: Pulling {} to reuse layers".format(dockerfile, repo_tag)
+    pull_cmd = run_local('docker pull {}/{}'.format('diginc', repo_tag))
+    if args['-v']:
+        print pull_cmd.stdout
+    print " ::: Building {} into {}".format(dockerfile, repo_tag)
+    build_cmd = run_local('docker build --pull -f {} -t {} .'.format(dockerfile, repo_tag))
+    if args['-v']:
+        print build_cmd.stdout
     if build_cmd.rc != 0:
+        print "     ::: Building {} encountered an error".format(dockerfile)
         print build_cmd.stderr
     assert build_cmd.rc == 0
 
 
 if __name__ == '__main__':
-    generate_dockerfiles()
-    build_everything()
+    args = docopt(__doc__, version='Dockerfile 0.1')
+    # print args
+
+    generate_dockerfiles(args)
+    build_dockerfiles(args)