quyun/backend/app/models/users.go

package models

import (
	"context"
	"database/sql"
	"time"

	"quyun/app/requests"
	"quyun/database/conds"
	"quyun/database/fields"
	"quyun/database/schemas/public/model"
	"quyun/database/schemas/public/table"

	. "github.com/go-jet/jet/v2/postgres"
	"github.com/go-jet/jet/v2/qrm"
	"github.com/pkg/errors"
	"github.com/samber/lo"
	"github.com/sirupsen/logrus"
)

func Transaction(ctx context.Context) (*sql.Tx, error) {
	return db.Begin()
}

// @provider
type usersModel struct {
	log *logrus.Entry `inject:"false"`
}

func (m *usersModel) Prepare() error {
	m.log = logrus.WithField("model", "usersModel")
	return nil
}

// GetByID
func (m *usersModel) GetByID(ctx context.Context, id int64) (*model.Users, error) {
	tbl := table.Users

	stmt := tbl.
		SELECT(tbl.AllColumns).
		WHERE(
			tbl.ID.EQ(Int64(id)),
		)
	m.log.Infof("sql: %s", stmt.DebugSql())

	var user model.Users
	err := stmt.QueryContext(ctx, db, &user)
	if err != nil {
		m.log.Errorf("error querying user by ID: %v", err)
		return nil, err
	}

	return &user, nil
}

func (m *usersModel) Own(ctx context.Context, userID, postID int64) error {
	tbl := table.UserPosts
	stmt := tbl.INSERT(tbl.MutableColumns).MODEL(&model.UserPosts{
		UserID:    userID,
		PostID:    postID,
		CreatedAt: time.Now(),
		UpdatedAt: time.Now(),
	})
	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error inserting user post: %v", err)
		return err
	}
	m.log.Infof("user post inserted successfully: userID=%d, postID=%d", userID, postID)

	return nil
}

// BuildConditionWithKey builds the WHERE clause for user queries
func (m *usersModel) BuildConditionWithKey(key *string) BoolExpression {
	tbl := table.Users

	cond := tbl.DeletedAt.IS_NULL()

	if key == nil || *key == "" {
		return cond
	}

	cond = cond.AND(
		tbl.Username.LIKE(String("%" + *key + "%")),
	)

	return cond
}

// countByCondition counts users matching the given condition
func (m *usersModel) countByCondition(ctx context.Context, expr BoolExpression) (int64, error) {
	var cnt struct {
		Cnt int64
	}

	tbl := table.Users
	stmt := SELECT(COUNT(tbl.ID).AS("cnt")).FROM(tbl).WHERE(expr)
	m.log.Infof("sql: %s", stmt.DebugSql())

	err := stmt.QueryContext(ctx, db, &cnt)
	if err != nil {
		m.log.Errorf("error counting users: %v", err)
		return 0, err
	}

	return cnt.Cnt, nil
}

// List returns a paginated list of users
func (m *usersModel) List(ctx context.Context, pagination *requests.Pagination, cond BoolExpression) (*requests.Pager, error) {
	pagination.Format()

	tbl := table.Users
	stmt := tbl.
		SELECT(tbl.AllColumns).
		WHERE(cond).
		ORDER_BY(tbl.ID.DESC()).
		LIMIT(pagination.Limit).
		OFFSET(pagination.Offset)
	m.log.Infof("sql: %s", stmt.DebugSql())

	var users []model.Users = make([]model.Users, 0)
	err := stmt.QueryContext(ctx, db, &users)
	if err != nil {
		m.log.Errorf("error querying users: %v", err)
		return nil, err
	}

	count, err := m.countByCondition(ctx, cond)
	if err != nil {
		m.log.Errorf("error getting user count: %v", err)
		return nil, err
	}

	return &requests.Pager{
		Items:      users,
		Total:      count,
		Pagination: *pagination,
	}, nil
}

// Create creates a new user
func (m *usersModel) Create(ctx context.Context, userModel *model.Users) (*model.Users, error) {
	userModel.CreatedAt = time.Now()
	userModel.UpdatedAt = time.Now()

	tbl := table.Users
	stmt := tbl.INSERT(tbl.MutableColumns).MODEL(userModel).RETURNING(tbl.AllColumns)
	m.log.Infof("sql: %s", stmt.DebugSql())

	var createdUser model.Users
	err := stmt.QueryContext(ctx, db, &createdUser)
	if err != nil {
		m.log.Errorf("error creating user: %v", err)
		return nil, err
	}
	return &createdUser, nil
}

