From e083919b33686722159a70d813e6731f9b2419a4 Mon Sep 17 00:00:00 2001
From: syuilo <syuilotan@yahoo.co.jp>
Date: Wed, 4 Jul 2018 13:21:30 +0900
Subject: [PATCH] wip

---
 elasticsearch/mappings.json              | 19 ++-----
 src/config/types.ts                      |  1 -
 src/index.ts                             |  4 +-
 src/server/api/endpoints/notes/search.ts | 63 ++++++++++++++++++++++++
 src/services/note/create.ts              | 17 ++++++-
 5 files changed, 85 insertions(+), 19 deletions(-)
 create mode 100644 src/server/api/endpoints/notes/search.ts

diff --git a/elasticsearch/mappings.json b/elasticsearch/mappings.json
index 654ab1745..80a007892 100644
--- a/elasticsearch/mappings.json
+++ b/elasticsearch/mappings.json
@@ -35,29 +35,16 @@
 				"bio": {
 					"type": "string",
 					"index": "analyzed",
-					"analyzer": "kuromoji"
+					"analyzer": "bigram"
 				}
 			}
 		},
-		"post": {
+		"note": {
 			"properties": {
 				"text": {
 					"type": "string",
 					"index": "analyzed",
-					"analyzer": "kuromoji"
-				}
-			}
-		},
-		"drive_file": {
-			"properties": {
-				"name": {
-					"type": "string",
-					"index": "analyzed",
-					"analyzer": "kuromoji"
-				},
-				"user": {
-					"type": "string",
-					"index": "not_analyzed"
+					"analyzer": "bigram"
 				}
 			}
 		}
diff --git a/src/config/types.ts b/src/config/types.ts
index 49eeac508..b0776fd9c 100644
--- a/src/config/types.ts
+++ b/src/config/types.ts
@@ -34,7 +34,6 @@ export type Source = {
 		pass: string;
 	};
 	elasticsearch: {
-		enable: boolean;
 		host: string;
 		port: number;
 		pass: string;
diff --git a/src/index.ts b/src/index.ts
index c89252dfc..1a7604457 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -19,7 +19,7 @@ import MachineInfo from './utils/machineInfo';
 import DependencyInfo from './utils/dependencyInfo';
 import serverStats from './daemons/server-stats';
 import notesStats from './daemons/notes-stats';
-
+import db from './db/mongodb';
 import loadConfig from './config/load';
 import { Config } from './config/types';
 
@@ -204,4 +204,6 @@ process.on('uncaughtException', err => {
 // Dying away...
 process.on('exit', code => {
 	Logger.info(`The process is going exit (${code})`);
+
+	db.close();
 });
diff --git a/src/server/api/endpoints/notes/search.ts b/src/server/api/endpoints/notes/search.ts
new file mode 100644
index 000000000..20c628b84
--- /dev/null
+++ b/src/server/api/endpoints/notes/search.ts
@@ -0,0 +1,63 @@
+import $ from 'cafy';
+import * as mongo from 'mongodb';
+import Note from '../../../../models/note';
+import { ILocalUser } from '../../../../models/user';
+import { pack } from '../../../../models/note';
+import es from '../../../../db/elasticsearch';
+
+module.exports = (params: any, me: ILocalUser) => new Promise(async (res, rej) => {
+	// Get 'query' parameter
+	const [query, queryError] = $.str.get(params.query);
+	if (queryError) return rej('invalid query param');
+
+	// Get 'offset' parameter
+	const [offset = 0, offsetErr] = $.num.optional().min(0).get(params.offset);
+	if (offsetErr) return rej('invalid offset param');
+
+	// Get 'limit' parameter
+	const [limit = 10, limitErr] = $.num.optional().range(1, 30).get(params.limit);
+	if (limitErr) return rej('invalid limit param');
+
+	es.search({
+		index: 'misskey',
+		type: 'note',
+		body: {
+			size: limit,
+			from: offset,
+			query: {
+				simple_query_string: {
+					fields: ['text'],
+					query: query,
+					default_operator: 'and'
+				}
+			},
+			sort: [
+				{ _doc: 'desc' }
+			]
+		}
+	}, async (error, response) => {
+		if (error) {
+			console.error(error);
+			return res(500);
+		}
+
+		if (response.hits.total === 0) {
+			return res([]);
+		}
+
+		const hits = response.hits.hits.map(hit => new mongo.ObjectID(hit._id));
+
+		// Fetch found notes
+		const notes = await Note.find({
+			_id: {
+				$in: hits
+			}
+		}, {
+			sort: {
+				_id: -1
+			}
+		});
+
+		res(await Promise.all(notes.map(note => pack(note, me))));
+	});
+});
diff --git a/src/services/note/create.ts b/src/services/note/create.ts
index a793c8e58..ea20b063d 100644
--- a/src/services/note/create.ts
+++ b/src/services/note/create.ts
@@ -18,6 +18,7 @@ import { IApp } from '../../models/app';
 import UserList from '../../models/user-list';
 import resolveUser from '../../remote/resolve-user';
 import Meta from '../../models/meta';
+import config from '../../config';
 
 type Type = 'reply' | 'renote' | 'quote' | 'mention';
 
@@ -366,7 +367,7 @@ export default async (user: IUser, data: {
 			watch(user._id, data.reply);
 		}
 
-		// (自分自身へのリプライでない限りは)通知を作成
+		// 通知
 		nm.push(data.reply.userId, 'reply');
 	}
 
@@ -427,4 +428,18 @@ export default async (user: IUser, data: {
 			});
 		}
 	}
+
+	// Register to search database
+	if (note.text && config.elasticsearch) {
+		const es = require('../../../db/elasticsearch');
+
+		es.index({
+			index: 'misskey',
+			type: 'note',
+			id: note._id.toString(),
+			body: {
+				text: note.text
+			}
+		});
+	}
 });