rclone/cmd/serve/httplib/httplib.go

// Package httplib provides common functionality for http servers
package httplib

import (
	"fmt"
	"log"
	"net/http"

	auth "github.com/abbot/go-http-auth"
	"github.com/ncw/rclone/fs"
	"github.com/spf13/pflag"
)

// Globals
var (
	bindAddress  = "localhost:8080"
	htPasswdFile string
)

// AddFlags adds the http server specific flags
func AddFlags(flagSet *pflag.FlagSet) {
	flagSet.StringVarP(&bindAddress, "addr", "", bindAddress, "IPaddress:Port to bind server to.")
	flagSet.StringVarP(&htPasswdFile, "htpasswd", "", htPasswdFile, "File to use for htpasswd authentication.")
}

// Help contains text describing the http server to add to the command
// help.
var Help = `
### Server options

Use --addr to specify which IP address and port the server should
listen on, eg --addr 1.2.3.4:8000 or --addr :8080 to listen to all
IPs.  By default it only listens on localhost.

Use --htpasswd /path/to/htpasswd to provide an htpasswd file.  This is
in standard apache format and supports MD5, SHA1 and BCrypt for basic
authentication.  Bcrypt is recommended.

To create an htpasswd file:

    touch htpasswd
    htpasswd -B htpasswd user
    htpasswd -B htpasswd anotherUser
`

// Server contains info about the running http server
type Server struct {
	bindAddress string
	httpServer  *http.Server
}

// NewServer creates an http server
func NewServer(handler http.Handler) *Server {
	// Use htpasswd if required on everything
	if htPasswdFile != "" {
		fs.Infof(nil, "Using %q as htpasswd storage", htPasswdFile)
		secretProvider := auth.HtpasswdFileProvider(htPasswdFile)
		authenticator := auth.NewBasicAuthenticator("rclone", secretProvider)
		handler = auth.JustCheck(authenticator, handler.ServeHTTP)
	}

	s := &Server{
		bindAddress: bindAddress,
	}
	// FIXME make a transport?
	s.httpServer = &http.Server{
		Addr:           s.bindAddress,
		Handler:        handler,
		MaxHeaderBytes: 1 << 20,
	}
	// go version specific initialisation
	initServer(s.httpServer)
	return s
}

// SetBindAddress overrides the config flag
func (s *Server) SetBindAddress(addr string) {
	s.bindAddress = addr
	s.httpServer.Addr = addr
}

// Serve runs the server - doesn't return
func (s *Server) Serve() {
	log.Fatal(s.httpServer.ListenAndServe())
}

// URL returns the serving address of this server
func (s *Server) URL() string {
	return fmt.Sprintf("http://%s/", s.bindAddress)
}