From ce6f9a174f59b6f90ed8ae24e78a42c3aece3d21 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Tue, 5 Aug 2025 17:49:52 +0800
Subject: [PATCH] :bug: Fix pub name

---
 DysonNetwork.Sphere/Poll/PollController.cs | 2 +-
 DysonNetwork.Sphere/Post/PostController.cs | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/DysonNetwork.Sphere/Poll/PollController.cs b/DysonNetwork.Sphere/Poll/PollController.cs
index 862b096..7dff222 100644
--- a/DysonNetwork.Sphere/Poll/PollController.cs
+++ b/DysonNetwork.Sphere/Poll/PollController.cs
@@ -107,7 +107,7 @@ public class PollController(AppDatabase db, PollService polls, PublisherService
 
     [HttpPost]
     [Authorize]
-    public async Task<ActionResult<Poll>> CreatePoll([FromBody] PollRequest request, [FromQuery] string pubName)
+    public async Task<ActionResult<Poll>> CreatePoll([FromBody] PollRequest request, [FromQuery(Name = "pub")] string pubName)
     {
         if (request.Questions is null) return BadRequest("Questions are required.");
         if (HttpContext.Items["CurrentUser"] is not Account currentUser) return Unauthorized();
diff --git a/DysonNetwork.Sphere/Post/PostController.cs b/DysonNetwork.Sphere/Post/PostController.cs
index 4f5e100..1d1e9ee 100644
--- a/DysonNetwork.Sphere/Post/PostController.cs
+++ b/DysonNetwork.Sphere/Post/PostController.cs
@@ -281,8 +281,7 @@ public class PostController(
     [RequiredPermission("global", "posts.create")]
     public async Task<ActionResult<Post>> CreatePost(
         [FromBody] PostRequest request,
-        [FromQuery(Name = "pub")] [FromHeader(Name = "X-Pub")]
-        string? pubName
+        [FromQuery(Name = "pub")] string? pubName
     )
     {
         request.Content = TextSanitizer.Sanitize(request.Content);
@@ -303,7 +302,7 @@ public class PostController(
         {
             publisher = await pub.GetPublisherByName(pubName);
             if (publisher is null) return BadRequest("Publisher was not found.");
-            if(!await pub.IsMemberWithRole(publisher.Id, accountId, PublisherMemberRole.Editor))
+            if (!await pub.IsMemberWithRole(publisher.Id, accountId, PublisherMemberRole.Editor))
                 return StatusCode(403, "You need at least be an editor to post as this publisher.");
         }