Курнат Андрей
100 lines
2.8 KiB
Go
100 lines
2.8 KiB
Go
package convert
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
opensearchgoAPI "github.com/opensearch-project/opensearch-go/v4/opensearchapi"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
|
|
"github.com/opencloud-eu/reva/v2/pkg/storagespace"
|
|
|
|
"github.com/qsfera/server/pkg/conversions"
|
|
searchMessage "github.com/qsfera/server/protogen/gen/qsfera/messages/search/v0"
|
|
"github.com/qsfera/server/services/search/pkg/search"
|
|
)
|
|
|
|
func OpenSearchHitToMatch(hit opensearchgoAPI.SearchHit) (*searchMessage.Match, error) {
|
|
resource, err := conversions.To[search.Resource](hit.Source)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to convert hit source: %w", err)
|
|
}
|
|
|
|
resourceRootID, err := storagespace.ParseID(resource.RootID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resourceID, err := storagespace.ParseID(resource.ID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resourceParentID, _ := storagespace.ParseID(resource.ParentID)
|
|
|
|
match := &searchMessage.Match{
|
|
Score: hit.Score,
|
|
Entity: &searchMessage.Entity{
|
|
Ref: &searchMessage.Reference{
|
|
ResourceId: &searchMessage.ResourceID{
|
|
StorageId: resourceRootID.GetStorageId(),
|
|
SpaceId: resourceRootID.GetSpaceId(),
|
|
OpaqueId: resourceRootID.GetOpaqueId(),
|
|
},
|
|
Path: resource.Path,
|
|
},
|
|
Id: &searchMessage.ResourceID{
|
|
StorageId: resourceID.GetStorageId(),
|
|
SpaceId: resourceID.GetSpaceId(),
|
|
OpaqueId: resourceID.GetOpaqueId(),
|
|
},
|
|
Name: resource.Name,
|
|
ParentId: &searchMessage.ResourceID{
|
|
StorageId: resourceParentID.GetStorageId(),
|
|
SpaceId: resourceParentID.GetSpaceId(),
|
|
OpaqueId: resourceParentID.GetOpaqueId(),
|
|
},
|
|
Size: resource.Size,
|
|
Type: resource.Type,
|
|
MimeType: resource.MimeType,
|
|
Deleted: resource.Deleted,
|
|
Tags: resource.Tags,
|
|
Highlights: func() string {
|
|
contentHighlights, ok := hit.Highlight["Content"]
|
|
if !ok {
|
|
return ""
|
|
}
|
|
|
|
return strings.Join(contentHighlights[:], "; ")
|
|
}(),
|
|
Audio: func() *searchMessage.Audio {
|
|
if !strings.HasPrefix(resource.MimeType, "audio/") {
|
|
return nil
|
|
}
|
|
|
|
audio, _ := conversions.To[*searchMessage.Audio](resource.Audio)
|
|
return audio
|
|
}(),
|
|
Image: func() *searchMessage.Image {
|
|
image, _ := conversions.To[*searchMessage.Image](resource.Image)
|
|
return image
|
|
}(),
|
|
Location: func() *searchMessage.GeoCoordinates {
|
|
geoCoordinates, _ := conversions.To[*searchMessage.GeoCoordinates](resource.Location)
|
|
return geoCoordinates
|
|
}(),
|
|
Photo: func() *searchMessage.Photo {
|
|
photo, _ := conversions.To[*searchMessage.Photo](resource.Photo)
|
|
return photo
|
|
}(),
|
|
},
|
|
}
|
|
|
|
if mtime, err := time.Parse(time.RFC3339, resource.Mtime); err == nil {
|
|
match.Entity.LastModifiedTime = ×tamppb.Timestamp{Seconds: mtime.Unix(), Nanos: int32(mtime.Nanosecond())}
|
|
}
|
|
|
|
return match, nil
|
|
}
|