org.go

/*
 * Copyright 2021 VMware, Inc.  All rights reserved.  Licensed under the Apache v2 License.
 */

package govcd

import (
	"errors"
	"fmt"
	"net/http"
	"net/url"
	"strings"

	"github.com/vmware/go-vcloud-director/v2/types/v56"
	"github.com/vmware/go-vcloud-director/v2/util"
)

type Org struct {
	Org           *types.Org
	client        *Client
	TenantContext *TenantContext
}

func NewOrg(client *Client) *Org {
	return &Org{
		Org:    new(types.Org),
		client: client,
	}
}

// Given an org with a valid HREF, the function refetches the org
// and updates the user's org data. Otherwise if the function fails,
// it returns an error. Users should use refresh whenever they have
// a stale org due to the creation/update/deletion of a resource
// within the org or the org itself.
func (org *Org) Refresh() error {
	if *org == (Org{}) {
		return fmt.Errorf("cannot refresh, Object is empty")
	}

	// Empty struct before a new unmarshal, otherwise we end up with duplicate
	// elements in slices.
	unmarshalledOrg := &types.Org{}

	_, err := org.client.ExecuteRequest(org.Org.HREF, http.MethodGet,
		"", "error refreshing organization: %s", nil, unmarshalledOrg)
	if err != nil {
		return err
	}
	org.Org = unmarshalledOrg

	// The request was successful
	return nil
}

// Given a valid catalog name, FindCatalog returns a Catalog object.
// If no catalog is found, then returns an empty catalog and no error.
// Otherwise it returns an error.
// Deprecated: use org.GetCatalogByName instead
func (org *Org) FindCatalog(catalogName string) (Catalog, error) {

	for _, link := range org.Org.Link {
		if link.Rel == "down" && link.Type == "application/vnd.vmware.vcloud.catalog+xml" && link.Name == catalogName {

			cat := NewCatalog(org.client)

			_, err := org.client.ExecuteRequest(link.HREF, http.MethodGet,
				"", "error retrieving catalog: %s", nil, cat.Catalog)

			return *cat, err
		}
	}

	return Catalog{}, nil
}

// GetVdcByName if user specifies valid vdc name then this returns a vdc object.
// If no vdc is found, then it returns an empty vdc and no error.
// Otherwise it returns an empty vdc and an error.
// Deprecated: use org.GetVDCByName instead
func (org *Org) GetVdcByName(vdcname string) (Vdc, error) {
	for _, link := range org.Org.Link {
		if link.Name == vdcname {
			vdc := NewVdc(org.client)
			vdc.parent = org

			_, err := org.client.ExecuteRequest(link.HREF, http.MethodGet,
				"", "error retrieving vdc: %s", nil, vdc.Vdc)

			return *vdc, err
		}
	}
	return Vdc{}, nil
}

// CreateCatalog creates a catalog with specified name and description
func CreateCatalog(client *Client, links types.LinkList, Name, Description string) (AdminCatalog, error) {
	adminCatalog, err := CreateCatalogWithStorageProfile(client, links, Name, Description, nil)
	if err != nil {
		return AdminCatalog{}, nil
	}
	return *adminCatalog, nil
}

// CreateCatalogWithStorageProfile is like CreateCatalog, but allows to specify storage profile
func CreateCatalogWithStorageProfile(client *Client, links types.LinkList, Name, Description string, storageProfiles *types.CatalogStorageProfiles) (*AdminCatalog, error) {
	reqCatalog := &types.Catalog{
		Name:        Name,
		Description: Description,
	}
	vcomp := &types.AdminCatalog{
		Xmlns:                  types.XMLNamespaceVCloud,
		Catalog:                *reqCatalog,
		CatalogStorageProfiles: storageProfiles,
	}

	var createOrgLink *types.Link
	for _, link := range links {
		if link.Rel == "add" && link.Type == types.MimeAdminCatalog {
			util.Logger.Printf("[TRACE] Create org - found the proper link for request, HREF: %s, "+
				"name: %s, type: %s, id: %s, rel: %s \n", link.HREF, link.Name, link.Type, link.ID, link.Rel)
			createOrgLink = link
		}
	}

	if createOrgLink == nil {
		return nil, fmt.Errorf("creating catalog failed to find url")
	}

	catalog := NewAdminCatalog(client)
	_, err := client.ExecuteRequest(createOrgLink.HREF, http.MethodPost,
		"application/vnd.vmware.admin.catalog+xml", "error creating catalog: %s", vcomp, catalog.AdminCatalog)
	if err != nil {
		return nil, err
	}

	err = catalog.WaitForTasks()
	if err != nil {
		return nil, err
	}
	return catalog, nil
}

