controller.py

#!/usr/bin/python3
# **************************************************************************** #
#                                                                              #
#                                                         :::      ::::::::    #
#    controller.py                                      :+:      :+:    :+:    #
#                                                     +:+ +:+         +:+      #
#    By: yoyassin <yoyassin@student.42.fr>          +#+  +:+       +#+         #
#                                                 +#+#+#+#+#+   +#+            #
#    Created: 2019/10/13 10:15:23 by yoyassin          #+#    #+#              #
#    Updated: 2019/10/13 10:15:23 by yoyassin         ###   ########.fr        #
#                                                                              #
# **************************************************************************** #

import socket, sys, readline, os, loader, re, pickle, rlcompleter
import signal, parse_cfgfile
from app import App
import threading

#initializing connection
names = []
def read(sock):
    strdata = "";
    while not strdata.endswith("end\n"):
        data = sock.recv(100)
        strdata = strdata + str(data, "UTF-8")
    strdata = strdata[:-4];
    print(strdata, end='')

def init_conn(_exit_):
    proc_list,socket_addr = parse_cfgfile.validate(_exit_)
    if proc_list == False :
        if _exit_:
            sys.exit(1)
        print("Something is wrong, couldn't reload config file.")
    if proc_list :
        for proc in proc_list :
            names.append(proc.name)
        for key in builtins:
            names.append(key)
    try:
        App.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        App.socket.connect(socket_addr)
    except socket.error:
        print("error connecting to socket: {}".format(socket_addr), file=sys.stderr)
        sys.exit(1)

def start(action):
    App.socket.sendall(action)
    read(App.socket)

def status(action):
    App.socket.sendall(action)
    read(App.socket)

def restart(action):
    App.socket.sendall(action)
    read(App.socket)

def stop(action):
    App.socket.sendall(action)
    read(App.socket)

def reload(action):
    App.socket.sendall(action)
    read(App.socket)
    App.socket.close()
    names = []
    init_conn(False)

def ft_exit(action):
    print("exit")
    sys.exit(0)

def shutdown(action):
    print("GETTING THE FUCK OUT...")
    App.socket.sendall(action)
    read(App.socket)
    ft_exit(None)


builtins = {'start': start,
            'status': status,
            'restart': restart,
            'stop': stop,
            'reload': reload,
            'exit': ft_exit,
            'shutdown': shutdown}

#controller loop


def autocomplete(text, state):
    actions = names
    options = [action for action in actions if action.startswith(text)]
    if state < len(actions):
        return options[state]
    else:
        return None

#autocomplete setup
readline.parse_and_bind("bind ^I rl_complete")
readline.set_completer(autocomplete)


def print_help(action):
    print("Usage:\n\tstatus\n\tstart [...]\n\tstop [...]\n\treload\n\texit\n")

def validate_line(line):
    actions = line.split()
    patt = re.compile('^[a-zA-Z]+$|^[a-zA-Z]+:?\s?[a-zA-Z0-9]+')
    for action in actions:
        if bool(re.match(patt, action)) :
            continue 
        else :
            return ""
    return actions

def sighandler(sig, frame):
    ft_exit(None)

signal.signal(signal.SIGINT, sighandler)
signal.signal(signal.SIGQUIT, signal.SIG_IGN)
init_conn(True)

while True:
    line = input('Taskmasterctl $> ')
    if not line:
        continue
    line = line.strip()
    action = list(validate_line(line))
    if not action:
        print("Invalid syntax.")
        continue
    if action[0] != "shutdown" and action[0] != "exit" and action[0] != "status" and action[0] != "reload" and len(action) < 2:
        print("Missing arguments")
        print_help(action)
        continue

    if builtins.get(action[0]):
        try:
            builtins[action[0]](pickle.dumps(action, pickle.HIGHEST_PROTOCOL))
        except IOError:
            print("can't connect to daemon")
        continue
    print("No such command: {}".format(action[0]), file=sys.stderr)