Added FTP support

Changes:
- added ftp support
- added ftp to every function which had already sftp interaction
- edited config
- added handle_ftp.py
fixed old configcheck() bugs
- added sftp/ftp to eggcrackingjar mechanism
- added new line input with enter

src/handlers/handle_config.py

@@ -11,16 +11,17 @@ class configurationValues:
         config = configparser.ConfigParser()
         config.sections()
         config.read("config.ini")
-        localPluginFolder = config['General']['LocalPluginFolder']
-        self.pathToPluginFolder = Path(config['General']['PathToPluginFolder'])
-        seperateDownloadPath = config['General']['SeperateDownloadPath']
-        self.pathToSeperateDownloadPath = Path(config['General']['PathToSeperateDownloadPath'])
+        localPluginFolder = config['General']['UseLocalPluginFolder']
+        self.pathToPluginFolder = Path(config['Local - This Machine']['PathToPluginFolder'])
+        seperateDownloadPath = config['Local - This Machine']['SeperateDownloadPath']
+        self.pathToSeperateDownloadPath = Path(config['Local - This Machine']['PathToSeperateDownloadPath'])
 
         self.sftp_server = config['SFTP - Remote Server']['Server']
         self.sftp_user = config['SFTP - Remote Server']['Username']
         self.sftp_password = config['SFTP - Remote Server']['Password']
-        sftp_port = config['SFTP - Remote Server']['Port']
+        sftp_port = config['SFTP - Remote Server']['SFTPPort']
         self.sftp_folderPath = config['SFTP - Remote Server']['PluginFolderOnServer']
+        sftp_useSftp = config['SFTP - Remote Server']['USE_SFTP']
         sftp_seperateDownloadPath = config['SFTP - Remote Server']['SeperateDownloadPath']
         self.sftp_pathToSeperateDownloadPath = config['SFTP - Remote Server']['PathToSeperateDownloadPath']
 
@@ -40,6 +41,11 @@ class configurationValues:
         else:
             self.sftp_seperateDownloadPath = False
 
+        if sftp_useSftp == 'True':
+            self.sftp_useSftp = True
+        else:
+            self.sftp_useSftp = False
+
 
 def checkConfig():
     configAvailable = os.path.isfile("config.ini")
@@ -53,23 +59,28 @@ def checkConfig():
 def createConfig():
     config = configparser.ConfigParser(allow_no_value=True)
     config['General'] = {}
-    config['General'][';'] = 'If a local plugin folder exists (True/False): (If False SFTP will be used)'
-    config['General']['LocalPluginFolder'] = 'True'
-    config['General']['PathToPluginFolder'] = 'C:/Users/USER/Desktop/plugins'
-    config['General'][';_'] = 'If you want a different folder to store the updated plugins change to (True/False) and the path below'
-    config['General']['SeperateDownloadPath'] = 'False'
-    config['General']['PathToSeperateDownloadPath'] = 'C:/Users/USER/Desktop/plugins'
+    config['General'][';'] = 'If a local plugin folder exists (True/False) (If False SFTP/FTP will be used):'
+    config['General']['UseLocalPluginFolder'] = 'True'
+
+    config['Local - This Machine'] = {}
+    config['Local - This Machine']['PathToPluginFolder'] = 'C:/Users/USER/Desktop/plugins'
+    config['Local - This Machine'][';'] = 'For a different folder to store the updated plugins change to (True/False) and the path below'
+    config['Local - This Machine']['SeperateDownloadPath'] = 'False'
+    config['Local - This Machine']['PathToSeperateDownloadPath'] = 'C:/Users/USER/Desktop/plugins'
 
     config['SFTP - Remote Server'] = {}
     config['SFTP - Remote Server']['Server'] = '0.0.0.0'
     config['SFTP - Remote Server']['Username'] = 'user'
     config['SFTP - Remote Server']['Password'] = 'password'
-    config['SFTP - Remote Server'][';'] = 'Normally you won*t need to change anything below this line'
-    config['SFTP - Remote Server']['Port'] = '22'
-    config['SFTP - Remote Server']['PluginFolderOnServer'] = '.\\plugins'
-    config['SFTP - Remote Server'][';_'] = 'If you want a different folder to store the updated plugins change to (True/False) and the path below'
+    config['SFTP - Remote Server'][';'] = 'If a different Port for SFTP needs to be used (Works only for SFTP)'
+    config['SFTP - Remote Server']['SFTPPort'] = '22'
+    config['SFTP - Remote Server'][';_'] = 'Change the path below if the plugin folder path is different on the SFTP/FTP server (Change only if you know what you are doing)'
+    config['SFTP - Remote Server']['PluginFolderOnServer'] = './plugins'
+    config['SFTP - Remote Server'][';__'] = 'If you want to use FTP instead of SFTP change to (False) else use (True)'
+    config['SFTP - Remote Server']['USE_SFTP'] = 'True'
+    config['SFTP - Remote Server'][';___'] = 'For a different folder to store the updated plugins (Only with the update command!) change to (True/False) and the path below'
     config['SFTP - Remote Server']['SeperateDownloadPath'] = 'False'
