Node readline 简单实现

Node readline 简单实现

function myReadline(callback) {
  const input = process.stdin; // 获取标准输入流
  const output = process.stdout; // 获取标准输出流

  let line = '' // 缓存输入

  function handleKeyPress(s){ // 键盘输入处理
    line += s // 缓存
    output.write(s) // 输出
    switch (s) { // 处理输入字符，使用 setRawMode(true) 后全部输入都要自己处理
      case '\r': // 判断回车
        callback(line) // 执行回调
        input.pause() // 关闭输入流
        break;
    }
  }

  input.on('keyPress', handleKeyPress) // 监听键盘输入
  emitKeyPressEvent(input) // 执行处理函数

  input.setRawMode(true) // 使用原生模式，对输入完全不做自动处理，Ctrl + c 也会无效，需要自己处理
  input.resume() // 回复正常模式
}

function emitKeyPressEvent(stream) {
  function onData(chunk) { // 按字符进行处理
    kp.next(chunk.toString()) // 传入输入字符，对 buffer 进行转换
  }

  let kp = keyPress(stream) // 拿到 generator 函数实例
  kp.next() // 执行 next()，启动函数，中断到 yield 位置

  stream.on('data', onData) // 监听键盘输入的核心，输入入口
}

function* keyPress(stream) { // 字符处理
  while(true) { // 真循环，持续接收
    const s = yield // 拿到字符
    stream.emit('keyPress', s) // 发布 keyPress 监听
  }
}

myReadline((input) => { // 回调
  console.log('log: ' + input)
})