neoss/build/ui/table.js

<div align="center"> <h1>neoss</h1> <i>Socket statistics, with a UI.</i> </div> <p align="center"> <a href="https://img.shields.io/github/v/release/pablolec/neoss" target="_blank"> <img src="https://img.shields.io/github/v/release/pablolec/neo

"use strict";
const blessed = require("neo-blessed"), Node = blessed.Node, Box = blessed.Box;
const getStats = require("../utils/getStats");
const helper = require("../utils/helper");
const popups = require("../ui/popups");
function Table(options) {
    var self = this;
    if (!(this instanceof Node)) {
        return new Table(options);
    }
    options = options || {};
    options.shrink = true;
    options.style = options.style || {};
    options.style.border = options.style.border || {};
    options.style.header = options.style.header || {};
    options.style.cell = options.style.cell || {};
    options.align = options.align || "center";
    delete options.height;
    Box.call(this, options);
    this.pad = options.pad != null ? options.pad : 2;
    this.selected = [0, 0];
    this.currentSocket = null;
    this.table = [];
    this.popupVisible = false;
    this.screenIsLocked = false;
    this.setData(options.rows || options.data);
    this.on("focus", function () {
        popups.focusPopup();
    });
    this.on("resize", function () {
        this.setData(self.table);
        self.screen.render();
    });
    this.key(["left"], function (ch, key) {
        if (this.screenIsLocked)
            return;
        if (this.selected[1] - 1 == -1) {
            return;
        }
        this.selected = [this.selected[0], this.selected[1] - 1];
        this.setData(this.table);
        self.screen.render();
    });
    this.key(["right"], function (ch, key) {
        if (this.screenIsLocked)
            return;
        if (this.selected[1] + 1 == Object.values(this.table.data[this.selected[0]]).length - 1) {
            return;
        }
        this.selected = [this.selected[0], this.selected[1] + 1];
        this.setData(this.table);
        self.screen.render();
    });
    this.key(["up"], function (ch, key) {
        if (this.screenIsLocked)
            return;
        if (this.selected[0] - 1 == -1) {
            return;
        }
        this.selected = [this.selected[0] - 1, this.selected[1]];
        this.currentSocket = this.table.data[this.selected[0]];
        this.setData(this.table);
        self.screen.render();
    });
    this.key(["down"], function (ch, key) {
        if (this.screenIsLocked)
            return;
        if (this.selected[0] + 1 == this.table.data.length) {
            return;
        }
        this.selected = [this.selected[0] + 1, this.selected[1]];
        this.currentSocket = this.table.data[this.selected[0]];
        this.setData(this.table);
        self.screen.render();
    });
    this.key(["s", "S"], function (ch, key) {
        if (this.screenIsLocked)
            return;
        this.table.data = helper.sortBy(this.selected[1], this.table.data);
        this.selected = [helper.retrieveSocket(this.currentSocket, this.table.data, this.selected[0]), this.selected[1]];
        this.setData(this.table);
        self.screen.render();
    });
    this.key(["enter"], function (ch, key) {
        if (this.screenIsLocked)
            return;
        let content = Object.values(this.table.data[this.selected[0]])[this.selected[1]];
        popups.handlePopup(this.selected[1], content);
    });
    this.key(["r", "R"], function (ch, key) {
        if (this.screenIsLocked)
            return;
        getStats.getStats();
    });
}
Table.prototype.__proto__ = Box.prototype;
Table.prototype.type = "table";
Table.prototype._calculateMaxes = function () {
    var self = this;
    var maxes = [];
    if (this.detached)
        return;
    this.rows = this.rows || [];
    this.rows.forEach(function (row) {
        row.forEach(function (cell, i) {
            var clen = self.strWidth(cell);
            if (!maxes[i] || maxes[i] < clen) {
                maxes[i] = clen;
            }
        });
    });
    var total = maxes.reduce(function (total, max) {
        return total + max;
    }, 0);
    total += maxes.length + 1;
    if (this.width < total) {
        return (this._maxes = null);
    }
    if (this.position.width != null) {
        var missing = this.width - total;
        var w = (missing / maxes.length) | 0;
        var wr = missing % maxes.length;
        maxes = maxes.map(function (max, i) {
            if (i === maxes.length - 1) {
                return max + w + wr;
            }
            return max + w;
        });
    }
    else {
        maxes = maxes.map(function (max) {
            return max + self.pad;
        });
    }
    return (this._maxes = maxes);
};
Table.prototype.setRows = Table.prototype.setData = function (table) {
    if (typeof table == "undefined") {
        return;
    }
    var self = this, text = "", align = this.align;
    this.table = table;
    var lines = Math.floor(this.screen.lines.length / 2) - 1;
    if (Array.isArray(table)) {
        this.rows = table || [];
    }
    else {
        var rows = [];
        Object.values(table.data).forEach(function (d) {
            let row = Object.values(d);
            delete row[8];
            delete row[9];
            row[8] = d["users"].text;
            rows.push(row);
        });
        this.rows = rows;
    }
    try {
        this.rows[this.selected[0]][this.selected[1]] =
            "{blue-bg}{bold}" + this.rows[this.selected[0]][this.selected[1]] + "{/bold}{/blue-bg}";
    }
    catch (TypeError) {
    }
    let scroll = helper.getScroll(this.selected[0], lines);
    this.rows = this.rows.slice(scroll[0], scroll[1]);
    if (typeof table.headers !== "undefined") {
        this.rows.unshift(table.headers);
    }
    this._calculateMaxes();
    if (!this._maxes) {
        this.setContent("Terminal width too small. Please resize your window.");
        this.screenIsLocked = true;
        popups.removePopup();
        return;
    }
    this.screenIsLocked = false;
    this.rows.forEach(function (row, i) {
        var isFooter = i === self.rows.length - 1;
        row.forEach(function (cell, i) {
            var width = self._maxes[i];
            var clen = self.strWidth(cell);
            if (i !== 0) {
                text += " ";
            }
            while (clen < width) {
                if (align === "center") {
                    cell = " " + cell + " ";
                    clen += 2;
                }
                else if (align === "left") {
                    cell = cell + " ";
                    clen += 1;
                }
                else if (align === "right") {
                    cell = " " + cell;
                    clen += 1;
                }
            }
            if (clen > width) {
                if (align === "center") {
                    cell = cell.substring(1);
                    clen--;
                }
                else if (align === "left") {
                    cell = cell.slice(0, -1);
                    clen--;
                }
                else if (align === "right") {
                    cell = cell.substring(1);
                    clen--;
                }
            }
            text += cell;
        });
        if (!isFooter) {
            text += "\n\n";
        }
    });
    delete this.align;
    this.setContent(text);
    this.align = align;
    this.focus();
    this.selected = [helper.retrieveSocket(this.currentSocket, this.table.data, this.selected[0]), this.selected[1]];
};
Table.prototype.render = function () {
    var self = this;
    var coords = this._render();
    if (!coords)
        return;
    this._calculateMaxes();
    if (!this._maxes) {
        return;
    }
    var lines = this.screen.lines, xi = coords.xi, yi = coords.yi, rx, ry, i;
    var dattr = this.sattr(this.style), hattr = this.sattr(this.style.header), cattr = this.sattr(this.style.cell), battr = this.sattr(this.style.border);
    var width = coords.xl - coords.xi - this.iright, height = coords.yl - coords.yi - this.ibottom;
    for (var y = this.itop; y < height; y++) {
        if (!lines[yi + y])
            break;
        for (var x = this.ileft; x < width; x++) {
            if (!lines[yi + y][xi + x])
                break;
            if (lines[yi + y][xi + x][0] !== dattr)
                continue;
            if (y === this.itop) {
                lines[yi + y][xi + x][0] = hattr;
            }
            else {
                lines[yi + y][xi + x][0] = cattr;
            }
            lines[yi + y].dirty = true;
        }
    }
    if (!this.border || this.options.noCellBorders)
        return coords;
    ry = 0;
    for (i = 0; i < self.rows.length + 1; i++) {
        if (!lines[yi + ry])
            break;
        rx = 0;
        self._maxes.forEach(function (max, i) {
            rx += max;
            if (i === 0) {
                if (!lines[yi + ry][xi + 0])
                    return;
                if (ry === 0) {
                    lines[yi + ry][xi + 0][0] = battr;
                }
                else if (ry / 2 === self.rows.length) {
                    lines[yi + ry][xi + 0][0] = battr;
                }
                else {
                    lines[yi + ry][xi + 0][0] = battr;
                    lines[yi + ry][xi + 0][1] = "\u251c";
                    if (!self.border.left) {
                        lines[yi + ry][xi + 0][1] = "\u2500";
                    }
                }
                lines[yi + ry].dirty = true;
            }
            else if (i === self._maxes.length - 1) {
                if (!lines[yi + ry][xi + rx + 1])
                    return;
                if (ry === 0) {
                    rx++;
                    lines[yi + ry][xi + rx][0] = battr;
                }
                else if (ry / 2 === self.rows.length) {
                    rx++;
                    lines[yi + ry][xi + rx][0] = battr;
                }
                else {
                    rx++;
                    lines[yi + ry][xi + rx][0] = battr;
                    lines[yi + ry][xi + rx][1] = "\u2524";
                    if (!self.border.right) {
                        lines[yi + ry][xi + rx][1] = "\u2500";
                    }
                }
                lines[yi + ry].dirty = true;
                return;
            }
            if (!lines[yi + ry][xi + rx + 1])
                return;
            if (ry === 0) {
                rx++;
                lines[yi + ry][xi + rx][0] = battr;
                lines[yi + ry][xi + rx][1] = "\u252c";
                if (!self.border.top) {
                    lines[yi + ry][xi + rx][1] = "\u2502";
                }
            }
            else if (ry / 2 === self.rows.length) {
                rx++;
                lines[yi + ry][xi + rx][0] = battr;
                lines[yi + ry][xi + rx][1] = "\u2534";
                if (!self.border.bottom) {
                    lines[yi + ry][xi + rx][1] = "\u2502";
                }
            }
            else {
                if (self.options.fillCellBorders) {
                    var lbg = (ry <= 2 ? hattr : cattr) & 0x1ff;
                    rx++;
                    lines[yi + ry][xi + rx][0] = (battr & ~0x1ff) | lbg;
                }
                else {
                    rx++;
                    lines[yi + ry][xi + rx][0] = battr;
                }
                lines[yi + ry][xi + rx][1] = "\u253c";
            }
            lines[yi + ry].dirty = true;
        });
        ry += 2;
    }
    for (ry = 1; ry < self.rows.length * 2; ry++) {
        if (!lines[yi + ry])
            break;
        rx = 0;
        self._maxes.slice(0, -1).forEach(function (max) {
            rx += max;
            if (!lines[yi + ry][xi + rx + 1])
                return;
            if (ry % 2 !== 0) {
                if (self.options.fillCellBorders) {
                    var lbg = (ry <= 2 ? hattr : cattr) & 0x1ff;
                    rx++;
                    lines[yi + ry][xi + rx][0] = (battr & ~0x1ff) | lbg;
                }
                else {
                    rx++;
                    lines[yi + ry][xi + rx][0] = battr;
                }
                lines[yi + ry][xi + rx][1] = "\u2502";
                lines[yi + ry].dirty = true;
            }
            else {
                rx++;
            }
        });
        rx = 1;
        self._maxes.forEach(function (max) {
            while (max--) {
                if (ry % 2 === 0) {
                    if (!lines[yi + ry])
                        break;
                    if (!lines[yi + ry][xi + rx + 1])
                        break;
                    if (self.options.fillCellBorders) {
                        var lbg = (ry <= 2 ? hattr : cattr) & 0x1ff;
                        lines[yi + ry][xi + rx][0] = (battr & ~0x1ff) | lbg;
                    }
                    else {
                        lines[yi + ry][xi + rx][0] = battr;
                    }
                    lines[yi + ry][xi + rx][1] = "\u2500";
                    lines[yi + ry].dirty = true;
                }
                rx++;
            }
            rx++;
        });
    }
    return coords;
};
module.exports = Table;