votes.go

package brutalinks

import (
	"time"

	vocab "github.com/go-ap/activitypub"
	"github.com/go-ap/errors"
)

const (
	ScoreMultiplier = 1
	ScoreMaxK       = 10000.0
	ScoreMaxM       = 10000000.0
	ScoreMaxB       = 10000000000.0
)

var ValidAppreciationTypes = vocab.ActivityVocabularyTypes{
	vocab.LikeType,
	vocab.DislikeType,
}

type VoteCollection []Vote

type VoteMetadata struct {
	IRI         string `json:"-"`
	OriginalIRI string `json:"-"`
}

type Vote struct {
	SubmittedBy *Account        `json:"-"`
	SubmittedAt time.Time       `json:"-"`
	UpdatedAt   time.Time       `json:"-"`
	Weight      int             `json:"weight"`
	Item        *Item           `json:"on"`
	Flags       FlagBits        `json:"-"`
	Metadata    *VoteMetadata   `json:"-"`
	Pub         *vocab.Activity `json:"-"`
}

func (v *Vote) ID() Hash {
	if v == nil {
		return AnonymousHash
	}
	return HashFromString(v.Metadata.IRI)
}

// HasMetadata
func (v Vote) HasMetadata() bool {
	return v.Metadata != nil
}

// IsValid
func (v *Vote) IsValid() bool {
	return v != nil && v.Item.IsValid()
}

// IsYay returns true if current vote is a Yay
func (v Vote) IsYay() bool {
	if v.Pub == nil {
		return false
	}
	return v.Pub.GetType() == vocab.LikeType
}

// IsNay returns true if current vote is a Nay
func (v Vote) IsNay() bool {
	if v.Pub == nil {
		return false
	}
	return v.Pub.GetType() == vocab.DislikeType
}

// AP returns the underlying actvitypub item
func (v *Vote) AP() vocab.Item {
	return v.Pub
}

// Type
func (v *Vote) Type() RenderType {
	return AppreciationType
}

// Date
func (v Vote) Date() time.Time {
	return v.SubmittedAt
}

func (v *Vote) Children() *RenderableList {
	return nil
}

func (v VoteCollection) Contains(vot Vote) bool {
	for _, vv := range v {
		if !vv.HasMetadata() || !vot.HasMetadata() {
			continue
		}
		if vv.Metadata.IRI == vot.Metadata.IRI {
			return true
		}
	}
	return false
}

type ScoreType int

const (
	ScoreItem = ScoreType(iota)
	ScoreAccount
)

func (v VoteCollection) First() (*Vote, error) {
	for _, vv := range v {
		return &vv, nil
	}
	return nil, errors.Errorf("empty %T", v)
}

// Score
func (v VoteCollection) Score() int {
	score := 0
	for _, vot := range v {
		score += vot.Weight
	}
	return score
}