79 lines
2.1 KiB
JavaScript
79 lines
2.1 KiB
JavaScript
|
'use strict';
|
||
|
|
||
|
const _ = {
|
||
|
sum: require('lodash/sum'),
|
||
|
flatten: require('lodash/flatten'),
|
||
|
};
|
||
|
const chalk = require('chalk');
|
||
|
|
||
|
/**
|
||
|
* The paginator returns a subset of the choices if the list is too long.
|
||
|
*/
|
||
|
|
||
|
class Paginator {
|
||
|
constructor(screen, options = {}) {
|
||
|
const { isInfinite = true } = options;
|
||
|
this.lastIndex = 0;
|
||
|
this.screen = screen;
|
||
|
this.isInfinite = isInfinite;
|
||
|
}
|
||
|
|
||
|
paginate(output, active, pageSize) {
|
||
|
pageSize = pageSize || 7;
|
||
|
let lines = output.split('\n');
|
||
|
|
||
|
if (this.screen) {
|
||
|
lines = this.screen.breakLines(lines);
|
||
|
active = _.sum(lines.map((lineParts) => lineParts.length).splice(0, active));
|
||
|
lines = _.flatten(lines);
|
||
|
}
|
||
|
|
||
|
// Make sure there's enough lines to paginate
|
||
|
if (lines.length <= pageSize) {
|
||
|
return output;
|
||
|
}
|
||
|
const visibleLines = this.isInfinite
|
||
|
? this.getInfiniteLines(lines, active, pageSize)
|
||
|
: this.getFiniteLines(lines, active, pageSize);
|
||
|
this.lastIndex = active;
|
||
|
return (
|
||
|
visibleLines.join('\n') +
|
||
|
'\n' +
|
||
|
chalk.dim('(Move up and down to reveal more choices)')
|
||
|
);
|
||
|
}
|
||
|
|
||
|
getInfiniteLines(lines, active, pageSize) {
|
||
|
if (this.pointer === undefined) {
|
||
|
this.pointer = 0;
|
||
|
}
|
||
|
const middleOfList = Math.floor(pageSize / 2);
|
||
|
// Move the pointer only when the user go down and limit it to the middle of the list
|
||
|
if (
|
||
|
this.pointer < middleOfList &&
|
||
|
this.lastIndex < active &&
|
||
|
active - this.lastIndex < pageSize
|
||
|
) {
|
||
|
this.pointer = Math.min(middleOfList, this.pointer + active - this.lastIndex);
|
||
|
}
|
||
|
|
||
|
// Duplicate the lines so it give an infinite list look
|
||
|
const infinite = _.flatten([lines, lines, lines]);
|
||
|
const topIndex = Math.max(0, active + lines.length - this.pointer);
|
||
|
|
||
|
return infinite.splice(topIndex, pageSize);
|
||
|
}
|
||
|
|
||
|
getFiniteLines(lines, active, pageSize) {
|
||
|
let topIndex = active - pageSize / 2;
|
||
|
if (topIndex < 0) {
|
||
|
topIndex = 0;
|
||
|
} else if (topIndex + pageSize > lines.length) {
|
||
|
topIndex = lines.length - pageSize;
|
||
|
}
|
||
|
return lines.splice(topIndex, pageSize);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
module.exports = Paginator;
|