apollo-backend/internal/api/watcher.go

package api

import (
	"context"
	"encoding/json"
	"net/http"
	"strconv"
	"strings"
	"time"

	validation "github.com/go-ozzo/ozzo-validation/v4"
	"github.com/gorilla/mux"

	"github.com/christianselig/apollo-backend/internal/domain"
)

type watcherCriteria struct {
	Author    string
	Subreddit string
	Upvotes   int64
	Keyword   string
	Flair     string
	Domain    string
}

type createWatcherRequest struct {
	Type      string
	User      string
	Subreddit string
	Label     string
	Criteria  watcherCriteria
}

func (cwr *createWatcherRequest) Validate() error {
	return validation.ValidateStruct(cwr,
		validation.Field(&cwr.Type, validation.Required),
		validation.Field(&cwr.User, validation.Required.When(cwr.Type == "user")),
		validation.Field(&cwr.Subreddit, validation.Required.When(cwr.Type == "subreddit" || cwr.Type == "trending")),
	)
}

type watcherCreatedResponse struct {
	ID int64 `json:"id"`
}

func (a *api) createWatcherHandler(w http.ResponseWriter, r *http.Request) {
	ctx := context.Background()

	vars := mux.Vars(r)
	apns := vars["apns"]
	redditID := vars["redditID"]

	cwr := &createWatcherRequest{
		Criteria: watcherCriteria{},
	}
	if err := json.NewDecoder(r.Body).Decode(cwr); err != nil {
		a.errorResponse(w, r, 500, err.Error())
		return
	}

	if err := cwr.Validate(); err != nil {
		a.errorResponse(w, r, 422, err.Error())
		return
	}

	dev, err := a.deviceRepo.GetByAPNSToken(ctx, apns)
	if err != nil {
		a.errorResponse(w, r, 422, err.Error())
		return
	}

	accs, err := a.accountRepo.GetByAPNSToken(ctx, apns)
	if err != nil || len(accs) == 0 {
		a.errorResponse(w, r, 422, err.Error())
		return
	}

	account := accs[0]
	found := false
	for _, acc := range accs {
		if acc.AccountID == redditID {
			found = true
			account = acc
		}
	}

	if !found {
		a.errorResponse(w, r, 422, "yeah nice try")
		return
	}

	ac := a.reddit.NewAuthenticatedClient(account.AccountID, account.RefreshToken, account.AccessToken)

	watcher := domain.Watcher{
		Label:     cwr.Label,
		DeviceID:  dev.ID,
		AccountID: account.ID,
		Author:    strings.ToLower(cwr.Criteria.Author),
		Subreddit: strings.ToLower(cwr.Criteria.Subreddit),
		Upvotes:   cwr.Criteria.Upvotes,
		Keyword:   strings.ToLower(cwr.Criteria.Keyword),
		Flair:     strings.ToLower(cwr.Criteria.Flair),
		Domain:    strings.ToLower(cwr.Criteria.Domain),
	}

	if cwr.Type == "subreddit" || cwr.Type == "trending" {
		srr, err := ac.SubredditAbout(ctx, cwr.Subreddit)
		if err != nil {
			a.errorResponse(w, r, 422, err.Error())
			return
		}

		sr, err := a.subredditRepo.GetByName(ctx, cwr.Subreddit)
		if err != nil {
			switch err {
			case domain.ErrNotFound:
				// Might be that we don't know about that subreddit yet
				sr = domain.Subreddit{SubredditID: srr.ID, Name: srr.Name}
				_ = a.subredditRepo.CreateOrUpdate(ctx, &sr)
			default:
				a.errorResponse(w, r, 500, err.Error())
				return
			}
		}

		switch cwr.Type {
		case "subreddit":
			watcher.Type = domain.SubredditWatcher
		case "trending":
			watcher.Label = "trending"
			watcher.Type = domain.TrendingWatcher
		}

		watcher.WatcheeID = sr.ID
	} else if cwr.Type == "user" {
		urr, err := ac.UserAbout(ctx, cwr.User)
		if err != nil {
			a.errorResponse(w, r, 500, err.Error())
			return
		}

		if !urr.AcceptFollowers {
			a.errorResponse(w, r, 422, "no followers accepted")
			return
		}

		u := domain.User{UserID: urr.ID, Name: urr.Name}
		err = a.userRepo.CreateOrUpdate(ctx, &u)

		if err != nil {
			a.errorResponse(w, r, 500, err.Error())
			return
		}

		watcher.Type = domain.UserWatcher
		watcher.WatcheeID = u.ID
	} else {
		a.errorResponse(w, r, 422, "unknown watcher type")
		return
	}

	if err := a.watcherRepo.Create(ctx, &watcher); err != nil {
		a.errorResponse(w, r, 422, err.Error())
		return
	}

	w.WriteHeader(http.StatusOK)
	w.Header().Set("Content-Type", "application/json")
	_ = json.NewEncoder(w).Encode(watcherCreatedResponse{ID: watcher.ID})
}

