installation working

Author: bipuladh

script.py

@@ -2,14 +2,11 @@
 from utils.configurations import loadConfigurations
 from utils.pull_secret_util import get_pull_secret
 from utils.first_run import isFirstRun, performFirstRun
-from utils.prompts import welcomeMessage, mainMenuPrompt, selectClusterPrompt
+from utils.prompts import welcomeMessage, mainMenuPrompt, selectClusterPrompt, localClustersPrompt
 from utils.setupCluster import setupCluster
 from utils.ClusterInquirer import ClusterInquirer
 from utils.dateHandler import getAvailableClusters
-
-'''
-Tell the user that we are gonna setup a config file
-'''
+from utils.run_ocs_install import LocalClusterHandler
 
 class Manager:
     def __init__(self):
@@ -18,29 +15,48 @@ def __init__(self):
             performFirstRun()
         self.configurations = loadConfigurations()
         self.clusterInquirer = ClusterInquirer(self.configurations)
+        self.LocalClusterHandler = LocalClusterHandler(self.clusterInquirer.getManager(),
+            self.configurations)
 
 
-    def setupCluster(self,selected):
-        self.clusterInquirer.setupKube(selected)
+    def setupCluster(self,selected, local=False):
+        if not local:
+            self.clusterInquirer.setupKube(selected)
+        else:
+            self.clusterInquirer.setupLocalKube(selected)
 
     def createCluster(self):
-        pass
+        self.LocalClusterHandler.startInstallation()
 
     def getClustersStatus(self):
         return self.clusterInquirer.getStatus()
 
 if __name__ == '__main__':
     #Declare a variable to maintain a program loop
     manager = Manager()
+    showLocalClusters = False
     while True:
-        choice = mainMenuPrompt()
+        localClusters = manager.LocalClusterHandler.getLocalUsableClusters()
+        if len(localClusters) > 0 :
+            showLocalClusters=True
+        choice = mainMenuPrompt(showLocalClusters)
+        #List available clusters
         if choice == 0:
+            #Get all clusters
             clusters = manager.getClustersStatus()
             healthyClusters = getAvailableClusters(clusters)
             selectedCluster = selectClusterPrompt( healthyClusters )
             manager.setupCluster(selectedCluster)
+            print("The selected cluster has been setup")
+        # Create new cluster
         if choice == 1:
-            #Create a new cluster
             manager.createCluster()
+        #Show local clusters
         if choice == 2:
+            #Selected cluster is the path to cluster
+            selectedCluster = localClustersPrompt(localClusters)
+            manager.setupCluster(selectedCluster, local=True)
+            print("Local cluster is setup!")
+        # Quit
+        if choice == 3:
             break

utils/ClusterInquirer.py

@@ -24,4 +24,19 @@ def setupKube(self, status):
         kubeconfig = self.manager.getExternalFile(owner,"kubeconfig")
         kubeadm = self.manager.getExternalFile(owner,"kubeadmin-password")
         writeKube(kubeconfig)
-        writeKubePass(kubeadm)
+        writeKubePass(kubeadm)
+    
+    def setupLocalKube(self, path):
+        authDir = os.path.join(path, 'auth')
+        kubeconfig = ''
+        kubeadmin_password = ''
+        with open( os.path.join(authDir, 'kubeconfig'), 'rb') as file:
+            kubeconfig = file.read()
+        with open( os.path.join(authDir, 'kubeadmin-password'), 'rb') as file:
+            kubeadmin_password = file.read()
+        writeKube(kubeconfig)
+        writeKubePass(kubeadmin_password)
+        
+
+    def getManager(self):
+        return self.manager

utils/__pycache__/consts.cpython-37.pyc

@@ -3,7 +3,7 @@
 HOME_DIR_PATH = os.getenv('HOME') 
 OCS_DIR_PATH = os.path.join(HOME_DIR_PATH, '.ocs_data')
 OCS_CONFIG_FILE = os.path.join(OCS_DIR_PATH, 'config')
-OCS_INSTALL_PATH = os.path.join(HOME_DIR_PATH, 'ocs-install')
+OCS_INSTALL_PATH = os.path.join(HOME_DIR_PATH, 'openshift-install')
 OCS_CLUSTERS_PATH = os.path.join(OCS_DIR_PATH, 'clusters')
 KUBE_PATH = os.path.join(HOME_DIR_PATH, 'awskube')
 

