Add hashtag extraction and management in CreatePost and EditPost logic; improve error handling

app/internal/logic/createpostlogic.go

@@ -43,6 +43,13 @@ func (l *CreatePostLogic) CreatePost(req *types.CreatePostReq) (resp *types.Crea
 		return nil, fmt.Errorf("invalid content format")
 	}
 
+	// Extract all hashtags from content
+	hashtags, err := utils.ExtractHashtagsFromContent(req.Content)
+	if err != nil {
+		l.Errorf("Failed to parse hashtags from content: %v", err)
+		return nil, fmt.Errorf("invalid content format")
+	}
+
 	// Migrate cover image from tmpfiles to files if provided
 	var coverID sql.NullInt64
 	if req.CoverImageKey != "" {
@@ -72,6 +79,36 @@ func (l *CreatePostLogic) CreatePost(req *types.CreatePostReq) (resp *types.Crea
 		return nil, err
 	}
 
+	// Create or get hashtag IDs and link them to the post
+	for _, tagName := range hashtags {
+		var tagID int64
+
+		// Try to get existing hashtag
+		getTagQuery := `SELECT id FROM hashtags WHERE name = $1`
+		err := tx.QueryRowContext(l.ctx, getTagQuery, tagName).Scan(&tagID)
+
+		if err == sql.ErrNoRows {
+			// Hashtag doesn't exist, create it
+			createTagQuery := `INSERT INTO hashtags (name) VALUES ($1) RETURNING id`
+			err = tx.QueryRowContext(l.ctx, createTagQuery, tagName).Scan(&tagID)
+			if err != nil {
+				l.Errorf("Failed to create hashtag %s: %v", tagName, err)
+				continue
+			}
+		} else if err != nil {
+			l.Errorf("Failed to get hashtag %s: %v", tagName, err)
+			continue
+		}
+
+		// Link hashtag to post
+		linkQuery := `INSERT INTO post_hashtags (post_id, hashtag_id) VALUES ($1, $2) ON CONFLICT DO NOTHING`
+		_, err = tx.ExecContext(l.ctx, linkQuery, postID, tagID)
+		if err != nil {
+			l.Errorf("Failed to link hashtag %s to post: %v", tagName, err)
+			// Continue with other hashtags
+		}
+	}
+
 	// Commit transaction
 	if err = tx.Commit(); err != nil {
 		l.Errorf("Failed to commit transaction: %v", err)

app/internal/logic/deletepostlogic.go

@@ -2,6 +2,7 @@ package logic
 
 import (
 	"context"
+	"database/sql"
 	"fmt"
 
 	"git.cialloo.com/CiallooWeb/Blog/app/internal/svc"
@@ -26,23 +27,57 @@ func NewDeletePostLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Delete
 }
 
 func (l *DeletePostLogic) DeletePost(req *types.DeletePostReq) (resp *types.DeletePostResp, err error) {
+	// Start a transaction
+	tx, err := l.svcCtx.DB.BeginTx(l.ctx, nil)
+	if err != nil {
+		l.Errorf("Failed to begin transaction: %v", err)
+		return nil, err
+	}
+	defer tx.Rollback()
+
+	// Get cover_id before deleting the post
+	var coverID sql.NullInt64
+	getCoverQuery := `SELECT cover_id FROM posts WHERE id = $1`
+	err = tx.QueryRowContext(l.ctx, getCoverQuery, req.PostId).Scan(&coverID)
+	if err != nil {
+		if err == sql.ErrNoRows {
+			l.Errorf("Post not found with id: %s", req.PostId)
+			return nil, fmt.Errorf("post not found")
+		}
+		l.Errorf("Failed to get post: %v", err)
+		return nil, err
+	}
+
+	// Delete hashtag associations (post_hashtags will be deleted by CASCADE)
+	deleteHashtagsQuery := `DELETE FROM post_hashtags WHERE post_id = $1`
+	_, err = tx.ExecContext(l.ctx, deleteHashtagsQuery, req.PostId)
+	if err != nil {
+		l.Errorf("Failed to delete hashtag associations: %v", err)
+		return nil, err
+	}
+
 	// Delete post from database
-	query := `DELETE FROM posts WHERE id = $1`
-	result, err := l.svcCtx.DB.ExecContext(l.ctx, query, req.PostId)
+	deleteQuery := `DELETE FROM posts WHERE id = $1`
+	_, err = tx.ExecContext(l.ctx, deleteQuery, req.PostId)
 	if err != nil {
 		l.Errorf("Failed to delete post: %v", err)
 		return nil, err
 	}
 
-	// Check if post exists
-	rowsAffected, err := result.RowsAffected()
-	if err != nil {
-		l.Errorf("Failed to get rows affected: %v", err)
-		return nil, err
+	// Delete cover file if exists
+	if coverID.Valid {
+		deleteFileQuery := `DELETE FROM files WHERE id = $1`
+		_, err = tx.ExecContext(l.ctx, deleteFileQuery, coverID.Int64)
+		if err != nil {
+			l.Errorf("Failed to delete cover file: %v", err)
+			// Don't fail the whole operation for this
+		}
 	}
-	if rowsAffected == 0 {
-		l.Errorf("Post not found with id: %s", req.PostId)
-		return nil, fmt.Errorf("post not found")
+
+	// Commit transaction
+	if err = tx.Commit(); err != nil {
+		l.Errorf("Failed to commit transaction: %v", err)
+		return nil, err
 	}
 
 	return &types.DeletePostResp{}, nil

app/internal/logic/editpostlogic.go

@@ -56,6 +56,13 @@ func (l *EditPostLogic) EditPost(req *types.EditPostReq) (resp *types.EditPostRe
 		return nil, fmt.Errorf("invalid content format")
 	}
 
+	// Extract all hashtags from content
+	hashtags, err := utils.ExtractHashtagsFromContent(req.Content)
+	if err != nil {
+		l.Errorf("Failed to parse hashtags from content: %v", err)
+		return nil, fmt.Errorf("invalid content format")
+	}
+
 	// Migrate new cover image from tmpfiles to files if provided
 	var coverID sql.NullInt64
 	if req.CoverImageKey != "" {
@@ -102,6 +109,45 @@ func (l *EditPostLogic) EditPost(req *types.EditPostReq) (resp *types.EditPostRe
 		}
 	}
 
+	// Update hashtags - remove old ones and add new ones
+	// First, remove all existing hashtag associations for this post
+	deleteHashtagsQuery := `DELETE FROM post_hashtags WHERE post_id = $1`
+	_, err = tx.ExecContext(l.ctx, deleteHashtagsQuery, req.PostId)
+	if err != nil {
+		l.Errorf("Failed to delete old hashtags: %v", err)
+		return nil, err
+	}
+
+	// Create or get hashtag IDs and link them to the post
+	for _, tagName := range hashtags {
+		var tagID int64
+
+		// Try to get existing hashtag
+		getTagQuery := `SELECT id FROM hashtags WHERE name = $1`
+		err := tx.QueryRowContext(l.ctx, getTagQuery, tagName).Scan(&tagID)
+
+		if err == sql.ErrNoRows {
+			// Hashtag doesn't exist, create it
+			createTagQuery := `INSERT INTO hashtags (name) VALUES ($1) RETURNING id`
+			err = tx.QueryRowContext(l.ctx, createTagQuery, tagName).Scan(&tagID)
+			if err != nil {
+				l.Errorf("Failed to create hashtag %s: %v", tagName, err)
+				continue
+			}
+		} else if err != nil {
+			l.Errorf("Failed to get hashtag %s: %v", tagName, err)
+			continue
+		}
+
+		// Link hashtag to post
+		linkQuery := `INSERT INTO post_hashtags (post_id, hashtag_id) VALUES ($1, $2) ON CONFLICT DO NOTHING`
+		_, err = tx.ExecContext(l.ctx, linkQuery, req.PostId, tagID)
+		if err != nil {
+			l.Errorf("Failed to link hashtag %s to post: %v", tagName, err)
+			// Continue with other hashtags
+		}
+	}
+
 	// Commit transaction
 	if err = tx.Commit(); err != nil {
 		l.Errorf("Failed to commit transaction: %v", err)