package proxy import ( "fmt" "net/http" "github.com/portainer/portainer/api/dataservices" "github.com/portainer/portainer/api/http/proxy/factory/kubernetes" cmap "github.com/orcaman/concurrent-map" "github.com/portainer/portainer/api/kubernetes/cli" portainer "github.com/portainer/portainer/api" "github.com/portainer/portainer/api/docker" "github.com/portainer/portainer/api/http/proxy/factory" ) // TODO: contain code related to legacy extension management type ( // Manager represents a service used to manage proxies to environments(endpoints) and extensions. Manager struct { proxyFactory *factory.ProxyFactory endpointProxies cmap.ConcurrentMap legacyExtensionProxies cmap.ConcurrentMap k8sClientFactory *cli.ClientFactory } ) // NewManager initializes a new proxy Service func NewManager(dataStore dataservices.DataStore, signatureService portainer.DigitalSignatureService, tunnelService portainer.ReverseTunnelService, clientFactory *docker.ClientFactory, kubernetesClientFactory *cli.ClientFactory, kubernetesTokenCacheManager *kubernetes.TokenCacheManager) *Manager { return &Manager{ endpointProxies: cmap.New(), legacyExtensionProxies: cmap.New(), k8sClientFactory: kubernetesClientFactory, proxyFactory: factory.NewProxyFactory(dataStore, signatureService, tunnelService, clientFactory, kubernetesClientFactory, kubernetesTokenCacheManager), } } // CreateAndRegisterEndpointProxy creates a new HTTP reverse proxy based on environment(endpoint) properties and and adds it to the registered proxies. // It can also be used to create a new HTTP reverse proxy and replace an already registered proxy. func (manager *Manager) CreateAndRegisterEndpointProxy(endpoint *portainer.Endpoint) (http.Handler, error) { proxy, err := manager.proxyFactory.NewEndpointProxy(endpoint) if err != nil { return nil, err } manager.endpointProxies.Set(fmt.Sprint(endpoint.ID), proxy) return proxy, nil } // CreateAgentProxyServer creates a new HTTP reverse proxy based on environment(endpoint) properties and and adds it to the registered proxies. // It can also be used to create a new HTTP reverse proxy and replace an already registered proxy. func (manager *Manager) CreateAgentProxyServer(endpoint *portainer.Endpoint) (*factory.ProxyServer, error) { return manager.proxyFactory.NewAgentProxy(endpoint) } // GetEndpointProxy returns the proxy associated to a key func (manager *Manager) GetEndpointProxy(endpoint *portainer.Endpoint) http.Handler { proxy, ok := manager.endpointProxies.Get(fmt.Sprint(endpoint.ID)) if !ok { return nil } return proxy.(http.Handler) } // DeleteEndpointProxy deletes the proxy associated to a key // and cleans the k8s environment(endpoint) client cache. DeleteEndpointProxy // is currently only called for edge connection clean up. func (manager *Manager) DeleteEndpointProxy(endpointID portainer.EndpointID) { manager.endpointProxies.Remove(fmt.Sprint(endpointID)) manager.k8sClientFactory.RemoveKubeClient(endpointID) } // CreateLegacyExtensionProxy creates a new HTTP reverse proxy for a legacy extension and adds it to the registered proxies func (manager *Manager) CreateLegacyExtensionProxy(key, extensionAPIURL string) (http.Handler, error) { proxy, err := manager.proxyFactory.NewLegacyExtensionProxy(extensionAPIURL) if err != nil { return nil, err } manager.legacyExtensionProxies.Set(key, proxy) return proxy, nil } // GetLegacyExtensionProxy returns a legacy extension proxy associated to a key func (manager *Manager) GetLegacyExtensionProxy(key string) http.Handler { proxy, ok := manager.legacyExtensionProxies.Get(key) if !ok { return nil } return proxy.(http.Handler) } // CreateGitlabProxy creates a new HTTP reverse proxy that can be used to send requests to the Gitlab API func (manager *Manager) CreateGitlabProxy(url string) (http.Handler, error) { return manager.proxyFactory.NewGitlabProxy(url) }