83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
# Modules
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
|
||
|
|
# Vars
|
||
|
|
allsv_path = '/etc/sv/'
|
||
|
|
enabledsv_path = '/var/service/'
|
||
|
|
allsv = os.listdir(allsv_path)
|
||
|
|
enabledsv = os.listdir(enabledsv_path)
|
||
|
|
|
||
|
|
# Functions
|
||
|
|
|
||
|
|
## Help
|
||
|
|
def helpmsg():
|
||
|
|
msg = '''
|
||
|
|
Usage:
|
||
|
|
---
|
||
|
|
vsc {e/enable/on/up} <service_name> - Run service and add it to autostart
|
||
|
|
vsc {d/disable/off/down <service_name> - Stop service and remove it from autostart
|
||
|
|
---'''
|
||
|
|
return msg
|
||
|
|
|
||
|
|
## Access
|
||
|
|
def su():
|
||
|
|
user = subprocess.run('whoami', shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
|
||
|
|
if user.stdout[:-1] == 'root':
|
||
|
|
return args()
|
||
|
|
else:
|
||
|
|
return 'Access denied'
|
||
|
|
|
||
|
|
## Service
|
||
|
|
def exec(option, service):
|
||
|
|
if option == 'on':
|
||
|
|
if service in enabledsv:
|
||
|
|
return 'Service alredy enabled'
|
||
|
|
if service in allsv:
|
||
|
|
subprocess.run(f'ln -s {allsv_path}{service} {enabledsv_path}', shell=True)
|
||
|
|
return f'Service {service} successfully enabled'
|
||
|
|
if not(service in allsv):
|
||
|
|
return f"Service {service} doesn't exist"
|
||
|
|
if option == 'off':
|
||
|
|
if not(service in allsv):
|
||
|
|
return f"Service {service} doesn't exist"
|
||
|
|
if not(service in enabledsv):
|
||
|
|
return f'Service {service} has already been disabled'
|
||
|
|
if service in enabledsv:
|
||
|
|
subprocess.run(f'rm {enabledsv_path}{service}', shell=True)
|
||
|
|
return f'Service {service} successfully disabled'
|
||
|
|
|
||
|
|
# Get argv
|
||
|
|
def getargs():
|
||
|
|
global service
|
||
|
|
global option
|
||
|
|
option = sys.argv[1]
|
||
|
|
service = sys.argv[2]
|
||
|
|
return options(option)
|
||
|
|
|
||
|
|
# Args
|
||
|
|
def args():
|
||
|
|
if len(sys.orig_argv[1:]) == 2:
|
||
|
|
if sys.orig_argv[2] == '--help':
|
||
|
|
return helpmsg()
|
||
|
|
elif len(sys.orig_argv[1:]) == 3:
|
||
|
|
return getargs()
|
||
|
|
else:
|
||
|
|
return 'Invalid usage, see --help'
|
||
|
|
|
||
|
|
def options(option):
|
||
|
|
options = {
|
||
|
|
'on': ['enable','e','on','up'],
|
||
|
|
'off': ['disable','d','off','down']
|
||
|
|
}
|
||
|
|
for i in options.keys():
|
||
|
|
if option in options.get(i):
|
||
|
|
return exec(i,service)
|
||
|
|
return f'Invalid option: {option}, see --help'
|
||
|
|
|
||
|
|
print(su())
|