2019-06-04 12:16:27 +09:00

171 lines
3.3 KiB
Go

package arn
import (
"errors"
"fmt"
"sort"
"github.com/aerogo/nano"
)
// Person represents a person in real life.
type Person struct {
Name PersonName `json:"name" editable:"true"`
Image Image `json:"image"`
hasID
hasPosts
hasCreator
hasEditor
hasLikes
hasDraft
}
// NewPerson creates a new person.
func NewPerson() *Person {
return &Person{
hasID: hasID{
ID: GenerateID("Person"),
},
hasCreator: hasCreator{
Created: DateTimeUTC(),
},
}
}
// Link returns the path to the person.
func (person *Person) Link() string {
return "/person/" + person.ID
}
// TitleByUser returns the preferred title for the given user.
func (person *Person) TitleByUser(user *User) string {
return person.Name.ByUser(user)
}
// String returns the default display name for the person.
func (person *Person) String() string {
return person.Name.ByUser(nil)
}
// TypeName returns the type name.
func (person *Person) TypeName() string {
return "Person"
}
// Self returns the object itself.
func (person *Person) Self() Loggable {
return person
}
// ImageLink ...
func (person *Person) ImageLink(size string) string {
extension := ".jpg"
if size == "original" {
extension = person.Image.Extension
}
return fmt.Sprintf("//%s/images/persons/%s/%s%s?%v", MediaHost, size, person.ID, extension, person.Image.LastModified)
}
// Publish publishes the person draft.
func (person *Person) Publish() error {
// No name
if person.Name.ByUser(nil) == "" {
return errors.New("No person name")
}
// No image
if !person.HasImage() {
return errors.New("No person image")
}
return publish(person)
}
// Unpublish turns the person into a draft.
func (person *Person) Unpublish() error {
return unpublish(person)
}
// HasImage returns true if the person has an image.
func (person *Person) HasImage() bool {
return person.Image.Extension != "" && person.Image.Width > 0
}
// GetPerson ...
func GetPerson(id string) (*Person, error) {
obj, err := DB.Get("Person", id)
if err != nil {
return nil, err
}
return obj.(*Person), nil
}
// DeleteImages deletes all images for the person.
func (person *Person) DeleteImages() {
deleteImages("persons", person.ID, person.Image.Extension)
}
// SortPersonsByLikes sorts the given slice of persons by the amount of likes.
func SortPersonsByLikes(persons []*Person) {
sort.Slice(persons, func(i, j int) bool {
aLikes := len(persons[i].Likes)
bLikes := len(persons[j].Likes)
if aLikes == bLikes {
return persons[i].Name.English.First < persons[j].Name.English.First
}
return aLikes > bLikes
})
}
// StreamPersons returns a stream of all persons.
func StreamPersons() <-chan *Person {
channel := make(chan *Person, nano.ChannelBufferSize)
go func() {
for obj := range DB.All("Person") {
channel <- obj.(*Person)
}
close(channel)
}()
return channel
}
// FilterPersons filters all persons by a custom function.
func FilterPersons(filter func(*Person) bool) []*Person {
var filtered []*Person
channel := DB.All("Person")
for obj := range channel {
realObject := obj.(*Person)
if filter(realObject) {
filtered = append(filtered, realObject)
}
}
return filtered
}
// AllPersons returns a slice of all persons.
func AllPersons() []*Person {
all := make([]*Person, 0, DB.Collection("Person").Count())
stream := StreamPersons()
for obj := range stream {
all = append(all, obj)
}
return all
}