initial iam

gisaia · Feb 6, 2024 · ca43219 · ca43219
1 parent 76917ea
commit ca43219
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 6 deletions.
diff --git a/arlas/cli/service.py b/arlas/cli/service.py
@@ -172,21 +172,24 @@ def __get_fields__(origin: list[str], properties: dict[str:dict]):
         return fields
 
     def __arlas__(arlas: str, suffix, post=None, put=None, delete=None):
+        __headers__ = Configuration.settings.arlas.get(arlas).server.headers.copy()
+        if Configuration.settings.arlas.get(arlas).authorization is not None:
+            __headers__["Authorization"] = "Bearer " + Service.__get_token__()
         endpoint = Configuration.settings.arlas.get(arlas)
         if endpoint is None:
             print("Error: arlas configuration ({}) not found among [{}].".format(arlas, ", ".join(Configuration.settings.arlas.keys())), file=sys.stderr)
             exit(1)
         url = "/".join([endpoint.server.location, suffix])
         if post:
-            r = requests.post(url, data=post, headers=endpoint.server.headers)
+            r = requests.post(url, data=post, headers=__headers__)
         else:
             if put:
-                r = requests.put(url, data=put, headers=endpoint.server.headers)
+                r = requests.put(url, data=put, headers=__headers__)
             else:
                 if delete:
-                    r = requests.delete(url, headers=endpoint.server.headers)
+                    r = requests.delete(url, headers=__headers__)
                 else:
-                    r = requests.get(url, headers=endpoint.server.headers)
+                    r = requests.get(url, headers=__headers__)
         if r.ok:
             return r.json()
         else:
@@ -238,3 +241,31 @@ def __fetch__(resource: Resource):
             print("Error: request failed with status {}: {}".format(str(r.status_code), r.content), file=sys.stderr)
             print("   url: {}".format(resource.location), file=sys.stderr)
             exit(1)
+
+    def __get_token__(arlas: str) -> str:
+        auth = Configuration.settings.arlas[arlas].authorization
+        if Configuration.settings.arlas[arlas].authorization.arlas_iam:
+            data = {
+                "email": auth.login,
+                "password": auth.password
+            }
+        else:
+            data = {
+                "client_id": auth.client_id,
+                "client_secret": auth.client_secret,
+                "username": auth.login,
+                "password": auth.password
+            }
+            if auth.grant_type:
+                data["grant_type"] = auth.grant_type
+
+        r = requests.post(
+            headers=auth.token_url.headers,
+            data=json.dumps(data),
+            url=auth.token_url.location)
+        if r.status_code >= 200 and r.status_code < 300:
+            return r.json()["accessToken"]
+        else:
+            print("Error: request to get token failed with status {}: {}".format(str(r.status_code), r.content), file=sys.stderr)
+            print("   url: {}".format(auth.token_url.location), file=sys.stderr)
+            exit(1)
diff --git a/arlas/cli/settings.py b/arlas/cli/settings.py
@@ -9,10 +9,19 @@ class Resource(BaseModel):
     headers: dict[str, str] | None = Field(default=None, title="List of headers, if needed, for http(s) requests")
 
 
+class AuthorizationService(BaseModel):
+    token_url: Resource = Field(default=None, title="Token URL of the authentication service")
+    login: str = Field(default=None, title="login")
+    password: str = Field(default=None, title="password")
+    client_id: str | None = Field(default=None, title="Client ID")
+    client_secret: str | None = Field(default=None, title="Client secret")
+    grant_type: str | None = Field(default=None, title="Grant type (e.g. password)")
+    arlas_iam: bool | None = Field(default=True, title="Is it an ARLAS IAM service?")
+
+
 class ARLAS(BaseModel):
     server: Resource = Field(title="ARLAS Server")
-    iam: Resource | None = Field(default=None, title="ARLAS IAM URL")
-    keycloak: Resource | None = Field(default=None, title="Keycloak URL")
+    authorization: AuthorizationService | None = Field(default=None, title="Keycloak URL")
     elastic: Resource | None = Field(default=None, title="dictionary of name/es resources")
     allow_delete: bool | None = Field(default=False, title="Is delete command allowed for this configuration?")