transfer.sh/vendor/github.com/google/martian/api/forwarder.go

// Copyright 2016 Google Inc. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Package api contains a forwarder to route system HTTP requests to the
// local API server.
package api

import (
	"fmt"
	"net/http"

	"github.com/google/martian"
	"github.com/google/martian/log"
)

// Forwarder is a request modifier that routes the request to the API server and
// marks the request for skipped logging.
type Forwarder struct {
	host string
	port int
}

// NewForwarder returns a Forwarder that rewrites requests to host.
func NewForwarder(host string, port int) *Forwarder {
	return &Forwarder{
		host: host,
		port: port,
	}
}

// ModifyRequest forwards the request to the local API server running at f.port,
// downgrades the scheme to http and marks the request context for skipped logging.
func (f *Forwarder) ModifyRequest(req *http.Request) error {
	ctx := martian.NewContext(req)
	ctx.APIRequest()
	ctx.SkipLogging()

	in := req.URL.String()
	req.URL.Scheme = "http"
	req.URL.Host = fmt.Sprintf("%s:%d", "localhost", f.port)
	out := req.URL.String()
	log.Infof("api.Forwarder: forwarding %s to %s", in, out)

	return nil
}