-    config['SFTP - Remote Server']['PathToSeperateDownloadPath'] = '.\\plugins'
+    config['SFTP - Remote Server']['PathToSeperateDownloadPath'] = './plugins'
 
 
     with open('config.ini', 'w') as configfile:

src/handlers/handle_ftp.py

@@ -0,0 +1,100 @@
+import os
+import sys
+import ftplib
+
+from utils.consoleoutput import oColors
+from handlers.handle_config import configurationValues
+
+
+def createFTPConnection():
+    configValues = configurationValues()
+    ftp = ftplib.FTP(configValues.sftp_server, user=configValues.sftp_user, \
+           passwd=configValues.sftp_password)
+    try:
+        return ftp
+    except UnboundLocalError:
+        print(oColors.brightRed + "[FTP]: Check your config.ini!" + oColors.standardWhite)
+        print(oColors.brightRed + "Exiting program..." + oColors.standardWhite)
+        sys.exit()
+
+
+def ftp_showPlugins(ftp):
+    configValues = configurationValues()
+    ftp.cwd(configValues.sftp_folderPath)
+    for attr in ftp.dir():
+        print(attr.filename, attr)
+
+
+def ftp_cdPluginDir(ftp):
+    configValues = configurationValues()
+    ftp.cwd(configValues.sftp_folderPath)
+
+
+def ftp_upload_file(ftp, itemPath):
+    configValues = configurationValues()
+    try:
+        ftp.cwd(configValues.sftp_folderPath)
+        itemPath = os.path.relpath(itemPath, 'TempSFTPFolder/')
+        itemPath = str(itemPath)
+        currentDirectory = os.getcwd()
+        os.chdir('TempSFTPFolder')
+        with open (itemPath, 'rb') as plugin_file:
+            ftp.storbinary('STOR '+ str(itemPath), plugin_file)
+    except FileNotFoundError:
+        print(oColors.brightRed + "The 'plugins' folder couldn*t be found on the remote host!" + oColors.standardWhite)
+        print(oColors.brightRed + "Aborting uploading." + oColors.standardWhite)
+    os.chdir(currentDirectory)
+    ftp.close()
+
+
+def ftp_upload_server_jar(ftp, itemPath):
+    try:
+        print(f"itempath: {itemPath}")
+        ftp.cwd('.')
+        itemPath = os.path.relpath(itemPath, 'TempSFTPFolder/')
+        itemPath = str(itemPath)
+        currentDirectory = os.getcwd()
+        os.chdir('TempSFTPFolder')
+        with open (itemPath, 'rb') as server_jar:
+            ftp.storbinary('STOR '+ str(itemPath), server_jar)
+    except FileNotFoundError:
+        print(oColors.brightRed + "The 'root' folder couldn*t be found on the remote host!" + oColors.standardWhite)
+        print(oColors.brightRed + "Aborting uploading." + oColors.standardWhite)
+    os.chdir(currentDirectory)
+    ftp.close()
+
+
+def ftp_listAll(ftp):
+    configValues = configurationValues()
+    try:
+        ftp.cwd(configValues.sftp_folderPath)
+        installedPlugins = ftp.nlst()
+    except FileNotFoundError:
+        print(oColors.brightRed + "The 'plugins' folder couldn*t be found on the remote host!" + oColors.standardWhite)
+
+    try:
+        return installedPlugins
+    except UnboundLocalError:
+        print(oColors.brightRed + "No plugins were found." + oColors.standardWhite)
+
+
+def ftp_listFilesInServerRoot(ftp):
+    try:
+        ftp.cwd('.')
+        filesInServerRoot = ftp.nlst()
+    except FileNotFoundError:
+        print(oColors.brightRed + "The 'root' folder couldn*t be found on the remote host!" + oColors.standardWhite)
+
+    try:
+        return filesInServerRoot
+    except UnboundLocalError:
+        print(oColors.brightRed + "No Serverjar was found." + oColors.standardWhite)
+
+
+def ftp_downloadFile(ftp, downloadPath, fileToDownload):
+    configValues = configurationValues()
+    ftp.cwd(configValues.sftp_folderPath)
+    filedata = open(downloadPath,'wb')
+    ftp.retrbinary('RETR '+fileToDownload, filedata.write)
+    filedata.close()
+    ftp.quit()

src/handlers/handle_input.py

@@ -82,18 +82,24 @@ def handleInput(inputCommand, inputSelectedObject, inputParams):
             break
         else:
             print(oColors.brightRed + "Error: Command not found. Please try again. :(" + oColors.standardWhite)
