Saner file names

viewmodels/agent_view.go

@@ -0,0 +1,58 @@
+package viewmodels
+
+import (
+	"strings"
+
+	"github.com/Theodor-Springmann-Stiftung/kgpz_web/providers/xmlprovider"
+)
+
+type AgentView struct {
+	Agents []xmlprovider.Agent
+	Works  map[string][]xmlprovider.Work
+	Pieces map[string][]xmlprovider.Piece
+}
+
+func AgentsView(letterorid string, lib *xmlprovider.Library) *AgentView {
+	res := AgentView{}
+	lib.Agents.Items.Range(func(key, value interface{}) bool {
+		k := key.(string)
+		if strings.HasPrefix(k, letterorid) {
+			agent := value.(xmlprovider.Agent)
+			res.Agents = append(res.Agents, agent)
+		}
+		return true
+	})
+
+	res.Works = make(map[string][]xmlprovider.Work)
+	res.Pieces = make(map[string][]xmlprovider.Piece)
+
+	lib.Works.Items.Range(func(key, value interface{}) bool {
+		w := value.(xmlprovider.Work)
+		for _, a := range res.Agents {
+			if strings.HasPrefix(a.ID, letterorid) {
+				_, ok := res.Works[a.ID]
+				if !ok {
+					res.Works[a.ID] = []xmlprovider.Work{}
+				}
+				res.Works[a.ID] = append(res.Works[a.ID], w)
+			}
+		}
+		return true
+	})
+
+	lib.Pieces.Items.Range(func(key, value interface{}) bool {
+		p := value.(xmlprovider.Piece)
+		for _, a := range res.Agents {
+			if strings.HasPrefix(a.ID, letterorid) {
+				_, ok := res.Pieces[a.ID]
+				if !ok {
+					res.Pieces[a.ID] = []xmlprovider.Piece{}
+				}
+				res.Pieces[a.ID] = append(res.Pieces[a.ID], p)
+			}
+		}
+		return true
+	})
+
+	return &res
+}