prologic-saltyim/internal/api.go

package internal

import (
	"net/http"

	"github.com/julienschmidt/httprouter"
	log "github.com/sirupsen/logrus"
	"github.com/unrolled/render"

	"go.mills.io/saltyim"
)

// API ...
type API struct {
	router *Router
	config *Config
	db     Store

	r *render.Render
}

// NewAPI ...
func NewAPI(router *Router, config *Config, db Store) *API {
	api := &API{router, config, db, render.New()}

	api.initRoutes()

	return api
}

func (a *API) initRoutes() {
	router := a.router.Group("/api/v1")

	router.GET("/ping", a.PingEndpoint())
	router.POST("/register", a.RegisterEndpoint())
}

// PingEndpoint ...
func (a *API) PingEndpoint() httprouter.Handle {
	return func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
		w.Header().Set("Content-Type", "application/json")
		_, _ = w.Write([]byte(`{}`))
	}
}

// RegisterEndpoint ...
func (a *API) RegisterEndpoint() httprouter.Handle {
	return func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
		req, signer, err := saltyim.NewRegisterRequest(r.Body)
		if err != nil {
			log.WithError(err).Error("error parsing register request")
			http.Error(w, "Bad Request", http.StatusBadRequest)
			return
		}
		if signer != req.Key {
			http.Error(w, "Bad Request", http.StatusBadRequest)
			return
		}

		if err := CreateConfig(a.config, req.Addr.Hash(), req.Key); err != nil {
			log.WithError(err).Errorf("error creating config for %s", req.Addr.String())
			http.Error(w, "Error", http.StatusInternalServerError)
			return
		}

		http.Error(w, "Account Created", http.StatusCreated)
	}
}