From 758f9dce90313a75e435a2ac99d32674d75e0c8d Mon Sep 17 00:00:00 2001
From: Super User <barankara5@gmail.com>
Date: Tue, 2 Jul 2024 13:06:13 +0300
Subject: [PATCH] Add service endpoint

---
 .../endpoints/__init__.py                     |   0
 .../endpoints/service.py                      | 125 ++++++++++++++++++
 2 files changed, 125 insertions(+)
 create mode 100644 webhook_to_fedora_messaging/endpoints/__init__.py
 create mode 100644 webhook_to_fedora_messaging/endpoints/service.py

diff --git a/webhook_to_fedora_messaging/endpoints/__init__.py b/webhook_to_fedora_messaging/endpoints/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/webhook_to_fedora_messaging/endpoints/service.py b/webhook_to_fedora_messaging/endpoints/service.py
new file mode 100644
index 0000000..190efa5
--- /dev/null
+++ b/webhook_to_fedora_messaging/endpoints/service.py
@@ -0,0 +1,125 @@
+from flask import Blueprint, Flask, request, Response, Request
+from ..database import db
+from ..models.service import Service
+from ..models.user import User
+from sqlalchemy_helpers import get_or_create
+from .util import not_found, success, bad_request, created, conflict, validate_request, exclude_from_val, unprocessable_entity
+
+
+app = Flask(__name__)
+service_endpoint = Blueprint("service_endpoint", __name__)
+
+
+@service_endpoint.route("/service", methods=["POST"])
+def create_service():
+    """
+    Used for creating a new service by sending a post request to /service/ path.
+    """
+    if not validate_request(request.json, fields=['username', 'type', 'desc', 'name']):
+        return unprocessable_entity()
+        
+    session = db.Session()
+    body = request.json
+    
+
+    user = session.query(User).filter(User.username == body['username']).first()
+    if user is None:
+        return not_found()
+    
+    service, is_created = get_or_create(session, Service, name=body['name'], type=body['type'], desc=body['desc'], user_id=user.id)
+    
+    if not is_created:
+        return conflict({'message': 'Service Already Exists'})
+    else:
+        return created({'message': 'Created', 'uuid': service.id})
+        
+    
+@service_endpoint.route("/service/search", methods=["GET"])
+def list_services():
+    """
+    Used for listing all services belong to a user by sending a get request to /service/search path
+    """
+    if not validate_request(request):
+        return unprocessable_entity()
+    
+    session = db.Session()
+    user = session.query(User).filter(User.username.like(request.json['username'])).first()
+    if user is None:
+        return not_found()
+    
+    services = session.query(Service).filter(Service.user_id == user.id).all()
+    
+    return success({'service_list': services})
+    
+    
+@service_endpoint.route("/service", methods=["GET"])
+def lookup_service():
+    """
+    Used for retrieving a service by it's uuid by sending a get request
+    to the /service path. 
+    
+    Request Body:
+        service_uuid: Service UUID
+    """
+    if not validate_request(request, ['service_uuid']):
+        return unprocessable_entity()
+    
+    session = db.Session()
+    service = session.query(Service).filter(Service.id == request.json['service_uuid']).first()
+    
+    if service is None:
+        return not_found()
+    else:
+        return success({'uuid': service.id, 'name': service.name, 'type': service.type, 'desc': service.desc})
+    
+    
+@service_endpoint.route("/service/revoke", methods=["PUT"])
+def revoke_service():
+    """
+    Used for revoking a service by sending a PUT request to /service/revoke path.
+    
+    Request Body:
+        service_uuid: Service UUID
+        username: Username of the user that servicce belongs to.
+    """
+    if not validate_request(request, fields=['username', 'service_uuid']):
+        return unprocessable_entity()
+    
+    session = db.Session()
+    user = session.query(User).filter(User.username == request.json['username']).first()
+    if user is None:
+        return not_found()
+    
+    service = session.query(Service).filter(Service.user_id == user.id).filter(Service.id == request.json['service_uuid']).first()
+    if service is None:
+        return not_found()
+    
+    service.disabled = True
+    session.commit()
+
+    return success({'uuid': service.id, 'is_valid': not service.disabled})
+    
+    
+@service_endpoint.route("/service", methods=["PUT"])
+def update_service():
+    """
+    Used for updating a service by sending a PUT request to /service path.
+    
+    Request Body:
+        uuid: UUID of the service
+        name: Updated name (optional)
+        mesg_body: Updated message body (optional)
+    
+    """
+    if not validate_request(request, fields=['uuid', 'name', 'mesg_body']):
+        return unprocessable_entity()
+    
+    session = db.Session()
+    service = session.query(Service).filter(Service.id == request.json['uuid']).first()
+    if service is None:
+        return not_found()
+    
+    service.name = request.json['name'] if request.json['name'] is not None and request.json['name'] != "" else service.name
+    service.desc = request.json['mesg_body'] if request.json['mesg_body'] is not None and request.json['mesg_body'] != "" else service.desc
+    session.commit()
+    return success({'uuid': service.id, 'name': service.name, 'mesg_body': service.desc, 'is_valid': not service.disabled})