@internetarchive/collection-browser 2.7.7-alpha.2 → 2.7.7

package/dist/src/collection-facets/smart-facets/heuristics/wikidata-heuristic.js

@@ -1,173 +1,173 @@
-// If wikidata describes the top query result as X, recommend facet Y, e.g.:
-// X              Y
-// written work   mt:texts
-// film           mt:movies
-// author         mt:texts and creator:<query>
-// filmmaker      mt:movies and creator:<query>
-// photographer   mt:image and creator:<query>
-// visual artist  mt:image and creator:<query>
-// etc.
-export class WikidataHeuristic {
-    async getRecommendedFacets(query) {
-        var _a;
-        const recommendations = [];
-        try {
-            const urlQuery = encodeURIComponent(query);
-            const wikidataResponse = await fetch(`https://www.wikidata.org/w/api.php?action=wbsearchentities&search=${urlQuery}&format=json&language=en&uselang=en&origin=*&type=item&limit=5`);
-            const searchResults = await wikidataResponse.json();
-            for (const [keyword, facets] of Object.entries(WikidataHeuristic.ENTITIES)) {
-                const keywordRegex = new RegExp(`\\b${keyword}\\b`);
-                if (keywordRegex.test((_a = searchResults.search[0]) === null || _a === void 0 ? void 0 : _a.description)) {
-                    const entityName = searchResults.search[0].label;
-                    recommendations.push(...facets.map(sf => {
-                        var _a;
-                        return ({
-                            label: (_a = sf.label) === null || _a === void 0 ? void 0 : _a.replace('__QUERY', entityName),
-                            facets: sf.facets.map(f => {
-                                var _a;
-                                const replaced = {
-                                    ...f,
-                                    bucketKey: f.bucketKey.replace('__QUERY', query),
-                                };
-                                if (f.displayText) {
-                                    replaced.displayText = (_a = replaced.displayText) === null || _a === void 0 ? void 0 : _a.replace('__QUERY', entityName);
-                                }
-                                return replaced;
-                            }),
-                        });
-                    }));
-                }
-            }
-            return recommendations;
-        }
-        catch (_b) {
-            return [];
-        }
-    }
-}
-WikidataHeuristic.ENTITIES = {
-    'written work': [
-        { facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] },
-    ],
-    literature: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],
-    book: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],
-    novel: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],
-    filmmaker: [
-        {
-            label: 'Films by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'movies' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    author: [
-        {
-            label: 'Writing by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'texts' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    writer: [
-        {
-            label: 'Writing by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'texts' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    poet: [
-        {
-            label: 'Writing by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'texts' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    photographer: [
-        {
-            label: 'Images by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'image' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    painter: [
-        {
-            label: 'Images by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'image' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    'visual artist': [
-        {
-            label: 'Images by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'image' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    'graphic artist': [
-        {
-            label: 'Images by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'image' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    singer: [
-        {
-            label: 'Music by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'audio' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    songwriter: [
-        {
-            label: 'Music by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'audio' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    musician: [
-        {
-            label: 'Music by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'audio' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    composer: [
-        {
-            label: 'Music by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'audio' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-    pianist: [
-        {
-            label: 'Music by __QUERY',
-            facets: [
-                { facetType: 'mediatype', bucketKey: 'audio' },
-                { facetType: 'creator', bucketKey: '__QUERY' },
-            ],
-        },
-    ],
-};
+// If wikidata describes the top query result as X, recommend facet Y, e.g.:
+// X              Y
+// written work   mt:texts
+// film           mt:movies
+// author         mt:texts and creator:<query>
+// filmmaker      mt:movies and creator:<query>
+// photographer   mt:image and creator:<query>
+// visual artist  mt:image and creator:<query>
+// etc.
+export class WikidataHeuristic {
+    async getRecommendedFacets(query) {
+        var _a;
+        const recommendations = [];
+        try {
+            const urlQuery = encodeURIComponent(query);
+            const wikidataResponse = await fetch(`https://www.wikidata.org/w/api.php?action=wbsearchentities&search=${urlQuery}&format=json&language=en&uselang=en&origin=*&type=item&limit=5`);
+            const searchResults = await wikidataResponse.json();
+            for (const [keyword, facets] of Object.entries(WikidataHeuristic.ENTITIES)) {
+                const keywordRegex = new RegExp(`\\b${keyword}\\b`);
+                if (keywordRegex.test((_a = searchResults.search[0]) === null || _a === void 0 ? void 0 : _a.description)) {
+                    const entityName = searchResults.search[0].label;
+                    recommendations.push(...facets.map(sf => {
+                        var _a;
+                        return ({
+                            label: (_a = sf.label) === null || _a === void 0 ? void 0 : _a.replace('__QUERY', entityName),
+                            facets: sf.facets.map(f => {
+                                var _a;
+                                const replaced = {
+                                    ...f,
+                                    bucketKey: f.bucketKey.replace('__QUERY', query),
+                                };
+                                if (f.displayText) {
+                                    replaced.displayText = (_a = replaced.displayText) === null || _a === void 0 ? void 0 : _a.replace('__QUERY', entityName);
+                                }
+                                return replaced;
+                            }),
+                        });
+                    }));
+                }
+            }
+            return recommendations;
+        }
+        catch (err) {
+            return [];
+        }
+    }
+}
+WikidataHeuristic.ENTITIES = {
+    'written work': [
+        { facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] },
+    ],
+    literature: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],
+    book: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],
+    novel: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],
+    filmmaker: [
+        {
+            label: 'Films by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'movies' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    author: [
+        {
+            label: 'Writing by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'texts' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    writer: [
+        {
+            label: 'Writing by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'texts' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    poet: [
+        {
+            label: 'Writing by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'texts' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    photographer: [
+        {
+            label: 'Images by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'image' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    painter: [
+        {
+            label: 'Images by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'image' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    'visual artist': [
+        {
+            label: 'Images by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'image' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    'graphic artist': [
+        {
+            label: 'Images by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'image' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    singer: [
+        {
+            label: 'Music by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'audio' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    songwriter: [
+        {
+            label: 'Music by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'audio' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    musician: [
+        {
+            label: 'Music by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'audio' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    composer: [
+        {
+            label: 'Music by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'audio' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+    pianist: [
+        {
+            label: 'Music by __QUERY',
+            facets: [
+                { facetType: 'mediatype', bucketKey: 'audio' },
+                { facetType: 'creator', bucketKey: '__QUERY' },
+            ],
+        },
+    ],
+};
 //# sourceMappingURL=wikidata-heuristic.js.map

package/dist/src/collection-facets/smart-facets/heuristics/wikidata-heuristic.js.map

@@ -1 +1 @@
-{"version":3,"file":"wikidata-heuristic.js","sourceRoot":"","sources":["../../../../../src/collection-facets/smart-facets/heuristics/wikidata-heuristic.ts"],"names":[],"mappings":"AAMA,4EAA4E;AAC5E,mBAAmB;AACnB,0BAA0B;AAC1B,2BAA2B;AAC3B,8CAA8C;AAC9C,+CAA+C;AAC/C,8CAA8C;AAC9C,8CAA8C;AAC9C,OAAO;AACP,MAAM,OAAO,iBAAiB;IA+H5B,KAAK,CAAC,oBAAoB,CAAC,KAAa;;QACtC,MAAM,eAAe,GAAiB,EAAE,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAE3C,MAAM,gBAAgB,GAAG,MAAM,KAAK,CAClC,qEAAqE,QAAQ,gEAAgE,CAC9I,CAAC;YACF,MAAM,aAAa,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAEpD,KAAK,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAC5C,iBAAiB,CAAC,QAAQ,CAC3B,EAAE,CAAC;gBACF,MAAM,YAAY,GAAG,IAAI,MAAM,CAAC,MAAM,OAAO,KAAK,CAAC,CAAC;gBACpD,IAAI,YAAY,CAAC,IAAI,CAAC,MAAA,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,0CAAE,WAAW,CAAC,EAAE,CAAC;oBAC5D,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;oBACjD,eAAe,CAAC,IAAI,CAClB,GAAG,MAAM,CAAC,GAAG,CACX,EAAE,CAAC,EAAE;;wBACH,OAAA,CAAC;4BACC,KAAK,EAAE,MAAA,EAAE,CAAC,KAAK,0CAAE,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC;4BAC/C,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;;gCACxB,MAAM,QAAQ,GAAG;oCACf,GAAG,CAAC;oCACJ,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC;iCACjD,CAAC;gCAEF,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;oCAClB,QAAQ,CAAC,WAAW,GAAG,MAAA,QAAQ,CAAC,WAAW,0CAAE,OAAO,CAClD,SAAS,EACT,UAAU,CACX,CAAC;gCACJ,CAAC;gCAED,OAAO,QAAQ,CAAC;4BAClB,CAAC,CAAC;yBACH,CAAe,CAAA;qBAAA,CACnB,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,OAAO,eAAe,CAAC;QACzB,CAAC;QAAC,WAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;;AA7KuB,0BAAQ,GAAoB;IAClD,cAAc,EAAE;QACd,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,EAAE;KAC7D;IACD,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IAC1E,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IACpE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IACrE,SAAS,EAAE;QACT;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE;gBAC/C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,MAAM,EAAE;QACN;YACE,KAAK,EAAE,oBAAoB;YAC3B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,MAAM,EAAE;QACN;YACE,KAAK,EAAE,oBAAoB;YAC3B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,IAAI,EAAE;QACJ;YACE,KAAK,EAAE,oBAAoB;YAC3B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,YAAY,EAAE;QACZ;YACE,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,OAAO,EAAE;QACP;YACE,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,eAAe,EAAE;QACf;YACE,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,gBAAgB,EAAE;QAChB;YACE,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,MAAM,EAAE;QACN;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,UAAU,EAAE;QACV;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,QAAQ,EAAE;QACR;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,QAAQ,EAAE;QACR;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,OAAO,EAAE;QACP;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;CACF,CAAC","sourcesContent":["import type {\n  SmartQueryHeuristic,\n  KeywordFacetMap,\n  SmartFacet,\n} from '../models';\n\n// If wikidata describes the top query result as X, recommend facet Y, e.g.:\n// X              Y\n// written work   mt:texts\n// film           mt:movies\n// author         mt:texts and creator:<query>\n// filmmaker      mt:movies and creator:<query>\n// photographer   mt:image and creator:<query>\n// visual artist  mt:image and creator:<query>\n// etc.\nexport class WikidataHeuristic implements SmartQueryHeuristic {\n  private static readonly ENTITIES: KeywordFacetMap = {\n    'written work': [\n      { facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] },\n    ],\n    literature: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],\n    book: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],\n    novel: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],\n    filmmaker: [\n      {\n        label: 'Films by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'movies' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    author: [\n      {\n        label: 'Writing by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'texts' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    writer: [\n      {\n        label: 'Writing by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'texts' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    poet: [\n      {\n        label: 'Writing by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'texts' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    photographer: [\n      {\n        label: 'Images by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'image' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    painter: [\n      {\n        label: 'Images by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'image' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    'visual artist': [\n      {\n        label: 'Images by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'image' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    'graphic artist': [\n      {\n        label: 'Images by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'image' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    singer: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    songwriter: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    musician: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    composer: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    pianist: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n  };\n\n  async getRecommendedFacets(query: string): Promise<SmartFacet[]> {\n    const recommendations: SmartFacet[] = [];\n\n    try {\n      const urlQuery = encodeURIComponent(query);\n\n      const wikidataResponse = await fetch(\n        `https://www.wikidata.org/w/api.php?action=wbsearchentities&search=${urlQuery}&format=json&language=en&uselang=en&origin=*&type=item&limit=5`,\n      );\n      const searchResults = await wikidataResponse.json();\n\n      for (const [keyword, facets] of Object.entries(\n        WikidataHeuristic.ENTITIES,\n      )) {\n        const keywordRegex = new RegExp(`\\\\b${keyword}\\\\b`);\n        if (keywordRegex.test(searchResults.search[0]?.description)) {\n          const entityName = searchResults.search[0].label;\n          recommendations.push(\n            ...facets.map(\n              sf =>\n                ({\n                  label: sf.label?.replace('__QUERY', entityName),\n                  facets: sf.facets.map(f => {\n                    const replaced = {\n                      ...f,\n                      bucketKey: f.bucketKey.replace('__QUERY', query),\n                    };\n\n                    if (f.displayText) {\n                      replaced.displayText = replaced.displayText?.replace(\n                        '__QUERY',\n                        entityName,\n                      );\n                    }\n\n                    return replaced;\n                  }),\n                }) as SmartFacet,\n            ),\n          );\n        }\n      }\n\n      return recommendations;\n    } catch {\n      return [];\n    }\n  }\n}\n"]}
+{"version":3,"file":"wikidata-heuristic.js","sourceRoot":"","sources":["../../../../../src/collection-facets/smart-facets/heuristics/wikidata-heuristic.ts"],"names":[],"mappings":"AAMA,4EAA4E;AAC5E,mBAAmB;AACnB,0BAA0B;AAC1B,2BAA2B;AAC3B,8CAA8C;AAC9C,+CAA+C;AAC/C,8CAA8C;AAC9C,8CAA8C;AAC9C,OAAO;AACP,MAAM,OAAO,iBAAiB;IA+H5B,KAAK,CAAC,oBAAoB,CAAC,KAAa;;QACtC,MAAM,eAAe,GAAiB,EAAE,CAAC;QAEzC,IAAI;YACF,MAAM,QAAQ,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAE3C,MAAM,gBAAgB,GAAG,MAAM,KAAK,CAClC,qEAAqE,QAAQ,gEAAgE,CAC9I,CAAC;YACF,MAAM,aAAa,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAEpD,KAAK,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAC5C,iBAAiB,CAAC,QAAQ,CAC3B,EAAE;gBACD,MAAM,YAAY,GAAG,IAAI,MAAM,CAAC,MAAM,OAAO,KAAK,CAAC,CAAC;gBACpD,IAAI,YAAY,CAAC,IAAI,CAAC,MAAA,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,0CAAE,WAAW,CAAC,EAAE;oBAC3D,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;oBACjD,eAAe,CAAC,IAAI,CAClB,GAAG,MAAM,CAAC,GAAG,CACX,EAAE,CAAC,EAAE;;wBACH,OAAA,CAAC;4BACC,KAAK,EAAE,MAAA,EAAE,CAAC,KAAK,0CAAE,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC;4BAC/C,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;;gCACxB,MAAM,QAAQ,GAAG;oCACf,GAAG,CAAC;oCACJ,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC;iCACjD,CAAC;gCAEF,IAAI,CAAC,CAAC,WAAW,EAAE;oCACjB,QAAQ,CAAC,WAAW,GAAG,MAAA,QAAQ,CAAC,WAAW,0CAAE,OAAO,CAClD,SAAS,EACT,UAAU,CACX,CAAC;iCACH;gCAED,OAAO,QAAQ,CAAC;4BAClB,CAAC,CAAC;yBACY,CAAA,CAAA;qBAAA,CACnB,CACF,CAAC;iBACH;aACF;YAED,OAAO,eAAe,CAAC;SACxB;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO,EAAE,CAAC;SACX;IACH,CAAC;;AA7KuB,0BAAQ,GAAoB;IAClD,cAAc,EAAE;QACd,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,EAAE;KAC7D;IACD,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IAC1E,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IACpE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IACrE,SAAS,EAAE;QACT;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE;gBAC/C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,MAAM,EAAE;QACN;YACE,KAAK,EAAE,oBAAoB;YAC3B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,MAAM,EAAE;QACN;YACE,KAAK,EAAE,oBAAoB;YAC3B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,IAAI,EAAE;QACJ;YACE,KAAK,EAAE,oBAAoB;YAC3B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,YAAY,EAAE;QACZ;YACE,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,OAAO,EAAE;QACP;YACE,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,eAAe,EAAE;QACf;YACE,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,gBAAgB,EAAE;QAChB;YACE,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,MAAM,EAAE;QACN;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,UAAU,EAAE;QACV;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,QAAQ,EAAE;QACR;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,QAAQ,EAAE;QACR;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;IACD,OAAO,EAAE;QACP;YACE,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE;gBACN,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE;gBAC9C,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/C;SACF;KACF;CACF,CAAC","sourcesContent":["import type {\n  SmartQueryHeuristic,\n  KeywordFacetMap,\n  SmartFacet,\n} from '../models';\n\n// If wikidata describes the top query result as X, recommend facet Y, e.g.:\n// X              Y\n// written work   mt:texts\n// film           mt:movies\n// author         mt:texts and creator:<query>\n// filmmaker      mt:movies and creator:<query>\n// photographer   mt:image and creator:<query>\n// visual artist  mt:image and creator:<query>\n// etc.\nexport class WikidataHeuristic implements SmartQueryHeuristic {\n  private static readonly ENTITIES: KeywordFacetMap = {\n    'written work': [\n      { facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] },\n    ],\n    literature: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],\n    book: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],\n    novel: [{ facets: [{ facetType: 'mediatype', bucketKey: 'texts' }] }],\n    filmmaker: [\n      {\n        label: 'Films by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'movies' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    author: [\n      {\n        label: 'Writing by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'texts' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    writer: [\n      {\n        label: 'Writing by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'texts' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    poet: [\n      {\n        label: 'Writing by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'texts' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    photographer: [\n      {\n        label: 'Images by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'image' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    painter: [\n      {\n        label: 'Images by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'image' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    'visual artist': [\n      {\n        label: 'Images by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'image' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    'graphic artist': [\n      {\n        label: 'Images by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'image' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    singer: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    songwriter: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    musician: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    composer: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n    pianist: [\n      {\n        label: 'Music by __QUERY',\n        facets: [\n          { facetType: 'mediatype', bucketKey: 'audio' },\n          { facetType: 'creator', bucketKey: '__QUERY' },\n        ],\n      },\n    ],\n  };\n\n  async getRecommendedFacets(query: string): Promise<SmartFacet[]> {\n    const recommendations: SmartFacet[] = [];\n\n    try {\n      const urlQuery = encodeURIComponent(query);\n\n      const wikidataResponse = await fetch(\n        `https://www.wikidata.org/w/api.php?action=wbsearchentities&search=${urlQuery}&format=json&language=en&uselang=en&origin=*&type=item&limit=5`\n      );\n      const searchResults = await wikidataResponse.json();\n\n      for (const [keyword, facets] of Object.entries(\n        WikidataHeuristic.ENTITIES\n      )) {\n        const keywordRegex = new RegExp(`\\\\b${keyword}\\\\b`);\n        if (keywordRegex.test(searchResults.search[0]?.description)) {\n          const entityName = searchResults.search[0].label;\n          recommendations.push(\n            ...facets.map(\n              sf =>\n                ({\n                  label: sf.label?.replace('__QUERY', entityName),\n                  facets: sf.facets.map(f => {\n                    const replaced = {\n                      ...f,\n                      bucketKey: f.bucketKey.replace('__QUERY', query),\n                    };\n\n                    if (f.displayText) {\n                      replaced.displayText = replaced.displayText?.replace(\n                        '__QUERY',\n                        entityName\n                      );\n                    }\n\n                    return replaced;\n                  }),\n                } as SmartFacet)\n            )\n          );\n        }\n      }\n\n      return recommendations;\n    } catch (err) {\n      return [];\n    }\n  }\n}\n"]}

package/dist/src/collection-facets/smart-facets/models.d.ts

@@ -1,26 +1,26 @@
-import type { FacetEventDetails, FacetOption } from '../../models';
-export interface FacetRef {
-    facetType: FacetOption;
-    bucketKey: string;
-    displayText?: string;
-}
-interface LabeledSmartFacet {
-    label: string;
-    facets: FacetRef[];
-    selected?: boolean;
-}
-interface UnlabeledSmartFacet {
-    label?: string;
-    facets: [FacetRef];
-    selected?: boolean;
-}
-export type SmartFacet = LabeledSmartFacet | UnlabeledSmartFacet;
-export interface SmartFacetEvent {
-    smartFacet: SmartFacet;
-    details: FacetEventDetails[];
-}
-export type KeywordFacetMap = Record<string, SmartFacet[]>;
-export interface SmartQueryHeuristic {
-    getRecommendedFacets(query: string): Promise<SmartFacet[]>;
-}
-export {};
+import type { FacetEventDetails, FacetOption } from '../../models';
+export interface FacetRef {
+    facetType: FacetOption;
+    bucketKey: string;
+    displayText?: string;
+}
+interface LabeledSmartFacet {
+    label: string;
+    facets: FacetRef[];
+    selected?: boolean;
+}
+interface UnlabeledSmartFacet {
+    label?: string;
+    facets: [FacetRef];
+    selected?: boolean;
+}
+export declare type SmartFacet = LabeledSmartFacet | UnlabeledSmartFacet;
+export interface SmartFacetEvent {
+    smartFacet: SmartFacet;
+    details: FacetEventDetails[];
+}
+export declare type KeywordFacetMap = Record<string, SmartFacet[]>;
+export interface SmartQueryHeuristic {
+    getRecommendedFacets(query: string): Promise<SmartFacet[]>;
+}
+export {};

package/dist/src/collection-facets/smart-facets/models.js

@@ -1,2 +1,2 @@
-export {};
+export {};
 //# sourceMappingURL=models.js.map

package/dist/src/collection-facets/smart-facets/smart-facet-bar.d.ts

@@ -1,31 +1,31 @@
-import { LitElement, TemplateResult, CSSResultGroup, PropertyValues } from 'lit';
-import type { Aggregation } from '@internetarchive/search-service';
-import type { CollectionTitles } from '../../data-source/models';
-import type { SelectedFacets } from '../../models';
-import './smart-facet-button';
-import './smart-facet-dropdown';
-export declare class SmartFacetBar extends LitElement {
-    query?: string;
-    aggregations?: Record<string, Aggregation>;
-    selectedFacets?: SelectedFacets;
-    /** The map from collection identifiers to their titles */
-    collectionTitles?: CollectionTitles;
-    filterToggleActive: boolean;
-    private heuristicRecs;
-    private smartFacets;
-    private lastAggregations?;
-    render(): TemplateResult<1>;
-    protected willUpdate(changed: PropertyValues): void;
-    private updateSmartFacets;
-    private makeSmartFacet;
-    private smartFacetButton;
-    private smartFacetDropdown;
-    private get filtersToggleTemplate();
-    private get facetsToDisplay();
-    private toSmartFacet;
-    private facetClicked;
-    private facetDropdownClicked;
-    private onDropdownClick;
-    private filterToggleClicked;
-    static get styles(): CSSResultGroup;
-}
+import { LitElement, TemplateResult, CSSResultGroup, PropertyValues } from 'lit';
+import type { Aggregation } from '@internetarchive/search-service';
+import type { CollectionTitles } from '../../data-source/models';
+import type { SelectedFacets } from '../../models';
+import './smart-facet-button';
+import './smart-facet-dropdown';
+export declare class SmartFacetBar extends LitElement {
+    query?: string;
+    aggregations?: Record<string, Aggregation>;
+    selectedFacets?: SelectedFacets;
+    /** The map from collection identifiers to their titles */
+    collectionTitles?: CollectionTitles;
+    filterToggleActive: boolean;
+    private heuristicRecs;
+    private smartFacets;
+    private lastAggregations?;
+    render(): TemplateResult<1>;
+    protected willUpdate(changed: PropertyValues): void;
+    private updateSmartFacets;
+    private makeSmartFacet;
+    private smartFacetButton;
+    private smartFacetDropdown;
+    private get filtersToggleTemplate();
+    private get facetsToDisplay();
+    private toSmartFacet;
+    private facetClicked;
+    private facetDropdownClicked;
+    private onDropdownClick;
+    private filterToggleClicked;
+    static get styles(): CSSResultGroup;
+}