nidus-sync/rmo/quick.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/h3utils"
	"github.com/Gleipnir-Technology/nidus-sync/html"
	"github.com/Gleipnir-Technology/nidus-sync/platform/report"
	"github.com/aarondl/opt/omit"
	"github.com/aarondl/opt/omitnull"
	"github.com/rs/zerolog/log"
)

type ContentQuick struct{}
type ContentQuickSubmitComplete struct {
	District *District
	ReportID string
}
type ContentRegisterNotificationsComplete struct {
	ReportID string
}
type District struct {
	LogoURL string
	Name    string
}

func getQuick(w http.ResponseWriter, r *http.Request) {
	html.RenderOrError(
		w,
		"rmo/quick.html",
		ContentQuick{},
	)
}
func getQuickSubmitComplete(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	report_id := r.URL.Query().Get("report")
	report, err := models.PublicreportQuicks.Query(
		models.SelectWhere.PublicreportQuicks.PublicID.EQ(report_id),
	).One(ctx, db.PGInstance.BobDB)
	if err != nil {
		respondError(w, "Failed to get report", err, http.StatusInternalServerError)
		return
	}
	var district *District
	if !report.OrganizationID.IsNull() {
		org_id := report.OrganizationID.MustGet()
		org, err := models.Organizations.Query(
			models.Preload.Organization.ImportDistrictGidDistrict(),
			models.SelectWhere.Organizations.ID.EQ(org_id),
		).One(ctx, db.PGInstance.BobDB)
		if err != nil {
			respondError(w, "Failed to get org", err, http.StatusInternalServerError)
			return
		}
		d := org.R.ImportDistrictGidDistrict
		log.Debug().Int32("org_id", org.ID).Int32("d_gid", d.Gid).Msg("Getting district")
		if d != nil {
			district = &District{
				LogoURL: config.MakeURLNidus("/api/district/%s/logo", org.Slug.GetOr("placeholder")),
				Name:    d.Agency.GetOr("Unknown"),
			}
		}
	}
	html.RenderOrError(
		w,
		"rmo/quick-submit-complete.html",
		ContentQuickSubmitComplete{
			District: district,
			ReportID: report.PublicID,
		},
	)
}
func getRegisterNotificationsComplete(w http.ResponseWriter, r *http.Request) {
	report := r.URL.Query().Get("report")
	html.RenderOrError(
		w,
		"rmo/register-notifications-complete.html",
		ContentRegisterNotificationsComplete{
			ReportID: report,
		},
	)
}
func postQuick(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
	}
	lat := r.FormValue("latitude")
	lng := r.FormValue("longitude")
	comments := r.FormValue("comments")

	var latitude float64
	if lat != "" {
		latitude, err = strconv.ParseFloat(lat, 64)
		if err != nil {
			respondError(w, "Failed to parse latitude", err, http.StatusBadRequest)
			return
		}
	}
	var longitude float64
	if lng != "" {
		longitude, err = strconv.ParseFloat(lng, 64)
		if err != nil {
			respondError(w, "Failed to parse longitude", err, http.StatusBadRequest)
			return
		}
	}
	u, err := report.GenerateReportID()
	if err != nil {
		respondError(w, "Failed to create quick 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)

	uploads, err := extractImageUploads(r)
	log.Info().Int("len", len(uploads)).Msg("extracted uploads")
	if err != nil {
		respondError(w, "Failed to extract image uploads", err, http.StatusInternalServerError)
		return
	}
	images, err := saveImageUploads(ctx, tx, uploads)
	if err != nil {
		respondError(w, "Failed to save image uploads", err, http.StatusInternalServerError)
		return
	}

	organization_id, err := matchDistrict(ctx, &longitude, &latitude, uploads)
	if err != nil {
		log.Warn().Err(err).Msg("Failed to match district")
	}

	log.Info().Int("len", len(images)).Msg("saved uploads")
	c, err := h3utils.GetCell(longitude, latitude, 15)
	if err != nil {
		respondError(w, "Failedt o get h3 cell", err, http.StatusInternalServerError)
	}
	setter := models.PublicreportQuickSetter{
		Address:        omit.From(""),
		Created:        omit.From(time.Now()),
		Comments:       omit.From(comments),
		OrganizationID: omitnull.FromPtr(organization_id),
		//Location: omitnull.From(fmt.Sprintf("ST_GeometryFromText(Point(%s %s))", longitude, latitude)),
		H3cell:        omitnull.From(c.String()),
		PublicID:      omit.From(u),
		ReporterEmail: omit.From(""),
		ReporterPhone: omit.From(""),
		Status:        omit.From(enums.PublicreportReportstatustypeReported),
	}
	quick, err := models.PublicreportQuicks.Insert(&setter).One(ctx, tx)
	if err != nil {
		respondError(w, "Failed to create database record", err, http.StatusInternalServerError)
		return
	}
	_, err = psql.Update(
		um.Table("publicreport.quick"),
		um.SetCol("location").To(fmt.Sprintf("ST_GeometryFromText('Point(%f %f)')", longitude, latitude)),
		um.Where(psql.Quote("id").EQ(psql.Arg(quick.ID))),
	).Exec(ctx, tx)
	if err != nil {
		respondError(w, "Failed to insert publicreport", err, http.StatusInternalServerError)
		return
	}
	log.Info().Float64("latitude", latitude).Float64("longitude", longitude).Msg("Got upload")
	if len(images) > 0 {
		setters := make([]*models.PublicreportQuickImageSetter, 0)
		for _, image := range images {
			setters = append(setters, &models.PublicreportQuickImageSetter{
				ImageID: omit.From(int32(image.ID)),
				QuickID: omit.From(int32(quick.ID)),
			})
		}
		_, err = models.PublicreportQuickImages.Insert(bob.ToMods(setters...)).Exec(ctx, tx)
		if err != nil {
			respondError(w, "Failed to save reference to images", err, http.StatusInternalServerError)
			return
		}
		log.Info().Int("len", len(images)).Msg("saved uploads")
	}
	tx.Commit(ctx)
	http.Redirect(w, r, fmt.Sprintf("/quick-submit-complete?report=%s", u), http.StatusFound)
}