rust-course/server/internal/story/posts.go

package story

import (
	"database/sql"
	"fmt"
	"net/http"
	"sort"
	"strings"

	"github.com/imdotdev/im.dev/server/internal/interaction"
	"github.com/imdotdev/im.dev/server/internal/tags"
	"github.com/imdotdev/im.dev/server/pkg/db"
	"github.com/imdotdev/im.dev/server/pkg/e"
	"github.com/imdotdev/im.dev/server/pkg/models"
)

const PostQueryPrefix = "select id,type,slug,title,url,cover,brief,creator,owner,created,updated from story "

func HomePosts(user *models.User, filter string) (models.Stories, *e.Error) {

	rows, err := db.Conn.Query(PostQueryPrefix+"where status=?", models.StatusPublished)
	if err != nil && err != sql.ErrNoRows {
		logger.Warn("get user posts error", "error", err)
		return nil, e.New(http.StatusInternalServerError, e.Internal)
	}

	posts := GetPosts(user, rows)
	sort.Sort(posts)

	return posts, nil
}

func UserPosts(tp string, user *models.User, uid string) (models.Stories, *e.Error) {
	var rows *sql.Rows
	var err error
	if tp == models.IDTypeUndefined {
		rows, err = db.Conn.Query(PostQueryPrefix+"where creator=? and status=?", uid, models.StatusPublished)
	} else {
		rows, err = db.Conn.Query(PostQueryPrefix+"where creator=? and type=? and status=?", uid, tp, models.StatusPublished)
	}

	if err != nil && err != sql.ErrNoRows {
		logger.Warn("get user posts error", "error", err)
		return nil, e.New(http.StatusInternalServerError, e.Internal)
	}

	posts := GetPosts(user, rows)

	sort.Sort(posts)

	pinned := make([]*models.Story, 0)
	unpinned := make([]*models.Story, 0)

	for _, post := range posts {
		post.Pinned = GetPinned(post.ID, user.ID)
		if post.Pinned {
			pinned = append(pinned, post)
		} else {
			unpinned = append(unpinned, post)
		}
	}

	newPosts := append(pinned, unpinned...)
	return newPosts, nil
}

func OrgPosts(tp string, user *models.User, orgID string) (models.Stories, *e.Error) {
	var rows *sql.Rows
	var err error
	if tp == models.IDTypeUndefined {
		rows, err = db.Conn.Query(PostQueryPrefix+"where owner=? and status=?", orgID, models.StatusPublished)
	} else {
		rows, err = db.Conn.Query(PostQueryPrefix+"where owner=? and type=? and status=?", orgID, tp, models.StatusPublished)
	}

	if err != nil && err != sql.ErrNoRows {
		logger.Warn("get user posts error", "error", err)
		return nil, e.New(http.StatusInternalServerError, e.Internal)
	}

	posts := GetPosts(user, rows)

	sort.Sort(posts)

	pinned := make([]*models.Story, 0)
	unpinned := make([]*models.Story, 0)

	for _, post := range posts {
		post.Pinned = GetPinned(post.ID, user.ID)
		if post.Pinned {
			pinned = append(pinned, post)
		} else {
			unpinned = append(unpinned, post)
		}
	}

	newPosts := append(pinned, unpinned...)
	return newPosts, nil
}

func UserDrafts(user *models.User, uid string) (models.Stories, *e.Error) {
	rows, err := db.Conn.Query(PostQueryPrefix+"where creator=? and status=?", uid, models.StatusDraft)
	if err != nil && err != sql.ErrNoRows {
		logger.Warn("get user drafts error", "error", err)
		return nil, e.New(http.StatusInternalServerError, e.Internal)
	}

	posts := GetPosts(user, rows)

	sort.Sort(posts)
	return posts, nil
}

func TagPosts(user *models.User, tagID string) (models.Stories, *e.Error) {
	// get post ids
	postIDs, err := tags.GetTargetIDs(tagID)
	if err != nil {
		logger.Warn("get user posts error", "error", err)
		return nil, e.New(http.StatusInternalServerError, e.Internal)
	}

	ids := strings.Join(postIDs, "','")

	q := fmt.Sprintf(PostQueryPrefix+"where id in ('%s') and status='%d'", ids, models.StatusPublished)
	rows, err := db.Conn.Query(q)
	if err != nil && err != sql.ErrNoRows {
		logger.Warn("get user posts error", "error", err)
		return nil, e.New(http.StatusInternalServerError, e.Internal)
	}

	posts := GetPosts(user, rows)

	sort.Sort(posts)
	return posts, nil
}

func BookmarkPosts(user *models.User, filter string) (models.Stories, *e.Error) {
	// get post ids
	rows, err := db.Conn.Query("select story_id from bookmarks where user_id=?", user.ID)
	if err != nil {
		logger.Warn("get bookmarks  error", "error", err)
		return nil, e.New(http.StatusInternalServerError, e.Internal)
	}

	postIDs := make([]string, 0)
	for rows.Next() {
		var id string
		rows.Scan(&id)
		postIDs = append(postIDs, id)
	}

	ids := strings.Join(postIDs, "','")

	q := fmt.Sprintf(PostQueryPrefix+"where id in ('%s')", ids)
	rows, err = db.Conn.Query(q)
	if err != nil && err != sql.ErrNoRows {
		logger.Warn("get user posts error", "error", err)
		return nil, e.New(http.StatusInternalServerError, e.Internal)
	}

	posts := GetPosts(user, rows)

	for _, post := range posts {
		_, rawTags, err := tags.GetTargetTags(post.ID)
		if err != nil {
			logger.Warn("get story tags error", "error", err)
			continue
		}

		post.RawTags = rawTags
	}

	sort.Sort(posts)
	return posts, nil
}

func GetPosts(user *models.User, rows *sql.Rows) models.Stories {
	posts := make(models.Stories, 0)
	for rows.Next() {
		ar := &models.Story{}
		err := rows.Scan(&ar.ID, &ar.Type, &ar.Slug, &ar.Title, &ar.URL, &ar.Cover, &ar.Brief, &ar.CreatorID, &ar.OwnerID, &ar.Created, &ar.Updated)
		if err != nil {
			logger.Warn("scan post error", "error", err)
			continue
		}

		// 获取作者信息
		creator := &models.UserSimple{ID: ar.CreatorID}
		creator.Query()
		ar.Creator = creator

		if ar.OwnerID != "" {
			owner := &models.UserSimple{ID: ar.OwnerID}
			owner.Query()
			ar.Owner = owner
		}

		// 获取评论信息
		ar.Comments = GetCommentCount(ar.ID)

		// 获取当前登录用户的like
		if user != nil {
			ar.Liked = interaction.GetLiked(ar.ID, user.ID)
			// 获取当前登录用户的bookmark
			ar.Bookmarked, _ = Bookmarked(user.ID, ar.ID)
		}
		ar.Likes = interaction.GetLikes(ar.ID)

		tags, rawTags, err := tags.GetTargetTags(ar.ID)
		if err != nil {
			logger.Warn("get tags error", "error", err)
		}
		ar.Tags = tags
		ar.RawTags = rawTags

		posts = append(posts, ar)
	}

	return posts
}