default.go

package _default

import (
	"context"
	"encoding/json"
	"github.com/mitchellh/mapstructure"
	"github.com/muety/telepush/config"
	"github.com/muety/telepush/model"
	"github.com/muety/telepush/util"
	"net/http"
	"strings"

	"github.com/muety/telepush/inlets"
)

type DefaultInlet struct{}

func (i *DefaultInlet) SupportedMethods() []string {
	return []string{http.MethodPost, http.MethodGet}
}

func (i *DefaultInlet) Name() string {
	return "default"
}

func (i *DefaultInlet) Handler(h http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		var m model.MessageWithOptions
		var err error

		m, err = i.tryParseBody(r)
		if err != nil {
			m, err = i.tryParseQuery(r)
		}
		if err != nil {
			w.WriteHeader(http.StatusBadRequest)
			w.Write([]byte(err.Error()))
			return
		}

		// make query params take precedence
		q := r.URL.Query()
		if strings.ToLower(q.Get("disable_link_previews")) == "true" {
			m.Options.DisableLinkPreviews = true
		}
		if strings.ToLower(q.Get("disable_markdown")) == "true" {
			m.Options.DisableMarkdown = true
		}

		if len(m.Origin) == 0 {
			m.Origin = model.DefaultOrigin
		}

		m.Text = "*" + util.EscapeMarkdown(m.Origin) + "* wrote:\n\n" + m.Text

		ctx := r.Context()
		ctx = context.WithValue(ctx, config.KeyMessage, &m.Message)
		ctx = context.WithValue(ctx, config.KeyParams, &m.Options)

		h.ServeHTTP(w, r.WithContext(ctx))
	})
}

func (i *DefaultInlet) tryParseBody(r *http.Request) (m model.MessageWithOptions, err error) {
	dec := json.NewDecoder(r.Body)
	err = dec.Decode(&m)
	return m, err
}

func (i *DefaultInlet) tryParseQuery(r *http.Request) (m model.MessageWithOptions, err error) {
	query := r.URL.Query()
	queryParams := make(map[string]string)
	for k := range query {
		queryParams[k] = query.Get(k)
	}
	decoder, _ := mapstructure.NewDecoder(&mapstructure.DecoderConfig{
		WeaklyTypedInput: true,
		Result:           &m,
	})
	err = decoder.Decode(queryParams)
	return m, err
}

func New() inlets.Inlet {
	return &DefaultInlet{}
}