main.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Copyright 2019 Red Hat, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http:#www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os
import argparse

from ocp.ocp import OCP
from istio.operator import Operator, ControlPlane


class Moitt(object):

    def __init__(self):
        self.profile = None
        self.pullsec = None
        self.crfile = None
        self.install = False
        self.uninstall = False
        self.component = None
        self.assets = None
        self.version = None
        self.quay = False
        self.release = None
        self.nightly = False
        
    def envParse(self):
        if 'AWS_PROFILE' in os.environ:
            self.profile = os.environ['AWS_PROFILE']
        if 'CR_FILE' in os.environ:
            self.crfile = os.environ['CR_FILE']

    
    def argParse(self):
        parser = argparse.ArgumentParser(description='Select an operation and component(s)')
        group = parser.add_mutually_exclusive_group(required=True)
        group.add_argument('-i', '--install', help='install operation', action='store_true')
        group.add_argument('-u', '--uninstall', help='uninstall operation', action='store_true')
        parser.add_argument('-c', '--component', type=str, choices=['ocp', 'istio'], help='Specify Component from ocp, istio')
        parser.add_argument('-d', '--directory', type=str, default='assets', help='OCP cluster config assets directory path')
        parser.add_argument('-v', '--version', type=str, default='4.6.17', help='OCP installer version')
        parser.add_argument('-q', '--quay', help='install istio operator from quay.io', action='store_true')
        parser.add_argument('-r', '--release', type=str, default='stable', help='OLM release channel')
        parser.add_argument('-b', '--bot', help='login cluster created by cluster-bot', action='store_true')
        parser.add_argument('-n', '--nightly', help='create nightly cluster for daily build', action='store_true')
        args = parser.parse_args()
        self.install = args.install
        self.uninstall = args.uninstall
        self.component = args.component
        self.assets = args.directory
        self.version = args.version
        self.quay = args.quay
        self.release = args.release
        self.bot = args.bot
        self.nightly = args.nightly
      

def main():
    moitt = Moitt()
    moitt.envParse()
    moitt.argParse()
    
    if not moitt.profile:
        raise KeyError("Missing AWS_PROFILE environment variable")

    ocp = OCP(profile=moitt.profile, assets=moitt.assets, version=moitt.version)
    os.environ['KUBECONFIG'] = moitt.assets + '/auth/kubeconfig'
    
    if moitt.component == 'ocp':
        if moitt.install:
            # Install ocp cluster
            ocp.install()
            
            # Read kubeadmin password
            with open(moitt.assets + '/auth/kubeadmin-password') as f:
                pw = f.read()
            ocp.login('kubeadmin', pw)
            ##ocp.login_bot()

            # Create testing users, qe1 and qe2
            ocp.create_users()
            ocp.logout()

        elif moitt.uninstall:
            ocp.uninstall()
    
    if moitt.component == 'istio':
        operator = Operator(maistra_branch="maistra-"+moitt.release, release=moitt.release)

        nslist = ['bookinfo', 'foo', 'bar', 'legacy']
        smmr = os.getcwd() + '/testdata/member-roll.yaml'
        sample = os.getcwd() + '/testdata/bookinfo.yaml'
        cp = ControlPlane("basic", "istio-system", "bookinfo", nslist, smmr, sample)
        if moitt.install:
            # deploy operators
            if moitt.bot:
                ocp.login_bot()
            else:
                # Read kubeadmin password
                with open(moitt.assets + '/auth/kubeadmin-password') as f:
                    pw = f.read()
                ocp.login('kubeadmin', pw)

            if moitt.quay:
                operator.update_quay_token()
                operator.apply_catalog_source()
            if moitt.nightly:
                ocp.logout()
                return

            operator.deploy_jaeger()
            operator.deploy_kiali()
            operator.deploy_istio()

            operator.check()
            ocp.logout()

            # deploy controlplane
            if moitt.bot:
                ocp.login_bot()
            else:
                ocp.login('qe1', os.getenv('QE1_PWD', 'qe1pw'))

            cp.install(cr_file=moitt.crfile)
            cp.create_ns(cp.nslist)
            cp.apply_smmr()
            cp.smoke_check()
            cp.check()
            ocp.logout()

        elif moitt.uninstall:
            # uninstall controlplane
            if moitt.bot:
                ocp.login_bot()
            else:
                ocp.login('qe1', os.getenv('QE1_PWD', 'qe1pw'))

            cp.uninstall(cr_file=moitt.crfile)
            ocp.logout()

            # uninstall operators
            if moitt.bot:
                ocp.login_bot()
            else:
                # Read kubeadmin password
                with open(moitt.assets + '/auth/kubeadmin-password') as f:
                    pw = f.read()
                ocp.login('kubeadmin', pw)

            operator.uninstall()

            ##if moitt.quay:
                #operator.uninstall_operator_source()
            ##    operator.uninstall_catalog_source()

            ocp.logout()

   
if __name__ == '__main__':
    main()