// CreateCatalog creates a catalog with given name and description under
// the given organization. Returns an Catalog that contains a creation
// task.
// API Documentation: https://code.vmware.com/apis/220/vcloud#/doc/doc/operations/POST-CreateCatalog.html
func (org *Org) CreateCatalog(name, description string) (Catalog, error) {
	catalog, err := org.CreateCatalogWithStorageProfile(name, description, nil)
	if err != nil {
		return Catalog{}, err
	}
	catalog.parent = org

	err = catalog.Refresh()
	if err != nil {
		return Catalog{}, err
	}
	// Make sure that the creation task is finished
	err = catalog.WaitForTasks()
	if err != nil {
		return Catalog{}, err
	}
	return *catalog, nil
}

// CreateCatalogWithStorageProfile is like CreateCatalog but additionally allows to specify storage profiles
func (org *Org) CreateCatalogWithStorageProfile(name, description string, storageProfiles *types.CatalogStorageProfiles) (*Catalog, error) {
	catalog := NewCatalog(org.client)
	adminCatalog, err := CreateCatalogWithStorageProfile(org.client, org.Org.Link, name, description, storageProfiles)
	if err != nil {
		return nil, err
	}
	catalog.Catalog = &adminCatalog.AdminCatalog.Catalog
	catalog.parent = org
	return catalog, nil
}

func validateVdcConfiguration(vdcDefinition *types.VdcConfiguration) error {
	if vdcDefinition.Name == "" {
		return errors.New("VdcConfiguration missing required field: Name")
	}
	if vdcDefinition.AllocationModel == "" {
		return errors.New("VdcConfiguration missing required field: AllocationModel")
	}
	if vdcDefinition.ComputeCapacity == nil {
		return errors.New("VdcConfiguration missing required field: ComputeCapacity")
	}
	if len(vdcDefinition.ComputeCapacity) != 1 {
		return errors.New("VdcConfiguration invalid field: ComputeCapacity must only have one element")
	}
	if vdcDefinition.ComputeCapacity[0] == nil {
		return errors.New("VdcConfiguration missing required field: ComputeCapacity[0]")
	}
	if vdcDefinition.ComputeCapacity[0].CPU == nil {
		return errors.New("VdcConfiguration missing required field: ComputeCapacity[0].CPU")
	}
	if vdcDefinition.ComputeCapacity[0].CPU.Units == "" {
		return errors.New("VdcConfiguration missing required field: ComputeCapacity[0].CPU.Units")
	}
	if vdcDefinition.ComputeCapacity[0].Memory == nil {
		return errors.New("VdcConfiguration missing required field: ComputeCapacity[0].Memory")
	}
	if vdcDefinition.ComputeCapacity[0].Memory.Units == "" {
		return errors.New("VdcConfiguration missing required field: ComputeCapacity[0].Memory.Units")
	}
	if vdcDefinition.VdcStorageProfile == nil || len(vdcDefinition.VdcStorageProfile) == 0 {
		return errors.New("VdcConfiguration missing required field: VdcStorageProfile")
	}
	if vdcDefinition.VdcStorageProfile[0].Units == "" {
		return errors.New("VdcConfiguration missing required field: VdcStorageProfile.Units")
	}
	if vdcDefinition.ProviderVdcReference == nil {
		return errors.New("VdcConfiguration missing required field: ProviderVdcReference")
	}
	if vdcDefinition.ProviderVdcReference.HREF == "" {
		return errors.New("VdcConfiguration missing required field: ProviderVdcReference.HREF")
	}
	return nil
}

// GetCatalogByHref  finds a Catalog by HREF
// On success, returns a pointer to the Catalog structure and a nil error
// On failure, returns a nil pointer and an error
func (org *Org) GetCatalogByHref(catalogHref string) (*Catalog, error) {
	cat := NewCatalog(org.client)

	_, err := org.client.ExecuteRequest(catalogHref, http.MethodGet,
		"", "error retrieving catalog: %s", nil, cat.Catalog)
	if err != nil {
		return nil, err
	}
	// The request was successful
	cat.parent = org
	return cat, nil
}

