From 16e4fc4158e46a3578854c6f71bcf49e3470fe28 Mon Sep 17 00:00:00 2001
From: Oscar Krause <oscar.krause@collinwebdesigns.de>
Date: Mon, 19 Dec 2022 13:15:19 +0100
Subject: [PATCH] added lease update request (PUT)

---
 app/main.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/app/main.py b/app/main.py
index ea9de3c..2af2722 100644
--- a/app/main.py
+++ b/app/main.py
@@ -18,6 +18,8 @@ from helper import load_key, private_bytes, public_key
 
 app = FastAPI()
 
+LEASE_EXPIRE_DELTA = relativedelta(minutes=15)  # days=90
+
 URL = '192.168.178.196'
 SITE_KEY_FILE = load_key('/opt/fastapi-dls/site.key')
 SITE_KEY_XID = '00000000-0000-0000-0000-000000000000'
@@ -216,7 +218,7 @@ async def lessor(request: Request):
             "lease": {
                 "ref": scope_ref,
                 "created": cur_time,
-                "expires": cur_time + relativedelta(minutes=15),  # days=90
+                "expires": cur_time + LEASE_EXPIRE_DELTA,
                 "recommended_lease_renewal": 0.15,
                 "offline_lease": "true",
                 "license_type": "CONCURRENT_COUNTED_SINGLE"
@@ -250,6 +252,25 @@ async def lease(request: Request):
     return response
 
 
+# venv/lib/python3.9/site-packages/nls_core_lease/lease_single.py
+@app.put('/leasing/v1/lease/{lease_ref}')
+async def lease_renew(request: Request, lease_ref: str):
+    print('> renew')
+
+    cur_time = datetime.utcnow()
+
+    response = {
+        "lease_ref": lease_ref,
+        "expires": cur_time + LEASE_EXPIRE_DELTA,
+        "recommended_lease_renewal": 0.16,
+        "offline_lease": True,
+        "prompts": None,
+        "sync_timestamp": cur_time
+    }
+
+    return response
+
+
 @app.delete('/leasing/v1/lessor/leases')
 async def lease_remove(request: Request, status_code=200):
     cur_time = datetime.utcnow()