[app] Use HalfAPI class to be able to use custom configuration

à

halfapi/app.py

@@ -23,7 +23,6 @@ from timing_asgi import TimingMiddleware
 from timing_asgi.integrations import StarletteScopeToName
 
 # module libraries
-from halfapi.conf import CONFIG, SECRET, PRODUCTION, DOMAINSDICT
 
 from .lib.domain_middleware import DomainMiddleware
 from .lib.timing import HTimingClient
@@ -34,10 +33,24 @@ from halfapi.lib.responses import (ORJSONResponse, UnauthorizedResponse,
     NotFoundResponse, InternalServerErrorResponse, NotImplementedResponse)
 
 from halfapi.lib.routes import gen_starlette_routes, debug_routes
-from halfapi.lib.schemas import get_api_routes, schema_json, get_acls
+from halfapi.lib.schemas import get_api_routes, get_api_domain_routes, schema_json, get_acls
 
 logger = logging.getLogger('uvicorn.asgi')
 
+
+class HalfAPI:
+    def __init__(self, config=None):
+        if config:
+            SECRET = config.get('SECRET')
+            PRODUCTION = config.get('PRODUCTION')
+            DOMAINS = config.get('DOMAINS')
+            CONFIG = config.get('CONFIG', {
+                'domains': DOMAINS
+            })
+        else:
+            from halfapi.conf import CONFIG, SECRET, PRODUCTION, DOMAINS
+
+
         routes = [ Route('/', get_api_routes) ]
 
 
@@ -57,12 +70,20 @@ if not PRODUCTION:
                 routes.append( route )
 
 
-if DOMAINSDICT:
-    for route in gen_starlette_routes(DOMAINSDICT()):
+        if DOMAINS:
+            for route in gen_starlette_routes(DOMAINS):
                 routes.append(route)
 
+            for domain in DOMAINS:
+                routes.append(
+                    Route(
+                        f'/{domain}',
+                        get_api_domain_routes(domain)
+                    )
+                )
 
-application = Starlette(
+
+        self.application = Starlette(
             debug=not PRODUCTION,
             routes=routes,
             exception_handlers={
@@ -73,22 +94,26 @@ application = Starlette(
             }
         )
 
-if DOMAINSDICT:
-    application.add_middleware(
+        print(f'HALFAPI CONFIG: {CONFIG}')
+        self.application.add_middleware(
             DomainMiddleware,
             config=CONFIG
         )
 
         if SECRET:
-    application.add_middleware(
+            self.application.add_middleware(
                 AuthenticationMiddleware,
                 backend=JWTAuthenticationBackend(secret_key=SECRET)
             )
 
         if not PRODUCTION:
-    application.add_middleware(
+            self.application.add_middleware(
                 TimingMiddleware,
                 client=HTimingClient(),
                 metric_namer=StarletteScopeToName(prefix="halfapi",
-        starlette_app=application)
+                starlette_app=self.application)
             )
+
+
+
+application = HalfAPI().application

halfapi/conf.py

@@ -53,16 +53,11 @@ LOGLEVEL = 'info'
 HOST = '127.0.0.1'
 PORT = '3000'
 SECRET = ''
+CONF_FILE = os.environ.get('HALFAPI_CONF_FILE', '.halfapi/config')
 
-IS_PROJECT = os.path.isfile('.halfapi/config')
+is_project = lambda: os.path.isfile(CONF_FILE)
 
 
-CONFIG = {
-    'project_name': PROJECT_NAME,
-    'production': PRODUCTION,
-    'secret': SECRET,
-    'domains': DOMAINS
-}
 
 default_config = {
     'project': {
@@ -84,7 +79,7 @@ HALFAPI_ETC_FILE=os.path.join(
 HALFAPI_DOT_FILE=os.path.join(
     os.getcwd(), '.halfapi', 'config')
 
-HALFAPI_CONFIG_FILES = [ HALFAPI_ETC_FILE, HALFAPI_DOT_FILE ]
+HALFAPI_CONFIG_FILES = [ CONF_FILE, HALFAPI_DOT_FILE ]
 
 def conf_files():
     return [
@@ -121,8 +116,11 @@ def read_config():
 
 
 
-if IS_PROJECT:
-    read_config()
+CONFIG = {}
+IS_PROJECT = False
+if is_project():
+    IS_PROJECT = True
+    CONFIG = read_config()
 
     PROJECT_NAME = config.get('project', 'name', fallback=PROJECT_NAME)
 
@@ -150,3 +148,11 @@ if IS_PROJECT:
     LOGLEVEL = config.get('project', 'loglevel').lower() or 'info'
 
     BASE_DIR = config.get('project', 'base_dir', fallback='.') #os.getcwd())
+
+    CONFIG = {
+        'project_name': PROJECT_NAME,
+        'production': PRODUCTION,
+        'secret': SECRET,
+        'domains': DOMAINS
+    }
+