resource_repository_conda_proxy.go

package repository

import (
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
	nexus "github.com/nduyphuong/go-nexus-client/nexus3"
	"github.com/nduyphuong/go-nexus-client/nexus3/schema/repository"
	"github.com/nduyphuong/terraform-provider-nexus/internal/schema/common"
	repositorySchema "github.com/nduyphuong/terraform-provider-nexus/internal/schema/repository"
	"github.com/nduyphuong/terraform-provider-nexus/internal/tools"
)

func ResourceRepositoryCondaProxy() *schema.Resource {
	return &schema.Resource{
		Description: "Use this resource to create an conda proxy repository.",

		Create: resourceCondaProxyRepositoryCreate,
		Delete: resourceCondaProxyRepositoryDelete,
		Exists: resourceCondaProxyRepositoryExists,
		Read:   resourceCondaProxyRepositoryRead,
		Update: resourceCondaProxyRepositoryUpdate,
		Importer: &schema.ResourceImporter{
			StateContext: schema.ImportStatePassthroughContext,
		},

		Schema: map[string]*schema.Schema{
			// Common schemas
			"id":     common.ResourceID,
			"name":   repositorySchema.ResourceName,
			"online": repositorySchema.ResourceOnline,
			// Proxy schemas
			"cleanup":                repositorySchema.ResourceCleanup,
			"http_client":            repositorySchema.ResourceHTTPClient,
			"negative_cache_enabled": repositorySchema.ResourceNegativeCacheEnabled,
			"negative_cache_ttl":     repositorySchema.ResourceNegativeCacheTTL,
			"proxy":                  repositorySchema.ResourceProxy,
			"routing_rule":           repositorySchema.ResourceRoutingRule,
			"storage":                repositorySchema.ResourceStorage,
		},
	}
}

func getCondaProxyRepositoryFromResourceData(resourceData *schema.ResourceData) repository.CondaProxyRepository {
	httpClientConfig := resourceData.Get("http_client").([]interface{})[0].(map[string]interface{})
	negativeCacheEnabled := resourceData.Get("negative_cache_enabled").(bool)
	negativeCacheTTL := resourceData.Get("negative_cache_ttl").(int)
	negativeCacheConfig := map[string]interface{}{
		"enabled": negativeCacheEnabled,
		"ttl":     negativeCacheTTL,
	}
	proxyConfig := resourceData.Get("proxy").([]interface{})[0].(map[string]interface{})
	storageConfig := resourceData.Get("storage").([]interface{})[0].(map[string]interface{})

	repo := repository.CondaProxyRepository{
		Name:   resourceData.Get("name").(string),
		Online: resourceData.Get("online").(bool),
		Storage: repository.Storage{
			BlobStoreName:               storageConfig["blob_store_name"].(string),
			StrictContentTypeValidation: storageConfig["strict_content_type_validation"].(bool),
		},
		HTTPClient: repository.HTTPClient{
			AutoBlock: httpClientConfig["auto_block"].(bool),
			Blocked:   httpClientConfig["blocked"].(bool),
		},
		NegativeCache: repository.NegativeCache{
			Enabled: negativeCacheConfig["enabled"].(bool),
			TTL:     negativeCacheConfig["ttl"].(int),
		},
		Proxy: repository.Proxy{
			ContentMaxAge:  proxyConfig["content_max_age"].(int),
			MetadataMaxAge: proxyConfig["metadata_max_age"].(int),
			RemoteURL:      proxyConfig["remote_url"].(string),
		},
	}

	if routingRule, ok := resourceData.GetOk("routing_rule"); ok {
		repo.RoutingRule = tools.GetStringPointer(routingRule.(string))
		repo.RoutingRuleName = tools.GetStringPointer(routingRule.(string))
	}

	cleanup, exists := resourceData.GetOk("cleanup")
	if exists {
		cleanupConfig := cleanup.([]interface{})[0].(map[string]interface{})
		if len(cleanupConfig) > 0 {
			policy_names, ok := cleanupConfig["policy_names"]
			if ok {
				repo.Cleanup = &repository.Cleanup{
					PolicyNames: tools.InterfaceSliceToStringSlice(policy_names.(*schema.Set).List()),
				}
			}
		}
	}

	if v, ok := httpClientConfig["authentication"]; ok {
		authList := v.([]interface{})
		if len(authList) == 1 && authList[0] != nil {
			authConfig := authList[0].(map[string]interface{})

			repo.HTTPClient.Authentication = &repository.HTTPClientAuthentication{
				NTLMDomain: authConfig["ntlm_domain"].(string),
				NTLMHost:   authConfig["ntlm_host"].(string),
				Type:       repository.HTTPClientAuthenticationType(authConfig["type"].(string)),
				Username:   authConfig["username"].(string),
				Password:   authConfig["password"].(string),
			}
		}
	}

	if v, ok := httpClientConfig["connection"]; ok {
		repo.HTTPClient.Connection = getHTTPClientConnection(v.([]interface{}))
	}

	return repo
}

