100 lines
2.8 KiB
Go
100 lines
2.8 KiB
Go
package itemevents
|
|
|
|
import (
|
|
"sort"
|
|
|
|
"gitea.maximumdirect.net/eric/notarius/internal/core/source"
|
|
"gitea.maximumdirect.net/eric/notarius/internal/modules/dnd"
|
|
"gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/shared"
|
|
)
|
|
|
|
type orderedItemEventResponse struct {
|
|
value itemEventResponse
|
|
earliest int
|
|
hasEvidence bool
|
|
}
|
|
|
|
func canonicalizeResponse(response *extractionResponse, order shared.SourceRefOrder, sourceID string) {
|
|
if response == nil {
|
|
return
|
|
}
|
|
ordered := make([]orderedItemEventResponse, len(response.Events))
|
|
for index := range response.Events {
|
|
earliest, hasEvidence := canonicalizeItemEvent(&response.Events[index], order, sourceID)
|
|
ordered[index] = orderedItemEventResponse{
|
|
value: response.Events[index],
|
|
earliest: earliest,
|
|
hasEvidence: hasEvidence,
|
|
}
|
|
}
|
|
sort.SliceStable(ordered, func(left, right int) bool {
|
|
if ordered[left].hasEvidence != ordered[right].hasEvidence {
|
|
return ordered[left].hasEvidence
|
|
}
|
|
if !ordered[left].hasEvidence {
|
|
return false
|
|
}
|
|
return ordered[left].earliest < ordered[right].earliest
|
|
})
|
|
for index := range ordered {
|
|
response.Events[index] = ordered[index].value
|
|
}
|
|
}
|
|
|
|
func canonicalizeItemEvent(event *itemEventResponse, order shared.SourceRefOrder, sourceID string) (int, bool) {
|
|
if event == nil {
|
|
return 0, false
|
|
}
|
|
refs := order.Canonicalize(itemEventSourceRefs(event.SourceRefs, sourceID))
|
|
event.SourceRefs = itemEventResponseRefs(refs)
|
|
return order.EarliestValid(refs)
|
|
}
|
|
|
|
func canonicalItemEventList(response extractionResponse, sourceID string) dnd.ItemEventList {
|
|
if response.Events == nil {
|
|
return dnd.ItemEventList{}
|
|
}
|
|
events := make([]dnd.ItemEvent, len(response.Events))
|
|
for index, event := range response.Events {
|
|
events[index] = dnd.ItemEvent{
|
|
Name: event.Name,
|
|
Kind: dnd.ItemEventKind(event.Kind),
|
|
Quantity: cloneQuantity(event.Quantity),
|
|
From: event.From,
|
|
To: event.To,
|
|
SourceRefs: itemEventSourceRefs(event.SourceRefs, sourceID),
|
|
}
|
|
}
|
|
return dnd.ItemEventList{Events: events}
|
|
}
|
|
|
|
func itemEventSourceRefs(refs []itemEventSourceRefResponse, sourceID string) []source.SourceRef {
|
|
if refs == nil {
|
|
return nil
|
|
}
|
|
values := make([]source.SourceRef, len(refs))
|
|
for index, ref := range refs {
|
|
values[index] = source.SourceRef{SourceID: sourceID, StartUnitID: ref.StartSegment, EndUnitID: ref.EndSegment}
|
|
}
|
|
return values
|
|
}
|
|
|
|
func itemEventResponseRefs(refs []source.SourceRef) []itemEventSourceRefResponse {
|
|
if refs == nil {
|
|
return nil
|
|
}
|
|
values := make([]itemEventSourceRefResponse, len(refs))
|
|
for index, ref := range refs {
|
|
values[index] = itemEventSourceRefResponse{StartSegment: ref.StartUnitID, EndSegment: ref.EndUnitID}
|
|
}
|
|
return values
|
|
}
|
|
|
|
func cloneQuantity(value *int) *int {
|
|
if value == nil {
|
|
return nil
|
|
}
|
|
quantity := *value
|
|
return &quantity
|
|
}
|