]> git.sesse.net Git - remoteglot/blobdiff - www/js/remoteglot.js
Handle streaming PGNs, like from Lichess (although this might break non-streaming...
[remoteglot] / www / js / remoteglot.js
index d6659f8e588cd8ee11ed6b3e9f321a78caee9f04..d36bf9e5117bed15911ea36263ed5084fc6f0c48 100644 (file)
@@ -5,7 +5,7 @@
  * Version of this script. If the server returns a version larger than
  * this, it is a sign we should reload to upgrade ourselves.
  *
- * @type {Number}
+ * @type {number}
  * @const
  * @private */
 let SCRIPT_VERSION = 2023122700;
@@ -75,17 +75,30 @@ let occupied_by_arrows = [];
 
 /** Currently displayed refutation lines (on-screen).
  * Can either come from the current_analysis_data, displayed_analysis_data,
- * or hash_refutation_lines.
+ * or hash_refutation_lines (choose_displayed_refutation_lines() chooses which one).
+ *
+ * @typedef {{
+ *     score: Array,
+ *     depth: string,
+ *     pv: Array.<string>,
+ *     move: string
+ * }}
+ * @private
  */
-let refutation_lines = [];
+var RefutationLine;
 
 /** Refutation lines from current hash probe.
- *
  * If non-null, will override refutation lines from the base position.
- * Note that these are relative to display_fen, not base_fen.
+ *
+ * @type {Array.<RefutationLine>}
  */
 let hash_refutation_lines = null;
 
+/**
+ * What FEN hash_refutation_lines is relative to.
+ */
+let hash_refutation_lines_base_fen = null;
+
 /** @type {number} @private */
 let ims = 0;
 
@@ -157,15 +170,19 @@ let display_fen = null;
 /** @typedef {{
  *    start_fen: string,
  *    pv: Array.<string>,
- *    move_num: number,
- *    toplay: string,
  *    scores: Array<{first_move: number, score: Object}>,
  *    start_display_move_num: number
- * }} DisplayLine
+ * }}
+ *
+ * "start_display_move_num" is the (half-)move number to start displaying the PV at,
+ * i.e., the index into pv.
  *
- * "start_display_move_num" is the (half-)move number to start displaying the PV at.
- * "score" is also evaluated at this point.
+ * "score" is also evaluated at this point. scores can be empty and is frequently
+ * sparse; it's generally only really useful for history (we obviously don't know
+ * much about how the score will * move during e.g. a PV, except that a mate/TB
+ * counter might go down).
  */
+var DisplayLine;
 
 /** All PVs that we currently know of.
  *
@@ -184,7 +201,11 @@ let current_display_line = null;
 /** @type {boolean} @private */
 let current_display_line_is_history = false;
 
-/** @type {?number} @private */
+/** @type {?number} @private
+ *
+ * The highlighted/used/shown move in current_display_line.pv, in terms of absolute index
+ * (not relative to e.g. the start FEN).
+ */
 let current_display_move = null;
 
 /**
@@ -200,7 +221,7 @@ let current_analysis_xhr = null;
  * The current timer to fire off a request to get main analysis (not history),
  * if any, so that we can abort it.
  *
- * @type {?Number}
+ * @type {?number}
  * @private
  */
 let current_analysis_request_timer = null;
@@ -225,7 +246,7 @@ let current_hash_xhr = null;
  * The current timer to display hash probe information (it could be waiting on the
  * board to stop animating), if any, so that we can abort it.
  *
- * @type {?Number}
+ * @type {?number}
  * @private
  */
 let current_hash_display_timer = null;
@@ -397,12 +418,10 @@ function redraw_arrows() {
 /** @param {!string} fen
  * @return {!string}
  *
- * Return whose side it is to play (W or B), given a FEN.
- * This is in uppercase due to historical reasons, which we
- * should get rid of.
+ * Return whose side it is to play (w or b), given a FEN.
  */
 function find_toplay(fen) {
-       return fen.split(' ')[1].toUpperCase();
+       return fen.split(' ')[1];
 }
 
 /** @param {!string} fen
@@ -414,6 +433,17 @@ function find_move_num(fen) {
        return parseInt(fen.split(' ')[5]);
 }
 
+/** @param {!string} fen
+ * @return {!number}
+ *
+ * Return the half-move clock, starting from 0 (and never resetting).
+ */
+function find_halfmove_num(fen) {
+       let move_num = find_move_num(fen);
+       let toplay = find_toplay(fen);
+       return (move_num - 1) * 2 + (toplay === 'w' ? 0 : 1);
+}
+
 /** @param {!number} x
  * @return {!number}
  */
@@ -689,7 +719,7 @@ function thousands(x) {
 /**
  * @param {!string} start_fen
  * @param {Array.<string>} pv
- * @param {Array<{ first_move: integer, score: Object }>} scores
+ * @param {Array<{ first_move: number, score: Object }>} scores
  * @param {number} start_display_move_num
  * @param {number=} opt_limit
  * @param {boolean=} opt_showlast
@@ -698,8 +728,6 @@ function add_pv(start_fen, pv, scores, start_display_move_num, opt_limit, opt_sh
        display_lines.push({
                start_fen: start_fen,
                pv: pv,
-               move_num: find_move_num(start_fen),
-               toplay: find_toplay(start_fen),
                scores: scores,
                start_display_move_num: start_display_move_num
        });
@@ -723,37 +751,15 @@ function add_pv(start_fen, pv, scores, start_display_move_num, opt_limit, opt_sh
 function print_pv(line_num, splicepos, opt_limit, opt_showlast) {
        let display_line = display_lines[line_num];
        let pv = display_line.pv;
-       let move_num = display_line.move_num;
-       let toplay = display_line.toplay;
-
-       // Truncate PV at the start if needed.
-       let start_display_move_num = display_line.start_display_move_num;
-       if (start_display_move_num > 0) {
-               pv = pv.slice(start_display_move_num);
-               let to_add = start_display_move_num;
-               if (toplay === 'B') {
-                       ++move_num;
-                       toplay = 'W';
-                       --to_add;
-               }
-               if (to_add % 2 == 1) {
-                       toplay = 'B';
-                       --to_add;
-               }
-               move_num += to_add / 2;
-               if (splicepos !== null && splicepos > 0) {
-                       --splicepos;
-               }
-       }
+       let halfmove_num = find_halfmove_num(display_line.start_fen) + 2;  // From two, for simplicity.
+       let start_halfmove_num = halfmove_num;
 
        let ret = document.createDocumentFragment();
-       let in_tb = false;
-       let i = 0;
-       if (opt_limit && opt_showlast && pv.length > opt_limit) {
-               // Truncate the PV at the beginning (instead of at the end).
-               // We assume here that toplay is 'W'. We also assume that if
-               // opt_showlast is set, then it is the history, and thus,
-               // the UI should be to expand the history.
+
+       // Truncate PV at the start if needed.
+       let to_skip = display_line.start_display_move_num;
+       if (opt_limit && opt_showlast && pv.length - to_skip > opt_limit) {
+               // Explicit (UI-visible) truncation from the start, for the history.
                ret.appendChild(document.createTextNode('('));
                let link = document.createElement('a');
                link.className = 'move';
@@ -761,25 +767,37 @@ function print_pv(line_num, splicepos, opt_limit, opt_showlast) {
                link.textContent = '…';
                ret.appendChild(link);
                ret.appendChild(document.createTextNode(') '));
-               i = pv.length - opt_limit;
-               if (i % 2 == 1) {
-                       ++i;
+               to_skip = pv.length - opt_limit;
+               to_skip += to_skip % 2;  // Make sure it starts on a white move.
+       }
+       if (to_skip > 0) {
+               pv = pv.slice(to_skip);
+               halfmove_num += to_skip;
+               if (splicepos !== null) {
+                       splicepos -= to_skip;
+                       if (splicepos < 0) {
+                               splicepos = 0;
+                       }
                }
-               move_num += i / 2;
-       } else if (toplay == 'B' && pv.length > 0) {
-               ret.appendChild(document.createTextNode(move_num + '. … '));
        }
-       for (; i < pv.length; ++i) {
+
+       // The initial move number needs to go before any (TB: …) marker.
+       if (halfmove_num % 2 == 1) {
+               // Black move.
+               ret.appendChild(document.createTextNode((halfmove_num - 1) / 2 + '. … '));
+       } else {
+               // White move.
+               ret.appendChild(document.createTextNode(halfmove_num / 2 + '. '));
+       }
+       let in_tb = false;
+       for (let i = 0; i < pv.length; ++i, ++halfmove_num) {
                let prefix = '';
                if (splicepos === i) {
-                       prefix = '(TB: ';
+                       prefix = '(TB:';
                        in_tb = true;
                }
 
-               if (toplay == 'B' && i == 0) {
-                       ++move_num;
-                       toplay = 'W';
-               } else if (toplay == 'W') {
+               if (halfmove_num % 2 == 0 && i != 0) {
                        if (i > opt_limit && !opt_showlast) {
                                if (in_tb) {
                                        prefix += ')';
@@ -787,20 +805,17 @@ function print_pv(line_num, splicepos, opt_limit, opt_showlast) {
                                ret.appendChild(document.createTextNode(prefix + ' (…)'));
                                return ret;
                        }
-                       prefix += ' ' + move_num + '. ';
-                       ++move_num;
-                       toplay = 'B';
+                       prefix += ' ' + (halfmove_num / 2) + '. ';
                } else {
                        prefix += ' ';
-                       toplay = 'W';
                }
                ret.appendChild(document.createTextNode(prefix));
 
                let link = document.createElement('a');
                link.className = 'move';
-               link.setAttribute('id', 'automove' + line_num + '-' + i);
+               link.setAttribute('id', 'automove' + line_num + '-' + (halfmove_num - start_halfmove_num));
                link.textContent = pv[i];
-               link.href = 'javascript:show_line(' + line_num + ', ' + i + ');';
+               link.href = 'javascript:show_line(' + line_num + ', ' + (halfmove_num - start_halfmove_num) + ');';
                ret.appendChild(link);
        }
        if (in_tb) {
@@ -848,12 +863,23 @@ function collapse_history(truncate_history) {
 }
 window['collapse_history'] = collapse_history;
 
-/** Update the HTML display of multi-PV from the global "refutation_lines".
+function choose_displayed_refutation_lines() {
+       if (hash_refutation_lines) {
+               // If we're in hash exploration, that takes precedence.
+               return [hash_refutation_lines, hash_refutation_lines_base_fen];
+       } else {
+               let data = displayed_analysis_data || current_analysis_data;
+               return [data['refutation_lines'], data['position']['fen']];
+       }
+}
+
+/** Update the HTML display of multi-PV.
  *
  * Also recreates the global "display_lines".
  */
 function update_refutation_lines() {
-       if (base_fen === null) {
+       const [refutation_lines, refutation_lines_base_fen] = choose_displayed_refutation_lines();
+       if (!refutation_lines) {
                return;
        }
        if (display_lines.length > 2) {
@@ -884,7 +910,7 @@ function update_refutation_lines() {
                moves.push(move);
        }
 
-       let invert = (find_toplay(base_fen) === 'B');
+       let invert = (find_toplay(refutation_lines_base_fen) === 'b');
        if (current_display_line && current_display_move % 2 == 0 && !current_display_line_is_history) {
                invert = !invert;
        }
@@ -921,7 +947,7 @@ function update_refutation_lines() {
                        let pv_td = document.createElement("td");
                        tr.appendChild(pv_td);
                        pv_td.classList.add("pv");
-                       pv_td.append(add_pv(base_fen, base_line.concat([ line['move'] ]), scores, start_display_move_num));
+                       pv_td.append(add_pv(refutation_lines_base_fen, base_line.concat([ line['move'] ]), scores, start_display_move_num));
 
                        tbl.append(tr);
                        continue;
@@ -950,7 +976,7 @@ function update_refutation_lines() {
                let pv_td = document.createElement("td");
                tr.appendChild(pv_td);
                pv_td.classList.add("pv");
-               pv_td.append(add_pv(base_fen, base_line.concat(line['pv']), scores, start_display_move_num, 10));
+               pv_td.append(add_pv(refutation_lines_base_fen, base_line.concat(line['pv']), scores, start_display_move_num, 10));
 
                tbl.append(tr);
        }
@@ -1178,18 +1204,18 @@ function update_board() {
                // Find the previous move.
                let previous_move_num, previous_toplay;
                let fen = data['position']['fen'];
-               if (find_toplay(fen) === 'B') {
+               if (find_toplay(fen) === 'b') {
                        previous_move_num = find_move_num(fen);
-                       previous_toplay = 'W';
+                       previous_toplay = 'w';
                } else {
                        previous_move_num = find_move_num(fen) - 1;
-                       previous_toplay = 'B';
+                       previous_toplay = 'b';
                }
 
                last_move = format_move_with_number(
                        data['position']['last_move'],
                        previous_move_num,
-                       previous_toplay == 'W');
+                       previous_toplay == 'w');
                headline += ' after ' + last_move;
        } else {
                last_move = null;
@@ -1240,7 +1266,6 @@ function update_board() {
                document.getElementById("refutationlines").replaceChildren();
                document.getElementById("whiteclock").replaceChildren();
                document.getElementById("blackclock").replaceChildren();
-               refutation_lines = [];
                update_refutation_lines();
                clear_arrows();
                update_displayed_line();
@@ -1294,7 +1319,7 @@ function update_board() {
                document.getElementById("searchstats").textContent = "";
        }
        if (admin_password !== null) {
-               document.getElementById("searchstats").innerHTML += " | <span style=\"color: red;\">ADMIN MODE (if password is right)</span>";
+               document.getElementById("searchstats").innerHTML += " | <span style=\"color: red;\">ADMIN MODE (if password is right) | <a href=\"javascript:undo_move()\">Undo move</a></span>";
        }
 
        // Update the board itself.
@@ -1327,7 +1352,7 @@ function update_board() {
                        hiddenboard.move(data['pv'][i + 1]);  // To keep continuity.
                }
 
-               let alt_moves = find_nonstupid_moves(data, 30, toplay === 'B');
+               let alt_moves = find_nonstupid_moves(data, 30, toplay === 'b');
                for (let i = 1; i < alt_moves.length && i < 3; ++i) {
                        hiddenboard = new Chess(base_fen);
                        let move = patch_move(hiddenboard.move(alt_moves[i]));
@@ -1339,7 +1364,7 @@ function update_board() {
 
        // See if all semi-reasonable moves have only one possible response.
        if (data['pv'].length >= 2) {
-               let nonstupid_moves = find_nonstupid_moves(data, 300, toplay === 'B');
+               let nonstupid_moves = find_nonstupid_moves(data, 300, toplay === 'b');
                let hiddenboard = new Chess(base_fen);
                hiddenboard.move(data['pv'][0]);
                let response = hiddenboard.move(data['pv'][1]);
@@ -1377,9 +1402,7 @@ function update_board() {
                }
        }
 
-       // Update the refutation lines.
        base_fen = data['position']['fen'];
-       refutation_lines = hash_refutation_lines || data['refutation_lines'];
        update_refutation_lines();
 
        // Update the sparkline last, since its size depends on how everything else reflowed.
@@ -1400,7 +1423,7 @@ function update_sparkline(data) {
                if (first_move_num !== undefined) {
                        let fen = data['position']['fen'];
                        let last_move_num = find_move_num(fen) * 2 - 3;
-                       if (find_toplay(fen) === 'B') {
+                       if (find_toplay(fen) === 'b') {
                                ++last_move_num;
                        }
 
@@ -1622,7 +1645,7 @@ function update_clock() {
 }
 
 /**
- * @param {Number} remaining_ms
+ * @param {number} remaining_ms
  * @param {boolean} show_seconds
  */
 function format_clock(remaining_ms, show_seconds) {
@@ -1648,7 +1671,7 @@ function format_clock(remaining_ms, show_seconds) {
 }
 
 /**
- * @param {Number} x
+ * @param {number} x
  */
 function format_2d(x) {
        if (x >= 10) {
@@ -1660,7 +1683,7 @@ function format_2d(x) {
 
 /**
  * @param {string} move
- * @param {Number} move_num Move number of this move.
+ * @param {number} move_num Move number of this move.
  * @param {boolean} white_to_play Whether white is to play this move.
  */
 function format_move_with_number(move, move_num, white_to_play) {
@@ -1676,7 +1699,7 @@ function format_move_with_number(move, move_num, white_to_play) {
 
 /**
  * @param {string} move
- * @param {Number} halfmove_num Half-move number that is to be played,
+ * @param {number} halfmove_num Half-move number that is to be played,
  *   starting from 0.
  */
 function format_halfmove_with_number(move, halfmove_num) {
@@ -1688,7 +1711,7 @@ function format_halfmove_with_number(move, halfmove_num) {
 
 /**
  * @param {Object} data
- * @param {Number} halfmove_num
+ * @param {number} halfmove_num
  */
 function format_tooltip(data, halfmove_num) {
        if (data['score_history'][halfmove_num + 1] ||
@@ -1724,15 +1747,6 @@ function format_tooltip(data, halfmove_num) {
        }
 }
 
-/**
- * @param {boolean} truncate_history
- */
-function set_truncate_history(truncate_history) {
-       truncate_display_history = truncate_history;
-       update_refutation_lines();
-}
-window['set_truncate_history'] = set_truncate_history;
-
 /**
  * @param {number} line_num
  * @param {number} move_num
@@ -1751,7 +1765,7 @@ function show_line(line_num, move_num) {
                return;
        } else {
                current_display_line = {...display_lines[line_num]};  // Shallow clone.
-               current_display_move = move_num + current_display_line.start_display_move_num;
+               current_display_move = move_num;
        }
        current_display_line_is_history = (line_num == 0);
 
@@ -1918,7 +1932,7 @@ function update_move_highlight() {
                        display_lines[1].pv = [];
                }
 
-               highlighted_move = document.getElementById("automove" + display_line_num + "-" + (current_display_move - current_display_line.start_display_move_num));
+               highlighted_move = document.getElementById("automove" + display_line_num + "-" + current_display_move);
                if (highlighted_move !== null) {
                        highlighted_move.classList.add('highlight');
                }
@@ -2067,6 +2081,7 @@ function show_explore_hash_results(data, fen) {
        }
        current_hash_display_timer = null;
        hash_refutation_lines = data['lines'];
+       hash_refutation_lines_base_fen = fen;
        update_board();
 }
 
@@ -2178,6 +2193,7 @@ function get_best_move(game, source, target, invert) {
        let best_move = null;
        let best_move_score = null;
 
+       let refutation_lines = choose_displayed_refutation_lines()[0];
        for (let move in refutation_lines) {
                let line = refutation_lines[move];
                if (!line['score']) {
@@ -2220,6 +2236,51 @@ function onDrop(source, target) {
        if (move === null) return 'snapback';
 }
 
+/**
+ * If we are in admin mode, send this move to the backend.
+ *
+ * @param {string} fen
+ * @param {string} move
+ */
+function send_chosen_move(fen, move) {
+       if (admin_password !== null) {
+               let history = current_analysis_data['position']['history'];
+               let url = '/manual-override.pl';
+               url += '?fen=' + encodeURIComponent(fen);
+               url += '&history=' + encodeURIComponent(JSON.stringify(history));
+               url += '&move=' + encodeURIComponent(move);
+               url += '&player_w=' + encodeURIComponent(current_analysis_data['position']['player_w']);
+               url += '&player_b=' + encodeURIComponent(current_analysis_data['position']['player_b']);
+               url += '&password=' + encodeURIComponent(admin_password);
+
+               console.log(fen, history);
+               fetch(url);  // Ignore the result.
+       }
+}
+
+function undo_move() {
+       if (admin_password !== null) {
+               let history = current_analysis_data['position']['history'];
+               history = history.slice(0, history.length - 1);
+
+               let position = current_analysis_data['position']['start_fen'];
+               let hiddenboard = chess_from(position, history, history.length);
+               let fen = hiddenboard.fen();
+
+               let url = '/manual-override.pl';
+               url += '?fen=' + encodeURIComponent(fen);
+               url += '&history=' + encodeURIComponent(JSON.stringify(history));
+               url += '&move=null';
+               url += '&player_w=' + encodeURIComponent(current_analysis_data['position']['player_w']);
+               url += '&player_b=' + encodeURIComponent(current_analysis_data['position']['player_b']);
+               url += '&password=' + encodeURIComponent(admin_password);
+
+               console.log(fen, history);
+               fetch(url);  // Ignore the result.
+       }
+}
+window['undo_move'] = undo_move;
+
 function onSnapEnd(source, target) {
        if (source === target && recommended_move !== null) {
                source = recommended_move.from;
@@ -2234,14 +2295,7 @@ function onSnapEnd(source, target) {
        });
 
        if (admin_password !== null) {
-               let url = '/manual-override.pl';
-               url += '?fen=' + encodeURIComponent(display_fen);
-               url += '&history=' + encodeURIComponent(JSON.stringify(current_analysis_data['position']['history']));
-               url += '&move=' + encodeURIComponent(move.san);
-               url += '&player_w=' + encodeURIComponent(current_analysis_data['position']['player_w']);
-               url += '&player_b=' + encodeURIComponent(current_analysis_data['position']['player_b']);
-               url += '&password=' + encodeURIComponent(admin_password);
-               fetch(url);
+               send_chosen_move(display_fen, move.san);
                return;
        }
 
@@ -2265,7 +2319,7 @@ function onSnapEnd(source, target) {
                }
                let line = display_lines[i];
                if (line.pv[line.start_display_move_num] === move.san) {
-                       show_line(i, 0);
+                       show_line(i, line.start_display_move_num);
                        return;
                }
        }