kgpz_web/viewmodels/agent_view.go

package viewmodels

import (
	"maps"
	"slices"
	"strings"

	"github.com/Theodor-Springmann-Stiftung/kgpz_web/providers/xmlprovider"
)

type AgentsListView struct {
	Search           string
	AvailableLetters []string
	Agents           map[string]AgentView
	Sorted           []string
}

type AgentView struct {
	xmlprovider.Agent
	Works  []WorkByAgent
	Pieces []PieceByAgent
}

type WorkByAgent struct {
	xmlprovider.Work
	Reference xmlprovider.AgentRef
}

type PieceByAgent struct {
	xmlprovider.Piece
	Reference xmlprovider.AgentRef
}

func AgentsView(letterorid string, lib *xmlprovider.Library) *AgentsListView {
	res := AgentsListView{Search: letterorid, Agents: make(map[string]AgentView)}
	av := make(map[string]bool)

	if len(letterorid) == 1 {
		// INFO: This is all persons beginning with a letter
		for _, a := range lib.Agents.Array {
			av[strings.ToUpper(a.ID[:1])] = true
			if strings.HasPrefix(a.ID, letterorid) {
				res.Sorted = append(res.Sorted, a.ID)
				res.Agents[a.ID] = AgentView{Agent: a}
			}
		}
	} else {
		// INFO: This is a specific person lookup by ID
		for _, a := range lib.Agents.Array {
			av[strings.ToUpper(a.ID[:1])] = true
			if a.ID == letterorid {
				res.Sorted = append(res.Sorted, a.ID)
				res.Agents[a.ID] = AgentView{Agent: a}
				break
			}
		}
	}

	// TODO: We won't need to lock the library if we take down all routes during parsing
	lib.Works.Lock()
	for _, w := range lib.Works.Array {
		if ref, ok := w.ReferencesAgent(letterorid); ok {
			if entry, ok := res.Agents[ref.Ref]; ok {
				entry.Works = append(entry.Works, WorkByAgent{Work: w, Reference: *ref})
				res.Agents[ref.Ref] = entry
			}
		}
	}
	lib.Works.Unlock()

	lib.Pieces.Lock()
	for _, p := range lib.Pieces.Array {
		if ref, ok := p.ReferencesAgent(letterorid); ok {
			if entry, ok := res.Agents[ref.Ref]; ok {
				entry.Pieces = append(entry.Pieces, PieceByAgent{Piece: p, Reference: *ref})
				res.Agents[ref.Ref] = entry
			}
		}
	}
	lib.Pieces.Unlock()

	res.AvailableLetters = slices.Collect(maps.Keys(av))
	slices.Sort(res.AvailableLetters)
	slices.Sort(res.Sorted)

	return &res
}