140 lines
3.3 KiB
Go
Raw Normal View History

2019-06-03 09:32:43 +00:00
package arn
import (
"errors"
"fmt"
"reflect"
"github.com/aerogo/aero"
"github.com/aerogo/api"
)
// Force interface implementations
var (
_ Publishable = (*Company)(nil)
_ fmt.Stringer = (*Company)(nil)
_ api.Newable = (*Company)(nil)
_ api.Editable = (*Company)(nil)
_ api.Deletable = (*Company)(nil)
_ api.ArrayEventListener = (*Company)(nil)
)
// Actions
func init() {
API.RegisterActions("Company", []*api.Action{
// Publish
PublishAction(),
// Unpublish
UnpublishAction(),
// Like
LikeAction(),
// Unlike
UnlikeAction(),
})
}
// Create sets the data for a new company with data we received from the API request.
func (company *Company) Create(ctx aero.Context) error {
user := GetUserFromContext(ctx)
if user == nil {
return errors.New("Not logged in")
}
company.ID = GenerateID("Company")
company.Created = DateTimeUTC()
company.CreatedBy = user.ID
company.Location = &Location{}
// Write log entry
logEntry := NewEditLogEntry(user.ID, "create", "Company", company.ID, "", "", "")
logEntry.Save()
return company.Unpublish()
}
// Edit creates an edit log entry.
func (company *Company) Edit(ctx aero.Context, key string, value reflect.Value, newValue reflect.Value) (consumed bool, err error) {
return edit(company, ctx, key, value, newValue)
}
// OnAppend saves a log entry.
func (company *Company) OnAppend(ctx aero.Context, key string, index int, obj interface{}) {
onAppend(company, ctx, key, index, obj)
}
// OnRemove saves a log entry.
func (company *Company) OnRemove(ctx aero.Context, key string, index int, obj interface{}) {
onRemove(company, ctx, key, index, obj)
}
// Save saves the company in the database.
func (company *Company) Save() {
DB.Set("Company", company.ID, company)
}
// DeleteInContext deletes the company in the given context.
func (company *Company) DeleteInContext(ctx aero.Context) error {
user := GetUserFromContext(ctx)
// Write log entry
logEntry := NewEditLogEntry(user.ID, "delete", "Company", company.ID, "", fmt.Sprint(company), "")
logEntry.Save()
return company.Delete()
}
// Delete deletes the object from the database.
func (company *Company) Delete() error {
if company.IsDraft {
draftIndex := company.Creator().DraftIndex()
draftIndex.CompanyID = ""
draftIndex.Save()
}
// Remove company ID from all anime
for anime := range StreamAnime() {
for index, id := range anime.StudioIDs {
if id == company.ID {
anime.StudioIDs = append(anime.StudioIDs[:index], anime.StudioIDs[index+1:]...)
break
}
}
for index, id := range anime.ProducerIDs {
if id == company.ID {
anime.ProducerIDs = append(anime.ProducerIDs[:index], anime.ProducerIDs[index+1:]...)
break
}
}
for index, id := range anime.LicensorIDs {
if id == company.ID {
anime.LicensorIDs = append(anime.LicensorIDs[:index], anime.LicensorIDs[index+1:]...)
break
}
}
}
DB.Delete("Company", company.ID)
return nil
}
// Authorize returns an error if the given API request is not authorized.
func (company *Company) Authorize(ctx aero.Context, action string) error {
user := GetUserFromContext(ctx)
if user == nil {
return errors.New("Not logged in")
}
if user.Role != "editor" && user.Role != "admin" {
return errors.New("Insufficient permissions")
}
return nil
}