#!/usr/bin/env python3.10
from subprocess import Popen, PIPE
from csv import DictReader
import os, socket



def read_csv(csv_file = str):
    with open(csv_file, newline='') as csv:
        read_server = DictReader(csv)
        server_liste, user_liste = [] , []
        for row in read_server:
            user_liste.append(row['user'])
            server_liste.append(row['\ufeffserver'])            
    return {
                'server' : server_liste, 
                'user' : user_liste
                }


def ssh_connection(con_server = str, bash_script = """ls -lisaR /home; apt list --installed"""):
    with Popen(['ssh', '-T', str(con_server)],
            stdin=PIPE, stdout=PIPE, stderr=PIPE,
            universal_newlines=True) as p:
            output, error = p.communicate(bash_script)
    print(output)
    print(error)
    print(p.returncode)

def ssh_check(con_server = str, check_port = 22):
    response = os.system("ping -c 1 " + con_server + '>& /dev/null')
    if response == 0:
        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
            return s.connect_ex((con_server, check_port)) == 0
    else:
        return False
    

if __name__ == "__main__":
    script_folder = os.path.dirname(os.path.realpath(__file__))
    server_csv = script_folder + os.sep + "test.csv"

    for count in range(0,len(read_csv(server_csv)['server'])):
        servername = read_csv(server_csv)['server'][count]
        con_username = read_csv(server_csv)['user'][count]
        user_server_con = str(con_username + '@' + servername)
        print(servername, "verbindng wird geprüft...",sep=' ')
        if ssh_check(con_server=servername) is True:
            print(servername, "erreichbar...", sep=' ')
            ssh_connection(con_server=user_server_con)
        else:
            print(servername, "nicht erreichbar oder kein ssh!!", sep=' ')
        #if ssh_check(con_server=servername) is True:
        #    print(servername, "port offen")
        #print("Server:", read_csv(server_csv)['server'][count], "User:", read_csv(server_csv)['user'][count], sep=" ")
        #ssh_connection(con_server=read_csv("test.csv")['server'][count])