package common

import (
	"github.com/rancher/naok/pkg/resources/schema"
	"github.com/rancher/norman/pkg/store/proxy"
	"github.com/rancher/norman/pkg/types"
	"github.com/rancher/norman/pkg/types/convert"
	"github.com/rancher/norman/pkg/types/values"
)

func Register(collection *schema.Collection, clientGetter proxy.ClientGetter) {
	collection.AddTemplate(&schema.Template{
		Store:     proxy.NewProxyStore(clientGetter),
		Formatter: Formatter,
		Mapper:    &DefaultColumns{},
	})
}

func Formatter(request *types.APIRequest, resource *types.RawResource) {
	selfLink := convert.ToString(values.GetValueN(resource.Values, "metadata", "selfLink"))
	if selfLink == "" {
		return
	}

	u := request.URLBuilder.RelativeToRoot(selfLink)
	resource.Links["view"] = u

	if _, ok := resource.Links["update"]; ok {
		resource.Links["update"] = u
	}
}