« get me outta code hell

generateArtistGroupContributionsInfo.js « dependencies « content « src - hsmusic-wiki - HSMusic - static wiki software cataloguing collaborative creation
about summary refs log tree commit diff
path: root/src/content/dependencies/generateArtistGroupContributionsInfo.js
blob: 0ecfaa3448d947cbbd66ad914ed727f11521a17f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import {stitchArrays, unique} from '../../util/sugar.js';

export default {
  contentDependencies: ['linkGroup'],
  extraDependencies: ['html', 'language', 'wikiData'],

  sprawl({groupCategoryData}) {
    return {
      groupOrder: groupCategoryData.flatMap(category => category.groups),
    }
  },

  query(sprawl, tracksAndAlbums) {
    const filteredAlbums = tracksAndAlbums.filter(thing => !thing.album);
    const filteredTracks = tracksAndAlbums.filter(thing => thing.album);

    const allAlbums = unique([
      ...filteredAlbums,
      ...filteredTracks.map(track => track.album),
    ]);

    const allGroupsUnordered = new Set(Array.from(allAlbums).flatMap(album => album.groups));
    const allGroupsOrdered = sprawl.groupOrder.filter(group => allGroupsUnordered.has(group));

    const mapTemplate = allGroupsOrdered.map(group => [group, 0]);
    const groupToCountMap = new Map(mapTemplate);
    const groupToDurationMap = new Map(mapTemplate);
    const groupToDurationCountMap = new Map(mapTemplate);

    for (const album of filteredAlbums) {
      for (const group of album.groups) {
        groupToCountMap.set(group, groupToCountMap.get(group) + 1);
      }
    }

    for (const track of filteredTracks) {
      for (const group of track.album.groups) {
        groupToCountMap.set(group, groupToCountMap.get(group) + 1);
        if (track.duration) {
          groupToDurationMap.set(group, groupToDurationMap.get(group) + track.duration);
          groupToDurationCountMap.set(group, groupToDurationCountMap.get(group) + 1);
        }
      }
    }

    const groupsSortedByCount =
      allGroupsOrdered
        .sort((a, b) => groupToCountMap.get(b) - groupToCountMap.get(a));

    const groupsSortedByDuration =
      allGroupsOrdered
        .filter(group => groupToDurationMap.get(group) > 0)
        .sort((a, b) => groupToDurationMap.get(b) - groupToDurationMap.get(a));

    const groupCounts =
      groupsSortedByCount
        .map(group => groupToCountMap.get(group));

    const groupDurations =
      groupsSortedByDuration
        .map(group => groupToDurationMap.get(group));

    const groupDurationsApproximate =
      groupsSortedByDuration
        .map(group => groupToDurationCountMap.get(group) > 1);

    return {
      groupsSortedByCount,
      groupsSortedByDuration,
      groupCounts,
      groupDurations,
      groupDurationsApproximate,
    };
  },

  relations(relation, query) {
    return {
      groupLinksSortedByCount:
        query.groupsSortedByCount
          .map(group => relation('linkGroup', group)),

      groupLinksSortedByDuration:
        query.groupsSortedByDuration
          .map(group => relation('linkGroup', group)),
    };
  },

  data(query) {
    return {
      groupCounts: query.groupCounts,
      groupDurations: query.groupDurations,
      groupDurationsApproximate: query.groupDurationsApproximate,
    };
  },

  slots: {
    mode: {
      validate: v => v.is('count', 'duration'),
    },
  },

  generate(data, relations, slots, {language}) {
    return (
      language.formatUnitList(
        (slots.mode === 'count'
          ? stitchArrays({
              groupLink: relations.groupLinksSortedByCount,
              count: data.groupCounts,
            }).map(({groupLink, count}) =>
                language.$('artistPage.groupsLine.item.withCount', {
                  group: groupLink,
                  count,
                }))
          : stitchArrays({
              groupLink: relations.groupLinksSortedByDuration,
              duration: data.groupDurations,
              approximate: data.groupDurationsApproximate,
            }).map(({groupLink, duration, approximate}) =>
                language.$('artistPage.groupsLine.item.withDuration', {
                  group: groupLink,
                  duration: language.formatDuration(duration, {approximate}),
                })))));
  },
};