mirror of https://github.com/portainer/portainer
138 lines
4.1 KiB
Go
138 lines
4.1 KiB
Go
package docker
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/docker/docker/client"
|
|
"github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/crypto"
|
|
)
|
|
|
|
const (
|
|
unsupportedEnvironmentType = portainer.Error("Environment not supported")
|
|
defaultDockerRequestTimeout = 60
|
|
dockerClientVersion = "1.40"
|
|
)
|
|
|
|
// ClientFactory is used to create Docker clients
|
|
type ClientFactory struct {
|
|
signatureService portainer.DigitalSignatureService
|
|
reverseTunnelService portainer.ReverseTunnelService
|
|
}
|
|
|
|
// NewClientFactory returns a new instance of a ClientFactory
|
|
func NewClientFactory(signatureService portainer.DigitalSignatureService, reverseTunnelService portainer.ReverseTunnelService) *ClientFactory {
|
|
return &ClientFactory{
|
|
signatureService: signatureService,
|
|
reverseTunnelService: reverseTunnelService,
|
|
}
|
|
}
|
|
|
|
// CreateClient is a generic function to create a Docker client based on
|
|
// a specific endpoint configuration. The nodeName parameter can be used
|
|
// with an agent enabled endpoint to target a specific node in an agent cluster.
|
|
func (factory *ClientFactory) CreateClient(endpoint *portainer.Endpoint, nodeName string) (*client.Client, error) {
|
|
if endpoint.Type == portainer.AzureEnvironment {
|
|
return nil, unsupportedEnvironmentType
|
|
} else if endpoint.Type == portainer.AgentOnDockerEnvironment {
|
|
return createAgentClient(endpoint, factory.signatureService, nodeName)
|
|
} else if endpoint.Type == portainer.EdgeAgentEnvironment {
|
|
return createEdgeClient(endpoint, factory.reverseTunnelService, nodeName)
|
|
}
|
|
|
|
if strings.HasPrefix(endpoint.URL, "unix://") || strings.HasPrefix(endpoint.URL, "npipe://") {
|
|
return createLocalClient(endpoint)
|
|
}
|
|
return createTCPClient(endpoint)
|
|
}
|
|
|
|
func createLocalClient(endpoint *portainer.Endpoint) (*client.Client, error) {
|
|
return client.NewClientWithOpts(
|
|
client.WithHost(endpoint.URL),
|
|
client.WithVersion(dockerClientVersion),
|
|
)
|
|
}
|
|
|
|
func createTCPClient(endpoint *portainer.Endpoint) (*client.Client, error) {
|
|
httpCli, err := httpClient(endpoint)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return client.NewClientWithOpts(
|
|
client.WithHost(endpoint.URL),
|
|
client.WithVersion(dockerClientVersion),
|
|
client.WithHTTPClient(httpCli),
|
|
)
|
|
}
|
|
|
|
func createEdgeClient(endpoint *portainer.Endpoint, reverseTunnelService portainer.ReverseTunnelService, nodeName string) (*client.Client, error) {
|
|
httpCli, err := httpClient(endpoint)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
headers := map[string]string{}
|
|
if nodeName != "" {
|
|
headers[portainer.PortainerAgentTargetHeader] = nodeName
|
|
}
|
|
|
|
tunnel := reverseTunnelService.GetTunnelDetails(endpoint.ID)
|
|
endpointURL := fmt.Sprintf("http://localhost:%d", tunnel.Port)
|
|
|
|
return client.NewClientWithOpts(
|
|
client.WithHost(endpointURL),
|
|
client.WithVersion(dockerClientVersion),
|
|
client.WithHTTPClient(httpCli),
|
|
client.WithHTTPHeaders(headers),
|
|
)
|
|
}
|
|
|
|
func createAgentClient(endpoint *portainer.Endpoint, signatureService portainer.DigitalSignatureService, nodeName string) (*client.Client, error) {
|
|
httpCli, err := httpClient(endpoint)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
signature, err := signatureService.CreateSignature(portainer.PortainerAgentSignatureMessage)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
headers := map[string]string{
|
|
portainer.PortainerAgentPublicKeyHeader: signatureService.EncodedPublicKey(),
|
|
portainer.PortainerAgentSignatureHeader: signature,
|
|
}
|
|
|
|
if nodeName != "" {
|
|
headers[portainer.PortainerAgentTargetHeader] = nodeName
|
|
}
|
|
|
|
return client.NewClientWithOpts(
|
|
client.WithHost(endpoint.URL),
|
|
client.WithVersion(dockerClientVersion),
|
|
client.WithHTTPClient(httpCli),
|
|
client.WithHTTPHeaders(headers),
|
|
)
|
|
}
|
|
|
|
func httpClient(endpoint *portainer.Endpoint) (*http.Client, error) {
|
|
transport := &http.Transport{}
|
|
|
|
if endpoint.TLSConfig.TLS {
|
|
tlsConfig, err := crypto.CreateTLSConfigurationFromDisk(endpoint.TLSConfig.TLSCACertPath, endpoint.TLSConfig.TLSCertPath, endpoint.TLSConfig.TLSKeyPath, endpoint.TLSConfig.TLSSkipVerify)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
transport.TLSClientConfig = tlsConfig
|
|
}
|
|
|
|
return &http.Client{
|
|
Transport: transport,
|
|
Timeout: defaultDockerRequestTimeout * time.Second,
|
|
}, nil
|
|
}
|