resource_repository_maven_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 ResourceRepositoryMavenProxy() *schema.Resource {
	return &schema.Resource{
		Description: "Use this resource to create a maven proxy repository.",

		Create: resourceMavenProxyRepositoryCreate,
		Delete: resourceMavenProxyRepositoryDelete,
		Exists: resourceMavenProxyRepositoryExists,
		Read:   resourceMavenProxyRepositoryRead,
		Update: resourceMavenProxyRepositoryUpdate,
		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.ResourceHTTPClientWithPreemptiveAuth,
			"negative_cache_enabled": repositorySchema.ResourceNegativeCacheEnabled,
			"negative_cache_ttl":     repositorySchema.ResourceNegativeCacheTTL,
			"proxy":                  repositorySchema.ResourceProxy,
			"routing_rule":           repositorySchema.ResourceRoutingRule,
			"storage":                repositorySchema.ResourceStorage,
			// Maven proxy schemas
			"maven": repositorySchema.ResourceMaven,
		},
	}
}

func getMavenProxyRepositoryFromResourceData(resourceData *schema.ResourceData) repository.MavenProxyRepository {
	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{})
	mavenConfig := resourceData.Get("maven").([]interface{})[0].(map[string]interface{})

	repo := repository.MavenProxyRepository{
		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.HTTPClientWithPreemptiveAuth{
			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),
		},
		Maven: repository.Maven{
			VersionPolicy: repository.MavenVersionPolicy(mavenConfig["version_policy"].(string)),
			LayoutPolicy:  repository.MavenLayoutPolicy(mavenConfig["layout_policy"].(string)),
		},
	}

	if mavenConfig["content_disposition"] != "" {
		contentDisposition := repository.MavenContentDisposition(mavenConfig["content_disposition"].(string))
		repo.Maven.ContentDisposition = &contentDisposition
	}

	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.HTTPClientAuthenticationWithPreemptive{
				NTLMDomain: authConfig["ntlm_domain"].(string),
				NTLMHost:   authConfig["ntlm_host"].(string),
				Type:       repository.HTTPClientAuthenticationType(authConfig["type"].(string)),
				Username:   authConfig["username"].(string),
				Password:   authConfig["password"].(string),
			}

			preemptive, ok := authConfig["preemptive"]
			if ok {
				repo.HTTPClient.Authentication.Preemptive = tools.GetBoolPointer(preemptive.(bool))
			}
		}
	}

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

	return repo
}

func setMavenProxyRepositoryToResourceData(repo *repository.MavenProxyRepository, 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("maven", flattenMaven(&repo.Maven)); err != nil {
		return err
	}

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

	if err := resourceData.Set("http_client", flattenHTTPClientWithPreemptiveAuth(&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 resourceMavenProxyRepositoryCreate(resourceData *schema.ResourceData, m interface{}) error {
	client := m.(*nexus.NexusClient)

	repo := getMavenProxyRepositoryFromResourceData(resourceData)

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

	return resourceMavenProxyRepositoryRead(resourceData, m)
}

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

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

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

	return setMavenProxyRepositoryToResourceData(repo, resourceData)
}

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

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

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

	return resourceMavenProxyRepositoryRead(resourceData, m)
}

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

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

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