// Update updates an existing user
func (m *usersModel) Update(ctx context.Context, id int64, userModel *model.Users) (*model.Users, error) {
	userModel.UpdatedAt = time.Now()

	tbl := table.Users
	stmt := tbl.
		UPDATE(
			tbl.MutableColumns.Except(
				tbl.CreatedAt,
				tbl.DeletedAt,
			),
		).
		MODEL(userModel).
		WHERE(tbl.ID.EQ(Int64(id))).
		RETURNING(tbl.AllColumns)
	m.log.Infof("sql: %s", stmt.DebugSql())

	var updatedUser model.Users
	if err := stmt.QueryContext(ctx, db, &updatedUser); err != nil {
		m.log.Errorf("error updating user: %v", err)
		return nil, err
	}
	return &updatedUser, nil
}

// DeleteByID soft deletes a user by ID
func (m *usersModel) DeleteByID(ctx context.Context, id int64) error {
	tbl := table.Users
	stmt := tbl.
		UPDATE(tbl.DeletedAt).
		SET(TimestampT(time.Now())).
		WHERE(
			tbl.ID.EQ(Int64(id)),
		)
	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error deleting user: %v", err)
		return err
	}
	return nil
}

// PostList returns a paginated list of posts for a user
func (m *usersModel) PostList(ctx context.Context, userId int64, pagination *requests.Pagination, conds ...conds.Cond) (*requests.Pager, error) {
	pagination.Format()

	tblUserPosts := table.UserPosts

	combineConds := tblUserPosts.UserID.EQ(Int64(userId))
	for _, c := range conds {
		combineConds = c(combineConds)
	}

	tbl := table.Posts
	stmt := SELECT(tbl.AllColumns).
		FROM(tbl.
			RIGHT_JOIN(
				tblUserPosts,
				tblUserPosts.PostID.EQ(tbl.ID),
			),
		).
		WHERE(combineConds).
		ORDER_BY(tblUserPosts.ID.DESC()).
		LIMIT(pagination.Limit).
		OFFSET(pagination.Offset)
	m.log.Infof("sql: %s", stmt.DebugSql())

	var posts []model.Posts
	err := stmt.QueryContext(ctx, db, &posts)
	if err != nil {
		if errors.Is(err, qrm.ErrNoRows) {
			return &requests.Pager{
				Items:      nil,
				Total:      0,
				Pagination: *pagination,
			}, nil
		}
		m.log.Errorf("error querying posts: %v", err)
		return nil, err
	}

	// total count
	var cnt struct {
		Cnt int64
	}

	stmtCnt := tblUserPosts.SELECT(COUNT(tblUserPosts.ID).AS("cnt")).WHERE(tblUserPosts.UserID.EQ(Int64(userId)))
	m.log.Infof("sql: %s", stmtCnt.DebugSql())

	if err := stmtCnt.QueryContext(ctx, db, &cnt); err != nil {
		m.log.Errorf("error counting users: %v", err)
		return nil, err
	}

	return &requests.Pager{
		Items:      posts,
		Total:      cnt.Cnt,
		Pagination: *pagination,
	}, nil
}

// GetUserIDByOpenID
func (m *usersModel) GetUserByOpenID(ctx context.Context, openID string) (*model.Users, error) {
	tbl := table.Users

	stmt := tbl.
		SELECT(tbl.AllColumns).
		WHERE(
			tbl.OpenID.EQ(String(openID)),
		)
	m.log.Infof("sql: %s", stmt.DebugSql())

	var user model.Users
	if err := stmt.QueryContext(ctx, db, &user); err != nil {
		m.log.Errorf("error querying user by OpenID: %v", err)
		return nil, err
	}

	return &user, nil
}

// GetUserByOpenIDOrCreate
func (m *usersModel) GetUserByOpenIDOrCreate(ctx context.Context, openID string, userModel *model.Users) (*model.Users, error) {
	user, err := m.GetUserByOpenID(ctx, openID)
	if err != nil {
		if errors.Is(err, qrm.ErrNoRows) {
			user, err = m.Create(ctx, userModel)
			if err != nil {
				return nil, errors.Wrap(err, "failed to create user")
			}
		} else {
			return nil, errors.Wrap(err, "failed to get user")
		}
	} else {
		userModel.OpenID = user.OpenID
		user, err = m.Update(ctx, user.ID, userModel)
		if err != nil {
			return nil, errors.Wrap(err, "failed to update user")
		}
	}
	return user, nil
}

