Rewrite to OOP

VoidServiceControl/classes.py

@@ -0,0 +1,110 @@
+# -*- coding: utf-8 -*-
+from enum import Enum
+from os import listdir as ls
+from subprocess import run as execute
+from subprocess import PIPE
+from sys import argv
+
+from VoidServiceControl.env import *
+
+
+class Help(Exception):
+    def __str__(self):
+        return ("Usage:\n---\nvsc {e/enable/on/up} <service_name> - Run service and add it to autostart\nvsc {"
+                "d/disable/off/down <service_name> - Stop service and remove it from autostart")
+
+
+def Su():
+    user = execute('whoami', shell=True, text=True, stdout=PIPE).stdout[:-1]
+    if user != 'root':
+        raise PermissionError("Error: Access denied")
+
+
+class Action(Enum):
+    ENABLE = ["enable", "e", "on", "up"]
+    DISABLE = ["disable", "d", "off", "down"]
+
+
+class Arg(Enum):
+    HELP = ["--help", "-h", "help"]
+
+    @classmethod
+    def all(cls):
+        return cls.HELP.value + cls.__action()
+
+    @classmethod
+    def __action(cls):
+        actions = list(map(lambda action: action.value, Action))
+        out = []
+        for lst in actions:
+            out.extend(lst)
+        return out
+
+
+class Args(object):
+
+    def __init__(self):
+        self.__arguments = argv[1:]
+        self.__check()
+
+    def __check(self) -> None:
+        if len(self.__arguments) != 2 and self.__arguments[0] not in Arg.HELP.value:
+            raise TypeError("Error: Bad Usage")
+        if self.__arguments[0] in Arg.HELP.value:
+            raise Help()
+        if self.__arguments[0] not in Arg.all():
+            raise TypeError("Error: Invalid args")
+
+    def __action(self) -> Action:
+        actions = list(Action)
+        for action in actions:
+            if self.__arguments[0] in action.value:
+                return action
+
+    def __iter__(self):
+        yield self.__action()
+        yield self.__arguments[1]
+
+
+class Service(object):
+    __enabled = False
+
+    def __init__(self, name: str) -> None:
+        self.__name = name
+        self.__check()
+
+    def __check(self) -> None:
+        all_services = ls(SV_PATH)
+        enabled_services = ls(ENABLED_PATH)
+        if self.__name not in all_services:
+            raise ValueError(f"Error: Service {self.__name} doesn't exists")
+        if self.__name in enabled_services:
+            self.__enabled = True
+
+    def getname(self) -> str:
+        return self.__name
+
+    def enable(self):
+        if self.__enabled:
+            raise ValueError(f"Error: Service {self.__name} already enabled")
+        execute(f'ln -s {SV_PATH}/{self.__name} {ENABLED_PATH}/', shell=True)
+
+    def disable(self):
+        if not self.__enabled:
+            raise ValueError(f"Error: Service {self.__name} already disabled")
+        execute(f'rm {ENABLED_PATH}/{self.__name}', shell=True)
+
+
+class Interface(object):
+
+    def __init__(self, service: str) -> None:
+        self.service = Service(service)
+
+    def action(self, action: Action) -> None:
+        match action:
+            case Action.ENABLE:
+                self.service.enable()
+                print(f"Service '{self.service.getname()}' successfully enabled")
+            case Action.DISABLE:
+                self.service.disable()
+                print(f"Service '{self.service.getname()}' successfully disabled")

VoidServiceControl/env.py

@@ -0,0 +1,3 @@
+# -*- coding: utf-8 -*-
+SV_PATH = '/etc/sv'
+ENABLED_PATH = '/var/service'