wiki-entity
Version:
Wiki entity fetcher
29 lines (28 loc) • 1.28 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.getEntityTypeByExtract = void 0;
const simpleEntity_1 = require("./simpleEntity");
const MAP = {
ro: [
{ reg: /\beste un (sat|oraș|județ|raion)\b/i, type: simpleEntity_1.SimpleEntityType.PLACE },
{ reg: /\beste o (comună|biserică|mănăstire)\b/i, type: simpleEntity_1.SimpleEntityType.PLACE },
{ reg: /\beste o (organizație)\b/i, type: simpleEntity_1.SimpleEntityType.ORG },
{ reg: /\beste un (om|scriitor|poet|cercetător|politician|businessman|cântăreț|muzician)\b/i, type: simpleEntity_1.SimpleEntityType.PERSON },
{ reg: /\beste o (scriitoare|poetă|cercetătoare|politiciană|cântăreață)\b/i, type: simpleEntity_1.SimpleEntityType.PERSON }
]
};
function getEntityTypeByExtract(extract, lang) {
if (extract) {
extract = extract.substr(0, 100);
const map = MAP[lang];
if (map && map.length) {
for (var i = 0; i < map.length; i++) {
const item = map[i];
if (item.reg.test(extract)) {
return item.type;
}
}
}
}
}
exports.getEntityTypeByExtract = getEntityTypeByExtract;