mirror of https://github.com/portainer/portainer
37 lines
1.2 KiB
Go
37 lines
1.2 KiB
Go
package agent
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
)
|
|
|
|
// Transport is an http.Transport wrapper that adds custom http headers to communicate to an Agent
|
|
type Transport struct {
|
|
httpTransport *http.Transport
|
|
signatureService portainer.DigitalSignatureService
|
|
}
|
|
|
|
// NewTransport returns a new transport that can be used to send signed requests to a Portainer agent
|
|
func NewTransport(signatureService portainer.DigitalSignatureService, httpTransport *http.Transport) *Transport {
|
|
transport := &Transport{
|
|
httpTransport: httpTransport,
|
|
signatureService: signatureService,
|
|
}
|
|
|
|
return transport
|
|
}
|
|
|
|
// RoundTrip is the implementation of the the http.RoundTripper interface
|
|
func (transport *Transport) RoundTrip(request *http.Request) (*http.Response, error) {
|
|
signature, err := transport.signatureService.CreateSignature(portainer.PortainerAgentSignatureMessage)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
request.Header.Set(portainer.PortainerAgentPublicKeyHeader, transport.signatureService.EncodedPublicKey())
|
|
request.Header.Set(portainer.PortainerAgentSignatureHeader, signature)
|
|
|
|
return transport.httpTransport.RoundTrip(request)
|
|
}
|