func setCondaProxyRepositoryToResourceData(repo *repository.CondaProxyRepository, resourceData *schema.ResourceData) error {
	resourceData.SetId(repo.Name)
	resourceData.Set("name", repo.Name)
	resourceData.Set("online", repo.Online)

	if repo.RoutingRuleName != nil {
		resourceData.Set("routing_rule", repo.RoutingRuleName)
	} else if repo.RoutingRule != nil {
		resourceData.Set("routing_rule", repo.RoutingRule)
	}

	if err := resourceData.Set("storage", flattenStorage(&repo.Storage)); err != nil {
		return err
	}

	if err := resourceData.Set("http_client", flattenHTTPClient(&repo.HTTPClient, resourceData)); err != nil {
		return err
	}

	if err := resourceData.Set("negative_cache_enabled", repo.NegativeCache.Enabled); err != nil {
		return err
	}

	if err := resourceData.Set("negative_cache_ttl", repo.NegativeCache.TTL); err != nil {
		return err
	}

	if err := resourceData.Set("proxy", flattenProxy(&repo.Proxy)); err != nil {
		return err
	}

	if repo.Cleanup != nil {
		if err := resourceData.Set("cleanup", flattenCleanup(repo.Cleanup)); err != nil {
			return err
		}
	}
	return nil
}

func resourceCondaProxyRepositoryCreate(resourceData *schema.ResourceData, m interface{}) error {
	client := m.(*nexus.NexusClient)

	repo := getCondaProxyRepositoryFromResourceData(resourceData)

	if err := client.Repository.Conda.Proxy.Create(repo); err != nil {
		return err
	}
	resourceData.SetId(repo.Name)

	return resourceCondaProxyRepositoryRead(resourceData, m)
}

func resourceCondaProxyRepositoryRead(resourceData *schema.ResourceData, m interface{}) error {
	client := m.(*nexus.NexusClient)

	repo, err := client.Repository.Conda.Proxy.Get(resourceData.Id())
	if err != nil {
		return err
	}

	if repo == nil {
		resourceData.SetId("")
		return nil
	}

	return setCondaProxyRepositoryToResourceData(repo, resourceData)
}

func resourceCondaProxyRepositoryUpdate(resourceData *schema.ResourceData, m interface{}) error {
	client := m.(*nexus.NexusClient)

	repoName := resourceData.Id()
	repo := getCondaProxyRepositoryFromResourceData(resourceData)

	if err := client.Repository.Conda.Proxy.Update(repoName, repo); err != nil {
		return err
	}

	return resourceCondaProxyRepositoryRead(resourceData, m)
}

func resourceCondaProxyRepositoryDelete(resourceData *schema.ResourceData, m interface{}) error {
	client := m.(*nexus.NexusClient)
	return client.Repository.Conda.Proxy.Delete(resourceData.Id())
}

func resourceCondaProxyRepositoryExists(resourceData *schema.ResourceData, m interface{}) (bool, error) {
	client := m.(*nexus.NexusClient)

	repo, err := client.Repository.Conda.Proxy.Get(resourceData.Id())
	return repo != nil, err
}