setup.go

package main

import (
	"net/http"
	"time"

	"github.com/gorilla/handlers"
	"github.com/gorilla/mux"
)

//Setup initialize the http server
func Setup(bindAddress string, allowedOrigins []string) *http.Server {

	router := mux.NewRouter()

	origins := handlers.AllowedOrigins(allowedOrigins)
	meths := handlers.AllowedMethods([]string{http.MethodGet, http.MethodOptions})
	heads := handlers.AllowedHeaders([]string{"x-user-platform", "x-user-id", "Content-Type", "content-type", "Origin"})

	// setting up routes
	router.HandleFunc("/random", random).Methods(http.MethodGet, http.MethodOptions)
	router.HandleFunc("/random/storage/{id}", servePost).Methods(http.MethodGet)
	router.HandleFunc("/health", healthCheck).Methods(http.MethodGet)

	srv := &http.Server{
		Handler: handlers.CORS(meths, origins, heads)(router),
		Addr:    bindAddress,

		WriteTimeout: time.Second * 15,
		ReadTimeout:  time.Second * 15,
		IdleTimeout:  time.Second * 60,
	}

	return srv
}