wiki-entity 0.5.12 → 0.6.1

package/lib/index.js

@@ -19,7 +19,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getEntities = exports.WikipediaApi = exports.simplifyEntity = void 0;
+exports.mapRedirects = exports.getEntities = exports.WikipediaApi = exports.simplifyEntity = void 0;
 const wikidata_1 = require("./wikidata");
 const api_1 = require("./wikipedia/api");
 Object.defineProperty(exports, "WikipediaApi", { enumerable: true, get: function () { return api_1.Api; } });
@@ -100,3 +100,27 @@ function getEntities(params) {
     });
 }
 exports.getEntities = getEntities;
+function mapRedirects(titles, lang) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const wikiApi = new api_1.Api();
+        return wikiApi
+            .redirects()
+            .query(lang, {
+            titles: titles.map((it) => it.replace(/\s+/g, "_")).join("|"),
+            redirects: "yes"
+        })
+            .then((apiResults) => {
+            const result = {};
+            apiResults.forEach((r) => {
+                var _a;
+                if ((_a = r.redirects) === null || _a === void 0 ? void 0 : _a.length) {
+                    const it = r.redirects.find((it) => titles.includes(it));
+                    if (it && it !== r.title)
+                        result[it] = r.title;
+                }
+            });
+            return result;
+        });
+    });
+}
+exports.mapRedirects = mapRedirects;

package/lib/wikidata/index.js

@@ -62,7 +62,7 @@ function exploreEntitiesProperties(entities, lang) {
                 }
             });
         });
-        return null;
+        return entities;
     });
 }
 function exploreEntityClaims(claims, params) {

package/lib/wikipedia/api.js

@@ -33,19 +33,19 @@ class Api {
             if (data && data.query && data.query.pages) {
                 return Object.keys(data.query.pages)
                     .map((id) => data.query.pages[id])
-                    .map((data) => {
+                    .map((page) => {
                     const item = {
-                        pageid: data.pageid,
-                        title: data.title
+                        pageid: page.pageid,
+                        title: page.title
                     };
-                    if (data["categories"]) {
-                        item.categories = data["categories"].map((it) => it.title);
+                    if (page["categories"]) {
+                        item.categories = page["categories"].map((it) => it.title);
                     }
-                    if (data["redirects"]) {
-                        item.redirects = data["redirects"].map((it) => it.title);
+                    if (page["redirects"]) {
+                        item.redirects = page["redirects"].map((it) => it.title);
                     }
-                    if (data["extract"]) {
-                        item.extract = data["extract"];
+                    if (page["extract"]) {
+                        item.extract = page["extract"];
                     }
                     return item;
                 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wiki-entity",
-  "version": "0.5.12",
+  "version": "0.6.1",
   "description": "Wiki entity fetcher",
   "main": "./lib/index.js",
   "types": "./types/index.d.ts",

package/types/index.d.ts

@@ -5,3 +5,4 @@ export { WikipediaApi };
 export * from "./simpleEntity";
 export * from "./types";
 export declare function getEntities(params: WikiEntitiesParams): Promise<WikiEntity[]>;
+export declare function mapRedirects(titles: string[], lang: string): Promise<Record<string, string>>;