#!/usr/bin/python
# Author: Athmane Madjoudj <athmanem@gmail.com>
#         Karanbir Singh <kbsingh@karan.org>
# Test default CentOS repos
# Note: since the -qa and CI setup will modify the
#       local repos, we need to run this tests
#       before those changes are made
from __future__ import print_function

import sys 
import datetime
import os

repos = []

try:
    import yum
    base = yum.YumBase()
    repos = base.repos.listEnabled()
except Exception:
    import dnf
    base = dnf.Base()
    base.read_all_repos()
    repos = list(base.repos.iter_enabled())


def getEnvironOpt(varname,defval):
    global now
    val=defval
    try:
        val = int(os.environ[varname])
    except KeyError:
        pass
    print("[+] %s -> %s:%d" % (now(),varname,val))
    return val

now = lambda: datetime.datetime.today().strftime("%c")
centos_default_repos = ['base']

with open('/etc/centos-release') as x:
    f = x.read()
    if 'Stream' in f:
        centos_default_repos = ['appstream', 'baseos', 'extras-common']


if getEnvironOpt('UPDATES',1):
    centos_default_repos.append('updates')
if getEnvironOpt('EXTRAS',1):
    centos_default_repos.append('extras')
if getEnvironOpt('CR',1):
    centos_default_repos.append('cr')
if getEnvironOpt('CENTOS_KERNEL',1):
    centos_default_repos.append('centos-kernel')
if getEnvironOpt('FASTTRACK',0):
    centos_default_repos.append('fasttrack')
if getEnvironOpt('CENTOSPLUS',0):
    centos_default_repos.append('centosplus')

print("[+] %s -> Check if non default repo is enabled" % now()) 
print(repos)
for repo in repos:
    if not repo.id in centos_default_repos:
        print('%s is enabled, should be disabled at this stage' % repo.id)
        print('[+] %s -> FAIL' % now())
        sys.exit(1)
print('[+] %s -> PASS' % now())
sys.exit(0)