github.com/minio/console@v1.4.1/api/tls.go (about)

     1  // This file is part of MinIO Console Server
     2  // Copyright (c) 2021 MinIO, Inc.
     3  //
     4  // This program is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU Affero General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // This program is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    12  // GNU Affero General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU Affero General Public License
    15  // along with this program.  If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package api
    18  
    19  import (
    20  	"net/http"
    21  )
    22  
    23  type ConsoleTransport struct {
    24  	Transport http.RoundTripper
    25  	ClientIP  string
    26  }
    27  
    28  func (t *ConsoleTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    29  	if t.ClientIP != "" {
    30  		// Do not set an empty x-forwarded-for
    31  		req.Header.Add(xForwardedFor, t.ClientIP)
    32  	}
    33  	return t.Transport.RoundTrip(req)
    34  }
    35  
    36  // PrepareSTSClientTransport :
    37  func PrepareSTSClientTransport(clientIP string) *ConsoleTransport {
    38  	return &ConsoleTransport{
    39  		Transport: GlobalTransport,
    40  		ClientIP:  clientIP,
    41  	}
    42  }
    43  
    44  // PrepareConsoleHTTPClient returns an http.Client with custom configurations need it by *credentials.STSAssumeRole
    45  // custom configurations include the use of CA certificates
    46  func PrepareConsoleHTTPClient(clientIP string) *http.Client {
    47  	// Return http client with default configuration
    48  	return &http.Client{
    49  		Transport: PrepareSTSClientTransport(clientIP),
    50  	}
    51  }