musenalm/migrations/seed/r_contents_agents.go

package seed

import (
	"slices"
	"strings"

	"github.com/Theodor-Springmann-Stiftung/musenalm/dbmodels"
	"github.com/Theodor-Springmann-Stiftung/musenalm/xmlmodels"
	"github.com/pocketbase/pocketbase/core"
)

func RecordsFromRelationInhalteAkteure(app core.App, relations xmlmodels.Relationen_Inhalte_Akteure) ([]*core.Record, error) {
	records := make([]*core.Record, 0, len(relations.Relationen))
	collection, err := app.FindCollectionByNameOrId(dbmodels.RelationTableName(dbmodels.CONTENTS_TABLE, dbmodels.AGENTS_TABLE))
	if err != nil {
		app.Logger().Error("Error finding collection", "error", err, "collection", dbmodels.RelationTableName(dbmodels.CONTENTS_TABLE, dbmodels.AGENTS_TABLE))
		return nil, err
	}

	for _, relation := range relations.Relationen {
		content, err := app.FindFirstRecordByData(dbmodels.CONTENTS_TABLE, dbmodels.MUSENALMID_FIELD, relation.Inhalt)
		if err != nil {
			app.Logger().Error("Error finding Inhalt", "error", err, "relation", relation)
			continue
		}

		agent, err := app.FindFirstRecordByData(dbmodels.AGENTS_TABLE, dbmodels.MUSENALMID_FIELD, relation.Akteur)
		if err != nil {
			app.Logger().Error("Error finding Content", "error", err, "relation", relation)
			continue
		}

		record := core.NewRecord(collection)
		record.Set(dbmodels.CONTENTS_TABLE, content.Id)
		record.Set(dbmodels.AGENTS_TABLE, agent.Id)

		switch relation.Relation {
		case "1":
			record.Set(dbmodels.RELATION_TYPE_FIELD, "Schöpfer")
			cat := content.GetStringSlice(dbmodels.MUSENALM_INHALTE_TYPE_FIELD)
			ber := agent.GetString(dbmodels.AGENTS_PROFESSION_FIELD)
			probt := 0
			probm := 0
			probg := 0

			if slices.ContainsFunc(cat, isProse) {
				probt += 1
			} else {
				probt -= 1
			}

			if Text(ber) {
				if probt > 0 {
					probt += 1
				}
				probt += 1
			}

			if slices.ContainsFunc(cat, isMusic) {
				probm += 1
			} else {
				probm -= 1
			}
			if Musik(ber) {
				if probm > 0 {
					probm += 1
				}
				probm += 1
			}

			if slices.ContainsFunc(cat, isGraph) {
				probg += 1
			} else {
				probg -= 1
			}
			if Graphiker(ber) {
				if probg > 0 {
					probg += 1
				}
				probg += 1
			}

			if probt == 3 && probm <= 1 && probg <= 1 {
				record.Set(dbmodels.RELATION_TYPE_FIELD, "Autor:in")
				break
			}

			if probm == 3 && probt <= 1 && probg <= 1 {
				record.Set(dbmodels.RELATION_TYPE_FIELD, "Komponist:in")
				break
			}

			if probg == 3 && probt <= 1 && probm <= 1 {
				record.Set(dbmodels.RELATION_TYPE_FIELD, "Künstler:in")
				break
			}

			record.Set(dbmodels.RELATION_TYPE_FIELD, "Schöpfer")
		case "2":
			record.Set(dbmodels.RELATION_TYPE_FIELD, "Autor:in")
		case "3":
			record.Set(dbmodels.RELATION_TYPE_FIELD, "Zeichner:in")
		case "4":
			record.Set(dbmodels.RELATION_TYPE_FIELD, "Kupferstecher:in")
		}

		rel := record.GetString(dbmodels.RELATION_TYPE_FIELD)
		ent := record.GetString(dbmodels.CONTENTS_TABLE)
		ser := record.GetString(dbmodels.AGENTS_TABLE)

		if strings.TrimSpace(rel) == "" || strings.TrimSpace(ent) == "" || strings.TrimSpace(ser) == "" {
			content.Set(dbmodels.EDITSTATE_FIELD, dbmodels.EDITORSTATE_VALUES[len(dbmodels.EDITORSTATE_VALUES)-2])
			_ = app.Save(content)
		}

		records = append(records, record)
	}

	return records, err
}

func isProse(cat string) bool {
	return cat == "Corrigenda" || cat == "Gedicht/Lied" || cat == "Motto" || cat == "Kalendarium" || cat == "Tabelle" || cat == "Inhaltsverzeichnis" || cat == "Text" || cat == "Prosa"
}

func isGraph(cat string) bool {
	return strings.HasPrefix(cat, "graph") || cat == "Graphik"
}

func Graphiker(beruf string) bool {
	return strings.Contains(beruf, "Graphik")
}

func Text(beruf string) bool {
	return strings.Contains(beruf, "Text")
}

func Musik(beruf string) bool {
	return strings.Contains(beruf, "Musik")
}

func isMusic(cat string) bool {
	return cat == "Gedicht/Lied" || cat == "Musikbeigabe"
}