ournet.data.stories/lib/search.js

Ournet stories data module

/*eslint no-underscore-dangle:1*/

'use strict';

var utils = require('./utils');
var Promise = utils.Promise;
var _ = utils._;
var elasticsearch = require('elasticsearch');
var ES_STORIES_INDEX = 'stories';
var ES_STORIES_TYPE = 'story';

var client = new elasticsearch.Client({
	host: process.env.NEWS_ES_HOST
});

Promise.promisifyAll(client);

// internals

function normalizeStory(data) {
	var story = _.pick(data, 'id', 'title', 'uniqueName', 'country', 'lang', 'createdAt', 'imageId');
	story['title_' + story.lang] = utils.atonic(story.title);

	return story;
}

function getStories(response) {
	var items = [];
	if (response.hits && response.hits.total > 0) {
		response.hits.hits.forEach(function(item) {
			items.push(item._source);
		});
	}

	return items;
}

// exports

/**
 * Create a story in elastic search
 */
exports.createStory = function(data) {
	var story = normalizeStory(data);

	return client.indexAsync({
		index: ES_STORIES_INDEX,
		type: ES_STORIES_TYPE,
		id: story.id,
		body: story,
		ttl: '90d'
	});
};

/**
 * Update a story
 */
exports.updateStory = function(id, data) {
	// console.log('updating story', id, data);
	return client.updateAsync({
		index: ES_STORIES_INDEX,
		type: ES_STORIES_TYPE,
		id: id,
		doc: data,
		body: {
			doc: data
		}
	});
};

/**
 * Search stories
 */
exports.searchStories = function(params) {
	var q = utils.atonic(params.q);
	var body = {
		'query': {
			'filtered': {
				'filter': {
					'bool': {
						'must': [{
							'term': {
								'country': params.country
							}
						}, {
							'term': {
								'lang': params.lang
							}
						}]
					}
				},
				'query': {
					'multi_match': {
						'query': q,
						'fields': ['title', 'title_' + params.lang]
					}
				}
			}
		}
	};

	if (params.min_score) {
		body.min_score = params.min_score;
	}
	// if (params.ignoreId) {
	//   body.query.filtered.filter.not = {
	//     _id: params.ignoreId
	//   };
	// }

	return client.searchAsync({
		index: ES_STORIES_INDEX,
		type: ES_STORIES_TYPE,
		body: body
	}).then(getStories);
};