mergekubeconfig.py

#!/usr/bin/python3
#
# Copyright 2017 Canonical, Ltd. Authored by Marco Ceppi
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.

import os
import sys
import yaml


def get_kubeconfig(cfg):
    if not os.path.exists(cfg):
        return

    with open(cfg) as f:
        return yaml.safe_load(f.read())


def merge_cfg(cfg, new_cfg):
    if not isinstance(cfg, dict) or not isinstance(new_cfg, dict):
        return None
    for k, v in new_cfg.iteritems():
        if k not in cfg:
            cfg[k] = v
        elif isinstance(v, list):
            cfg[k] = cfg[k] + v
        else:
            cfg[k] = merge_cfg(cfg[k], v)

    return cfg


def merge_config(kubecfg_file, newcfg_file, name):
    kubecfg = get_kubeconfig(kubecfg_file)
    newcfg = get_kubeconfig(newcfg_file)

    if not kubecfg or not newcfg:
        print(kubecfg, kubecfg_file)
        return 1

    newcfg['clusters'][0]['name'] = name
    newcfg['users'][0]['name'] = name
    newcfg['contexts'][0]['name'] = name
    newcfg['contexts'][0]['context']['cluster'] = name
    newcfg['contexts'][0]['context']['user'] = name

    kubecfg = merge_cfg(kubecfg, newcfg)

    with open(kubecfg_file, 'r+') as f:
        f.seek(0)
        f.write(yaml.dump(kubecfg, default_flow_style=False))
        f.truncate()


if __name__ == '__main__':
    KUBEHOME = os.path.expanduser('~/.kube')
    KUBECONFIG_FILE = os.path.join(KUBEHOME, 'config')

    if len(sys.argv) != 3:
       print('Where da args at?' if len(sys.argv) < 3 else 'Too many args bruv')
       sys.exit(1)

    NEWCONFIG_FILE = os.path.expanduser(sys.argv[1])
    JUJU_MODEL = sys.argv[2]

    rc = merge_config(KUBECONFIG_FILE, NEWCONFIG_FILE, JUJU_MODEL) or 0
    sys.exit(rc)

step-01_get-kubectl.sh

#!/bin/bash

# Path to executing script
SCRIPT=$(realpath $0)

# Directory housing script
SCRIPTPATH=$(dirname $SCRIPT)

. $SCRIPTPATH/common.sh

mkdir -p $HOME/.kube
mkdir -p $HOME/bin

cmd_to_run="~/bin/kubectl"
if [ ! -f ~/.kube/config ]; then
    juju scp kubernetes-master/0:config ~/.kube/config
else
    juju scp kubernetes-master/0:config ~/.kube/config.conjure-up
    cmd_to_run="$cmd_to_run --kubeconfig=~/.kube/config.conjure-up"
fi

juju scp kubernetes-master/0:kubectl $HOME/bin/kubectl

exposeResult "The Kubernetes client utility is now available at '$cmd_to_run'" 0 "true"

python3 $SCRIPTPATH/mergekubeconfig.py ~/.kube/config.conjure-up $JUJU_MODEL