@internetarchive/collection-browser 2.8.1-alpha-webdev7002.2 → 2.9.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/src/collection-facets/smart-facets/dedupe.js +2 -1
- package/dist/src/collection-facets/smart-facets/dedupe.js.map +1 -1
- package/dist/src/collection-facets/smart-facets/heuristics/query-keywords/query-keywords-map.js +24 -8
- package/dist/src/collection-facets/smart-facets/heuristics/query-keywords/query-keywords-map.js.map +1 -1
- package/dist/src/collection-facets/smart-facets/heuristics/wikidata/wikidata-heuristic.js +2 -8
- package/dist/src/collection-facets/smart-facets/heuristics/wikidata/wikidata-heuristic.js.map +1 -1
- package/package.json +1 -1
- package/src/collection-facets/smart-facets/dedupe.ts +2 -1
- package/src/collection-facets/smart-facets/heuristics/query-keywords/query-keywords-map.ts +24 -8
- package/src/collection-facets/smart-facets/heuristics/wikidata/wikidata-heuristic.ts +0 -6
|
@@ -11,7 +11,8 @@ export function dedupe(facets) {
|
|
|
11
11
|
if (!Array.isArray(facets[0])) {
|
|
12
12
|
const facetsUnnested = facets;
|
|
13
13
|
let result = [...facetsUnnested];
|
|
14
|
-
for (
|
|
14
|
+
for (let i = 0; i < result.length; i++) {
|
|
15
|
+
const curFacet = result[i];
|
|
15
16
|
result = result.filter(sf => curFacet === sf || !smartFacetEquals(curFacet, sf));
|
|
16
17
|
}
|
|
17
18
|
return result;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"dedupe.js","sourceRoot":"","sources":["../../../../src/collection-facets/smart-facets/dedupe.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAExD;;;;;;;GAOG;AACH,MAAM,UAAU,MAAM,CAA0C,MAAS;IACvE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9B,MAAM,cAAc,GAAG,MAAsB,CAAC;QAE9C,IAAI,MAAM,GAAiB,CAAC,GAAG,cAAc,CAAC,CAAC;QAC/C,KAAK,MAAM,
|
|
1
|
+
{"version":3,"file":"dedupe.js","sourceRoot":"","sources":["../../../../src/collection-facets/smart-facets/dedupe.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAExD;;;;;;;GAOG;AACH,MAAM,UAAU,MAAM,CAA0C,MAAS;IACvE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9B,MAAM,cAAc,GAAG,MAAsB,CAAC;QAE9C,IAAI,MAAM,GAAiB,CAAC,GAAG,cAAc,CAAC,CAAC;QAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAC3B,MAAM,GAAG,MAAM,CAAC,MAAM,CACpB,EAAE,CAAC,EAAE,CAAC,QAAQ,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,EAAE,CAAC,CACzD,CAAC;QACJ,CAAC;QAED,OAAO,MAAW,CAAC;IACrB,CAAC;IAED,MAAM,YAAY,GAAG,MAAwB,CAAC;IAE9C,MAAM,MAAM,GAAmB,EAAE,CAAC;IAClC,KAAK,MAAM,aAAa,IAAI,YAAY,EAAE,CAAC;QACzC,MAAM,SAAS,GAAiB,EAAE,CAAC;QACnC,KAAK,MAAM,QAAQ,IAAI,aAAa,EAAE,CAAC;YACrC,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CACjC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,gBAAgB,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAC/C,CAAC;YACF,IAAI,CAAC,QAAQ;gBAAE,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1C,CAAC;QACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,OAAO,MAAW,CAAC;AACrB,CAAC","sourcesContent":["import type { SmartFacet } from './models';\nimport { smartFacetEquals } from './smart-facet-equals';\n\n/**\n * Removes any duplicated smart facets from the given array.\n * Smart facets are equal if they have the same `label` and same\n * set of facet refs. Only the first occurrence of a given smart\n * facet is kept.\n * @param facets The array of smart facets to deduplicate\n * @returns A new array containing the deduplicated set of facets\n */\nexport function dedupe<T extends SmartFacet[] | SmartFacet[][]>(facets: T): T {\n if (!Array.isArray(facets[0])) {\n const facetsUnnested = facets as SmartFacet[];\n\n let result: SmartFacet[] = [...facetsUnnested];\n for (let i = 0; i < result.length; i++) {\n const curFacet = result[i];\n result = result.filter(\n sf => curFacet === sf || !smartFacetEquals(curFacet, sf),\n );\n }\n\n return result as T;\n }\n\n const facetsNested = facets as SmartFacet[][];\n\n const result: SmartFacet[][] = [];\n for (const curFacetArray of facetsNested) {\n const subresult: SmartFacet[] = [];\n for (const curFacet of curFacetArray) {\n const existing = result.find(sfa =>\n sfa.find(sf => smartFacetEquals(curFacet, sf)),\n );\n if (!existing) subresult.push(curFacet);\n }\n if (subresult.length > 0) {\n result.push(subresult);\n }\n }\n\n return result as T;\n}\n"]}
|
package/dist/src/collection-facets/smart-facets/heuristics/query-keywords/query-keywords-map.js
CHANGED
|
@@ -1,11 +1,27 @@
|
|
|
1
|
-
const TEXTS = {
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
const
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
const
|
|
8
|
-
|
|
1
|
+
const TEXTS = {
|
|
2
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'texts' }],
|
|
3
|
+
};
|
|
4
|
+
const AUDIO = {
|
|
5
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'audio' }],
|
|
6
|
+
};
|
|
7
|
+
const MOVIES = {
|
|
8
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'movies' }],
|
|
9
|
+
};
|
|
10
|
+
const IMAGE = {
|
|
11
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'image' }],
|
|
12
|
+
};
|
|
13
|
+
const SOFTWARE = {
|
|
14
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'software' }],
|
|
15
|
+
};
|
|
16
|
+
const ETREE = {
|
|
17
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'etree' }],
|
|
18
|
+
};
|
|
19
|
+
const WEB = {
|
|
20
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'web' }],
|
|
21
|
+
};
|
|
22
|
+
const DATA = {
|
|
23
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'data' }],
|
|
24
|
+
};
|
|
9
25
|
/**
|
|
10
26
|
* Map from keywords found in the search query to an array of
|
|
11
27
|
* likely-relevant "smart facets" for those keywords.
|
package/dist/src/collection-facets/smart-facets/heuristics/query-keywords/query-keywords-map.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"query-keywords-map.js","sourceRoot":"","sources":["../../../../../../src/collection-facets/smart-facets/heuristics/query-keywords/query-keywords-map.ts"],"names":[],"mappings":"AAEA,MAAM,KAAK,GAAe,
|
|
1
|
+
{"version":3,"file":"query-keywords-map.js","sourceRoot":"","sources":["../../../../../../src/collection-facets/smart-facets/heuristics/query-keywords/query-keywords-map.ts"],"names":[],"mappings":"AAEA,MAAM,KAAK,GAAe;IACxB,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;CACzD,CAAC;AACF,MAAM,KAAK,GAAe;IACxB,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;CACzD,CAAC;AACF,MAAM,MAAM,GAAe;IACzB,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;CAC1D,CAAC;AACF,MAAM,KAAK,GAAe;IACxB,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;CACzD,CAAC;AACF,MAAM,QAAQ,GAAe;IAC3B,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;CAC5D,CAAC;AACF,MAAM,KAAK,GAAe;IACxB,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;CACzD,CAAC;AACF,MAAM,GAAG,GAAe;IACtB,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;CACvD,CAAC;AACF,MAAM,IAAI,GAAe;IACvB,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;CACxD,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,cAAc,GAA8B;IACvD,IAAI,EAAE,CAAC,KAAK,CAAC;IACb,IAAI,EAAE,CAAC,KAAK,CAAC;IACb,KAAK,EAAE,CAAC,KAAK,CAAC;IACd,QAAQ,EAAE,CAAC,KAAK,CAAC;IACjB,SAAS,EAAE,CAAC,KAAK,CAAC;IAClB,GAAG,EAAE,CAAC,KAAK,CAAC;IACZ,IAAI,EAAE,CAAC,KAAK,CAAC;IACb,KAAK,EAAE,CAAC,KAAK,CAAC;IACd,IAAI,EAAE,CAAC,KAAK,CAAC;IACb,KAAK,EAAE,CAAC,KAAK,CAAC;IACd,MAAM,EAAE,CAAC,KAAK,CAAC;IACf,OAAO,EAAE,CAAC,KAAK,CAAC;IAChB,KAAK,EAAE,CAAC,KAAK,CAAC;IACd,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,KAAK,EAAE,CAAC,MAAM,CAAC;IACf,KAAK,EAAE,CAAC,MAAM,CAAC;IACf,IAAI,EAAE,CAAC,MAAM,CAAC;IACd,SAAS,EAAE,CAAC,MAAM,CAAC;IACnB,OAAO,EAAE,CAAC,MAAM,CAAC;IACjB,KAAK,EAAE,CAAC,KAAK,CAAC;IACd,KAAK,EAAE,CAAC,KAAK,CAAC;IACd,OAAO,EAAE,CAAC,KAAK,CAAC;IAChB,QAAQ,EAAE,CAAC,KAAK,CAAC;IACjB,GAAG,EAAE,CAAC,KAAK,CAAC;IACZ,IAAI,EAAE,CAAC,KAAK,CAAC;IACb,GAAG,EAAE,CAAC,KAAK,CAAC;IACZ,GAAG,EAAE,CAAC,KAAK,CAAC;IACZ,QAAQ,EAAE,CAAC,QAAQ,CAAC;IACpB,GAAG,EAAE,CAAC,QAAQ,CAAC;IACf,OAAO,EAAE,CAAC,QAAQ,CAAC;IACnB,IAAI,EAAE,CAAC,QAAQ,CAAC;IAChB,MAAM,EAAE,CAAC,QAAQ,CAAC;IAClB,KAAK,EAAE,CAAC,KAAK,CAAC;IACd,OAAO,EAAE,CAAC,KAAK,CAAC;IAChB,YAAY,EAAE,CAAC,KAAK,CAAC;IACrB,WAAW,EAAE,CAAC,GAAG,CAAC;IAClB,OAAO,EAAE,CAAC,IAAI,CAAC;CAChB,CAAC","sourcesContent":["import type { KeywordFacetMap, SmartFacet } from '../../models';\n\nconst TEXTS: SmartFacet = {\n facets: [{ facetType: 'mediatype', bucketKey: 'texts' }],\n};\nconst AUDIO: SmartFacet = {\n facets: [{ facetType: 'mediatype', bucketKey: 'audio' }],\n};\nconst MOVIES: SmartFacet = {\n facets: [{ facetType: 'mediatype', bucketKey: 'movies' }],\n};\nconst IMAGE: SmartFacet = {\n facets: [{ facetType: 'mediatype', bucketKey: 'image' }],\n};\nconst SOFTWARE: SmartFacet = {\n facets: [{ facetType: 'mediatype', bucketKey: 'software' }],\n};\nconst ETREE: SmartFacet = {\n facets: [{ facetType: 'mediatype', bucketKey: 'etree' }],\n};\nconst WEB: SmartFacet = {\n facets: [{ facetType: 'mediatype', bucketKey: 'web' }],\n};\nconst DATA: SmartFacet = {\n facets: [{ facetType: 'mediatype', bucketKey: 'data' }],\n};\n\n/**\n * Map from keywords found in the search query to an array of\n * likely-relevant \"smart facets\" for those keywords.\n */\nexport const QUERY_KEYWORDS: Readonly<KeywordFacetMap> = {\n text: [TEXTS],\n book: [TEXTS],\n novel: [TEXTS],\n magazine: [TEXTS],\n newspaper: [TEXTS],\n pdf: [TEXTS],\n epub: [TEXTS],\n audio: [AUDIO],\n song: [AUDIO],\n music: [AUDIO],\n listen: [AUDIO],\n podcast: [AUDIO],\n radio: [AUDIO],\n stream: [AUDIO, MOVIES],\n video: [MOVIES],\n movie: [MOVIES],\n film: [MOVIES],\n animation: [MOVIES],\n youtube: [MOVIES],\n image: [IMAGE],\n photo: [IMAGE],\n picture: [IMAGE],\n painting: [IMAGE],\n jpg: [IMAGE],\n jpeg: [IMAGE],\n png: [IMAGE],\n gif: [IMAGE],\n software: [SOFTWARE],\n app: [SOFTWARE],\n program: [SOFTWARE],\n game: [SOFTWARE],\n arcade: [SOFTWARE],\n etree: [ETREE],\n concert: [ETREE],\n 'live music': [ETREE],\n 'web crawl': [WEB],\n dataset: [DATA],\n};\n"]}
|
|
@@ -52,24 +52,18 @@ export class WikidataHeuristic {
|
|
|
52
52
|
* @inheritdoc
|
|
53
53
|
*/
|
|
54
54
|
async getRecommendedFacets(query) {
|
|
55
|
-
var _a
|
|
55
|
+
var _a;
|
|
56
56
|
const recommendations = [];
|
|
57
57
|
try {
|
|
58
58
|
const wikidataURL = this.getWikidataURL(query);
|
|
59
59
|
const wikidataResponse = await fetch(wikidataURL);
|
|
60
60
|
const searchResults = await wikidataResponse.json();
|
|
61
|
-
console.log('wikidata results', searchResults);
|
|
62
61
|
for (const [keyword, facets] of Object.entries(WIKIDATA_ENTITIES)) {
|
|
63
|
-
console.log('testing keyword', keyword);
|
|
64
62
|
const keywordRegex = new RegExp(`\\b${keyword}\\b`);
|
|
65
|
-
|
|
66
|
-
console.log('description', (_a = searchResults.search[0]) === null || _a === void 0 ? void 0 : _a.description);
|
|
67
|
-
console.log('test', keywordRegex.test((_b = searchResults.search[0]) === null || _b === void 0 ? void 0 : _b.description));
|
|
68
|
-
if (keywordRegex.test((_c = searchResults.search[0]) === null || _c === void 0 ? void 0 : _c.description)) {
|
|
63
|
+
if (keywordRegex.test((_a = searchResults.search[0]) === null || _a === void 0 ? void 0 : _a.description)) {
|
|
69
64
|
const entityName = searchResults.search[0].label;
|
|
70
65
|
recommendations.push(...this.replaceQueryPlaceholders(facets, entityName));
|
|
71
66
|
}
|
|
72
|
-
console.log('recs', recommendations);
|
|
73
67
|
}
|
|
74
68
|
return recommendations;
|
|
75
69
|
}
|
package/dist/src/collection-facets/smart-facets/heuristics/wikidata/wikidata-heuristic.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"wikidata-heuristic.js","sourceRoot":"","sources":["../../../../../../src/collection-facets/smart-facets/heuristics/wikidata/wikidata-heuristic.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAE1D,4EAA4E;AAC5E,mBAAmB;AACnB,0BAA0B;AAC1B,2BAA2B;AAC3B,8CAA8C;AAC9C,+CAA+C;AAC/C,8CAA8C;AAC9C,8CAA8C;AAC9C,OAAO;AACP,MAAM,OAAO,iBAAiB;IAA9B;QACmB,sBAAiB,GAAG,oCAAoC,CAAC;QAEzD,0BAAqB,GACpC,wFAAwF,CAAC;
|
|
1
|
+
{"version":3,"file":"wikidata-heuristic.js","sourceRoot":"","sources":["../../../../../../src/collection-facets/smart-facets/heuristics/wikidata/wikidata-heuristic.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAE1D,4EAA4E;AAC5E,mBAAmB;AACnB,0BAA0B;AAC1B,2BAA2B;AAC3B,8CAA8C;AAC9C,+CAA+C;AAC/C,8CAA8C;AAC9C,8CAA8C;AAC9C,OAAO;AACP,MAAM,OAAO,iBAAiB;IAA9B;QACmB,sBAAiB,GAAG,oCAAoC,CAAC;QAEzD,0BAAqB,GACpC,wFAAwF,CAAC;IAyE7F,CAAC;IAvEC;;;OAGG;IACK,cAAc,CAAC,KAAa;QAClC,MAAM,QAAQ,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAC3C,OAAO,GAAG,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,qBAAqB,WAAW,QAAQ,EAAE,CAAC;IACrF,CAAC;IAED;;;;;OAKG;IACK,wBAAwB,CAC9B,WAAyB,EACzB,KAAa;QAEb,OAAO,WAAW,CAAC,GAAG,CACpB,UAAU,CAAC,EAAE;;YACX,OAAA,CAAC;gBACC,oDAAoD;gBACpD,KAAK,EAAE,MAAA,UAAU,CAAC,KAAK,0CAAE,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC;gBAClD,6EAA6E;gBAC7E,MAAM,EAAE,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;;oBACpC,MAAM,QAAQ,GAAG;wBACf,GAAG,KAAK;wBACR,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC;qBACrD,CAAC;oBAEF,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;wBACtB,QAAQ,CAAC,WAAW,GAAG,MAAA,QAAQ,CAAC,WAAW,0CAAE,OAAO,CAClD,SAAS,EACT,KAAK,CACN,CAAC;oBACJ,CAAC;oBAED,OAAO,QAAQ,CAAC;gBAClB,CAAC,CAAC;aACH,CAAe,CAAA;SAAA,CACnB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB,CAAC,KAAa;;QACtC,MAAM,eAAe,GAAiB,EAAE,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAC/C,MAAM,gBAAgB,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC;YAClD,MAAM,aAAa,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAEpD,KAAK,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBAClE,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,IAAI,CAAC,wBAAwB,CAAC,MAAM,EAAE,UAAU,CAAC,CACrD,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,OAAO,eAAe,CAAC;QACzB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAClB,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CACF","sourcesContent":["import type { SmartQueryHeuristic, SmartFacet } from '../../models';\nimport { WIKIDATA_ENTITIES } from './wikidata-entity-map';\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 readonly WIKIDATA_BASE_URL = 'https://www.wikidata.org/w/api.php';\n\n private readonly WIKIDATA_DEFAULT_ARGS =\n '?action=wbsearchentities&format=json&language=en&uselang=en&origin=*&type=item&limit=5';\n\n /**\n * Returns the full URL for a Wikidata search for the given query\n * @param query The query to search for\n */\n private getWikidataURL(query: string): string {\n const urlQuery = encodeURIComponent(query);\n return `${this.WIKIDATA_BASE_URL}${this.WIKIDATA_DEFAULT_ARGS}&search=${urlQuery}`;\n }\n\n /**\n * Replaces query placeholders with an actual query string, within a collection of smart facets\n * @param smartFacets The array of smart facets\n * @param query The query string to replace placeholders with\n * @returns A new array of smart facets with all query placeholders replaced\n */\n private replaceQueryPlaceholders(\n smartFacets: SmartFacet[],\n query: string,\n ): SmartFacet[] {\n return smartFacets.map(\n smartFacet =>\n ({\n // Replace placeholders within the smart facet label\n label: smartFacet.label?.replace('__QUERY', query),\n // Replace placeholders within the facets themselves (buckets & display text)\n facets: smartFacet.facets.map(facet => {\n const replaced = {\n ...facet,\n bucketKey: facet.bucketKey.replace('__QUERY', query),\n };\n\n if (facet.displayText) {\n replaced.displayText = replaced.displayText?.replace(\n '__QUERY',\n query,\n );\n }\n\n return replaced;\n }),\n }) as SmartFacet,\n );\n }\n\n /**\n * @inheritdoc\n */\n async getRecommendedFacets(query: string): Promise<SmartFacet[]> {\n const recommendations: SmartFacet[] = [];\n\n try {\n const wikidataURL = this.getWikidataURL(query);\n const wikidataResponse = await fetch(wikidataURL);\n const searchResults = await wikidataResponse.json();\n\n for (const [keyword, facets] of Object.entries(WIKIDATA_ENTITIES)) {\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 ...this.replaceQueryPlaceholders(facets, entityName),\n );\n }\n }\n\n return recommendations;\n } catch (err) {\n console.warn(err);\n return [];\n }\n }\n}\n"]}
|
package/package.json
CHANGED
|
@@ -14,7 +14,8 @@ export function dedupe<T extends SmartFacet[] | SmartFacet[][]>(facets: T): T {
|
|
|
14
14
|
const facetsUnnested = facets as SmartFacet[];
|
|
15
15
|
|
|
16
16
|
let result: SmartFacet[] = [...facetsUnnested];
|
|
17
|
-
for (
|
|
17
|
+
for (let i = 0; i < result.length; i++) {
|
|
18
|
+
const curFacet = result[i];
|
|
18
19
|
result = result.filter(
|
|
19
20
|
sf => curFacet === sf || !smartFacetEquals(curFacet, sf),
|
|
20
21
|
);
|
|
@@ -1,13 +1,29 @@
|
|
|
1
1
|
import type { KeywordFacetMap, SmartFacet } from '../../models';
|
|
2
2
|
|
|
3
|
-
const TEXTS: SmartFacet = {
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
const
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
const
|
|
10
|
-
|
|
3
|
+
const TEXTS: SmartFacet = {
|
|
4
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'texts' }],
|
|
5
|
+
};
|
|
6
|
+
const AUDIO: SmartFacet = {
|
|
7
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'audio' }],
|
|
8
|
+
};
|
|
9
|
+
const MOVIES: SmartFacet = {
|
|
10
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'movies' }],
|
|
11
|
+
};
|
|
12
|
+
const IMAGE: SmartFacet = {
|
|
13
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'image' }],
|
|
14
|
+
};
|
|
15
|
+
const SOFTWARE: SmartFacet = {
|
|
16
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'software' }],
|
|
17
|
+
};
|
|
18
|
+
const ETREE: SmartFacet = {
|
|
19
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'etree' }],
|
|
20
|
+
};
|
|
21
|
+
const WEB: SmartFacet = {
|
|
22
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'web' }],
|
|
23
|
+
};
|
|
24
|
+
const DATA: SmartFacet = {
|
|
25
|
+
facets: [{ facetType: 'mediatype', bucketKey: 'data' }],
|
|
26
|
+
};
|
|
11
27
|
|
|
12
28
|
/**
|
|
13
29
|
* Map from keywords found in the search query to an array of
|
|
@@ -70,21 +70,15 @@ export class WikidataHeuristic implements SmartQueryHeuristic {
|
|
|
70
70
|
const wikidataURL = this.getWikidataURL(query);
|
|
71
71
|
const wikidataResponse = await fetch(wikidataURL);
|
|
72
72
|
const searchResults = await wikidataResponse.json();
|
|
73
|
-
console.log('wikidata results', searchResults);
|
|
74
73
|
|
|
75
74
|
for (const [keyword, facets] of Object.entries(WIKIDATA_ENTITIES)) {
|
|
76
|
-
console.log('testing keyword', keyword);
|
|
77
75
|
const keywordRegex = new RegExp(`\\b${keyword}\\b`);
|
|
78
|
-
console.log('regex', keywordRegex);
|
|
79
|
-
console.log('description', searchResults.search[0]?.description);
|
|
80
|
-
console.log('test', keywordRegex.test(searchResults.search[0]?.description));
|
|
81
76
|
if (keywordRegex.test(searchResults.search[0]?.description)) {
|
|
82
77
|
const entityName = searchResults.search[0].label;
|
|
83
78
|
recommendations.push(
|
|
84
79
|
...this.replaceQueryPlaceholders(facets, entityName),
|
|
85
80
|
);
|
|
86
81
|
}
|
|
87
|
-
console.log('recs', recommendations);
|
|
88
82
|
}
|
|
89
83
|
|
|
90
84
|
return recommendations;
|