utils/__pycache__/first_run.cpython-37.pyc

@@ -1,5 +1,5 @@
 import os
-from utils.consts import OCS_DIR_PATH, KUBE_PATH, HOME_DIR_PATH
+from utils.consts import OCS_DIR_PATH, KUBE_PATH, HOME_DIR_PATH, OCS_CLUSTERS_PATH
 from utils.configurations import DEFAULT_CONFIG, writeConfigurations
 from utils.prompts import getGitInformation, teamGitIdPrompt
 
@@ -12,22 +12,25 @@ def isFirstRun():
 def performFirstRun():
     setupDirectories()
     setupKubeconfig()
-    username, password = getGitInformation()
+    username, password, enc_key = getGitInformation()
     userList = teamGitIdPrompt()
-    setupConfigurations( username, password, userList)
+    setupConfigurations( username, password, userList, enc_key)
 
-def setupConfigurations(gitUsername, gitPassword, userList):
+def setupConfigurations(gitUsername, gitPassword, userList, enc_key):
     config = DEFAULT_CONFIG
     config['username'] = gitUsername
     config['password'] = gitPassword
     config['friends_git_id'] = userList
+    config['team_enc_key'] = enc_key
     writeConfigurations(config)
 
 def setupDirectories():
     if not os.path.exists(OCS_DIR_PATH):
         os.mkdir(OCS_DIR_PATH)
     if not os.path.exists(KUBE_PATH):
         os.mkdir(KUBE_PATH)
+    if not os.path.exists(OCS_CLUSTERS_PATH):
+        os.mkdir(OCS_CLUSTERS_PATH)
 
 def setupKubeconfig():
     line = "export KUBECONFIG=" + KUBE_PATH + "/kubeconfig"

utils/__pycache__/gitHelper.cpython-37.pyc

@@ -1,10 +1,7 @@
-from git import Repo, GitCommandError
 from utils.consts import REPO_NAME
-from datetime import date
-import os
-from utils.utils import repo_name_from_url
-from shutil import copyfile
 from github import Github
+import json
+from utils.security import decryptByte
 
 
 
@@ -15,28 +12,27 @@ def __init__( self, username, password):
 
 
 class ExternRepository(Repository):
-
-    def __init__(self, username, password, externalUser):
+    def __init__(self, username, password, externalUser, teamEncKey):
         super(ExternRepository,self).__init__(username, password)
         repo_name = externalUser + "/" + REPO_NAME
         self.repo = self.gitUser.get_repo(repo_name)
+        self.teamEncKey = teamEncKey
 
     def getStatus(self):
         status = self.repo.get_contents("status.json")
         return status.decoded_content
 
     def getFile(self, fileName):
         contents = self.repo.get_contents(fileName)
-        return contents.decoded_content
+        return decryptByte(self.teamEncKey, contents.decoded_content)
 
     def getOwner(self):
         return (self.repo.owner).login
 
 class InternalRepository(Repository):
-
     def __init__(self,username,password):
         super(InternalRepository,self).__init__(username, password)
-        repo_name = username + "/" + 'badhikar_ocs'
+        repo_name = username + "/" + REPO_NAME
         self.repo = self.gitUser.get_repo(repo_name)
 
     def addFile(self, filename, data):
@@ -53,12 +49,22 @@ def addFile(self, filename, data):
             self.createFile(filename, data)
 
     def updateFile(self, filename, data):
-        filePath = "/{}".format(filename)
+        filePath = "{}".format(filename)
         contents = self.repo.get_contents(filePath)
         commit_msg = "Updated file {}".format(filename) 
         self.repo.update_file(contents.path, commit_msg, data, contents.sha)
 
     def createFile(self, filename, data):
-        filePath = "/{}".format(filename)
+        filePath = "{}".format(filename)
         commit_msg = "Created file {}".format(filename)
-        self.repo.create_file(filePath, commit_msg, data)
+        self.repo.create_file(filePath, commit_msg, data)
+
+    def getStatus(self):
+        file = {}
+        try:
+            file = self.repo.get_contents("status.json")
+            file = file.decoded_content
+            file = json.loads(file)
+        except Exception:
+            pass
+        return file