143 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
		
		
			
		
	
	
			143 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
|  | 'use strict'; | ||
|  | 
 | ||
|  | const color = require('kleur'); | ||
|  | 
 | ||
|  | const Prompt = require('./prompt'); | ||
|  | 
 | ||
|  | const _require = require('../util'), | ||
|  |       style = _require.style, | ||
|  |       clear = _require.clear, | ||
|  |       figures = _require.figures; | ||
|  | 
 | ||
|  | const _require2 = require('sisteransi'), | ||
|  |       erase = _require2.erase, | ||
|  |       cursor = _require2.cursor; | ||
|  | /** | ||
|  |  * SelectPrompt Base Element | ||
|  |  * @param {Object} opts Options | ||
|  |  * @param {String} opts.message Message | ||
|  |  * @param {Array} opts.choices Array of choice objects | ||
|  |  * @param {String} [opts.hint] Hint to display | ||
|  |  * @param {Number} [opts.initial] Index of default value | ||
|  |  * @param {Stream} [opts.stdin] The Readable stream to listen to | ||
|  |  * @param {Stream} [opts.stdout] The Writable stream to write readline data to | ||
|  |  */ | ||
|  | 
 | ||
|  | 
 | ||
|  | class SelectPrompt extends Prompt { | ||
|  |   constructor(opts = {}) { | ||
|  |     super(opts); | ||
|  |     this.msg = opts.message; | ||
|  |     this.hint = opts.hint || '- Use arrow-keys. Return to submit.'; | ||
|  |     this.warn = opts.warn || '- This option is disabled'; | ||
|  |     this.cursor = opts.initial || 0; | ||
|  |     this.choices = opts.choices.map((ch, idx) => { | ||
|  |       if (typeof ch === 'string') ch = { | ||
|  |         title: ch, | ||
|  |         value: idx | ||
|  |       }; | ||
|  |       return { | ||
|  |         title: ch && (ch.title || ch.value || ch), | ||
|  |         value: ch && (ch.value || idx), | ||
|  |         selected: ch && ch.selected, | ||
|  |         disabled: ch && ch.disabled | ||
|  |       }; | ||
|  |     }); | ||
|  |     this.value = (this.choices[this.cursor] || {}).value; | ||
|  |     this.clear = clear(''); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   moveCursor(n) { | ||
|  |     this.cursor = n; | ||
|  |     this.value = this.choices[n].value; | ||
|  |     this.fire(); | ||
|  |   } | ||
|  | 
 | ||
|  |   reset() { | ||
|  |     this.moveCursor(0); | ||
|  |     this.fire(); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   abort() { | ||
|  |     this.done = this.aborted = true; | ||
|  |     this.fire(); | ||
|  |     this.render(); | ||
|  |     this.out.write('\n'); | ||
|  |     this.close(); | ||
|  |   } | ||
|  | 
 | ||
|  |   submit() { | ||
|  |     if (!this.selection.disabled) { | ||
|  |       this.done = true; | ||
|  |       this.aborted = false; | ||
|  |       this.fire(); | ||
|  |       this.render(); | ||
|  |       this.out.write('\n'); | ||
|  |       this.close(); | ||
|  |     } else this.bell(); | ||
|  |   } | ||
|  | 
 | ||
|  |   first() { | ||
|  |     this.moveCursor(0); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   last() { | ||
|  |     this.moveCursor(this.choices.length - 1); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   up() { | ||
|  |     if (this.cursor === 0) return this.bell(); | ||
|  |     this.moveCursor(this.cursor - 1); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   down() { | ||
|  |     if (this.cursor === this.choices.length - 1) return this.bell(); | ||
|  |     this.moveCursor(this.cursor + 1); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   next() { | ||
|  |     this.moveCursor((this.cursor + 1) % this.choices.length); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   _(c, key) { | ||
|  |     if (c === ' ') return this.submit(); | ||
|  |   } | ||
|  | 
 | ||
|  |   get selection() { | ||
|  |     return this.choices[this.cursor]; | ||
|  |   } | ||
|  | 
 | ||
|  |   render() { | ||
|  |     if (this.closed) return; | ||
|  |     if (this.firstRender) this.out.write(cursor.hide);else this.out.write(erase.lines(this.choices.length + 1)); | ||
|  |     super.render(); // Print prompt
 | ||
|  | 
 | ||
|  |     this.out.write([style.symbol(this.done, this.aborted), color.bold(this.msg), style.delimiter(false), this.done ? this.selection.title : this.selection.disabled ? color.yellow(this.warn) : color.gray(this.hint)].join(' ')); // Print choices
 | ||
|  | 
 | ||
|  |     if (!this.done) { | ||
|  |       this.out.write('\n' + this.choices.map((v, i) => { | ||
|  |         let title, prefix; | ||
|  | 
 | ||
|  |         if (v.disabled) { | ||
|  |           title = this.cursor === i ? color.gray().underline(v.title) : color.strikethrough().gray(v.title); | ||
|  |           prefix = this.cursor === i ? color.bold().gray(figures.pointer) + ' ' : '  '; | ||
|  |         } else { | ||
|  |           title = this.cursor === i ? color.cyan().underline(v.title) : v.title; | ||
|  |           prefix = this.cursor === i ? color.cyan(figures.pointer) + ' ' : '  '; | ||
|  |         } | ||
|  | 
 | ||
|  |         return `${prefix} ${title}`; | ||
|  |       }).join('\n')); | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  | } | ||
|  | 
 | ||
|  | module.exports = SelectPrompt; |