feat(media meta): reverse geocoding from mapbox (#2922)
Aaron Liu
1 day ago
parent
668b542c59
commit
5e5dca40c4
@ -1 +1 @@
|
||||
Subproject commit dece1c7098de2efe38aaa25d6cafc41a2de568ff
|
||||
Subproject commit fc7791cde1444e1be0935f1fbc32d956fa6eb756
|
||||
@ -0,0 +1,236 @@
|
||||
package mediameta
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/manager/entitysource"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/logging"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/request"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/setting"
|
||||
)
|
||||
|
||||
const mapBoxURL = "https://api.mapbox.com/search/geocode/v6/reverse"
|
||||
|
||||
const (
|
||||
Street = "street"
|
||||
Locality = "locality"
|
||||
Place = "place"
|
||||
District = "district"
|
||||
Region = "region"
|
||||
Country = "country"
|
||||
)
|
||||
|
||||
type geocodingExtractor struct {
|
||||
settings setting.Provider
|
||||
l logging.Logger
|
||||
client request.Client
|
||||
}
|
||||
|
||||
func newGeocodingExtractor(settings setting.Provider, l logging.Logger, client request.Client) *geocodingExtractor {
|
||||
return &geocodingExtractor{
|
||||
settings: settings,
|
||||
l: l,
|
||||
client: client,
|
||||
}
|
||||
}
|
||||
|
||||
func (e *geocodingExtractor) Exts() []string {
|
||||
return exifExts
|
||||
}
|
||||
|
||||
func (e *geocodingExtractor) Extract(ctx context.Context, ext string, source entitysource.EntitySource, opts ...optionFunc) ([]driver.MediaMeta, error) {
|
||||
option := &option{}
|
||||
for _, opt := range opts {
|
||||
opt.apply(option)
|
||||
}
|
||||
|
||||
// Find GPS info from extracted
|
||||
var latStr, lngStr string
|
||||
for _, meta := range option.extracted {
|
||||
if meta.Key == GpsLat {
|
||||
latStr = meta.Value
|
||||
}
|
||||
if meta.Key == GpsLng {
|
||||
lngStr = meta.Value
|
||||
}
|
||||
}
|
||||
|
||||
if latStr == "" || lngStr == "" {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
lat, err := strconv.ParseFloat(latStr, 64)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("geocoding: failed to parse latitude: %w", err)
|
||||
}
|
||||
|
||||
lng, err := strconv.ParseFloat(lngStr, 64)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("geocoding: failed to parse longitude: %w", err)
|
||||
}
|
||||
|
||||
metas, err := e.getGeocoding(ctx, lat, lng, option.language)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("geocoding: failed to get geocoding: %w", err)
|
||||
}
|
||||
|
||||
for i, _ := range metas {
|
||||
metas[i].Type = driver.MetaTypeGeocoding
|
||||
}
|
||||
|
||||
return metas, nil
|
||||
}
|
||||
|
||||
func (e *geocodingExtractor) getGeocoding(ctx context.Context, lat, lng float64, language string) ([]driver.MediaMeta, error) {
|
||||
values := url.Values{}
|
||||
values.Add("longitude", fmt.Sprintf("%f", lng))
|
||||
values.Add("latitude", fmt.Sprintf("%f", lat))
|
||||
values.Add("limit", "1")
|
||||
values.Add("access_token", e.settings.MediaMetaGeocodingMapboxAK(ctx))
|
||||
if language != "" {
|
||||
values.Add("language", language)
|
||||
}
|
||||
|
||||
resp, err := e.client.Request(
|
||||
"GET",
|
||||
mapBoxURL+"?"+values.Encode(),
|
||||
nil,
|
||||
request.WithContext(ctx),
|
||||
request.WithLogger(e.l),
|
||||
).CheckHTTPResponse(http.StatusOK).GetResponse()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to get geocoding from mapbox: %w", err)
|
||||
}
|
||||
|
||||
var geocoding MapboxGeocodingResponse
|
||||
if err := json.Unmarshal([]byte(resp), &geocoding); err != nil {
|
||||
return nil, fmt.Errorf("failed to unmarshal geocoding from mapbox: %w", err)
|
||||
}
|
||||
|
||||
if len(geocoding.Features) == 0 {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
metas := make([]driver.MediaMeta, 0)
|
||||
contexts := geocoding.Features[0].Properties.Context
|
||||
if contexts.Street != nil {
|
||||
metas = append(metas, driver.MediaMeta{
|
||||
Key: Street,
|
||||
Value: contexts.Street.Name,
|
||||
})
|
||||
}
|
||||
if contexts.Locality != nil {
|
||||
metas = append(metas, driver.MediaMeta{
|
||||
Key: Locality,
|
||||
Value: contexts.Locality.Name,
|
||||
})
|
||||
}
|
||||
if contexts.Place != nil {
|
||||
metas = append(metas, driver.MediaMeta{
|
||||
Key: Place,
|
||||
Value: contexts.Place.Name,
|
||||
})
|
||||
}
|
||||
if contexts.District != nil {
|
||||
metas = append(metas, driver.MediaMeta{
|
||||
Key: District,
|
||||
Value: contexts.District.Name,
|
||||
})
|
||||
}
|
||||
if contexts.Region != nil {
|
||||
metas = append(metas, driver.MediaMeta{
|
||||
Key: Region,
|
||||
Value: contexts.Region.Name,
|
||||
})
|
||||
}
|
||||
if contexts.Country != nil {
|
||||
metas = append(metas, driver.MediaMeta{
|
||||
Key: Country,
|
||||
Value: contexts.Country.Name,
|
||||
})
|
||||
}
|
||||
|
||||
return metas, nil
|
||||
}
|
||||
|
||||
// MapboxGeocodingResponse represents the response from Mapbox Geocoding API
|
||||
type MapboxGeocodingResponse struct {
|
||||
Type string `json:"type"` // "FeatureCollection"
|
||||
Features []Feature `json:"features"` // Array of feature objects
|
||||
Attribution string `json:"attribution"` // Attribution to Mapbox
|
||||
}
|
||||
|
||||
// Feature represents a feature object in the geocoding response
|
||||
type Feature struct {
|
||||
ID string `json:"id"` // Feature ID (same as mapbox_id)
|
||||
Type string `json:"type"` // "Feature"
|
||||
Geometry Geometry `json:"geometry"` // Spatial geometry of the feature
|
||||
Properties Properties `json:"properties"` // Feature details
|
||||
}
|
||||
|
||||
// Geometry represents the spatial geometry of a feature
|
||||
type Geometry struct {
|
||||
Type string `json:"type"` // "Point"
|
||||
Coordinates []float64 `json:"coordinates"` // [longitude, latitude]
|
||||
}
|
||||
|
||||
// Properties contains the feature's detailed information
|
||||
type Properties struct {
|
||||
MapboxID string `json:"mapbox_id"` // Unique feature identifier
|
||||
FeatureType string `json:"feature_type"` // Type of feature (country, region, etc.)
|
||||
Name string `json:"name"` // Formatted address string
|
||||
NamePreferred string `json:"name_preferred"` // Canonical or common alias
|
||||
PlaceFormatted string `json:"place_formatted"` // Formatted context string
|
||||
FullAddress string `json:"full_address"` // Full formatted address
|
||||
Context Context `json:"context"` // Hierarchy of parent features
|
||||
Coordinates Coordinates `json:"coordinates"` // Geographic position and accuracy
|
||||
BBox []float64 `json:"bbox,omitempty"` // Bounding box [minLon,minLat,maxLon,maxLat]
|
||||
MatchCode MatchCode `json:"match_code"` // Metadata about result matching
|
||||
}
|
||||
|
||||
// Context represents the hierarchy of encompassing parent features
|
||||
type Context struct {
|
||||
Country *ContextFeature `json:"country,omitempty"`
|
||||
Region *ContextFeature `json:"region,omitempty"`
|
||||
Postcode *ContextFeature `json:"postcode,omitempty"`
|
||||
District *ContextFeature `json:"district,omitempty"`
|
||||
Place *ContextFeature `json:"place,omitempty"`
|
||||
Locality *ContextFeature `json:"locality,omitempty"`
|
||||
Neighborhood *ContextFeature `json:"neighborhood,omitempty"`
|
||||
Street *ContextFeature `json:"street,omitempty"`
|
||||
}
|
||||
|
||||
// ContextFeature represents a feature in the context hierarchy
|
||||
type ContextFeature struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
NamePreferred string `json:"name_preferred,omitempty"`
|
||||
MapboxID string `json:"mapbox_id"`
|
||||
}
|
||||
|
||||
// Coordinates represents geographical position and accuracy information
|
||||
type Coordinates struct {
|
||||
Longitude float64 `json:"longitude"` // Longitude of result
|
||||
Latitude float64 `json:"latitude"` // Latitude of result
|
||||
Accuracy string `json:"accuracy,omitempty"` // Accuracy metric for address results
|
||||
RoutablePoints []RoutablePoint `json:"routable_points,omitempty"` // Array of routable points
|
||||
}
|
||||
|
||||
// RoutablePoint represents a routable point for an address feature
|
||||
type RoutablePoint struct {
|
||||
Name string `json:"name"` // Name of the routable point
|
||||
Longitude float64 `json:"longitude"` // Longitude coordinate
|
||||
Latitude float64 `json:"latitude"` // Latitude coordinate
|
||||
}
|
||||
|
||||
// MatchCode contains metadata about how result components match the input query
|
||||
type MatchCode struct {
|
||||
// Add specific match code fields as needed based on Mapbox documentation
|
||||
// This structure may vary depending on the specific match codes returned
|
||||
}
|
||||
Loading…
Reference in new issue