初始化

康凯

2022-05-20 aa4c235a8ca67ea8b731f90c951a465e92c0a865

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
'use strict';
const stdin = process.stdin;
 
module.exports = () => {
    let ret = '';
 
    return new Promise(resolve => {
        if (stdin.isTTY) {
            resolve(ret);
            return;
        }
 
        stdin.setEncoding('utf8');
 
        stdin.on('readable', () => {
            let chunk;
 
            while ((chunk = stdin.read())) {
                ret += chunk;
            }
        });
 
        stdin.on('end', () => {
            resolve(ret);
        });
    });
};
 
module.exports.buffer = () => {
    const ret = [];
    let len = 0;
 
    return new Promise(resolve => {
        if (stdin.isTTY) {
            resolve(Buffer.concat([]));
            return;
        }
 
        stdin.on('readable', () => {
            let chunk;
 
            while ((chunk = stdin.read())) {
                ret.push(chunk);
                len += chunk.length;
            }
        });
 
        stdin.on('end', () => {
            resolve(Buffer.concat(ret, len));
        });
    });
};