Redmage/server/routes/subreddit_create.go

package routes

import (
	"encoding/json"
	"errors"
	"fmt"
	"net/http"
	"strconv"

	"github.com/robfig/cron/v3"
	"github.com/tigorlazuardi/redmage/api"
	"github.com/tigorlazuardi/redmage/api/reddit"
	"github.com/tigorlazuardi/redmage/models"
	"github.com/tigorlazuardi/redmage/pkg/errs"
	"github.com/tigorlazuardi/redmage/pkg/log"
	"github.com/tigorlazuardi/redmage/views/components"
	"github.com/tigorlazuardi/redmage/views/subreddits/put"
)

func (routes *Routes) SubredditsCreateAPI(rw http.ResponseWriter, req *http.Request) {
	ctx, span := tracer.Start(req.Context(), "*Routes.SubredditsCreate")
	defer span.End()

	var (
		body *models.Subreddit
		enc  = json.NewEncoder(rw)
	)

	if err := json.NewDecoder(req.Body).Decode(&body); err != nil {
		rw.WriteHeader(http.StatusBadRequest)
		_ = enc.Encode(map[string]string{"error": fmt.Sprintf("failed to decode json body: %s", err)})
		return
	}

	if err := validateSubredditsCreate(body); err != nil {
		rw.WriteHeader(http.StatusBadRequest)
		_ = enc.Encode(map[string]string{"error": err.Error()})
		return
	}

	actual, err := routes.API.SubredditCheck(ctx, api.SubredditCheckParam{
		Subreddit:     body.Name,
		SubredditType: reddit.SubredditType(body.Subtype),
	})
	if err != nil {
		log.New(ctx).Err(err).Error("subreddit check returns error")
		code, message := errs.HTTPMessage(err)
		rw.WriteHeader(code)
		_ = enc.Encode(map[string]string{"error": message})
		return
	}

	body.Name = actual

	sub, err := routes.API.SubredditsCreate(ctx, body)
	if err != nil {
		log.New(ctx).Err(err).Error("failed to create subreddit")
		code, message := errs.HTTPMessage(err)
		rw.WriteHeader(code)
		_ = enc.Encode(map[string]string{"error": message})
		return
	}

	rw.WriteHeader(http.StatusCreated)
	if err := enc.Encode(sub); err != nil {
		log.New(ctx).Err(err).Error("failed to encode subreddit into json")
	}
}

func (routes *Routes) SubredditsCreateHTMX(rw http.ResponseWriter, r *http.Request) {
	ctx, span := tracer.Start(r.Context(), "*Routes.SubredditsCreateHTMX")
	defer span.End()

	sub, err := subredditsDataFromRequest(r)
	if err != nil {
		code, message := errs.HTTPMessage(err)
		rw.WriteHeader(code)
		if err := components.ErrorNotication(message).Render(ctx, rw); err != nil {
			log.New(ctx).Err(err).Error("failed to render error notification")
		}
		return
	}
	actual, err := routes.API.SubredditCheck(ctx, api.SubredditCheckParam{
		Subreddit:     sub.Name,
		SubredditType: reddit.SubredditType(sub.Subtype),
	})
	if err != nil {
		rw.WriteHeader(http.StatusBadRequest)
		log.New(ctx).Err(err).Error("subreddit check returns error")
		renderer := put.NameInput(put.NameInputData{
			Value: sub.Name,
			Error: err.Error(),
		})
		if err := renderer.Render(ctx, rw); err != nil {
			log.New(ctx).Err(err).Error("failed to render error")
		}
		return
	}
	sub.Name = actual

	_, err = routes.API.SubredditsCreate(ctx, sub)
	if err != nil {
		log.New(ctx).Err(err).Error("failed to create subreddit")
		code, message := errs.HTTPMessage(err)
		rw.Header().Set("HX-Retarget", components.NotificationContainerID)
		rw.WriteHeader(code)
		if err := components.ErrorNotication(message).Render(ctx, rw); err != nil {
			log.New(ctx).Err(err).Error("failed to render error")
		}
		return
	}
	if fetch, _ := strconv.ParseBool(r.FormValue("fetch")); fetch {
		_ = routes.API.PubsubStartDownloadSubreddit(ctx, api.PubsubStartDownloadSubredditParams{
			Subreddit: sub.Name,
		})
	}
	rw.Header().Set("HX-Redirect", "/subreddits")
	rw.WriteHeader(http.StatusCreated)
	_, _ = rw.Write([]byte("Subreddit created"))
}

func subredditsDataFromRequest(r *http.Request) (sub *models.Subreddit, err error) {
	sub = &models.Subreddit{}

	var t reddit.SubredditType
	err = t.Parse(r.FormValue("type"))
	if err != nil {
		return nil, errs.
			Wrapw(err, "invalid subreddit type", "type", r.FormValue("type")).
			Code(http.StatusBadRequest)
	}
	sub.Subtype = int32(t)

	sub.Name = r.FormValue("name")
	if sub.Name == "" {
		return nil, errs.Fail("name is required").Code(http.StatusBadRequest)
	}

	enableSchedule, _ := strconv.Atoi(r.FormValue("enable_schedule"))
	sub.EnableSchedule = int32(enableSchedule)
	if sub.EnableSchedule > 1 {
		sub.EnableSchedule = 1
	} else if sub.EnableSchedule < 0 {
		sub.EnableSchedule = 0
	}
	if sub.EnableSchedule == 0 {
		sub.Schedule = "@daily"
	}

	if sub.EnableSchedule == 1 {
		schedule := r.FormValue("schedule")
		_, err = cron.ParseStandard(schedule)
		if err != nil {
			return nil, errs.Wrapf(err, "invalid cron schedule: %s", err).Code(http.StatusBadRequest)
		}
		sub.Schedule = schedule
	}

	countback, _ := strconv.Atoi(r.FormValue("countback"))
	sub.Countback = int32(countback)
	if sub.Countback < 1 {
		return nil, errs.Fail("countback must be 1 or higher").Code(http.StatusBadRequest)
	}

	return sub, nil
}

var cronParser = cron.NewParser(cron.Minute | cron.Hour | cron.Dom | cron.Month | cron.Dow | cron.Descriptor)

func validateSubredditsCreate(body *models.Subreddit) error {
	if body.Name == "" {
		return errors.New("name is required")
	}
	if body.EnableSchedule > 1 {
		body.EnableSchedule = 1
	} else if body.EnableSchedule < 0 {
		body.EnableSchedule = 0
	}
	if body.Subtype > 1 {
		body.Subtype = 1
	} else if body.Subtype < 0 {
		body.Subtype = 0
	}
	if body.Schedule == "" {
		return errors.New("schedule is required")
	}
	_, err := cronParser.Parse(body.Schedule)
	if err != nil {
		return fmt.Errorf("bad cron schedule: %w", err)
	}
	if body.Countback < 1 {
		return errors.New("countback must be 1 or higher")
	}
	return nil
}