80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
from subprocess import check_output
|
|
from urllib.parse import parse_qsl
|
|
|
|
from http.server import BaseHTTPRequestHandler
|
|
from adduser import adduser
|
|
from ldappass import code
|
|
|
|
|
|
class api(BaseHTTPRequestHandler):
|
|
"""User management HTTP server"""
|
|
|
|
def send(self, code, body):
|
|
"""Send HTTP response and body"""
|
|
|
|
print(code, body)
|
|
self.send_response(code)
|
|
self.send_header('Content-Type', 'text/plain')
|
|
self.send_header('Content-Length', str(len(body)))
|
|
self.end_headers()
|
|
self.wfile.write(body.encode('utf-8'))
|
|
|
|
def usernew(self, data):
|
|
"""Register a new user"""
|
|
|
|
if data['code'] != code():
|
|
self.send(403, 'Nice try, but that secret code is totally wrong')
|
|
return
|
|
|
|
if not all(c.isdigit() or c.islower() for c in data['username']) or data['username'][0].isdigit():
|
|
self.send(403, 'I don\'t like that username')
|
|
return
|
|
|
|
# Add the user
|
|
adduser(data['username'], data['firstname'],
|
|
data['lastname'], data['email'], data['password'])
|
|
self.send(200, 'Well I think it worked...')
|
|
|
|
def info(self):
|
|
"""Return info about the server"""
|
|
|
|
self.send(200, check_output('neofetch | sed \'s/\\x1B\\[[0-9;\\?]*[a-zA-Z]//g\'', shell=True).decode('utf-8'))
|
|
|
|
def packages(self):
|
|
"""Return the packages installed on the server"""
|
|
|
|
self.send(200, check_output(['pacman', '-Q']).decode('utf-8'))
|
|
|
|
def do_GET(self):
|
|
"""Handle GET requests"""
|
|
|
|
if self.path == '/api/server/info':
|
|
self.info()
|
|
return
|
|
elif self.path == '/api/server/packages':
|
|
self.packages()
|
|
return
|
|
|
|
self.send(501, 'We don\'t know how to do that yet')
|
|
|
|
def do_POST(self):
|
|
"""Handle API POST requests"""
|
|
|
|
content_length = int(self.headers['Content-Length'])
|
|
try:
|
|
data = dict(parse_qsl(self.rfile.read(
|
|
content_length).decode('utf-8')))
|
|
except:
|
|
self.send(400, 'Oops, that request doesn\'t look quite right')
|
|
|
|
# Print debug data
|
|
debug = data.copy()
|
|
debug.pop('password')
|
|
debug.pop('confirmpassword')
|
|
print(debug)
|
|
|
|
if self.path == '/api/user/new':
|
|
self.usernew(data)
|
|
return
|
|
|
|
self.send(501, 'We don\'t know how to do that yet')
|