Dragon-Knight/internal/models/news/news.go

package news

import (
	"fmt"
	"strings"
	"time"

	nigiri "git.sharkk.net/Sharkk/Nigiri"
)

// News represents a news post in the game
type News struct {
	ID      int    `json:"id"`
	Author  int    `json:"author" db:"index"`
	Posted  int64  `json:"posted"`
	Content string `json:"content" db:"required"`
}

// Global store
var store *nigiri.BaseStore[News]
var db *nigiri.Collection

// Init sets up the Nigiri store and indices
func Init(collection *nigiri.Collection) {
	db = collection
	store = nigiri.NewBaseStore[News]()

	// Register custom indices
	store.RegisterIndex("byAuthor", nigiri.BuildIntGroupIndex(func(n *News) int {
		return n.Author
	}))

	store.RegisterIndex("allByPosted", nigiri.BuildSortedListIndex(func(a, b *News) bool {
		if a.Posted != b.Posted {
			return a.Posted > b.Posted // DESC
		}
		return a.ID > b.ID // DESC
	}))

	store.RebuildIndices()
}

// GetStore returns the news store
func GetStore() *nigiri.BaseStore[News] {
	if store == nil {
		panic("news store not initialized - call Init first")
	}
	return store
}

// Creates a new News with sensible defaults
func New() *News {
	return &News{
		Author:  0,                 // No author by default
		Posted:  time.Now().Unix(), // Current time
		Content: "",                // Empty content
	}
}

// Validate checks if news has valid values
func (n *News) Validate() error {
	if n.Posted < 0 {
		return fmt.Errorf("news Posted timestamp cannot be negative")
	}
	if strings.TrimSpace(n.Content) == "" {
		return fmt.Errorf("news Content cannot be empty")
	}
	return nil
}

// CRUD operations
func (n *News) Save() error {
	if n.ID == 0 {
		id, err := store.Create(n)
		if err != nil {
			return err
		}
		n.ID = id
		return nil
	}
	return store.Update(n.ID, n)
}

func (n *News) Delete() error {
	store.Remove(n.ID)
	return nil
}

// Insert with ID assignment
func (n *News) Insert() error {
	id, err := store.Create(n)
	if err != nil {
		return err
	}
	n.ID = id
	return nil
}

// Query functions
func Find(id int) (*News, error) {
	news, exists := store.Find(id)
	if !exists {
		return nil, fmt.Errorf("news with ID %d not found", id)
	}
	return news, nil
}

func All() ([]*News, error) {
	return store.AllSorted("allByPosted"), nil
}

func ByAuthor(authorID int) ([]*News, error) {
	return store.GroupByIndex("byAuthor", authorID), nil
}

func Recent(limit int) ([]*News, error) {
	all := store.AllSorted("allByPosted")
	if limit > len(all) {
		limit = len(all)
	}
	return all[:limit], nil
}

func Since(since int64) ([]*News, error) {
	return store.FilterByIndex("allByPosted", func(n *News) bool {
		return n.Posted >= since
	}), nil
}

func Between(start, end int64) ([]*News, error) {
	return store.FilterByIndex("allByPosted", func(n *News) bool {
		return n.Posted >= start && n.Posted <= end
	}), nil
}

func Search(term string) ([]*News, error) {
	lowerTerm := strings.ToLower(term)
	return store.FilterByIndex("allByPosted", func(n *News) bool {
		return strings.Contains(strings.ToLower(n.Content), lowerTerm)
	}), nil
}

// Helper methods
func (n *News) PostedTime() time.Time {
	return time.Unix(n.Posted, 0)
}

func (n *News) SetPostedTime(t time.Time) {
	n.Posted = t.Unix()
}

func (n *News) IsRecent() bool {
	return time.Since(n.PostedTime()) < 24*time.Hour
}

func (n *News) Age() time.Duration {
	return time.Since(n.PostedTime())
}

func (n *News) ReadableTime() string {
	return n.PostedTime().Format("Jan 2, 2006 3:04 PM")
}

func (n *News) IsAuthor(userID int) bool {
	return n.Author == userID
}

func (n *News) Preview(maxLength int) string {
	if len(n.Content) <= maxLength {
		return n.Content
	}

	if maxLength < 3 {
		return n.Content[:maxLength]
	}

	return n.Content[:maxLength-3] + "..."
}

func (n *News) WordCount() int {
	if n.Content == "" {
		return 0
	}

	// Simple word count by splitting on spaces
	words := 0
	inWord := false

	for _, char := range n.Content {
		if char == ' ' || char == '\t' || char == '\n' || char == '\r' {
			if inWord {
				words++
				inWord = false
			}
		} else {
			inWord = true
		}
	}

	if inWord {
		words++
	}

	return words
}

func (n *News) Length() int {
	return len(n.Content)
}

func (n *News) Contains(term string) bool {
	return strings.Contains(strings.ToLower(n.Content), strings.ToLower(term))
}

func (n *News) IsEmpty() bool {
	return strings.TrimSpace(n.Content) == ""
}

// Legacy compatibility functions (will be removed later)
func LoadData(dataPath string) error {
	// No longer needed - Nigiri handles this
	return nil
}

func SaveData(dataPath string) error {
	// No longer needed - Nigiri handles this
	return nil
}