feat: add category selection to post edit form and badges to post list

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

cmds/server.go

@@ -109,7 +109,7 @@ Starting weiro without any arguments will start the server.
 
 			ih := handlers.IndexHandler{SiteService: svcs.Sites}
 			lh := handlers.LoginHandler{Config: cfg, AuthService: svcs.Auth}
-			ph := handlers.PostsHandler{PostService: svcs.Posts}
+			ph := handlers.PostsHandler{PostService: svcs.Posts, CategoryService: svcs.Categories}
 			uh := handlers.UploadsHandler{UploadsService: svcs.Uploads}
 			ssh := handlers.SiteSettingsHandler{SiteService: svcs.Sites}
 			ch := handlers.CategoriesHandler{CategoryService: svcs.Categories}

handlers/posts.go

@@ -6,11 +6,13 @@ import (
 
 	"github.com/gofiber/fiber/v3"
 	"lmika.dev/lmika/weiro/models"
+	"lmika.dev/lmika/weiro/services/categories"
 	"lmika.dev/lmika/weiro/services/posts"
 )
 
 type PostsHandler struct {
 	PostService     *posts.Service
+	CategoryService *categories.Service
 }
 
 func (ph PostsHandler) Index(c fiber.Ctx) error {
@@ -42,8 +44,15 @@ func (ph PostsHandler) New(c fiber.Ctx) error {
 		State: models.StateDraft,
 	}
 
+	cats, err := ph.CategoryService.ListCategories(c.Context())
+	if err != nil {
+		return err
+	}
+
 	return c.Render("posts/edit", fiber.Map{
 		"post":               p,
+		"categories":         cats,
+		"selectedCategories": map[int64]bool{},
 	})
 }
 
@@ -62,11 +71,28 @@ func (ph PostsHandler) Edit(c fiber.Ctx) error {
 		return err
 	}
 
+	cats, err := ph.CategoryService.ListCategories(c.Context())
+	if err != nil {
+		return err
+	}
+
+	postCats, err := ph.PostService.GetPostCategories(c.Context(), postID)
+	if err != nil {
+		return err
+	}
+
+	selectedCategories := make(map[int64]bool)
+	for _, pc := range postCats {
+		selectedCategories[pc.ID] = true
+	}
+
 	return accepts(c, json(func() any {
 		return post
 	}), html(func(c fiber.Ctx) error {
 		return c.Render("posts/edit", fiber.Map{
 			"post":               post,
+			"categories":         cats,
+			"selectedCategories": selectedCategories,
 		})
 	}))
 }

services/posts/create.go