// GetCatalogByName  finds a Catalog by Name
// On success, returns a pointer to the Catalog structure and a nil error
// On failure, returns a nil pointer and an error
//
// refresh has no effect here, but is kept to preserve signature
func (org *Org) GetCatalogByName(catalogName string, refresh bool) (*Catalog, error) {
	vdcQuery, err := org.queryCatalogByName(catalogName)
	if ContainsNotFound(err) {
		return nil, ErrorEntityNotFound
	}
	if err != nil {
		return nil, fmt.Errorf("error querying Catalog: %s", err)
	}
	// This is not an AdminOrg and admin HREF must be removed if it exists
	href := strings.Replace(vdcQuery.HREF, "/api/admin", "/api", 1)
	return org.GetCatalogByHref(href)
}

// GetCatalogById finds a Catalog by ID
// On success, returns a pointer to the Catalog structure and a nil error
// On failure, returns a nil pointer and an error
func (org *Org) GetCatalogById(catalogId string, refresh bool) (*Catalog, error) {
	vdcQuery, err := org.queryCatalogById(catalogId)
	if ContainsNotFound(err) {
		return nil, ErrorEntityNotFound
	}
	if err != nil {
		return nil, fmt.Errorf("error querying Catalog: %s", err)
	}

	// This is not an AdminOrg and admin HREF must be removed if it exists
	href := strings.Replace(vdcQuery.HREF, "/api/admin", "/api", 1)
	return org.GetCatalogByHref(href)
}

// GetCatalogByNameOrId finds a Catalog by name or ID
// On success, returns a pointer to the Catalog structure and a nil error
// On failure, returns a nil pointer and an error
func (org *Org) GetCatalogByNameOrId(identifier string, refresh bool) (*Catalog, error) {
	getByName := func(name string, refresh bool) (interface{}, error) {
		return org.GetCatalogByName(name, refresh)
	}
	getById := func(id string, refresh bool) (interface{}, error) {
		return org.GetCatalogById(id, refresh)
	}
	entity, err := getEntityByNameOrIdSkipNonId(getByName, getById, identifier, refresh)
	if entity == nil {
		return nil, err
	}
	return entity.(*Catalog), err
}

// GetVDCByHref finds a VDC by HREF
// On success, returns a pointer to the VDC structure and a nil error
// On failure, returns a nil pointer and an error
func (org *Org) GetVDCByHref(vdcHref string) (*Vdc, error) {
	vdc := NewVdc(org.client)
	_, err := org.client.ExecuteRequest(vdcHref, http.MethodGet,
		"", "error retrieving VDC: %s", nil, vdc.Vdc)
	if err != nil {
		return nil, err
	}
	// The request was successful
	vdc.parent = org
	return vdc, nil
}

// GetVDCByName finds a VDC by Name
// On success, returns a pointer to the VDC structure and a nil error
// On failure, returns a nil pointer and an error
//
// refresh has no effect and is kept to preserve signature
func (org *Org) GetVDCByName(vdcName string, refresh bool) (*Vdc, error) {
	vdcQuery, err := org.queryOrgVdcByName(vdcName)
	if ContainsNotFound(err) {
		return nil, ErrorEntityNotFound
	}
	if err != nil {
		return nil, fmt.Errorf("error querying VDC: %s", err)
	}
	// This is not an AdminOrg and admin HREF must be removed if it exists
	href := strings.Replace(vdcQuery.HREF, "/api/admin", "/api", 1)
	return org.GetVDCByHref(href)
}

// GetVDCById finds a VDC by ID
// On success, returns a pointer to the VDC structure and a nil error
// On failure, returns a nil pointer and an error
//
// refresh has no effect and is kept to preserve signature
func (org *Org) GetVDCById(vdcId string, refresh bool) (*Vdc, error) {
	vdcQuery, err := org.queryOrgVdcById(vdcId)
	if ContainsNotFound(err) {
		return nil, ErrorEntityNotFound
	}
	if err != nil {
		return nil, fmt.Errorf("error querying VDC: %s", err)
	}

	// This is not an AdminOrg and admin HREF must be removed if it exists
	href := strings.Replace(vdcQuery.HREF, "/api/admin", "/api", 1)
	return org.GetVDCByHref(href)
}

// GetVDCByNameOrId finds a VDC by name or ID
// On success, returns a pointer to the VDC structure and a nil error
// On failure, returns a nil pointer and an error
//
// refresh has no effect and is kept to preserve signature
func (org *Org) GetVDCByNameOrId(identifier string, refresh bool) (*Vdc, error) {
	getByName := func(name string, refresh bool) (interface{}, error) {
		return org.GetVDCByName(name, refresh)
	}
	getById := func(id string, refresh bool) (interface{}, error) {
		return org.GetVDCById(id, refresh)
	}
	entity, err := getEntityByNameOrIdSkipNonId(getByName, getById, identifier, refresh)
	if entity == nil {
		return nil, err
	}
	return entity.(*Vdc), err
}

