140 lines
3.3 KiB
Go
140 lines
3.3 KiB
Go
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
|
|
}
|