wiki-entity 0.6.1 → 0.7.0

package/README.md

@@ -14,6 +14,24 @@ getEntities({ language: 'en', titles: ['Europe'] }).then(entities => {});
 getEntities({ language: 'en', ids: ['Q46'] }).then(entities => {});
 ```
 
+## User-Agent
+
+[Wikimedia's User-Agent policy](https://meta.wikimedia.org/wiki/User-Agent_policy) requires every client to send an identifying `User-Agent` header. Generic UAs are rate-limited or blocked (HTTP `429`).
+
+Since `0.7.0`, `wiki-entity` sends a default UA built from `package.json`, but you should override it with your app's name and a contact URL/email:
+
+```ts
+import { setUserAgent } from "wiki-entity";
+
+setUserAgent("MyApp/1.0 (https://myapp.example; contact@myapp.example)");
+```
+
+Or set it via environment variable:
+
+```
+WIKI_ENTITY_USER_AGENT="MyApp/1.0 (https://myapp.example; contact@myapp.example)"
+```
+
 ## API
 
 ### getEntities(params): Promise<WikiEntity[]>

package/lib/index.js

@@ -1,126 +1,129 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-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; } });
-const utils_1 = require("./utils");
-const get_entity_types_1 = require("./wikidata/get_entity_types");
-var simplify_entity_1 = require("./wikidata/simplify_entity");
-Object.defineProperty(exports, "simplifyEntity", { enumerable: true, get: function () { return simplify_entity_1.simplifyEntity; } });
-__exportStar(require("./simpleEntity"), exports);
-__exportStar(require("./types"), exports);
-function getEntities(params) {
-    return __awaiter(this, void 0, void 0, function* () {
-        const lang = params.language || "en";
-        const entities = yield (0, wikidata_1.getEntities)(params);
-        const ids = Object.keys(entities).filter((id) => (0, utils_1.isValidWikiId)(id));
-        if (ids.length === 0)
-            return [];
-        const wikiApi = new api_1.Api({}, params.httpTimeout);
-        if (params.extract)
-            wikiApi.extract(params.extract);
-        if (params.redirects)
-            wikiApi.redirects();
-        if (params.categories)
-            wikiApi.categories();
-        const useWikiApi = params.extract ||
-            params.redirects ||
-            params.categories ||
-            params.wikiPageId !== false;
-        const tasks = [];
-        if (useWikiApi && entities[ids[0]] && entities[ids[0]].sitelinks) {
-            const titleIds = ids.reduce((prev, id) => {
-                if (entities[id] &&
-                    entities[id].sitelinks &&
-                    entities[id].sitelinks[lang]) {
-                    prev[entities[id].sitelinks[lang]] = id;
-                }
-                return prev;
-            }, {});
-            tasks.push(wikiApi
-                .query(lang, {
-                titles: ids
-                    .map((id) => (entities[id] &&
-                    entities[id].sitelinks &&
-                    entities[id].sitelinks[lang]) ||
-                    null)
-                    .filter((item) => !!item)
-                    .join("|")
-            })
-                .then((apiResults) => apiResults.forEach((result) => {
-                const entity = entities[titleIds[result.title]];
-                entity.pageid = result.pageid;
-                if (params.extract)
-                    entity.extract = result.extract;
-                if (params.redirects)
-                    entity.redirects = result.redirects;
-                if (params.categories)
-                    entity.categories = result.categories;
-            })));
-        }
-        if (params.types === true || Array.isArray(params.types)) {
-            const prefixes = Array.isArray(params.types)
-                ? params.types
-                : null;
-            ids.forEach((id) => {
-                if (entities[id] && entities[id].sitelinks) {
-                    const enName = entities[id].sitelinks["en"];
-                    if (enName) {
-                        tasks.push((0, get_entity_types_1.getEntityTypesByName)(enName, prefixes).then((types) => {
-                            entities[id].types = types;
-                        }));
-                    }
-                }
-            });
-        }
-        yield Promise.all(tasks);
-        return Object.keys(entities)
-            .map((id) => entities[id])
-            .filter((it) => !!it);
-    });
-}
-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;
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mapRedirects = exports.getEntities = exports.WikipediaApi = exports.getUserAgent = exports.setUserAgent = 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; } });
+const utils_1 = require("./utils");
+const get_entity_types_1 = require("./wikidata/get_entity_types");
+var simplify_entity_1 = require("./wikidata/simplify_entity");
+Object.defineProperty(exports, "simplifyEntity", { enumerable: true, get: function () { return simplify_entity_1.simplifyEntity; } });
+var request_1 = require("./request");
+Object.defineProperty(exports, "setUserAgent", { enumerable: true, get: function () { return request_1.setUserAgent; } });
+Object.defineProperty(exports, "getUserAgent", { enumerable: true, get: function () { return request_1.getUserAgent; } });
+__exportStar(require("./simpleEntity"), exports);
+__exportStar(require("./types"), exports);
+function getEntities(params) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const lang = params.language || "en";
+        const entities = yield (0, wikidata_1.getEntities)(params);
+        const ids = Object.keys(entities).filter((id) => (0, utils_1.isValidWikiId)(id));
+        if (ids.length === 0)
+            return [];
+        const wikiApi = new api_1.Api({}, params.httpTimeout);
+        if (params.extract)
+            wikiApi.extract(params.extract);
+        if (params.redirects)
+            wikiApi.redirects();
+        if (params.categories)
+            wikiApi.categories();
+        const useWikiApi = params.extract ||
+            params.redirects ||
+            params.categories ||
+            params.wikiPageId !== false;
+        const tasks = [];
+        if (useWikiApi && entities[ids[0]] && entities[ids[0]].sitelinks) {
+            const titleIds = ids.reduce((prev, id) => {
+                if (entities[id] &&
+                    entities[id].sitelinks &&
+                    entities[id].sitelinks[lang]) {
+                    prev[entities[id].sitelinks[lang]] = id;
+                }
+                return prev;
+            }, {});
+            tasks.push(wikiApi
+                .query(lang, {
+                titles: ids
+                    .map((id) => (entities[id] &&
+                    entities[id].sitelinks &&
+                    entities[id].sitelinks[lang]) ||
+                    null)
+                    .filter((item) => !!item)
+                    .join("|")
+            })
+                .then((apiResults) => apiResults.forEach((result) => {
+                const entity = entities[titleIds[result.title]];
+                entity.pageid = result.pageid;
+                if (params.extract)
+                    entity.extract = result.extract;
+                if (params.redirects)
+                    entity.redirects = result.redirects;
+                if (params.categories)
+                    entity.categories = result.categories;
+            })));
+        }
+        if (params.types === true || Array.isArray(params.types)) {
+            const prefixes = Array.isArray(params.types)
+                ? params.types
+                : null;
+            ids.forEach((id) => {
+                if (entities[id] && entities[id].sitelinks) {
+                    const enName = entities[id].sitelinks["en"];
+                    if (enName) {
+                        tasks.push((0, get_entity_types_1.getEntityTypesByName)(enName, prefixes).then((types) => {
+                            entities[id].types = types;
+                        }));
+                    }
+                }
+            });
+        }
+        yield Promise.all(tasks);
+        return Object.keys(entities)
+            .map((id) => entities[id])
+            .filter((it) => !!it);
+    });
+}
+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/request.js

@@ -1,19 +1,37 @@
-"use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const axios_1 = require("axios");
-function default_1(url, options) {
-    return __awaiter(this, void 0, void 0, function* () {
-        const response = yield (0, axios_1.default)(url, Object.assign(Object.assign({ method: "GET", responseType: "json", headers: { "User-Agent": "entity-finder" } }, options), { timeout: options.timeout || 10 * 1000 }));
-        return response.data;
-    });
-}
-exports.default = default_1;
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getUserAgent = exports.setUserAgent = void 0;
+const axios_1 = require("axios");
+const pkg = require("../package.json");
+const DEFAULT_USER_AGENT = process.env.WIKI_ENTITY_USER_AGENT ||
+    `${pkg.name}/${pkg.version} (${pkg.homepage ||
+        (pkg.bugs && pkg.bugs.url) ||
+        "https://github.com/entitizer/wiki-entity"})`;
+let userAgent = DEFAULT_USER_AGENT;
+function setUserAgent(ua) {
+    if (!ua || typeof ua !== "string") {
+        throw new Error("User-Agent must be a non-empty string");
+    }
+    userAgent = ua;
+}
+exports.setUserAgent = setUserAgent;
+function getUserAgent() {
+    return userAgent;
+}
+exports.getUserAgent = getUserAgent;
+function request(url, options) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const response = yield (0, axios_1.default)(url, Object.assign(Object.assign({ method: "GET", responseType: "json", timeout: 15 * 1000 }, options), { headers: Object.assign({ "User-Agent": userAgent, "Accept-Encoding": "gzip" }, (options.headers || {})) }));
+        return response.data;
+    });
+}
+exports.default = request;

package/lib/simpleEntity/convertToSimpleEntity.js

@@ -1,53 +1,53 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.convertToSimpleEntity = void 0;
-const getEntityType_1 = require("./getEntityType");
-const getEntityInstanceType_1 = require("./getEntityInstanceType");
-const getEntityTypeByExtract_1 = require("./getEntityTypeByExtract");
-const getEntityData_1 = require("./getEntityData");
-const getEntityCountry_1 = require("./getEntityCountry");
-const utils_1 = require("../utils");
-function convertToSimpleEntity(wikiEntity, lang, options) {
-    options = options || {};
-    const entity = {};
-    entity.lang = lang.toLowerCase();
-    entity.wikiDataId = wikiEntity.id;
-    entity.name = wikiEntity.label;
-    entity.description = wikiEntity.description;
-    entity.wikiPageId = wikiEntity.pageid;
-    entity.about = wikiEntity.extract;
-    if (wikiEntity.redirectsToId)
-        entity.redirectsToId = wikiEntity.redirectsToId;
-    if (wikiEntity.redirectsFromId)
-        entity.redirectsFromId = wikiEntity.redirectsFromId;
-    if (wikiEntity.types) {
-        entity.types = (0, utils_1.uniq)(wikiEntity.types.filter((item) => !/:(Thing|Agent)$/.test(item)));
-    }
-    entity.type = (0, getEntityType_1.getEntityType)(wikiEntity);
-    if (!entity.type) {
-        entity.type = (0, getEntityInstanceType_1.getEntityType)(wikiEntity);
-        if (!entity.type && entity.about && !entity.types) {
-            entity.type = (0, getEntityTypeByExtract_1.getEntityTypeByExtract)(entity.about, lang);
-        }
-        if (!entity.type && options.defaultType) {
-            entity.type = options.defaultType;
-        }
-    }
-    if (wikiEntity.sitelinks) {
-        entity.wikiPageTitle = wikiEntity.sitelinks[lang];
-    }
-    entity.categories = createCategories(wikiEntity);
-    if (wikiEntity.claims) {
-        const ids = Object.keys(wikiEntity.claims);
-        if (ids.length) {
-            entity.data = (0, getEntityData_1.getEntityData)(wikiEntity);
-        }
-        entity.countryCodes = (0, getEntityCountry_1.getEntityCountryCode)(wikiEntity);
-    }
-    return entity;
-}
-exports.convertToSimpleEntity = convertToSimpleEntity;
-function createCategories(entity) {
-    let categories = entity.categories || [];
-    return (0, utils_1.uniq)(categories);
-}
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.convertToSimpleEntity = void 0;
+const getEntityType_1 = require("./getEntityType");
+const getEntityInstanceType_1 = require("./getEntityInstanceType");
+const getEntityTypeByExtract_1 = require("./getEntityTypeByExtract");
+const getEntityData_1 = require("./getEntityData");
+const getEntityCountry_1 = require("./getEntityCountry");
+const utils_1 = require("../utils");
+function convertToSimpleEntity(wikiEntity, lang, options) {
+    options = options || {};
+    const entity = {};
+    entity.lang = lang.toLowerCase();
+    entity.wikiDataId = wikiEntity.id;
+    entity.name = wikiEntity.label;
+    entity.description = wikiEntity.description;
+    entity.wikiPageId = wikiEntity.pageid;
+    entity.about = wikiEntity.extract;
+    if (wikiEntity.redirectsToId)
+        entity.redirectsToId = wikiEntity.redirectsToId;
+    if (wikiEntity.redirectsFromId)
+        entity.redirectsFromId = wikiEntity.redirectsFromId;
+    if (wikiEntity.types) {
+        entity.types = (0, utils_1.uniq)(wikiEntity.types.filter((item) => !/:(Thing|Agent)$/.test(item)));
+    }
+    entity.type = (0, getEntityType_1.getEntityType)(wikiEntity);
+    if (!entity.type) {
+        entity.type = (0, getEntityInstanceType_1.getEntityType)(wikiEntity);
+        if (!entity.type && entity.about && !entity.types) {
+            entity.type = (0, getEntityTypeByExtract_1.getEntityTypeByExtract)(entity.about, lang);
+        }
+        if (!entity.type && options.defaultType) {
+            entity.type = options.defaultType;
+        }
+    }
+    if (wikiEntity.sitelinks) {
+        entity.wikiPageTitle = wikiEntity.sitelinks[lang];
+    }
+    entity.categories = createCategories(wikiEntity);
+    if (wikiEntity.claims) {
+        const ids = Object.keys(wikiEntity.claims);
+        if (ids.length) {
+            entity.data = (0, getEntityData_1.getEntityData)(wikiEntity);
+        }
+        entity.countryCodes = (0, getEntityCountry_1.getEntityCountryCode)(wikiEntity);
+    }
+    return entity;
+}
+exports.convertToSimpleEntity = convertToSimpleEntity;
+function createCategories(entity) {
+    let categories = entity.categories || [];
+    return (0, utils_1.uniq)(categories);
+}

package/lib/simpleEntity/getEntityCountry.js

@@ -1,25 +1,25 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getEntityCountryCode = void 0;
-const countries = require("../../data/countries.json");
-function getEntityCountryCode(wikiEntity) {
-    if (!(wikiEntity === null || wikiEntity === void 0 ? void 0 : wikiEntity.claims)) {
-        return null;
-    }
-    const countryIds = getEntityCountryIds(wikiEntity);
-    const list = [];
-    for (let id of countryIds) {
-        if (countries[id]) {
-            list.push(countries[id].cc2.toLowerCase());
-        }
-    }
-    return list.length ? list : null;
-}
-exports.getEntityCountryCode = getEntityCountryCode;
-function getEntityCountryIds(wikiEntity) {
-    let prop = wikiEntity.claims.P17 ||
-        wikiEntity.claims.P27 ||
-        wikiEntity.claims.P495 ||
-        wikiEntity.claims.P1532;
-    return (prop && prop.values.map((item) => item.value)) || [];
-}
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getEntityCountryCode = void 0;
+const countries = require("../../data/countries.json");
+function getEntityCountryCode(wikiEntity) {
+    if (!(wikiEntity === null || wikiEntity === void 0 ? void 0 : wikiEntity.claims)) {
+        return null;
+    }
+    const countryIds = getEntityCountryIds(wikiEntity);
+    const list = [];
+    for (let id of countryIds) {
+        if (countries[id]) {
+            list.push(countries[id].cc2.toLowerCase());
+        }
+    }
+    return list.length ? list : null;
+}
+exports.getEntityCountryCode = getEntityCountryCode;
+function getEntityCountryIds(wikiEntity) {
+    let prop = wikiEntity.claims.P17 ||
+        wikiEntity.claims.P27 ||
+        wikiEntity.claims.P495 ||
+        wikiEntity.claims.P1532;
+    return (prop && prop.values.map((item) => item.value)) || [];
+}

package/lib/simpleEntity/getEntityData.js

@@ -1,27 +1,27 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getEntityData = void 0;
-function getEntityData(wikiEntity) {
-    if (!wikiEntity.claims) {
-        return null;
-    }
-    const data = {};
-    for (let key in wikiEntity.claims) {
-        const values = wikiEntity.claims[key].values;
-        data[key] = [];
-        values.forEach((value) => {
-            if (!value.value_string && value.value === null) {
-                throw new Error(`WikiEntity claim value cannot be null: ${wikiEntity.id}:${key}`);
-            }
-            const v = {
-                value: value.value_string || value.value.toString()
-            };
-            if (value.label && v.value !== value.label) {
-                v.label = value.label;
-            }
-            data[key].push(v.value);
-        });
-    }
-    return data;
-}
-exports.getEntityData = getEntityData;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getEntityData = void 0;
+function getEntityData(wikiEntity) {
+    if (!wikiEntity.claims) {
+        return null;
+    }
+    const data = {};
+    for (let key in wikiEntity.claims) {
+        const values = wikiEntity.claims[key].values;
+        data[key] = [];
+        values.forEach((value) => {
+            if (!value.value_string && value.value === null) {
+                throw new Error(`WikiEntity claim value cannot be null: ${wikiEntity.id}:${key}`);
+            }
+            const v = {
+                value: value.value_string || value.value.toString()
+            };
+            if (value.label && v.value !== value.label) {
+                v.label = value.label;
+            }
+            data[key].push(v.value);
+        });
+    }
+    return data;
+}
+exports.getEntityData = getEntityData;

package/lib/simpleEntity/getEntityInstanceType.js

@@ -1,32 +1,32 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getEntityType = void 0;
-const simpleEntity_1 = require("./simpleEntity");
-function getEntityType(wikiEntity) {
-    if (!wikiEntity.claims) {
-        return null;
-    }
-    const type = getTypeByProp(wikiEntity, 'P31');
-    if (type) {
-        return type;
-    }
-}
-exports.getEntityType = getEntityType;
-function getTypeByProp(wikiEntity, prop) {
-    const instanceOf = wikiEntity.claims[prop];
-    if (!instanceOf) {
-        return null;
-    }
-    const ENTITY_TYPES = require('../../data/entity_types.json');
-    const types = simpleEntity_1.SIMPLE_ENTITY_TYPES;
-    for (let i = 0; i < types.length; i++) {
-        const type = types[i];
-        for (let j = 0; j < instanceOf.values.length; j++) {
-            const value = instanceOf.values[j].value;
-            if (~ENTITY_TYPES[type].indexOf(value)) {
-                return type;
-            }
-        }
-    }
-    return null;
-}
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getEntityType = void 0;
+const simpleEntity_1 = require("./simpleEntity");
+function getEntityType(wikiEntity) {
+    if (!wikiEntity.claims) {
+        return null;
+    }
+    const type = getTypeByProp(wikiEntity, 'P31');
+    if (type) {
+        return type;
+    }
+}
+exports.getEntityType = getEntityType;
+function getTypeByProp(wikiEntity, prop) {
+    const instanceOf = wikiEntity.claims[prop];
+    if (!instanceOf) {
+        return null;
+    }
+    const ENTITY_TYPES = require('../../data/entity_types.json');
+    const types = simpleEntity_1.SIMPLE_ENTITY_TYPES;
+    for (let i = 0; i < types.length; i++) {
+        const type = types[i];
+        for (let j = 0; j < instanceOf.values.length; j++) {
+            const value = instanceOf.values[j].value;
+            if (~ENTITY_TYPES[type].indexOf(value)) {
+                return type;
+            }
+        }
+    }
+    return null;
+}