Document.go

/*
 * This file is automatically generated
 */

package catalog

import (
	"encoding/json"
	"fmt"

	apiv1 "github.com/Axway/agent-sdk/pkg/apic/apiserver/models/api/v1"

	"github.com/Axway/agent-sdk/pkg/util/log"
)

var (
	DocumentCtx log.ContextField = "document"

	_DocumentGVK = apiv1.GroupVersionKind{
		GroupKind: apiv1.GroupKind{
			Group: "catalog",
			Kind:  "Document",
		},
		APIVersion: "v1",
	}

	DocumentScopes = []string{"Product", "ProductRelease"}
)

const DocumentResourceName = "documents"

func DocumentGVK() apiv1.GroupVersionKind {
	return _DocumentGVK
}

func init() {
	apiv1.RegisterGVK(_DocumentGVK, DocumentScopes[0], DocumentResourceName)
	log.RegisterContextField(DocumentCtx)
}

// Document Resource
type Document struct {
	apiv1.ResourceMeta
	Icon  interface{}  `json:"icon"`
	Owner *apiv1.Owner `json:"owner"`
	Spec  DocumentSpec `json:"spec"`
	// Status DocumentStatus `json:"status"`
	Status *apiv1.ResourceStatus `json:"status"`
}

// NewDocument creates an empty *Document
func NewDocument(name, scopeKind, scopeName string) (*Document, error) {
	validScope := false
	for _, s := range DocumentScopes {
		if scopeKind == s {
			validScope = true
			break
		}
	}
	if !validScope {
		return nil, fmt.Errorf("scope '%s' not valid for Document kind", scopeKind)
	}

	return &Document{
		ResourceMeta: apiv1.ResourceMeta{
			Name:             name,
			GroupVersionKind: _DocumentGVK,
			Metadata: apiv1.Metadata{
				Scope: apiv1.MetadataScope{
					Name: scopeName,
					Kind: scopeKind,
				},
			},
		},
	}, nil
}

// DocumentFromInstanceArray converts a []*ResourceInstance to a []*Document
func DocumentFromInstanceArray(fromArray []*apiv1.ResourceInstance) ([]*Document, error) {
	newArray := make([]*Document, 0)
	for _, item := range fromArray {
		res := &Document{}
		err := res.FromInstance(item)
		if err != nil {
			return make([]*Document, 0), err
		}
		newArray = append(newArray, res)
	}

	return newArray, nil
}

// AsInstance converts a Document to a ResourceInstance
func (res *Document) AsInstance() (*apiv1.ResourceInstance, error) {
	meta := res.ResourceMeta
	meta.GroupVersionKind = DocumentGVK()
	res.ResourceMeta = meta

	m, err := json.Marshal(res)
	if err != nil {
		return nil, err
	}

	instance := apiv1.ResourceInstance{}
	err = json.Unmarshal(m, &instance)
	if err != nil {
		return nil, err
	}

	return &instance, nil
}

// FromInstance converts a ResourceInstance to a Document
func (res *Document) FromInstance(ri *apiv1.ResourceInstance) error {
	if ri == nil {
		res = nil
		return nil
	}
	var err error
	rawResource := ri.GetRawResource()
	if rawResource == nil {
		rawResource, err = json.Marshal(ri)
		if err != nil {
			return err
		}
	}
	err = json.Unmarshal(rawResource, res)
	return err
}

// MarshalJSON custom marshaller to handle sub resources
func (res *Document) MarshalJSON() ([]byte, error) {
	m, err := json.Marshal(&res.ResourceMeta)
	if err != nil {
		return nil, err
	}

	var out map[string]interface{}
	err = json.Unmarshal(m, &out)
	if err != nil {
		return nil, err
	}

	out["icon"] = res.Icon
	out["owner"] = res.Owner
	out["spec"] = res.Spec
	out["status"] = res.Status

	return json.Marshal(out)
}

// UnmarshalJSON custom unmarshaller to handle sub resources
func (res *Document) UnmarshalJSON(data []byte) error {
	var err error

	aux := &apiv1.ResourceInstance{}
	err = json.Unmarshal(data, aux)
	if err != nil {
		return err
	}

	res.ResourceMeta = aux.ResourceMeta
	res.Owner = aux.Owner

	// ResourceInstance holds the spec as a map[string]interface{}.
	// Convert it to bytes, then convert to the spec type for the resource.
	sr, err := json.Marshal(aux.Spec)
	if err != nil {
		return err
	}

	err = json.Unmarshal(sr, &res.Spec)
	if err != nil {
		return err
	}

	// marshalling subresource Icon
	if v, ok := aux.SubResources["icon"]; ok {
		sr, err = json.Marshal(v)
		if err != nil {
			return err
		}

		delete(aux.SubResources, "icon")
		err = json.Unmarshal(sr, &res.Icon)
		if err != nil {
			return err
		}
	}

	// marshalling subresource Status
	if v, ok := aux.SubResources["status"]; ok {
		sr, err = json.Marshal(v)
		if err != nil {
			return err
		}

		delete(aux.SubResources, "status")
		// err = json.Unmarshal(sr, &res.Status)
		res.Status = &apiv1.ResourceStatus{}
		err = json.Unmarshal(sr, res.Status)
		if err != nil {
			return err
		}
	}

	return nil
}

// PluralName returns the plural name of the resource
func (res *Document) PluralName() string {
	return DocumentResourceName
}