nidus-sync/rmo/pool.go

package rmo

import (
	"fmt"
	"net/http"
	"strconv"
	"time"

	"github.com/Gleipnir-Technology/bob"
	"github.com/Gleipnir-Technology/bob/dialect/psql"
	"github.com/Gleipnir-Technology/bob/dialect/psql/um"
	"github.com/Gleipnir-Technology/nidus-sync/config"
	"github.com/Gleipnir-Technology/nidus-sync/db"
	"github.com/Gleipnir-Technology/nidus-sync/db/enums"
	"github.com/Gleipnir-Technology/nidus-sync/db/models"
	"github.com/Gleipnir-Technology/nidus-sync/html"
	"github.com/aarondl/opt/omit"
	"github.com/rs/zerolog/log"
)

type ContextPool struct {
	MapboxToken string
}
type ContextPoolSubmitComplete struct {
	ReportID string
}

var (
	Pool               = buildTemplate("pool", "base")
	PoolSubmitComplete = buildTemplate("pool-submit-complete", "base")
)

func getPool(w http.ResponseWriter, r *http.Request) {
	html.RenderOrError(
		w,
		Pool,
		ContextPool{
			MapboxToken: config.MapboxToken,
		},
	)
}
func getPoolSubmitComplete(w http.ResponseWriter, r *http.Request) {
	report := r.URL.Query().Get("report")
	html.RenderOrError(
		w,
		PoolSubmitComplete,
		ContextPoolSubmitComplete{
			ReportID: report,
		},
	)
}
func postPool(w http.ResponseWriter, r *http.Request) {
	err := r.ParseMultipartForm(32 << 10) // 32 MB buffer
	if err != nil {
		respondError(w, "Failed to parse form", err, http.StatusBadRequest)
		return
	}
	access_comments := r.FormValue("access-comments")
	access_gate := boolFromForm(r, "access-gate")
	access_fence := boolFromForm(r, "access-fence")
	access_locked := boolFromForm(r, "access-locked")
	access_dog := boolFromForm(r, "access-dog")
	access_other := boolFromForm(r, "access-other")
	address := r.FormValue("address")
	address_country := r.FormValue("address-country")
	address_postcode := r.FormValue("address-postcode")
	address_place := r.FormValue("address-place")
	address_region := r.FormValue("address-region")
	address_street := r.FormValue("address-street")
	comments := r.FormValue("comments")
	has_adult := boolFromForm(r, "has-adult")
	has_larvae := boolFromForm(r, "has-larvae")
	has_pupae := boolFromForm(r, "has-pupae")
	map_zoom_str := r.FormValue("map-zoom")
	owner_email := r.FormValue("owner-email")
	owner_name := r.FormValue("owner-name")
	owner_phone := r.FormValue("owner-phone")
	reporter_email := r.FormValue("reporter-email")
	reporter_name := r.FormValue("reporter-name")
	reporter_phone := r.FormValue("reporter-phone")
	subscribe := boolFromForm(r, "subscribe")

	map_zoom, err := strconv.ParseFloat(map_zoom_str, 32)
	if err != nil {
		respondError(w, "Failed to parse zoom level", err, http.StatusBadRequest)
		return
	}
	public_id, err := GenerateReportID()
	if err != nil {
		respondError(w, "Failed to create pool report public ID", err, http.StatusInternalServerError)
		return
	}

	ctx := r.Context()
	tx, err := db.PGInstance.BobDB.BeginTx(ctx, nil)
	if err != nil {
		respondError(w, "Failed to create transaction", err, http.StatusInternalServerError)
		return
	}
	defer tx.Rollback(ctx)

	setter := models.PublicreportPoolSetter{
		AccessComments:  omit.From(access_comments),
		AccessGate:      omit.From(access_gate),
		AccessFence:     omit.From(access_fence),
		AccessLocked:    omit.From(access_locked),
		AccessDog:       omit.From(access_dog),
		AccessOther:     omit.From(access_other),
		Address:         omit.From(address),
		AddressCountry:  omit.From(address_country),
		AddressPostCode: omit.From(address_postcode),
		AddressPlace:    omit.From(address_place),
		AddressStreet:   omit.From(address_street),
		AddressRegion:   omit.From(address_region),
		Comments:        omit.From(comments),
		Created:         omit.From(time.Now()),
		//H3cell: add later
		HasAdult:  omit.From(has_adult),
		HasLarvae: omit.From(has_larvae),
		HasPupae:  omit.From(has_pupae),
		//Location: add later
		MapZoom:       omit.From(map_zoom),
		OwnerEmail:    omit.From(owner_email),
		OwnerName:     omit.From(owner_name),
		OwnerPhone:    omit.From(owner_phone),
		PublicID:      omit.From(public_id),
		ReporterEmail: omit.From(reporter_email),
		ReporterName:  omit.From(reporter_name),
		ReporterPhone: omit.From(reporter_phone),
		Status:        omit.From(enums.PublicreportReportstatustypeReported),
		Subscribe:     omit.From(subscribe),
	}
	pool, err := models.PublicreportPools.Insert(&setter).One(ctx, tx)
	if err != nil {
		respondError(w, "Failed to create database record", err, http.StatusInternalServerError)
		return
	}

	geospatial, err := geospatialFromForm(r)
	if err != nil {
		respondError(w, "Failed to handle geospatial data", err, http.StatusInternalServerError)
		return
	}
	if geospatial.Populated {
		_, err = psql.Update(
			um.Table("publicreport.pool"),
			um.SetCol("h3cell").ToArg(geospatial.Cell),
			um.SetCol("location").To(geospatial.GeometryQuery),
			um.Where(psql.Quote("id").EQ(psql.Arg(pool.ID))),
		).Exec(ctx, tx)
		if err != nil {
			respondError(w, "Failed to insert publicreport.pool", err, http.StatusInternalServerError)
			return
		}
	}
	log.Info().Int32("id", pool.ID).Str("public_id", pool.PublicID).Msg("Created pool report")
	uploads, err := extractImageUploads(r)
	if err != nil {
		respondError(w, "Failed to extract image uploads", err, http.StatusInternalServerError)
		return
	}
	images, err := saveImageUploads(r.Context(), tx, uploads)
	setters := make([]*models.PublicreportPoolImageSetter, 0)
	for _, image := range images {
		setters = append(setters, &models.PublicreportPoolImageSetter{
			ImageID: omit.From(int32(image.ID)),
			PoolID:  omit.From(int32(pool.ID)),
		})
	}
	_, err = models.PublicreportPoolImages.Insert(bob.ToMods(setters...)).Exec(r.Context(), tx)
	if err != nil {
		respondError(w, "Failed to save upload relationships", err, http.StatusInternalServerError)
		return
	}
	tx.Commit(ctx)
	http.Redirect(w, r, fmt.Sprintf("/pool-submit-complete?report=%s", public_id), http.StatusFound)
}