func (a *api) editWatcherHandler(w http.ResponseWriter, r *http.Request) {
	ctx := context.Background()

	vars := mux.Vars(r)
	id, err := strconv.ParseInt(vars["watcherID"], 10, 64)
	if err != nil {
		a.errorResponse(w, r, 422, err.Error())
		return
	}

	watcher, err := a.watcherRepo.GetByID(ctx, id)
	if err != nil || watcher.Device.APNSToken != vars["apns"] {
		a.errorResponse(w, r, 422, "nice try")
		return
	}

	ewr := &createWatcherRequest{
		Criteria: watcherCriteria{},
	}

	if err := json.NewDecoder(r.Body).Decode(ewr); err != nil {
		a.errorResponse(w, r, 500, err.Error())
		return
	}

	watcher.Label = ewr.Label
	watcher.Author = strings.ToLower(ewr.Criteria.Author)
	watcher.Subreddit = strings.ToLower(ewr.Criteria.Subreddit)
	watcher.Upvotes = ewr.Criteria.Upvotes
	watcher.Keyword = strings.ToLower(ewr.Criteria.Keyword)
	watcher.Flair = strings.ToLower(ewr.Criteria.Flair)
	watcher.Domain = strings.ToLower(ewr.Criteria.Domain)

	if err := a.watcherRepo.Update(ctx, &watcher); err != nil {
		a.errorResponse(w, r, 500, err.Error())
		return
	}

	w.WriteHeader(http.StatusOK)
}

func (a *api) deleteWatcherHandler(w http.ResponseWriter, r *http.Request) {
	ctx := context.Background()

	vars := mux.Vars(r)
	id, err := strconv.ParseInt(vars["watcherID"], 10, 64)
	if err != nil {
		a.errorResponse(w, r, 422, err.Error())
		return
	}

	watcher, err := a.watcherRepo.GetByID(ctx, id)
	if err != nil || watcher.Device.APNSToken != vars["apns"] {
		a.errorResponse(w, r, 422, "nice try")
		return
	}

	_ = a.watcherRepo.Delete(ctx, id)
	w.WriteHeader(http.StatusOK)
}

type watcherItem struct {
	ID          int64     `json:"id"`
	CreatedAt   time.Time `json:"created_at"`
	Type        string    `json:"type"`
	Label       string    `json:"label"`
	SourceLabel string    `json:"source_label"`
	Upvotes     *int64    `json:"upvotes,omitempty"`
	Keyword     string    `json:"keyword,omitempty"`
	Flair       string    `json:"flair,omitempty"`
	Domain      string    `json:"domain,omitempty"`
	Hits        int64     `json:"hits"`
	Author      string    `json:"author,omitempty"`
}

func (a *api) listWatchersHandler(w http.ResponseWriter, r *http.Request) {
	ctx := context.Background()

	vars := mux.Vars(r)
	apns := vars["apns"]
	redditID := vars["redditID"]

	watchers, err := a.watcherRepo.GetByDeviceAPNSTokenAndAccountRedditID(ctx, apns, redditID)
	if err != nil {
		a.errorResponse(w, r, 400, err.Error())
		return
	}

	wis := make([]watcherItem, len(watchers))
	for i, watcher := range watchers {
		wi := watcherItem{
			ID:          watcher.ID,
			CreatedAt:   watcher.CreatedAt,
			Type:        watcher.Type.String(),
			Label:       watcher.Label,
			SourceLabel: watcher.WatcheeLabel,
			Keyword:     watcher.Keyword,
			Flair:       watcher.Flair,
			Domain:      watcher.Domain,
			Hits:        watcher.Hits,
			Author:      watcher.Author,
		}

		if watcher.Upvotes != 0 {
			wi.Upvotes = &watcher.Upvotes
		}

		wis[i] = wi
	}
	w.WriteHeader(http.StatusOK)
	w.Header().Set("Content-Type", "application/json")
	_ = json.NewEncoder(w).Encode(wis)
}