// GetUsersMapByIDs
func (m *usersModel) GetUsersMapByIDs(ctx context.Context, ids []int64) (map[int64]model.Users, error) {
	if len(ids) == 0 {
		return nil, nil
	}

	tbl := table.Users
	stmt := tbl.
		SELECT(tbl.AllColumns).
		WHERE(
			tbl.ID.IN(lo.Map(ids, func(id int64, _ int) Expression { return Int64(id) })...),
		)

	m.log.Infof("sql: %s", stmt.DebugSql())

	var users []model.Users = make([]model.Users, 0)
	err := stmt.QueryContext(ctx, db, &users)
	if err != nil {
		m.log.Errorf("error querying users: %v", err)
		return nil, err
	}

	return lo.SliceToMap(users, func(item model.Users) (int64, model.Users) {
		return item.ID, item
	}), nil
}

func (m *usersModel) BatchCheckHasBought(ctx context.Context, userID int64, postIDs []int64) (map[int64]bool, error) {
	tbl := table.UserPosts
	stmt := tbl.SELECT(tbl.PostID.AS("post_id")).WHERE(
		tbl.UserID.EQ(Int64(userID)).AND(
			tbl.PostID.IN(lo.Map(postIDs, func(id int64, _ int) Expression { return Int64(id) })...),
		),
	)

	var userPosts []struct {
		PostID int64
	}
	if err := stmt.QueryContext(ctx, db, &userPosts); err != nil {
		m.log.Errorf("error querying user posts: %v", err)
		return nil, err
	}

	result := make(map[int64]bool)
	for _, post := range userPosts {
		result[post.PostID] = true
	}
	return result, nil
}

// HasBought
func (m *usersModel) HasBought(ctx context.Context, userID, postID int64) (bool, error) {
	tbl := table.UserPosts
	stmt := tbl.
		SELECT(tbl.ID).
		WHERE(
			tbl.UserID.EQ(Int64(userID)).AND(
				tbl.PostID.EQ(Int64(postID)),
			),
		)

	m.log.Infof("sql: %s", stmt.DebugSql())

	var userPost model.UserPosts
	if err := stmt.QueryContext(ctx, db, &userPost); err != nil {
		if errors.Is(err, qrm.ErrNoRows) {
			return false, nil
		}
		m.log.Errorf("error querying user post: %v", err)
		return false, err
	}

	return userPost.ID > 0, nil
}

// Count
func (m *usersModel) Count(ctx context.Context, cond BoolExpression) (int64, error) {
	tbl := table.Users
	stmt := tbl.
		SELECT(COUNT(tbl.ID).AS("cnt")).
		WHERE(cond)

	m.log.Infof("sql: %s", stmt.DebugSql())

	var cnt struct {
		Cnt int64
	}

	if err := stmt.QueryContext(ctx, db, &cnt); err != nil {
		m.log.Errorf("error counting users: %v", err)
		return 0, err
	}

	return cnt.Cnt, nil
}

// UpdateUsername
func (m *usersModel) UpdateUsername(ctx context.Context, id int64, username string) error {
	tbl := table.Users
	stmt := tbl.
		UPDATE(tbl.Username).
		SET(String(username)).
		WHERE(
			tbl.ID.EQ(Int64(id)),
		)
	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error updating username: %v", err)
		return err
	}
	return nil
}

// UpdateUserToken
func (m *usersModel) UpdateUserToken(ctx context.Context, id int64, token fields.UserAuthToken) error {
	tbl := table.Users
	stmt := tbl.
		UPDATE(tbl.AuthToken).
		SET(fields.ToJson(token)).
		WHERE(
			tbl.ID.EQ(Int64(id)),
		)
	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error updating user token: %v", err)
		return err
	}
	return nil
}

// BuyPosts
func (m *usersModel) BuyPosts(ctx context.Context, userID, postID, price int64) error {
	tbl := table.UserPosts
	stmt := tbl.
		INSERT(tbl.MutableColumns).
		MODEL(&model.UserPosts{
			UserID:    userID,
			PostID:    postID,
			Price:     price,
			CreatedAt: time.Now(),
			UpdatedAt: time.Now(),
		})

	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error inserting user post: %v", err)
		return err
	}
	return nil
}

func (m *usersModel) RevokePosts(ctx context.Context, userID, postID int64) error {
	tbl := table.UserPosts
	stmt := tbl.
		DELETE().
		WHERE(
			tbl.UserID.EQ(Int64(userID)).AND(
				tbl.PostID.EQ(Int64(postID)),
			),
		)
	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error revoking user post: %v", err)
		return err
	}
	return nil
}

// SetBalance
func (m *usersModel) SetBalance(ctx context.Context, id int64, balance int64) error {
	tbl := table.Users
	stmt := tbl.
		UPDATE(tbl.Balance).
		SET(Int64(balance)).
		WHERE(
			tbl.ID.EQ(Int64(id)),
		)
	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error updating user balance: %v", err)
		return err
	}
	return nil
}