minio/pkg/server/router.go

/*
 * Minimalist Object Storage, (C) 2014 Minio, Inc.
 *
 * 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 server

import (
	"net/http"

	router "github.com/gorilla/mux"
	"github.com/minio/minio/pkg/server/api"
	"github.com/minio/minio/pkg/server/rpc"
)

// registerAPI - register all the object API handlers to their respective paths
func registerAPI(mux *router.Router, a api.Minio) http.Handler {
	mux.HandleFunc("/", a.ListBucketsHandler).Methods("GET")
	mux.HandleFunc("/{bucket}", a.ListObjectsHandler).Methods("GET")
	mux.HandleFunc("/{bucket}", a.PutBucketHandler).Methods("PUT")
	mux.HandleFunc("/{bucket}", a.HeadBucketHandler).Methods("HEAD")
	mux.HandleFunc("/{bucket}/{object:.*}", a.HeadObjectHandler).Methods("HEAD")
	mux.HandleFunc("/{bucket}/{object:.*}", a.PutObjectPartHandler).Queries("partNumber", "{partNumber:[0-9]+}", "uploadId", "{uploadId:.*}").Methods("PUT")
	mux.HandleFunc("/{bucket}/{object:.*}", a.ListObjectPartsHandler).Queries("uploadId", "{uploadId:.*}").Methods("GET")
	mux.HandleFunc("/{bucket}/{object:.*}", a.CompleteMultipartUploadHandler).Queries("uploadId", "{uploadId:.*}").Methods("POST")
	mux.HandleFunc("/{bucket}/{object:.*}", a.NewMultipartUploadHandler).Methods("POST")
	mux.HandleFunc("/{bucket}/{object:.*}", a.AbortMultipartUploadHandler).Queries("uploadId", "{uploadId:.*}").Methods("DELETE")
	mux.HandleFunc("/{bucket}/{object:.*}", a.GetObjectHandler).Methods("GET")
	mux.HandleFunc("/{bucket}/{object:.*}", a.PutObjectHandler).Methods("PUT")

	// not implemented yet
	mux.HandleFunc("/{bucket}", a.DeleteBucketHandler).Methods("DELETE")

	// unsupported API
	mux.HandleFunc("/{bucket}/{object:.*}", a.DeleteObjectHandler).Methods("DELETE")

	return mux
}

// add a handlerFunc typedef
type handlerFunc func(http.Handler) http.Handler

// chain struct to hold handlers
type chain struct {
	handlers []handlerFunc
}

// loop through handlers and return a final one
func (c chain) final(mux http.Handler) http.Handler {
	var f http.Handler
	if mux != nil {
		f = mux
	} else {
		f = http.DefaultServeMux
	}
	for _, handler := range c.handlers {
		f = handler(f)
	}
	return f
}

// registerChain - register an array of handlers in a chain of style -> handler(handler(handler(handler...)))
func registerChain(handlers ...handlerFunc) chain {
	ch := chain{}
	ch.handlers = append(ch.handlers, handlers...)
	return ch
}

// registerCustomMiddleware register all available custom middleware
func registerCustomMiddleware(mux http.Handler, conf api.Config) http.Handler {
	ch := registerChain(
		api.ValidContentTypeHandler,
		api.TimeValidityHandler,
		api.IgnoreResourcesHandler,
		api.ValidateAuthHeaderHandler,
		api.LoggingHandler,
		// Add new your new middleware here
	)

	mux = ch.final(mux)
	mux = api.RateLimitHandler(mux, conf.RateLimit)
	return mux
}

// registerRPC - register rpc handlers
func registerRPC(mux *router.Router, s *rpc.Server) http.Handler {
	mux.Handle("/rpc", s)
	return mux
}

// getAPIHandler api handler
func getAPIHandler(conf api.Config) (http.Handler, api.Minio) {
	mux := router.NewRouter()
	minioAPI := api.New()
	apiHandler := registerAPI(mux, minioAPI)
	apiHandler = registerCustomMiddleware(apiHandler, conf)
	return apiHandler, minioAPI
}

// getRPCHandler rpc handler
func getRPCHandler() http.Handler {
	s := rpc.NewServer()
	s.RegisterJSONCodec()
	s.RegisterService(new(rpc.VersionService), "Version")
	s.RegisterService(new(rpc.SysInfoService), "SysInfo")
	s.RegisterService(new(rpc.MemStatsService), "MemStats")
	s.RegisterService(new(rpc.DiskInfoService), "DiskInfo")
	s.RegisterService(new(rpc.DonutService), "Donut")
	// Add new RPC services here
	return registerRPC(router.NewRouter(), s)
}