// QueryCatalogList returns a list of catalogs for this organization
func (org *Org) QueryCatalogList() ([]*types.CatalogRecord, error) {
	util.Logger.Printf("[DEBUG] QueryCatalogList with Org HREF %s", org.Org.HREF)
	filter := map[string]string{
		"org": org.Org.HREF,
	}
	return queryCatalogList(org.client, filter)
}

// GetTaskList returns Tasks for Organization and error.
func (org *Org) GetTaskList() (*types.TasksList, error) {

	for _, link := range org.Org.Link {
		if link.Rel == "down" && link.Type == "application/vnd.vmware.vcloud.tasksList+xml" {

			tasksList := &types.TasksList{}

			_, err := org.client.ExecuteRequest(link.HREF, http.MethodGet, "",
				"error getting taskList: %s", nil, tasksList)
			if err != nil {
				return nil, err
			}

			return tasksList, nil
		}
	}

	return nil, fmt.Errorf("link not found")
}

// queryOrgVdcByName returns a single QueryResultOrgVdcRecordType
func (org *Org) queryOrgVdcByName(vdcName string) (*types.QueryResultOrgVdcRecordType, error) {
	filterFields := map[string]string{
		"org":     org.Org.HREF,
		"orgName": org.Org.Name,
		"name":    vdcName,
	}
	allVdcs, err := queryOrgVdcList(org.client, filterFields)
	if err != nil {
		return nil, err
	}

	if allVdcs == nil || len(allVdcs) < 1 {
		return nil, ErrorEntityNotFound
	}

	if len(allVdcs) > 1 {
		return nil, fmt.Errorf("found more than 1 VDC with Name '%s'", vdcName)
	}

	return allVdcs[0], nil
}

// queryOrgVdcById returns a single QueryResultOrgVdcRecordType
func (org *Org) queryOrgVdcById(vdcId string) (*types.QueryResultOrgVdcRecordType, error) {
	filterMap := map[string]string{
		"org":     org.Org.HREF,
		"orgName": org.Org.Name,
		"id":      vdcId,
	}
	allVdcs, err := queryOrgVdcList(org.client, filterMap)

	if err != nil {
		return nil, err
	}

	if len(allVdcs) < 1 {
		return nil, ErrorEntityNotFound
	}

	return allVdcs[0], nil
}

// queryCatalogByName returns a single CatalogRecord
func (org *Org) queryCatalogByName(catalogName string) (*types.CatalogRecord, error) {
	filterMap := map[string]string{
		// Not injecting `org` or `orgName` here because shared catalogs may also appear here and they would have different
		// parent Org
		// "org":     org.Org.HREF,
		// "orgName": org.Org.Name,
		"name": catalogName,
	}
	allCatalogs, err := queryCatalogList(org.client, filterMap)
	if err != nil {
		return nil, err
	}

	if allCatalogs == nil || len(allCatalogs) < 1 {
		return nil, ErrorEntityNotFound
	}

	// To conform with this API standard it would be best to return an error if more than 1 item is found, but because
	// previous method of getting Catalog by Name returned the first result we are doing the same here
	// if len(allCatalogs) > 1 {
	// 	return nil, fmt.Errorf("found more than 1 Catalog with Name '%s'", catalogName)
	// }

	var localCatalog *types.CatalogRecord
	// if multiple results are found - return the one defined in `org` (local)
	if len(allCatalogs) > 1 {
		util.Logger.Printf("[DEBUG] org.queryCatalogByName found %d Catalogs by name '%s'", len(allCatalogs), catalogName)
		for _, catalog := range allCatalogs {
			util.Logger.Printf("[DEBUG] org.queryCatalogByName found a Catalog by name '%s' in Org '%s'", catalogName, catalog.OrgName)
			if catalog.OrgName == org.Org.Name {
				util.Logger.Printf("[DEBUG] org.queryCatalogByName Catalog '%s' is local for Org '%s'. Prioritising it",
					catalogName, org.Org.Name)
				// Not interrupting the loop here to still dump all results to logs
				localCatalog = catalog
			}
		}
	}

	// local catalog was found - return it
	if localCatalog != nil {
		return localCatalog, nil
	}

	// If only one catalog is found or multiple catalogs with no local ones - return the first one
	return allCatalogs[0], nil
}

