added get_thread_messages and create_forum_thread endpoints

server/api/api.py

@@ -2,7 +2,6 @@ import flask
 
 from . import user_api
 from . import forum_api
-from . import thread_api
 
 api = flask.Blueprint(
     name='api',
@@ -11,4 +10,3 @@ api = flask.Blueprint(
 )
 api.register_blueprint(user_api.user)
 api.register_blueprint(forum_api.forum)
-api.register_blueprint(thread_api.thread)

server/api/forum_api.py

@@ -16,7 +16,7 @@ forum = flask.Blueprint(
 )
 
 
-@forum.route('/forum/send_message', methods=['POST'])
+@forum.route('/send_message', methods=['POST'])
 @flask_login.login_required
 def send_message() -> tuple[str, int]:
 
@@ -42,16 +42,16 @@ def send_message() -> tuple[str, int]:
 
     return '', 201
 
-@forum.route('/forum/get_threads')
+@forum.route('/get_threads')
 @flask_login.login_required
 def get_forum_threads() -> tuple[flask.Response, int]:
     """
     Get all threads from forums the current user has access to.
     """
-    user_id: str = flask_login.current_user.get_id()
+    user_id: int = flask_login.current_user.get_id()
 
     # Get all forums the user has access to
-    user_forums: list[mgU.Forum] = database.get_user_forums(int(user_id))
+    user_forums: list[mgU.Forum] = database.get_user_forums(user_id)
 
     if not user_forums:
         return flask.jsonify({"threads": []}), 200
@@ -68,3 +68,53 @@ def get_forum_threads() -> tuple[flask.Response, int]:
 
     return flask.jsonify(threads_dict), 200
 
+@forum.route('/get_messages')
+@flask_login.login_required
+def get_thread_messages() -> tuple[flask.Response, int]:
+    thread_id = flask.request.args.get("thread", type=int)
+    if thread_id is None:
+        return flask.jsonify({"error": "Missing thread parameter"}), 400
+
+    messages = database.get_messages_by_thread(thread_id)
+    messages_dicts = {"messages" : [dataclasses.asdict(msg) for msg in messages]}
+    return flask.jsonify(messages_dicts), 200
+
+@forum.route('/create_thread', methods=['POST'])
+@flask_login.login_required
+def create_forum_thread() -> tuple[flask.Response, int] | tuple[str, int]:
+
+    data = flask.request.json
+
+    if not data:
+        return '', 400
+
+    if any([not key in data for key in ['forum_id', 'name']]):
+        return '', 400
+
+    if type(data['forum_id']) != int:
+        return '', 400
+
+    forum_id: int = data.get('forum_id')
+    thread_name: str = data.get('name')
+
+    if not thread_name or not thread_name.strip():
+        return flask.jsonify({"error": "Thread name cannot be empty"}), 400
+
+    user_id: int = flask_login.current_user.get_id()
+
+    user_owns_forum: bool = helpers.user_owns_forum(user_id, forum_id)
+
+    if not user_owns_forum:
+        return flask.jsonify({"error": "You don't have permission to create threads in this forum"}), 403
+
+    # Create the thread
+    new_thread = database.create_thread(thread_name.strip(), forum_id)
+
+    if new_thread is None:
+        return flask.jsonify({"error": "Failed to create thread"}), 500
+
+    # Return the created thread
+    thread_dict = dataclasses.asdict(new_thread)
+    return flask.jsonify({"thread": thread_dict, "message": "Thread created successfully"}), 201
+
+

server/api/helpers.py

@@ -24,3 +24,11 @@ def user_in_thread(user_id: int, thread_id: int) -> bool:
                 return True
 
     return False
+
+def user_owns_forum(user_id: int, forum_id: int) -> bool:
+    forum_owner_ids: list[int] = database.get_forum_owners(forum_id)
+
+    if not forum_owner_ids:
+        return False
+
+    return user_id in forum_owner_ids

server/api/thread_api.py

@@ -1,24 +0,0 @@
-import dataclasses
-
-import flask
-import flask_login
-
-import database
-
-
-thread = flask.Blueprint(
-    name='thread_api',
-    import_name=__name__,
-    url_prefix='/thread/'
-)
-
-@thread.route('/get_messages')
-@flask_login.login_required
-def get_thread_messages() -> tuple[flask.Response, int]:
-    thread_id = flask.request.args.get("thread", type=int)
-    if thread_id is None:
-        return flask.jsonify({"error": "Missing thread parameter"}), 400
-
-    messages = database.get_messages_by_thread(thread_id)
-    messages_dicts = {"messages" : [dataclasses.asdict(msg) for msg in messages]}
-    return flask.jsonify(messages_dicts), 200