package data

import (
	"context"
	"encoding/json"
	"errors"
	"time"

	"kra/internal/biz"

	"gorm.io/gorm"
)

type announcementPO struct {
	ID          uint `gorm:"primaryKey"`
	CreatedAt   time.Time
	UpdatedAt   time.Time
	DeletedAt   gorm.DeletedAt `gorm:"index"`
	Title       string
	Content     string `gorm:"type:text"`
	UserID      *uint  `gorm:"column:user_id"`
	Attachments jsonPO
}

func (announcementPO) TableName() string { return "kra_announcements_info" }

type announcementRepo struct{ data *Data }

func NewAnnouncementRepo(data *Data) biz.AnnouncementRepo { return &announcementRepo{data: data} }

func newAnnouncement(item *biz.Announcement) announcementPO {
	return announcementPO{ID: item.ID, Title: item.Title, Content: item.Content, UserID: item.UserID, Attachments: jsonPO(item.Attachments)}
}

func announcementToBiz(item announcementPO) *biz.Announcement {
	return &biz.Announcement{ID: item.ID, CreatedAt: item.CreatedAt, UpdatedAt: item.UpdatedAt, Title: item.Title, Content: item.Content, UserID: item.UserID, Attachments: json.RawMessage(item.Attachments)}
}

func (r *announcementRepo) Create(ctx context.Context, item *biz.Announcement) error {
	po := newAnnouncement(item)
	if err := r.data.gormDB.WithContext(ctx).Create(&po).Error; err != nil {
		return err
	}
	item.ID, item.CreatedAt, item.UpdatedAt = po.ID, po.CreatedAt, po.UpdatedAt
	return nil
}

func (r *announcementRepo) Delete(ctx context.Context, id uint) error {
	return r.data.gormDB.WithContext(ctx).Delete(&announcementPO{}, id).Error
}

func (r *announcementRepo) DeleteByIDs(ctx context.Context, ids []uint) error {
	return r.data.gormDB.WithContext(ctx).Where("id IN ?", ids).Delete(&announcementPO{}).Error
}

func (r *announcementRepo) Update(ctx context.Context, item *biz.Announcement) error {
	po := newAnnouncement(item)
	return r.data.gormDB.WithContext(ctx).Model(&announcementPO{}).Where("id = ?", item.ID).Updates(&po).Error
}

func (r *announcementRepo) Find(ctx context.Context, id uint) (*biz.Announcement, error) {
	var po announcementPO
	if err := r.data.gormDB.WithContext(ctx).First(&po, id).Error; err != nil {
		return nil, err
	}
	return announcementToBiz(po), nil
}

func (r *announcementRepo) List(ctx context.Context, filter biz.AnnouncementFilter) ([]*biz.Announcement, int64, error) {
	db := r.data.gormDB.WithContext(ctx).Model(&announcementPO{})
	if filter.StartCreatedAt != nil && filter.EndCreatedAt != nil {
		db = db.Where("created_at BETWEEN ? AND ?", filter.StartCreatedAt, filter.EndCreatedAt)
	}
	var total int64
	if err := db.Count(&total).Error; err != nil {
		return nil, 0, err
	}
	if filter.PageSize > 0 {
		db = applyPagination(db, filter.Page, filter.PageSize, 100)
	}
	var pos []announcementPO
	if err := db.Find(&pos).Error; err != nil {
		return nil, 0, err
	}
	items := make([]*biz.Announcement, 0, len(pos))
	for _, po := range pos {
		items = append(items, announcementToBiz(po))
	}
	return items, total, nil
}

func (r *announcementRepo) UserOptions(ctx context.Context) ([]biz.UserOption, error) {
	var rows []struct {
		Label string
		Value uint
	}
	err := r.data.gormDB.WithContext(ctx).Table("sys_users").Select("nick_name AS label, id AS value").Scan(&rows).Error
	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, err
	}
	items := make([]biz.UserOption, 0, len(rows))
	for _, row := range rows {
		items = append(items, biz.UserOption{Label: row.Label, Value: row.Value})
	}
	return items, nil
}