1 | 'use strict';
|
2 |
|
3 | const _ = {
|
4 | sum: require('lodash/sum'),
|
5 | flatten: require('lodash/flatten'),
|
6 | };
|
7 | const chalk = require('chalk');
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 | class Paginator {
|
14 | constructor(screen, options = {}) {
|
15 | const { isInfinite = true } = options;
|
16 | this.lastIndex = 0;
|
17 | this.screen = screen;
|
18 | this.isInfinite = isInfinite;
|
19 | }
|
20 |
|
21 | paginate(output, active, pageSize) {
|
22 | pageSize = pageSize || 7;
|
23 | let lines = output.split('\n');
|
24 |
|
25 | if (this.screen) {
|
26 | lines = this.screen.breakLines(lines);
|
27 | active = _.sum(lines.map((lineParts) => lineParts.length).splice(0, active));
|
28 | lines = _.flatten(lines);
|
29 | }
|
30 |
|
31 |
|
32 | if (lines.length <= pageSize) {
|
33 | return output;
|
34 | }
|
35 | const visibleLines = this.isInfinite
|
36 | ? this.getInfiniteLines(lines, active, pageSize)
|
37 | : this.getFiniteLines(lines, active, pageSize);
|
38 | this.lastIndex = active;
|
39 | return (
|
40 | visibleLines.join('\n') +
|
41 | '\n' +
|
42 | chalk.dim('(Move up and down to reveal more choices)')
|
43 | );
|
44 | }
|
45 |
|
46 | getInfiniteLines(lines, active, pageSize) {
|
47 | if (this.pointer === undefined) {
|
48 | this.pointer = 0;
|
49 | }
|
50 | const middleOfList = Math.floor(pageSize / 2);
|
51 |
|
52 | if (
|
53 | this.pointer < middleOfList &&
|
54 | this.lastIndex < active &&
|
55 | active - this.lastIndex < pageSize
|
56 | ) {
|
57 | this.pointer = Math.min(middleOfList, this.pointer + active - this.lastIndex);
|
58 | }
|
59 |
|
60 |
|
61 | const infinite = _.flatten([lines, lines, lines]);
|
62 | const topIndex = Math.max(0, active + lines.length - this.pointer);
|
63 |
|
64 | return infinite.splice(topIndex, pageSize);
|
65 | }
|
66 |
|
67 | getFiniteLines(lines, active, pageSize) {
|
68 | let topIndex = active - pageSize / 2;
|
69 | if (topIndex < 0) {
|
70 | topIndex = 0;
|
71 | } else if (topIndex + pageSize > lines.length) {
|
72 | topIndex = lines.length - pageSize;
|
73 | }
|
74 | return lines.splice(topIndex, pageSize);
|
75 | }
|
76 | }
|
77 |
|
78 | module.exports = Paginator;
|