Added arn to the main repository

arn/DraftIndex.go

@@ -0,0 +1,61 @@
+package arn
+
+import (
+	"errors"
+	"reflect"
+)
+
+// DraftIndex has references to unpublished drafts a user created.
+type DraftIndex struct {
+	UserID       string `json:"userId"`
+	GroupID      string `json:"groupId"`
+	SoundTrackID string `json:"soundTrackId"`
+	CompanyID    string `json:"companyId"`
+	QuoteID      string `json:"quoteId"`
+	CharacterID  string `json:"characterId"`
+	AnimeID      string `json:"animeId"`
+	AMVID        string `json:"amvId"`
+}
+
+// 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
+}