// queryCatalogById returns a single QueryResultOrgVdcRecordType
func (org *Org) queryCatalogById(catalogId string) (*types.CatalogRecord, error) {
	filterMap := map[string]string{
		// Not injecting `org` or `orgName` here because shared catalogs may also appear here and they would have different
		// parent Org
		// "org":     org.Org.HREF,
		// "orgName": org.Org.Name,
		"id": catalogId,
	}
	allCatalogs, err := queryCatalogList(org.client, filterMap)

	if err != nil {
		return nil, err
	}

	if len(allCatalogs) < 1 {
		return nil, ErrorEntityNotFound
	}

	return allCatalogs[0], nil
}

// QueryOrgVdcList returns all Org VDCs using query endpoint
//
// Note. Being a 'System' user it will not return any VDC
func (org *Org) QueryOrgVdcList() ([]*types.QueryResultOrgVdcRecordType, error) {
	filter := map[string]string{
		"org": org.Org.HREF,
	}

	return queryOrgVdcList(org.client, filter)
}

// queryOrgVdcList performs an `orgVdc` or `adminOrgVdc` (for System user) and optionally applies filterFields
func queryOrgVdcList(client *Client, filterFields map[string]string) ([]*types.QueryResultOrgVdcRecordType, error) {
	util.Logger.Printf("[DEBUG] queryOrgVdcList with filter %#v", filterFields)
	queryType := client.GetQueryType(types.QtOrgVdc)

	filter := map[string]string{
		"type": queryType,
	}

	// When a map of filters with non empty keys and values is supplied - apply it
	if filterFields != nil {
		filterSlice := make([]string, 0)

		for filterFieldName, filterFieldValue := range filterFields {
			// Do not inject 'org' filter for System user as API returns an error
			if !client.IsSysAdmin && filterFieldName == "org" {
				continue
			}

			if filterFieldName != "" && filterFieldValue != "" {
				filterText := fmt.Sprintf("%s==%s", filterFieldName, url.QueryEscape(filterFieldValue))
				filterSlice = append(filterSlice, filterText)
			}
		}

		if len(filterSlice) > 0 {
			filter["filter"] = strings.Join(filterSlice, ";")
			filter["filterEncoded"] = "true"
		}
	}

	results, err := client.cumulativeQuery(queryType, nil, filter)
	if err != nil {
		return nil, fmt.Errorf("error querying Org VDCs %s", err)
	}

	if client.IsSysAdmin {
		return results.Results.OrgVdcAdminRecord, nil
	} else {
		return results.Results.OrgVdcRecord, nil
	}
}

func queryCatalogList(client *Client, filterFields map[string]string) ([]*types.CatalogRecord, error) {
	util.Logger.Printf("[DEBUG] queryCatalogList with filter %#v", filterFields)
	queryType := client.GetQueryType(types.QtCatalog)

	filter := map[string]string{
		"type": queryType,
	}

	// When a map of filters with non empty keys and values is supplied - apply it
	if filterFields != nil {
		filterSlice := make([]string, 0)

		for filterFieldName, filterFieldValue := range filterFields {
			// Do not inject 'org' filter for System user as API returns an error
			if !client.IsSysAdmin && filterFieldName == "org" {
				continue
			}

			if filterFieldName != "" && filterFieldValue != "" {
				filterText := fmt.Sprintf("%s==%s", filterFieldName, url.QueryEscape(filterFieldValue))
				filterSlice = append(filterSlice, filterText)
			}
		}

		if len(filterSlice) > 0 {
			filter["filter"] = strings.Join(filterSlice, ";")
			filter["filterEncoded"] = "true"
		}
	}

	results, err := client.cumulativeQuery(queryType, nil, filter)
	if err != nil {
		return nil, err
	}

	var catalogs []*types.CatalogRecord

	if client.IsSysAdmin {
		catalogs = results.Results.AdminCatalogRecord
	} else {
		catalogs = results.Results.CatalogRecord
	}
	util.Logger.Printf("[DEBUG] QueryCatalogList returned with : %#v and error: %s", catalogs, err)
	return catalogs, nil
}

// GetVappByHref returns a vApp reference by running a VCD API call
// If no valid vApp is found, it returns a nil VApp reference and an error
func (org *Org) GetVAppByHref(vappHref string) (*VApp, error) {
	newVapp := NewVApp(org.client)

	_, err := org.client.ExecuteRequest(vappHref, http.MethodGet,
		"", "error retrieving vApp: %s", nil, newVapp.VApp)

	if err != nil {
		return nil, err
	}
	return newVapp, nil
}