111 lines
3.7 KiB
Python
Executable File
111 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# Daemon.py - PVC HTTP API daemon
|
|
# Part of the Parallel Virtual Cluster (PVC) system
|
|
#
|
|
# Copyright (C) 2018-2024 Joshua M. Boniface <joshua@boniface.me>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, version 3.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
#
|
|
###############################################################################
|
|
|
|
|
|
from ssl import SSLContext, TLSVersion
|
|
|
|
from distutils.util import strtobool as dustrtobool
|
|
|
|
import daemon_lib.config as cfg
|
|
|
|
# Daemon version
|
|
version = "0.9.99"
|
|
|
|
# API version
|
|
API_VERSION = 1.0
|
|
|
|
|
|
##########################################################
|
|
# Helper Functions
|
|
##########################################################
|
|
|
|
|
|
def strtobool(stringv):
|
|
if stringv is None:
|
|
return False
|
|
if isinstance(stringv, bool):
|
|
return bool(stringv)
|
|
try:
|
|
return bool(dustrtobool(stringv))
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
##########################################################
|
|
# Configuration Parsing
|
|
##########################################################
|
|
|
|
|
|
# Get our configuration
|
|
config = cfg.get_configuration()
|
|
config["daemon_name"] = "pvcapid"
|
|
config["daemon_version"] = version
|
|
|
|
|
|
##########################################################
|
|
# Entrypoint
|
|
##########################################################
|
|
|
|
|
|
def entrypoint():
|
|
import pvcapid.flaskapi as pvc_api # noqa: E402
|
|
|
|
if config["api_ssl_enabled"]:
|
|
context = SSLContext()
|
|
context.minimum_version = TLSVersion.TLSv1
|
|
context.get_ca_certs()
|
|
context.load_cert_chain(
|
|
config["api_ssl_cert_file"], keyfile=config["api_ssl_key_file"]
|
|
)
|
|
else:
|
|
context = None
|
|
|
|
# Print our startup messages
|
|
print("")
|
|
print("|--------------------------------------------------------------|")
|
|
print("| |")
|
|
print("| ███████████ ▜█▙ ▟█▛ █████ █ █ █ |")
|
|
print("| ██ ▜█▙ ▟█▛ ██ |")
|
|
print("| ███████████ ▜█▙ ▟█▛ ██ |")
|
|
print("| ██ ▜█▙▟█▛ ███████████ |")
|
|
print("| |")
|
|
print("|--------------------------------------------------------------|")
|
|
print("| Parallel Virtual Cluster API daemon v{0: <23} |".format(version))
|
|
print("| Debug: {0: <53} |".format(str(config["debug"])))
|
|
print("| API version: v{0: <46} |".format(API_VERSION))
|
|
print(
|
|
"| Listen: {0: <52} |".format(
|
|
"{}:{}".format(config["api_listen_address"], config["api_listen_port"])
|
|
)
|
|
)
|
|
print("| SSL: {0: <55} |".format(str(config["api_ssl_enabled"])))
|
|
print("| Authentication: {0: <44} |".format(str(config["api_auth_enabled"])))
|
|
print("|--------------------------------------------------------------|")
|
|
print("")
|
|
|
|
pvc_api.celery_startup()
|
|
pvc_api.app.run(
|
|
config["api_listen_address"],
|
|
config["api_listen_port"],
|
|
threaded=True,
|
|
ssl_context=context,
|
|
)
|