74 lines
2.3 KiB
Go
74 lines
2.3 KiB
Go
package npcinteractions
|
|
|
|
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"
|
|
)
|
|
|
|
func canonicalizeResponse(response *extractionResponse, order shared.SourceRefOrder, sourceID string) {
|
|
if response == nil {
|
|
return
|
|
}
|
|
for index := range response.Interactions {
|
|
canonicalizeInteraction(&response.Interactions[index], order, sourceID)
|
|
}
|
|
sort.SliceStable(response.Interactions, func(i, j int) bool {
|
|
left, leftOK := order.EarliestValid(canonicalSourceRefs(response.Interactions[i].SourceRefs, sourceID))
|
|
right, rightOK := order.EarliestValid(canonicalSourceRefs(response.Interactions[j].SourceRefs, sourceID))
|
|
if leftOK != rightOK {
|
|
return leftOK
|
|
}
|
|
if !leftOK {
|
|
return false
|
|
}
|
|
return left < right
|
|
})
|
|
}
|
|
|
|
func canonicalizeInteraction(interaction *interactionResponse, order shared.SourceRefOrder, sourceID string) {
|
|
if interaction == nil {
|
|
return
|
|
}
|
|
interaction.SourceRefs = interactionResponseRefs(order.Canonicalize(canonicalSourceRefs(interaction.SourceRefs, sourceID)))
|
|
}
|
|
|
|
func canonicalInteractionList(response extractionResponse, sourceID string) dnd.NPCInteractionList {
|
|
interactions := make([]dnd.NPCInteraction, len(response.Interactions))
|
|
for index, interaction := range response.Interactions {
|
|
interactions[index] = dnd.NPCInteraction{
|
|
Name: interaction.Name,
|
|
Kind: dnd.NPCInteractionKind(interaction.Kind),
|
|
SourceRefs: canonicalSourceRefs(interaction.SourceRefs, sourceID),
|
|
}
|
|
}
|
|
if response.Interactions == nil {
|
|
interactions = nil
|
|
}
|
|
return dnd.NPCInteractionList{Interactions: interactions}
|
|
}
|
|
|
|
func canonicalSourceRefs(refs []interactionSourceRefResponse, sourceID string) []source.SourceRef {
|
|
if refs == nil {
|
|
return nil
|
|
}
|
|
out := make([]source.SourceRef, len(refs))
|
|
for index, ref := range refs {
|
|
out[index] = source.SourceRef{SourceID: sourceID, StartUnitID: ref.StartUnitID, EndUnitID: ref.EndUnitID}
|
|
}
|
|
return out
|
|
}
|
|
|
|
func interactionResponseRefs(refs []source.SourceRef) []interactionSourceRefResponse {
|
|
if refs == nil {
|
|
return nil
|
|
}
|
|
values := make([]interactionSourceRefResponse, len(refs))
|
|
for index, ref := range refs {
|
|
values[index] = interactionSourceRefResponse{StartUnitID: ref.StartUnitID, EndUnitID: ref.EndUnitID}
|
|
}
|
|
return values
|
|
}
|