terraform-provider-unifi/internal/provider/lazy_client.go

package provider

import (
	"crypto/tls"
	"net"
	"net/http"
	"net/http/cookiejar"
	"sync"
	"time"

	"github.com/hashicorp/terraform/helper/logging"
	"github.com/paultyng/go-unifi/unifi"
)

type lazyClient struct {
	baseURL string
	user    string
	pass    string

	once  sync.Once
	inner *unifi.Client
}

func setHTTPClient(c *unifi.Client) {
	httpClient := &http.Client{}
	httpClient.Transport = &http.Transport{
		Proxy: http.ProxyFromEnvironment,
		DialContext: (&net.Dialer{
			Timeout:   30 * time.Second,
			KeepAlive: 30 * time.Second,
			DualStack: true,
		}).DialContext,
		MaxIdleConns:          100,
		IdleConnTimeout:       90 * time.Second,
		TLSHandshakeTimeout:   10 * time.Second,
		ExpectContinueTimeout: 1 * time.Second,

		// TODO: make this opt-in
		TLSClientConfig: &tls.Config{
			InsecureSkipVerify: true,
		},
	}

	httpClient.Transport = logging.NewTransport("Unifi", httpClient.Transport)

	jar, _ := cookiejar.New(nil)
	httpClient.Jar = jar

	c.SetHTTPClient(httpClient)
}

func (c *lazyClient) init() error {
	var err error
	c.once.Do(func() {
		c.inner = &unifi.Client{}
		setHTTPClient(c.inner)

		err = c.inner.SetBaseURL(c.baseURL)
		if err != nil {
			return
		}

		err = c.inner.Login(c.user, c.pass)
	})
	return err
}

func (c *lazyClient) ListUserGroup(site string) ([]unifi.UserGroup, error) {
	c.init()
	return c.inner.ListUserGroup(site)
}
func (c *lazyClient) ListWLANGroup(site string) ([]unifi.WLANGroup, error) {
	c.init()
	return c.inner.ListWLANGroup(site)
}
func (c *lazyClient) DeleteNetwork(site, id, name string) error {
	c.init()
	return c.inner.DeleteNetwork(site, id, name)
}
func (c *lazyClient) CreateNetwork(site string, d *unifi.Network) (*unifi.Network, error) {
	c.init()
	return c.inner.CreateNetwork(site, d)
}
func (c *lazyClient) GetNetwork(site, id string) (*unifi.Network, error) {
	c.init()
	return c.inner.GetNetwork(site, id)
}
func (c *lazyClient) UpdateNetwork(site string, d *unifi.Network) (*unifi.Network, error) {
	c.init()
	return c.inner.UpdateNetwork(site, d)
}
func (c *lazyClient) DeleteWLAN(site, id string) error {
	c.init()
	return c.inner.DeleteWLAN(site, id)
}
func (c *lazyClient) CreateWLAN(site string, d *unifi.WLAN) (*unifi.WLAN, error) {
	c.init()
	return c.inner.CreateWLAN(site, d)
}
func (c *lazyClient) GetWLAN(site, id string) (*unifi.WLAN, error) {
	c.init()
	return c.inner.GetWLAN(site, id)
}
func (c *lazyClient) UpdateWLAN(site string, d *unifi.WLAN) (*unifi.WLAN, error) {
	c.init()
	return c.inner.UpdateWLAN(site, d)
}
func (c *lazyClient) DeleteUserGroup(site, id string) error {
	c.init()
	return c.inner.DeleteUserGroup(site, id)
}
func (c *lazyClient) CreateUserGroup(site string, d *unifi.UserGroup) (*unifi.UserGroup, error) {
	c.init()
	return c.inner.CreateUserGroup(site, d)
}
func (c *lazyClient) GetUserGroup(site, id string) (*unifi.UserGroup, error) {
	c.init()
	return c.inner.GetUserGroup(site, id)
}
func (c *lazyClient) UpdateUserGroup(site string, d *unifi.UserGroup) (*unifi.UserGroup, error) {
	c.init()
	return c.inner.UpdateUserGroup(site, d)
}
func (c *lazyClient) GetUser(site, id string) (*unifi.User, error) {
	c.init()
	return c.inner.GetUser(site, id)
}
func (c *lazyClient) GetUserByMAC(site, mac string) (*unifi.User, error) {
	c.init()
	return c.inner.GetUserByMAC(site, mac)
}
func (c *lazyClient) CreateUser(site string, d *unifi.User) (*unifi.User, error) {
	c.init()
	return c.inner.CreateUser(site, d)
}
func (c *lazyClient) UpdateUser(site string, d *unifi.User) (*unifi.User, error) {
	c.init()
	return c.inner.UpdateUser(site, d)
}
func (c *lazyClient) DeleteUserByMAC(site, mac string) error {
	c.init()
	return c.inner.DeleteUserByMAC(site, mac)
}
func (c *lazyClient) BlockUserByMAC(site, mac string) error {
	c.init()
	return c.inner.BlockUserByMAC(site, mac)
}
func (c *lazyClient) UnblockUserByMAC(site, mac string) error {
	c.init()
	return c.inner.UnblockUserByMAC(site, mac)
}
func (c *lazyClient) ListFirewallGroup(site string) ([]unifi.FirewallGroup, error) {
	c.init()
	return c.inner.ListFirewallGroup(site)
}
func (c *lazyClient) DeleteFirewallGroup(site, id string) error {
	c.init()
	return c.inner.DeleteFirewallGroup(site, id)
}
func (c *lazyClient) CreateFirewallGroup(site string, d *unifi.FirewallGroup) (*unifi.FirewallGroup, error) {
	c.init()
	return c.inner.CreateFirewallGroup(site, d)
}
func (c *lazyClient) GetFirewallGroup(site, id string) (*unifi.FirewallGroup, error) {
	c.init()
	return c.inner.GetFirewallGroup(site, id)
}
func (c *lazyClient) UpdateFirewallGroup(site string, d *unifi.FirewallGroup) (*unifi.FirewallGroup, error) {
	c.init()
	return c.inner.UpdateFirewallGroup(site, d)
}
func (c *lazyClient) GetPortForward(site, id string) (*unifi.PortForward, error) {
	c.init()
	return c.inner.GetPortForward(site, id)
}
func (c *lazyClient) DeletePortForward(site, id string) error {
	c.init()
	return c.inner.DeletePortForward(site, id)
}
func (c *lazyClient) CreatePortForward(site string, d *unifi.PortForward) (*unifi.PortForward, error) {
	c.init()
	return c.inner.CreatePortForward(site, d)
}
func (c *lazyClient) UpdatePortForward(site string, d *unifi.PortForward) (*unifi.PortForward, error) {
	c.init()
	return c.inner.UpdatePortForward(site, d)
}