+            print(oColors.brightRed + "Use: '" + oColors.standardWhite +"help command" + oColors.brightRed +"' to get all available commands" + oColors.standardWhite)
             getInput()
     getInput()
 
 
 def getInput():
+    inputCommand = None
     while True:
         try:
             inputCommand, inputSelectedObject, *inputParams = input("pluGET >> ").split()
             break
         except ValueError:
-            print(oColors.brightRed + "Wrong input! Use: > 'command' 'selectedObject' [optionalParams]" + oColors.standardWhite)
-            print(oColors.brightRed + "Use: '" + oColors.standardWhite +"help command" + oColors.brightRed +"' to get all available commands" + oColors.standardWhite)
-
+            if inputCommand == None:
+                continue
+            else:
+                print(oColors.brightRed + "Wrong input! Use: > 'command' 'selectedObject' [optionalParams]" + oColors.standardWhite)
+                print(oColors.brightRed + "Use: '" + oColors.standardWhite +"help command" + oColors.brightRed +"' to get all available commands" + oColors.standardWhite)
+        except KeyboardInterrupt:
+            sys.exit()
     inputParams = inputParams[0] if inputParams else None
     handleInput(inputCommand, inputSelectedObject, inputParams)

src/handlers/handle_sftp.py

@@ -1,17 +1,20 @@
 import sys
+import os
 import pysftp
 import paramiko
+import ftplib
 
 from utils.consoleoutput import oColors
-from handlers.handle_config import checkConfig
+from handlers.handle_config import configurationValues
 
 
 def createSFTPConnection():
+    configValues = configurationValues()
     cnopts = pysftp.CnOpts()
     cnopts.hostkeys = None # TODO fix this
     try:
-        sftp = pysftp.Connection(checkConfig().sftp_server, username=checkConfig().sftp_user, \
-               password=checkConfig().sftp_password, port=checkConfig().sftp_port, cnopts=cnopts)
+        sftp = pysftp.Connection(configValues.sftp_server, username=configValues.sftp_user, \
+               password=configValues.sftp_password, port=configValues.sftp_port, cnopts=cnopts)
     except paramiko.ssh_exception.AuthenticationException:
         print(oColors.brightRed + "[SFTP]: Wrong Username/Password" + oColors.standardWhite)
     except paramiko.ssh_exception.SSHException:
@@ -41,7 +44,7 @@ def sftp_upload_file(sftp, itemPath):
 
     except FileNotFoundError:
         print(oColors.brightRed + "The 'plugins' folder couldn*t be found on the remote host!" + oColors.standardWhite)
-        print(oColors.brightRed + "Aborting installation." + oColors.standardWhite)
+        print(oColors.brightRed + "Aborting uploading." + oColors.standardWhite)
     sftp.close()
 
 
@@ -49,10 +52,9 @@ def sftp_upload_server_jar(sftp, itemPath):
     try:
         sftp.chdir('.')
         sftp.put(itemPath)
-
     except FileNotFoundError:
-        print(oColors.brightRed + "The 'plugins' folder couldn*t be found on the remote host!" + oColors.standardWhite)
-        print(oColors.brightRed + "Aborting installation." + oColors.standardWhite)
+        print(oColors.brightRed + "The 'root' folder couldn*t be found on the remote host!" + oColors.standardWhite)
+        print(oColors.brightRed + "Aborting uploading." + oColors.standardWhite)
     sftp.close()
 
 
@@ -60,7 +62,6 @@ def sftp_listAll(sftp):
     try:
         sftp.chdir('plugins')
         installedPlugins = sftp.listdir()
-
     except FileNotFoundError:
         print(oColors.brightRed + "The 'plugins' folder couldn*t be found on the remote host!" + oColors.standardWhite)
 
@@ -72,13 +73,22 @@ def sftp_listAll(sftp):
 
 def sftp_listFilesInServerRoot(sftp):
     try:
-        #sftp.chdir('plugins')
         filesInServerRoot = sftp.listdir()
-
     except FileNotFoundError:
         print(oColors.brightRed + "The 'root' folder couldn*t be found on the remote host!" + oColors.standardWhite)
 
     try:
         return filesInServerRoot
     except UnboundLocalError:
-        print(oColors.brightRed + "No Serverjar was found." + oColors.standardWhite)
+        print(oColors.brightRed + "No Serverjar was found." + oColors.standardWhite)
+
+
+def sftp_downloadFile(sftp, downloadPath, fileToDownload):
+    configValues = configurationValues()
+    sftp.cwd(configValues.sftp_folderPath)
+    currentDirectory = os.getcwd()
+    os.chdir('TempSFTPFolder')
+    sftp.get(fileToDownload)
+    sftp.close()
+    os.chdir(currentDirectory)
+