2019-06-03 09:32:43 +00:00
|
|
|
package arn
|
|
|
|
|
|
|
|
import (
|
|
|
|
"errors"
|
|
|
|
|
|
|
|
"github.com/aerogo/nano"
|
|
|
|
)
|
|
|
|
|
2019-11-18 05:01:13 +00:00
|
|
|
// CompanyID represents a company ID.
|
|
|
|
type CompanyID = ID
|
|
|
|
|
2019-06-03 09:32:43 +00:00
|
|
|
// Company represents an anime studio, producer or licensor.
|
|
|
|
type Company struct {
|
|
|
|
Name CompanyName `json:"name" editable:"true"`
|
|
|
|
Description string `json:"description" editable:"true" type:"textarea"`
|
|
|
|
Email string `json:"email" editable:"true"`
|
|
|
|
Links []*Link `json:"links" editable:"true"`
|
|
|
|
|
|
|
|
// Mixins
|
|
|
|
hasID
|
|
|
|
hasMappings
|
|
|
|
hasLikes
|
|
|
|
hasDraft
|
|
|
|
|
|
|
|
// Other editable fields
|
|
|
|
Location *Location `json:"location" editable:"true"`
|
|
|
|
Tags []string `json:"tags" editable:"true"`
|
|
|
|
|
|
|
|
// Editing dates
|
|
|
|
hasCreator
|
|
|
|
hasEditor
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewCompany creates a new company.
|
|
|
|
func NewCompany() *Company {
|
|
|
|
return &Company{
|
|
|
|
hasID: hasID{
|
|
|
|
ID: GenerateID("Company"),
|
|
|
|
},
|
|
|
|
Name: CompanyName{},
|
|
|
|
Links: []*Link{},
|
|
|
|
Tags: []string{},
|
|
|
|
hasCreator: hasCreator{
|
|
|
|
Created: DateTimeUTC(),
|
|
|
|
},
|
|
|
|
hasMappings: hasMappings{
|
|
|
|
Mappings: []*Mapping{},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Link returns a single company.
|
|
|
|
func (company *Company) Link() string {
|
|
|
|
return "/company/" + company.ID
|
|
|
|
}
|
|
|
|
|
|
|
|
// Anime returns the anime connected with this company.
|
|
|
|
func (company *Company) Anime() (studioAnime []*Anime, producedAnime []*Anime, licensedAnime []*Anime) {
|
|
|
|
for anime := range StreamAnime() {
|
|
|
|
if Contains(anime.StudioIDs, company.ID) {
|
|
|
|
studioAnime = append(studioAnime, anime)
|
|
|
|
}
|
|
|
|
|
|
|
|
if Contains(anime.ProducerIDs, company.ID) {
|
|
|
|
producedAnime = append(producedAnime, anime)
|
|
|
|
}
|
|
|
|
|
|
|
|
if Contains(anime.LicensorIDs, company.ID) {
|
|
|
|
licensedAnime = append(licensedAnime, anime)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
SortAnimeByQuality(studioAnime)
|
|
|
|
SortAnimeByQuality(producedAnime)
|
|
|
|
SortAnimeByQuality(licensedAnime)
|
|
|
|
|
|
|
|
return studioAnime, producedAnime, licensedAnime
|
|
|
|
}
|
|
|
|
|
|
|
|
// Publish publishes the company draft.
|
|
|
|
func (company *Company) Publish() error {
|
|
|
|
// No title
|
|
|
|
if company.Name.English == "" {
|
|
|
|
return errors.New("No English company name")
|
|
|
|
}
|
|
|
|
|
|
|
|
return publish(company)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Unpublish turns the company into a draft.
|
|
|
|
func (company *Company) Unpublish() error {
|
|
|
|
return unpublish(company)
|
|
|
|
}
|
|
|
|
|
|
|
|
// String implements the default string serialization.
|
|
|
|
func (company *Company) String() string {
|
|
|
|
return company.Name.English
|
|
|
|
}
|
|
|
|
|
|
|
|
// TypeName returns the type name.
|
|
|
|
func (company *Company) TypeName() string {
|
|
|
|
return "Company"
|
|
|
|
}
|
|
|
|
|
|
|
|
// Self returns the object itself.
|
|
|
|
func (company *Company) Self() Loggable {
|
|
|
|
return company
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetCompany returns a single company.
|
2019-11-18 06:13:51 +00:00
|
|
|
func GetCompany(id CompanyID) (*Company, error) {
|
2019-06-03 09:32:43 +00:00
|
|
|
obj, err := DB.Get("Company", id)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return obj.(*Company), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// StreamCompanies returns a stream of all companies.
|
|
|
|
func StreamCompanies() <-chan *Company {
|
|
|
|
channel := make(chan *Company, nano.ChannelBufferSize)
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
for obj := range DB.All("Company") {
|
|
|
|
channel <- obj.(*Company)
|
|
|
|
}
|
|
|
|
|
|
|
|
close(channel)
|
|
|
|
}()
|
|
|
|
|
|
|
|
return channel
|
|
|
|
}
|
|
|
|
|
|
|
|
// FilterCompanies filters all companies by a custom function.
|
|
|
|
func FilterCompanies(filter func(*Company) bool) []*Company {
|
|
|
|
var filtered []*Company
|
|
|
|
|
|
|
|
channel := DB.All("Company")
|
|
|
|
|
|
|
|
for obj := range channel {
|
|
|
|
realObject := obj.(*Company)
|
|
|
|
|
|
|
|
if filter(realObject) {
|
|
|
|
filtered = append(filtered, realObject)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return filtered
|
|
|
|
}
|
|
|
|
|
|
|
|
// AllCompanies returns a slice of all companies.
|
|
|
|
func AllCompanies() []*Company {
|
|
|
|
all := make([]*Company, 0, DB.Collection("Company").Count())
|
|
|
|
|
|
|
|
stream := StreamCompanies()
|
|
|
|
|
|
|
|
for obj := range stream {
|
|
|
|
all = append(all, obj)
|
|
|
|
}
|
|
|
|
|
|
|
|
return all
|
|
|
|
}
|