Prev/next

2025-10-29 17:25:32 +00:00 · 2025-06-03 16:30:22 +02:00
parent 6be9663617
commit 0978ceb0c2
8 changed files with 104 additions and 15 deletions
--- a/dbmodels/agents_functions.go
+++ b/dbmodels/agents_functions.go
@@ -0,0 +1,317 @@
+package dbmodels
+
+import (
+	"github.com/pocketbase/dbx"
+	"github.com/pocketbase/pocketbase/core"
+	"golang.org/x/text/collate"
+	"golang.org/x/text/language"
+)
+
+type AgentsEntries map[string][]*REntriesAgents
+type AgentsContents map[string][]*RContentsAgents
+
+func FTS5SearchAgents(app core.App, query string) ([]*Agent, error) {
+	a := []*Agent{}
+	q := NormalizeQuery(query)
+	req := IntoQueryRequests([]string{AGENTS_NAME_FIELD, AGENTS_PSEUDONYMS_FIELD, REFERENCES_FIELD, AGENTS_BIOGRAPHICAL_DATA_FIELD, ANNOTATION_FIELD}, q)
+
+	if len(req) == 0 {
+		return a, nil
+	}
+
+	ids, err := FTS5Search(app, AGENTS_TABLE, req...)
+
+	if err != nil {
+		return nil, err
+	}
+
+	idany := []any{}
+	for _, id := range ids {
+		idany = append(idany, id.ID)
+	}
+
+	err = app.RecordQuery(AGENTS_TABLE).
+		Where(dbx.HashExp{ID_FIELD: idany}).
+		All(&a)
+	if err != nil {
+		return nil, err
+	}
+
+	return a, nil
+}
+
+func AgentsForEntries(app core.App, entries []*Entry) (map[string]*Agent, AgentsEntries, error) {
+	eids := []any{}
+	for _, e := range entries {
+		eids = append(eids, e.Id)
+	}
+
+	relations := []*REntriesAgents{}
+	err := app.RecordQuery(RelationTableName(ENTRIES_TABLE, AGENTS_TABLE)).
+		Where(dbx.HashExp{ENTRIES_TABLE: eids}).
+		All(&relations)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	agentIds := []any{}
+	for _, r := range relations {
+		agentIds = append(agentIds, r.Agent())
+	}
+
+	agents := []*Agent{}
+	err = app.RecordQuery(AGENTS_TABLE).
+		Where(dbx.HashExp{ID_FIELD: agentIds}).
+		All(&agents)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	agentsMap := make(map[string]*Agent, len(agents))
+	for _, a := range agents {
+		agentsMap[a.Id] = a
+	}
+
+	relationMap := make(map[string][]*REntriesAgents, len(entries))
+	for _, r := range relations {
+		relationMap[r.Entry()] = append(relationMap[r.Entry()], r)
+	}
+
+	return agentsMap, relationMap, nil
+}
+
+func AgentsForContents(app core.App, contents []*Content) (map[string]*Agent, AgentsContents, error) {
+	cids := []any{}
+	for _, c := range contents {
+		cids = append(cids, c.Id)
+	}
+
+	relations := []*RContentsAgents{}
+	err := app.RecordQuery(RelationTableName(CONTENTS_TABLE, AGENTS_TABLE)).
+		Where(dbx.HashExp{CONTENTS_TABLE: cids}).
+		All(&relations)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	agentIds := []any{}
+	for _, r := range relations {
+		agentIds = append(agentIds, r.Agent())
+	}
+
+	agents := []*Agent{}
+	err = app.RecordQuery(AGENTS_TABLE).
+		Where(dbx.HashExp{ID_FIELD: agentIds}).
+		All(&agents)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	agentsMap := make(map[string]*Agent, len(agents))
+	for _, a := range agents {
+		agentsMap[a.Id] = a
+	}
+
+	relationMap := make(map[string][]*RContentsAgents, len(contents))
+	for _, r := range relations {
+		relationMap[r.Content()] = append(relationMap[r.Content()], r)
+	}
+
+	return agentsMap, relationMap, nil
+}
+
+func LettersForAgents(app core.App, filter string) ([]string, error) {
+	letters := []core.Record{}
+	ids := []string{}
+
+	if filter == "" || filter == "noorg" {
+		err := app.RecordQuery(AGENTS_TABLE).
+			Select("upper(substr(" + AGENTS_NAME_FIELD + ", 1, 1)) AS id").
+			Distinct(true).
+			Where(dbx.HashExp{AGENTS_CORP_FIELD: false}).
+			All(&letters)
+		if err != nil {
+			return nil, err
+		}
+	} else if filter == "org" {
+		err := app.RecordQuery(AGENTS_TABLE).
+			Select("upper(substr(" + AGENTS_NAME_FIELD + ", 1, 1)) AS id").
+			Distinct(true).
+			Where(dbx.HashExp{AGENTS_CORP_FIELD: true}).
+			All(&letters)
+		if err != nil {
+			return nil, err
+		}
+	} else if filter == "musik" {
+		err := app.RecordQuery(AGENTS_TABLE).
+			Select("upper(substr(" + AGENTS_NAME_FIELD + ", 1, 1)) AS id").
+			Distinct(true).
+			Where(dbx.Like(AGENTS_PROFESSION_FIELD, "Musik").Match(true, true)).
+			All(&letters)
+		if err != nil {
+			return nil, err
+		}
+	} else if filter == "text" {
+		err := app.RecordQuery(AGENTS_TABLE).
+			Select("upper(substr(" + AGENTS_NAME_FIELD + ", 1, 1)) AS id").
+			Distinct(true).
+			Where(dbx.Like(AGENTS_PROFESSION_FIELD, "Text").Match(true, true)).
+			All(&letters)
+		if err != nil {
+			return nil, err
+		}
+	} else if filter == "graphik" {
+		err := app.RecordQuery(AGENTS_TABLE).
+			Select("upper(substr(" + AGENTS_NAME_FIELD + ", 1, 1)) AS id").
+			Distinct(true).
+			Where(dbx.Like(AGENTS_PROFESSION_FIELD, "Graphik").Match(true, true)).
+			All(&letters)
+		if err != nil {
+			return nil, err
+		}
+	} else if filter == "hrsg" {
+		err := app.RecordQuery(AGENTS_TABLE).
+			Select("upper(substr(" + AGENTS_NAME_FIELD + ", 1, 1)) AS id").
+			Distinct(true).
+			Where(dbx.Like(AGENTS_PROFESSION_FIELD, "Hrsg").Match(true, true)).
+			All(&letters)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	for _, l := range letters {
+		ids = append(ids, l.GetString("id"))
+	}
+
+	collator := collate.New(language.German, collate.Loose)
+	collator.SortStrings(ids)
+
+	return ids, nil
+}
+
+func AgentsForLetter(app core.App, letter string) ([]*Agent, error) {
+	agents := []*Agent{}
+	err := app.RecordQuery(AGENTS_TABLE).
+		Where(dbx.Like(AGENTS_NAME_FIELD, letter).Match(false, true)).
+		OrderBy(AGENTS_NAME_FIELD).
+		All(&agents)
+	if err != nil {
+		return nil, err
+	}
+
+	return agents, nil
+}
+
+func BasicSearchAgents(app core.App, query string) ([]*Agent, []*Agent, error) {
+	agents, err := TitleSearchAgents(app, query)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	altagents, err := AltSearchAgents(app, query)
+	if err != nil {
+		return nil, nil, err
+	}
+	return agents, altagents, nil
+}
+
+func TitleSearchAgents(app core.App, query string) ([]*Agent, error) {
+	agents := []*Agent{}
+	err := app.RecordQuery(AGENTS_TABLE).
+		Where(dbx.Like(AGENTS_NAME_FIELD, query).Match(true, true)).
+		OrWhere(dbx.Like(AGENTS_PSEUDONYMS_FIELD, query).Match(true, true)).
+		OrderBy(AGENTS_NAME_FIELD).
+		All(&agents)
+	if err != nil {
+		return nil, err
+	}
+
+	return agents, nil
+}
+
+func AltSearchAgents(app core.App, query string) ([]*Agent, error) {
+	agents := []*Agent{}
+	err := app.RecordQuery(AGENTS_TABLE).
+		Where(dbx.Like(ANNOTATION_FIELD, query).Match(true, true)).
+		OrderBy(AGENTS_NAME_FIELD).
+		All(&agents)
+	if err != nil {
+		return nil, err
+	}
+
+	return agents, nil
+}
+
+func AgentsForProfession(app core.App, profession string, letter string) ([]*Agent, error) {
+	agents := []*Agent{}
+	err := app.RecordQuery(AGENTS_TABLE).
+		Where(dbx.Like(AGENTS_NAME_FIELD, letter).Match(false, true)).
+		AndWhere(dbx.Like(AGENTS_PROFESSION_FIELD, profession).Match(true, true)).
+		OrderBy(AGENTS_NAME_FIELD).
+		All(&agents)
+	if err != nil {
+		return nil, err
+	}
+
+	return agents, nil
+}
+
+func AgentsForOrg(app core.App, org bool, letter string) ([]*Agent, error) {
+	agents := []*Agent{}
+	err := app.RecordQuery(AGENTS_TABLE).
+		Where(dbx.Like(AGENTS_NAME_FIELD, letter).Match(false, true)).
+		AndWhere(dbx.HashExp{AGENTS_CORP_FIELD: org}).
+		OrderBy(AGENTS_NAME_FIELD).
+		All(&agents)
+	if err != nil {
+		return nil, err
+	}
+
+	return agents, nil
+}
+
+type AgentCount struct {
+	Count int    `db:"count"`
+	ID    string `db:"id"`
+}
+
+func CountAgentsBaende(app core.App, ids []any) (map[string]int, error) {
+	couns := []AgentCount{}
+	err := app.RecordQuery(RelationTableName(ENTRIES_TABLE, AGENTS_TABLE)).
+		Select("count(*) as count, " + AGENTS_TABLE + " as id").
+		Where(dbx.HashExp{AGENTS_TABLE: ids}).
+		GroupBy(AGENTS_TABLE).
+		All(&couns)
+
+	if err != nil {
+		return nil, err
+	}
+
+	ret := make(map[string]int, len(couns))
+	for _, c := range couns {
+		ret[c.ID] = c.Count
+	}
+
+	return ret, nil
+}
+
+func CountAgentsContents(app core.App, ids []any) (map[string]int, error) {
+	couns := []AgentCount{}
+	err := app.RecordQuery(RelationTableName(CONTENTS_TABLE, AGENTS_TABLE)).
+		Select("count(*) as count, " + AGENTS_TABLE + " as id").
+		Where(dbx.HashExp{AGENTS_TABLE: ids}).
+		GroupBy(AGENTS_TABLE).
+		All(&couns)
+
+	if err != nil {
+		return nil, err
+	}
+
+	ret := make(map[string]int, len(couns))
+	for _, c := range couns {
+		ret[c.ID] = c.Count
+	}
+
+	return ret, nil
+}