62 lines
1.4 KiB
Go
Raw Normal View History

2019-06-03 09:32:43 +00:00
package arn
import (
"errors"
"reflect"
)
// DraftIndex has references to unpublished drafts a user created.
type DraftIndex struct {
UserID string `json:"userId" primary:"true"`
2019-06-07 01:50:38 +00:00
GroupID string `json:"groupId"`
SoundTrackID string `json:"soundTrackId"`
CompanyID string `json:"companyId"`
QuoteID string `json:"quoteId"`
CharacterID string `json:"characterId"`
AnimeID AnimeID `json:"animeId"`
AMVID string `json:"amvId"`
2019-06-03 09:32:43 +00:00
}
// NewDraftIndex ...
func NewDraftIndex(userID UserID) *DraftIndex {
return &DraftIndex{
UserID: userID,
}
}
// GetID gets the ID for the given type name.
func (index *DraftIndex) GetID(typeName string) (string, error) {
v := reflect.ValueOf(index).Elem()
fieldValue := v.FieldByName(typeName + "ID")
if !fieldValue.IsValid() {
return "", errors.New("Invalid draft index ID type: " + typeName)
}
return fieldValue.String(), nil
}
// SetID sets the ID for the given type name.
func (index *DraftIndex) SetID(typeName string, id string) error {
v := reflect.ValueOf(index).Elem()
fieldValue := v.FieldByName(typeName + "ID")
if !fieldValue.IsValid() {
return errors.New("Invalid draft index ID type: " + typeName)
}
fieldValue.SetString(id)
return nil
}
// GetDraftIndex ...
func GetDraftIndex(id string) (*DraftIndex, error) {
obj, err := DB.Get("DraftIndex", id)
if err != nil {
return nil, err
}
return obj.(*DraftIndex), nil
}