Eli Ribble
ffd424df12
This avoids the problem of having to assign the compliance report later when we get location data and image data.
87 lines
2.7 KiB
Go
87 lines
2.7 KiB
Go
package platform
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/Gleipnir-Technology/bob/dialect/psql"
|
|
"github.com/Gleipnir-Technology/bob/dialect/psql/sm"
|
|
"github.com/Gleipnir-Technology/nidus-sync/db"
|
|
"github.com/Gleipnir-Technology/nidus-sync/db/models"
|
|
"github.com/Gleipnir-Technology/nidus-sync/platform/types"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
func DistrictCatchall(ctx context.Context) (*models.Organization, error) {
|
|
return models.Organizations.Query(
|
|
models.SelectWhere.Organizations.IsCatchall.EQ(true),
|
|
).One(ctx, db.PGInstance.BobDB)
|
|
}
|
|
func DistrictForLocation(ctx context.Context, lng float64, lat float64) (*models.Organization, error) {
|
|
organizations, err := models.Organizations.Query(
|
|
sm.Where(
|
|
psql.F("ST_Contains", psql.Raw("service_area_geometry"), psql.F("ST_SetSRID", psql.F("ST_MakePoint", psql.Arg(lng), psql.Arg(lat)), psql.Arg(4326))),
|
|
),
|
|
).All(ctx, db.PGInstance.BobDB)
|
|
|
|
log.Debug().Int("len", len(organizations)).Float64("lng", lng).Float64("lat", lat).Msg("Attempting district match")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to query organization: %w", err)
|
|
}
|
|
switch len(organizations) {
|
|
case 0:
|
|
return nil, nil
|
|
case 1:
|
|
org := organizations[0]
|
|
return org, nil
|
|
default:
|
|
return nil, errors.New("too many organizations")
|
|
}
|
|
}
|
|
func matchDistrict(ctx context.Context, location *types.Location, images []ImageUpload) (int32, error) {
|
|
var err error
|
|
var org *models.Organization
|
|
for _, image := range images {
|
|
if image.Exif == nil {
|
|
continue
|
|
}
|
|
if image.Exif.GPS == nil {
|
|
continue
|
|
}
|
|
org, err = DistrictForLocation(ctx, image.Exif.GPS.Longitude, image.Exif.GPS.Latitude)
|
|
if err != nil {
|
|
log.Warn().Err(err).Msg("Failed to get district for location")
|
|
continue
|
|
}
|
|
if org != nil {
|
|
return org.ID, nil
|
|
}
|
|
}
|
|
if location != nil {
|
|
if location.Longitude == 0 || location.Latitude == 0 {
|
|
org, err = DistrictCatchall(ctx)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("get catchall: %w", err)
|
|
}
|
|
log.Debug().Int32("id", org.ID).Msg("No location from images, no latlng for the report itself, using catchall")
|
|
return org.ID, nil
|
|
}
|
|
org, err = DistrictForLocation(ctx, location.Longitude, location.Latitude)
|
|
if err != nil {
|
|
log.Warn().Err(err).Msg("Failed to get district for location")
|
|
return 0, fmt.Errorf("Failed to get district for location: %w", err)
|
|
}
|
|
}
|
|
if org == nil {
|
|
org, err = DistrictCatchall(ctx)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("get catchall: %w", err)
|
|
}
|
|
log.Debug().Err(err).Int32("id", org.ID).Msg("No district match by report location, using catchall")
|
|
return org.ID, nil
|
|
}
|
|
log.Debug().Err(err).Int32("org_id", org.ID).Msg("Found district match by report location")
|
|
return org.ID, nil
|
|
}
|