package model

import (
	"context"
	"time"

	"gorm.io/gorm"

	"spark-server/internal/module/friend/entity"
	"spark-server/internal/pkg/errs"
)

// UserBlock maps table user_blocks.
type UserBlock struct {
	UserID    int64     `gorm:"column:user_id;primaryKey;comment:拉黑人 user_id"`
	BlockedID int64     `gorm:"column:blocked_id;primaryKey;comment:被拉黑人 user_id"`
	CreatedAt time.Time `gorm:"column:created_at;comment:拉黑时间"`
}

func (UserBlock) TableName() string { return "user_blocks" }

func (m *Repo) IsBlocked(ctx context.Context, userID, targetID int64) (bool, error) {
	db, err := m.session(ctx)
	if err != nil {
		return false, err
	}
	var n int64
	if err := db.Model(&UserBlock{}).Where("user_id = ? AND blocked_id = ?", userID, targetID).Count(&n).Error; err != nil {
		return false, mapDBErr(err)
	}
	return n > 0, nil
}

func (m *Repo) BlockedEither(ctx context.Context, a, b int64) (bool, error) {
	me, err := m.IsBlocked(ctx, a, b)
	if err != nil || me {
		return me, err
	}
	return m.IsBlocked(ctx, b, a)
}

func (m *Repo) CreateBlock(ctx context.Context, b *entity.Block) error {
	if b == nil || b.UserID <= 0 || b.BlockedID <= 0 {
		return errs.ErrInvalidArgument
	}
	db, err := m.session(ctx)
	if err != nil {
		return err
	}
	now := b.CreatedAt
	if now.IsZero() {
		now = time.Now()
	}
	return db.Transaction(func(tx *gorm.DB) error {
		row := UserBlock{UserID: b.UserID, BlockedID: b.BlockedID, CreatedAt: now}
		if err := tx.Where("user_id = ? AND blocked_id = ?", b.UserID, b.BlockedID).
			Attrs(row).FirstOrCreate(&row).Error; err != nil {
			return mapDBErr(err)
		}
		small, big := entity.OrderUsers(b.UserID, b.BlockedID)
		if err := tx.Where("user_small = ? AND user_big = ?", small, big).Delete(&FriendRelation{}).Error; err != nil {
			return mapDBErr(err)
		}
		if err := deleteRemarksBetween(tx, b.UserID, b.BlockedID); err != nil {
			return err
		}
		return rejectPairRequests(tx, b.UserID, b.BlockedID, now)
	})
}

func (m *Repo) DeleteBlock(ctx context.Context, userID, blockedID int64) error {
	db, err := m.session(ctx)
	if err != nil {
		return err
	}
	return mapDBErr(db.Where("user_id = ? AND blocked_id = ?", userID, blockedID).Delete(&UserBlock{}).Error)
}