test: debug jwt

Author: vancauwe

README.md

@@ -97,9 +97,9 @@ the output should be: `{"data":{"type":"test","id":"1","attributes":{"message":"
 6. Try the inference endpoint:
 
 ```bash
-curl -X GET "http://localhost:8000/v1/inference/epfl/YOUR_DB" \
+curl -X GET "http://localhost:8000/v1/inference/epfl/neo4j" \
   -H "accept: application/json" \
-  -H "Authorization: Bearer YOUR_JWT_TOKEN"
+  -H "Authorization: Bearer YOUR_TOKEN"
 ```
 
 #### Inference

src/inference_api.py

@@ -17,18 +17,17 @@
 
 
 def verify_jwt(credentials: HTTPAuthorizationCredentials = Depends(security)):
+    print(JWT_SECRET_KEY)
     token = credentials.credentials
     print("token")
     print(token)
-    payload = jwt.decode(token, JWT_SECRET_KEY, algorithms=[JWT_ALGORITHM])
-    print("payload")
-    print(payload)
-    print("service: ", payload.get("service"))
+    header = jwt.get_unverified_header(token)
+    print(header)
     try:
-        payload = jwt.decode(token, JWT_SECRET_KEY, algorithms=[JWT_ALGORITHM])
-        if payload.get("service") != "airflow":
+        claims = jwt.decode(token, JWT_SECRET_KEY, algorithms=[JWT_ALGORITHM])
+        if claims.get("service") != "airflow":
             raise HTTPException(status_code=403, detail="Invalid service identity")
-        return payload
+        return claims
     except JWTError as e:
         print("JWT decode error:", e)
         raise HTTPException(status_code=403, detail="Invalid token")

src/utils/api_token.py

@@ -22,22 +22,21 @@
 
 # FOR RUNAI INFERENCE API
 print("All configurations for token in place:")
-print(JWT_SECRET_KEY, JWT_ALGORITHM)
-payload = {
+print(datetime.utcnow() + timedelta(hours=3))
+claims = {
     "sub": AIRFLOW_SUB,
     "service": AIRFLOW_SERVICE,
-    "exp": datetime.utcnow() + timedelta(hours=1),
+    "exp": datetime.utcnow() + timedelta(hours=3),
     "iat": datetime.utcnow(),
 }
-token = jwt.encode(payload, JWT_SECRET_KEY, algorithm=JWT_ALGORITHM)
+token = jwt.encode(claims, JWT_SECRET_KEY, algorithm=JWT_ALGORITHM)
 print("Validating token")
 credentials = HTTPAuthorizationCredentials(scheme="Bearer", credentials=token)
-print(credentials)
-payload = verify_jwt(credentials)
-print(payload)
+claims = verify_jwt(credentials)
+print(claims)
 print(f"Generated token: {token}")
 
 # for AIRFLOW API
-# payload = {"username": AIRFLOW_USERNAME, "password": AIRFLOW_PASSWORD}
+# claims = {"username": AIRFLOW_USERNAME, "password": AIRFLOW_PASSWORD}
 # headers = {"Authorization": f"Bearer {token}"}
 # res = requests.get(INFERENCE_URL + f"/inference/{NEO4J_DATABASE}", headers=headers)