Move the Node.js stuff out of the public www directory.
[remoteglot] / server / hash-lookup.js
1 var grpc = require('grpc');
2 var Chess = require('../www/js/chess.min.js').Chess;
3
4 var PROTO_PATH = __dirname + '/hashprobe.proto';
5 var hashprobe_proto = grpc.load(PROTO_PATH).hashprobe;
6
7 // TODO: Make destination configurable.
8 var client = new hashprobe_proto.HashProbe('localhost:50051', grpc.credentials.createInsecure());
9
10 var board = new Chess();
11
12 var handle_request = function(fen, response) {
13         if (!board.validate_fen(fen).valid) {
14                 response.writeHead(400, {});
15                 response.end();
16                 return;
17         }
18         client.probe({fen: fen}, function(err, probe_response) {
19                 if (err) {
20                         response.writeHead(500, {});
21                         response.end();
22                 } else {
23                         handle_response(fen, response, probe_response);
24                 }
25         });
26 }
27 exports.handle_request = handle_request;
28
29 var handle_response = function(fen, response, probe_response) {
30         var lines = {};
31
32         var root = translate_line(board, fen, probe_response['root']);
33         for (var i = 0; i < probe_response['line'].length; ++i) {
34                 var line = probe_response['line'][i];
35                 var uci_move = line['move']['from_sq'] + line['move']['to_sq'] + line['move']['promotion'];
36                 lines[uci_move] = translate_line(board, fen, line);
37         }
38
39         var text = JSON.stringify({
40                 root: root,
41                 lines: lines
42         });
43         var headers = {
44                 'Content-Type': 'text/json; charset=utf-8'
45                 //'Content-Length': text.length
46         };
47         response.writeHead(200, headers);
48         response.write(text);
49         response.end();
50 }
51
52 var translate_line = function(board, fen, line) {
53         var r = {};
54         board.load(fen);
55         var toplay = board.turn();
56
57         if (line['move'] && line['move']['from_sq']) {
58                 var promo = line['move']['promotion'];
59                 if (promo) {
60                         r['pretty_move'] = board.move({ from: line['move']['from_sq'], to: line['move']['to_sq'], promotion: promo.toLowerCase() }).san;
61                 } else {
62                         r['pretty_move'] = board.move({ from: line['move']['from_sq'], to: line['move']['to_sq'] }).san;
63                 }
64         } else {
65                 r['pretty_move'] = '';
66         }
67         r['sort_key'] = r['pretty_move'];
68         if (!line['found']) {
69                 r['pv_pretty'] = [];
70                 return r;
71         }
72         r['depth'] = line['depth'];
73
74         // Convert the PV.
75         var pv = [];
76         if (r['pretty_move']) {
77                 pv.push(r['pretty_move']);
78         }
79         for (var j = 0; j < line['pv'].length; ++j) {
80                 var move = line['pv'][j];
81                 var decoded = board.move({ from: move['from_sq'], to: move['to_sq'], promotion: move['promotion'] });
82                 if (decoded === null) {
83                         break;
84                 }
85                 pv.push(decoded.san);
86         }
87         r['pv_pretty'] = pv;
88
89         // Convert the score. Use the static eval if no search.
90         var value = line['value'] || line['eval'];
91         var score = null;
92         if (value['score_type'] === 'SCORE_CP') {
93                 score = ['cp', value['score_cp']];
94         } else if (value['score_type'] === 'SCORE_MATE') {
95                 score = ['m', value['score_mate']];
96         }
97         if (score) {
98                 if (line['bound'] === 'BOUND_UPPER') {
99                         score.push('≤');
100                 } else if (line['bound'] === 'BOUND_LOWER') {
101                         score.push('≥');
102                 }
103         }
104
105         r['score'] = score;
106
107         return r;
108 }