@@ -14,6 +14,7 @@ type CreatePostParams struct {
 	Title       string  `form:"title" json:"title"`
 	Body        string  `form:"body" json:"body"`
 	Action      string  `form:"action" json:"action"`
+	CategoryIDs []int64 `form:"category_ids" json:"category_ids"`
 }
 
 func (s *Service) UpdatePost(ctx context.Context, params CreatePostParams) (*models.Post, error) {
@@ -53,7 +54,21 @@ func (s *Service) UpdatePost(ctx context.Context, params CreatePostParams) (*mod
 		// Leave unchanged
 	}
 
-	if err := s.db.SavePost(ctx, post); err != nil {
+	// Use a transaction for atomicity of post save + category reassignment
+	tx, err := s.db.BeginTx(ctx)
+	if err != nil {
+		return nil, err
+	}
+	defer tx.Rollback()
+
+	txDB := s.db.QueriesWithTx(tx)
+	if err := txDB.SavePost(ctx, post); err != nil {
+		return nil, err
+	}
+	if err := txDB.SetPostCategories(ctx, post.ID, params.CategoryIDs); err != nil {
+		return nil, err
+	}
+	if err := tx.Commit(); err != nil {
 		return nil, err
 	}
 

services/posts/list.go

@@ -7,7 +7,12 @@ import (
 	"lmika.dev/lmika/weiro/providers/db"
 )
 
-func (s *Service) ListPosts(ctx context.Context, showDeleted bool) ([]*models.Post, error) {
+type PostWithCategories struct {
+	*models.Post
+	Categories []*models.Category
+}
+
+func (s *Service) ListPosts(ctx context.Context, showDeleted bool) ([]*PostWithCategories, error) {
 	site, ok := models.GetSite(ctx)
 	if !ok {
 		return nil, models.SiteRequiredError
@@ -21,7 +26,15 @@ func (s *Service) ListPosts(ctx context.Context, showDeleted bool) ([]*models.Po
 		return nil, err
 	}
 
-	return posts, nil
+	result := make([]*PostWithCategories, len(posts))
+	for i, post := range posts {
+		cats, err := s.db.SelectCategoriesOfPost(ctx, post.ID)
+		if err != nil {
+			return nil, err
+		}
+		result[i] = &PostWithCategories{Post: post, Categories: cats}
+	}
+	return result, nil
 }
 
 func (s *Service) GetPost(ctx context.Context, pid int64) (*models.Post, error) {
@@ -32,3 +45,7 @@ func (s *Service) GetPost(ctx context.Context, pid int64) (*models.Post, error)
 
 	return post, nil
 }
+
+func (s *Service) GetPostCategories(ctx context.Context, postID int64) ([]*models.Category, error) {
+	return s.db.SelectCategoriesOfPost(ctx, postID)
+}

views/posts/edit.html

@@ -4,6 +4,8 @@
     data-controller="postedit"
         data-action="keydown.meta+s->postedit#save keydown.meta+enter->postedit#publish"
         data-postedit-save-action-value="{{ if $isPublished }}Update{{ else }}Save Draft{{ end }}">
+    <div class="row">
+      <div class="col-md-9">
         <input type="hidden" name="guid" value="{{ .post.GUID }}">
         <div class="mb-2">
           <input type="text" name="title" class="form-control" placeholder="Title" value="{{ .post.Title }}">
@@ -19,5 +21,24 @@
             <input type="submit" name="action" class="btn btn-secondary mt-2" value="Save Draft">
           {{ end }}
         </div>
+      </div>
+      <div class="col-md-3">
+        <div class="card">
+          <div class="card-header">Categories</div>
+          <div class="card-body">
+            {{ range .categories }}
+              <div class="form-check">
+                <input class="form-check-input" type="checkbox" name="category_ids"
+                       value="{{ .ID }}" id="cat-{{ .ID }}"
+                       {{ if index $.selectedCategories .ID }}checked{{ end }}>
+                <label class="form-check-label" for="cat-{{ .ID }}">{{ .Name }}</label>
+              </div>
+            {{ else }}
+              <span class="text-muted">No categories yet.</span>
+            {{ end }}
+          </div>
+        </div>
+      </div>
+    </div>
   </form>
 </main>

views/posts/index.html

@@ -26,11 +26,14 @@
         {{ if $p.Title }}<h4 class="mb-3">{{ $p.Title }}</h4>{{ end }}
         {{ markdown $p.Body $.site }}
 
-        <div class="mb-3 d-flex align-items-center">
-          {{ if eq .State 1 }}
-            <span class="text-muted">{{ $.user.FormatTime .UpdatedAt }}</span> <span class="ms-3 badge text-primary-emphasis bg-primary-subtle border border-primary-subtle">Draft</span>
+        <div class="mb-3 d-flex align-items-center flex-wrap gap-1">
+          {{ if eq $p.State 1 }}
+            <span class="text-muted">{{ $.user.FormatTime $p.UpdatedAt }}</span> <span class="ms-3 badge text-primary-emphasis bg-primary-subtle border border-primary-subtle">Draft</span>
           {{ else }}
-            <span class="text-muted">{{ $.user.FormatTime .PublishedAt }}</span>
+            <span class="text-muted">{{ $.user.FormatTime $p.PublishedAt }}</span>
+          {{ end }}
+          {{ range $p.Categories }}
+            <span class="ms-1 badge bg-secondary-subtle text-secondary-emphasis border border-secondary-subtle">{{ .Name }}</span>
           {{ end }}
         </div>