management_api.py

#!/usr/bin/env python3

import os, sys, subprocess, time, json, urllib.request, ssl
from http.server import BaseHTTPRequestHandler, HTTPServer

api_file = open("/home/ec2-user/.api_key", "r")
API_KEY = api_file.read().strip()
api_file.close()

def runLocally(commandString):
  p = subprocess.Popen(commandString, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, shell=True)
  output = b""
  for line in p.stdout:
    output += line
  return output

class RequestHandler(BaseHTTPRequestHandler):
  def _set_headers(self):
    self.send_response(200)
    self.send_header('Access-Control-Allow-Origin','*')
    self.send_header('Access-Control-Allow-Methods','GET')
    self.send_header('Content-type','text/json')
    self.end_headers()

  def do_GET(self):
    self._set_headers()
    if self.path == "/minecraft-status":
      status = runLocally("/home/ec2-user/scripts/minecraft_get_status.sh")
      self.wfile.write(bytes(str(status, 'utf-8'), 'utf-8'))
    if self.path == "/map-status":
      response = urllib.request.urlopen("https://minecraft.deltaidea.com/overviewer.js")
      status = response.status
      self.wfile.write(bytes('{{"status":{}}}'.format(status), 'utf-8'))
    return

  def do_POST(self):
    self._set_headers()
    content_len = int(self.headers.get('Content-Length'))
    post_body = str(self.rfile.read(content_len), 'utf-8').strip()

    if self.path == "/system-shutdown":
      if post_body == "key=" + API_KEY:
        runLocally("screen /home/ec2-user/scripts/system_safe_shutdown.sh")
        self.wfile.write(bytes('{"result":"success","message":"The server is shutting down in 1 minute. Minecraft is going offline in 10 seconds."}', 'utf-8'))
      else:
        self.wfile.write(bytes('{"result":"error","message":"Incorrect API key! Request body must contain \\\"key=...\\\""}', 'utf-8'))
    return

def run():
  print('Starting API server...')
  httpd = HTTPServer(('0.0.0.0', 5000), RequestHandler)
  httpd.socket = ssl.wrap_socket (httpd.socket,
    keyfile="/etc/letsencrypt/live/minecraft.deltaidea.com/privkey.pem",
    certfile="/etc/letsencrypt/live/minecraft.deltaidea.com/fullchain.pem", server_side=True)
  print('Running API server...')
  httpd.serve_forever()

try:
  run()
except Exception as e:
  print(e)

from http.server import HTTPServer, BaseHTTPRequestHandler