database

type Message struct {
	ID          uint         `db:"id"`
	ChatID      int64        `db:"chat_id"`
	UserID      int64        `db:"user_id"`
	Content     string       `db:"content"`
	Timestamp   time.Time    `db:"timestamp"`
	CreatedAt   time.Time    `db:"created_at"`
	UpdatedAt   time.Time    `db:"updated_at"`
	ProcessedAt sql.NullTime `db:"processed_at"`
}

type Store interface {
	Ping(ctx context.Context) error

	SaveMessage(ctx context.Context, message *Message) error

	RunSQLMaintenance(ctx context.Context) error

	GetUserProfile(ctx context.Context, userID int64) (*UserProfile, error)

	GetAllUserProfiles(ctx context.Context) (map[int64]*UserProfile, error)

	SaveUserProfile(ctx context.Context, profile *UserProfile) error

	GetUnprocessedMessages(ctx context.Context) ([]*Message, error)

	MarkMessagesAsProcessed(ctx context.Context, messageIDs []uint) error

	GetRecentMessages(ctx context.Context, chatID int64, limit int, beforeID uint) ([]*Message, error)

	DeleteAllMessagesAndProfiles(ctx context.Context) error
}

type UserProfile struct {
	ID              uint      `db:"id"`
	UserID          int64     `db:"user_id"`
	Aliases         string    `db:"aliases"`
	OriginLocation  string    `db:"origin_location"`
	CurrentLocation string    `db:"current_location"`
	AgeRange        string    `db:"age_range"`
	Traits          string    `db:"traits"`
	CreatedAt       time.Time `db:"created_at"`
	UpdatedAt       time.Time `db:"updated_at"`
}