resource_repository_pypi_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 ResourceRepositoryPypiProxy() *schema.Resource {
	return &schema.Resource{
		Description: "Use this resource to create an NPM proxy repository.",

		Create: resourcePypiProxyRepositoryCreate,
		Delete: resourcePypiProxyRepositoryDelete,
		Exists: resourcePypiProxyRepositoryExists,
		Read:   resourcePypiProxyRepositoryRead,
		Update: resourcePypiProxyRepositoryUpdate,
		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 getPypiProxyRepositoryFromResourceData(resourceData *schema.ResourceData) repository.PypiProxyRepository {
	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.PypiProxyRepository{
		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 setPypiProxyRepositoryToResourceData(repo *repository.PypiProxyRepository, 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 resourcePypiProxyRepositoryCreate(resourceData *schema.ResourceData, m interface{}) error {
	client := m.(*nexus.NexusClient)

	repo := getPypiProxyRepositoryFromResourceData(resourceData)

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

	return resourcePypiProxyRepositoryRead(resourceData, m)
}

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

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

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

	return setPypiProxyRepositoryToResourceData(repo, resourceData)
}

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

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

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

	return resourcePypiProxyRepositoryRead(resourceData, m)
}

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

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

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