terraform-provider-proxmox/proxmoxtf/config.go

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

package proxmoxtf

import (
	"errors"
	"os"

	"github.com/bpg/terraform-provider-proxmox/proxmox"
	"github.com/bpg/terraform-provider-proxmox/proxmox/api"
	"github.com/bpg/terraform-provider-proxmox/proxmox/ssh"
)

// ProviderConfiguration is the configuration for the provider.
type ProviderConfiguration struct {
	apiClient      api.Client
	sshClient      ssh.Client
	tmpDirOverride string
}

// NewProviderConfiguration creates a new provider configuration.
func NewProviderConfiguration(
	apiClient api.Client,
	sshClient ssh.Client,
	tmpDirOverride string,
) ProviderConfiguration {
	return ProviderConfiguration{
		apiClient:      apiClient,
		sshClient:      sshClient,
		tmpDirOverride: tmpDirOverride,
	}
}

// GetClient returns the Proxmox API client.
func (c *ProviderConfiguration) GetClient() (proxmox.Client, error) {
	if c.apiClient == nil {
		return nil, errors.New(
			"you must specify the API access details in the provider configuration",
		)
	}

	if c.sshClient == nil {
		return nil, errors.New(
			"you must specify the SSH access details in the provider configuration",
		)
	}

	return proxmox.NewClient(c.apiClient, c.sshClient, c.tmpDirOverride), nil
}

// TempDir returns (possibly overridden) os.TempDir().
func (c *ProviderConfiguration) TempDir() string {
	if c.tmpDirOverride != "" {
		return c.tmpDirOverride
	}

	return os.TempDir()
}