package common

import "database/sql"
import "github.com/Azareal/Gosora/query_gen"

var Activity ActivityStream

type ActivityStream interface {
	Add(alert Alert) (int, error)
	Get(id int) (Alert, error)
	Count() (count int)
}

type DefaultActivityStream struct {
	add *sql.Stmt
	get *sql.Stmt
	count *sql.Stmt
}

func NewDefaultActivityStream(acc *qgen.Accumulator) (*DefaultActivityStream, error) {
	return &DefaultActivityStream{
		add: acc.Insert("activity_stream").Columns("actor, targetUser, event, elementType, elementID, createdAt").Fields("?,?,?,?,?,UTC_TIMESTAMP()").Prepare(),
		get: acc.Select("activity_stream").Columns("actor, targetUser, event, elementType, elementID, createdAt").Where("asid = ?").Prepare(),
		count: acc.Count("activity_stream").Prepare(),
	}, acc.FirstError()
}

func (s *DefaultActivityStream) Add(alert Alert) (int, error) {
	res, err := s.add.Exec(alert.ActorID, alert.TargetUserID, alert.Event, alert.ElementType, alert.ElementID)
	if err != nil {
		return 0, err
	}

	lastID, err := res.LastInsertId()
	return int(lastID), err
}

func (s *DefaultActivityStream) Get(id int) (Alert, error) {
	var a = Alert{ASID: id}
	err := s.get.QueryRow(id).Scan(&a.ActorID, &a.TargetUserID, &a.Event, &a.ElementType, &a.ElementID, &a.CreatedAt)
	return a, err
}

// TODO: Write a test for this
// Count returns the total number of activity stream items
func (s *DefaultActivityStream) Count() (count int) {
	err := s.count.QueryRow().Scan(&count)
	if err != nil {
		LogError(err)
	}
	return count
}