1 files changed, 113 insertions, 0 deletions
diff --git a/moon_engine/moon_engine/__main__.py b/moon_engine/moon_engine/__main__.py
new file mode 100644
index 00000000..16f1d322
--- /dev/null
+++ b/moon_engine/moon_engine/__main__.py
@@ -0,0 +1,113 @@
+# Software Name: MOON
+
+# Version: 5.4
+
+# SPDX-FileCopyrightText: Copyright (c) 2018-2020 Orange and its contributors
+# SPDX-License-Identifier: Apache-2.0
+
+# This software is distributed under the 'Apache License 2.0',
+# the text of which is available at 'http://www.apache.org/licenses/LICENSE-2.0.txt'
+# or see the "LICENSE" file for more details.
+
+
+import logging
+import subprocess  # nosec
+import os
+import sys
+import hug.interface
+from moon_engine.api import configuration
+
+
+LOGGER = logging.getLogger("moon.engine")
+
+
+@hug.cli("start_server")
+@hug.local()
+def start_server(conf_file):
+    """ Start the server of the engine """
+
+    try:
+        guni_conf_file = get_info(conf_file, "moon").strip('"\n')
+        port = get_info(conf_file, "bind").split(":")[1].strip('"\n')
+        log_dir = get_info(conf_file, "pid_file_dir").strip('"\n')
+    except ValueError:
+        return
+
+    configuration.init_logging(guni_conf_file)
+    LOGGER.setLevel(logging.ERROR)
+
+    pid_filename = log_dir + port + ".pid"
+    _command = ["gunicorn", "moon_engine.server:__hug_wsgi__", "-D", "-p", pid_filename, "-c", conf_file]
+    subprocess.Popen(_command, stdout=subprocess.PIPE, close_fds=True)  # nosec
+
+
+@hug.cli("stop_server")
+@hug.local()
+def stop_server(conf_file):
+    """ Stop the server of the engine """
+
+    try:
+        guni_conf_file = get_info(conf_file, "moon").strip('"\n')
+        port = get_info(conf_file, "bind").split(":")[1].strip('"\n')
+        log_dir = get_info(conf_file, "pid_file_dir").strip('"\n')
+    except ValueError:
+        return
+
+    configuration.init_logging(guni_conf_file)
+    LOGGER.setLevel(logging.ERROR)
+
+    pid_filename = log_dir + port + ".pid"
+
+    try:
+        pid_file = open(pid_filename, 'r')
+    except FileNotFoundError:
+        LOGGER.error(f"File {pid_filename} not found. Server on port {port} not running?")
+        return
+
+    try:
+        pid = int(pid_file.read())
+    except ValueError:
+        LOGGER.error(f"The pid found in {pid_filename} is not valid")
+        return
+
+    os.kill(pid, 15)
+
+
+def get_info(conf, key):
+    with open(conf) as config:
+        lines = config.readlines()
+        for line in lines:
+            if line.startswith(key):
+                return line.split("=")[1].strip()
+        LOGGER.error(f"Key \"{key}\" missing from Gunicorn configuration file")
+        raise ValueError
+
+
+def run():
+    if len(sys.argv) > 1:
+
+        command = sys.argv[1]
+        sys.argv.pop(1)
+        # if command == "conf":
+        #     configuration.get_configuration.interface.cli()
+        # elif command == "db":
+        #     configuration.init_database.interface.cli()
+        if command == "start":
+            start_server.interface.cli()
+        elif command == "stop":
+            stop_server.interface.cli()
+        else:
+            LOGGER.critical("Unknown command {}".format(command))
+
+    else:
+        # TODO: update the command management by using argparse
+        print("""Possible commands are:
+        # - conf
+        # - db
+        - start
+        - stop
+        """)
+
+
+if __name__ == "__main__":
+    run()