server.go

package radium

import (
	"encoding/json"
	"net/http"
	"reflect"

	"github.com/gorilla/mux"
)

// NewServer initializes the http API server with given
// instance of Radium
func NewServer(ins *Instance, defaultStrategy string) *Server {
	srv := &Server{}

	srv.ins = ins
	srv.Logger = ins.Logger
	srv.defStrategy = defaultStrategy
	srv.router = mux.NewRouter()
	srv.router.HandleFunc("/search", srv.handleSearch)
	srv.router.HandleFunc("/sources", srv.handleSources)

	return srv
}

// Server represents an instance of HTTP API server
type Server struct {
	Logger

	ins         *Instance
	router      *mux.Router
	defStrategy string
}

func (srv Server) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
	srv.router.ServeHTTP(wr, req)
}

func (srv Server) handleSearch(wr http.ResponseWriter, req *http.Request) {
	wr.Header().Set("Content-type", "application/json")

	query := Query{}
	strategy := req.FormValue("strategy")
	if strategy == "" {
		strategy = srv.defStrategy
	}

	query.Text = req.FormValue("q")
	query.Attribs = map[string]string{}

	for key, val := range req.URL.Query() {
		if key != "q" && len(val) > 0 {
			query.Attribs[key] = val[0]
		}
	}

	ctx := req.Context()
	rs, err := srv.ins.Search(ctx, query, strategy)
	if err != nil {
		srv.Warnf("error occurred during search: %s", err)
		wr.WriteHeader(http.StatusNotFound)
		json.NewEncoder(wr).Encode(map[string]interface{}{
			"error": err.Error(),
		})
		return
	}

	srv.Infof("found %d result(s) for '%s'", len(rs), query.Text)
	json.NewEncoder(wr).Encode(rs)
}

func (srv Server) handleSources(wr http.ResponseWriter, req *http.Request) {
	sources := map[string]string{}
	for _, src := range srv.ins.GetSources() {
		ty := reflect.TypeOf(src.Source)
		sources[src.Name] = ty.String()
	}
	wr.Header().Set("Content-type", "application/json")
	json.NewEncoder(wr).Encode(sources)
}