276 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
		
		
			
		
	
	
			276 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
|  | 'use strict'; | ||
|  | 
 | ||
|  | function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } } | ||
|  | 
 | ||
|  | function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; } | ||
|  | 
 | ||
|  | const color = require('kleur'); | ||
|  | 
 | ||
|  | const Prompt = require('./prompt'); | ||
|  | 
 | ||
|  | const _require = require('sisteransi'), | ||
|  |       cursor = _require.cursor; | ||
|  | 
 | ||
|  | const _require2 = require('../util'), | ||
|  |       style = _require2.style, | ||
|  |       clear = _require2.clear, | ||
|  |       figures = _require2.figures, | ||
|  |       strip = _require2.strip; | ||
|  | 
 | ||
|  | const getVal = (arr, i) => arr[i] && (arr[i].value || arr[i].title || arr[i]); | ||
|  | 
 | ||
|  | const getTitle = (arr, i) => arr[i] && (arr[i].title || arr[i].value || arr[i]); | ||
|  | 
 | ||
|  | const getIndex = (arr, valOrTitle) => { | ||
|  |   const index = arr.findIndex(el => el.value === valOrTitle || el.title === valOrTitle); | ||
|  |   return index > -1 ? index : undefined; | ||
|  | }; | ||
|  | /** | ||
|  |  * TextPrompt Base Element | ||
|  |  * @param {Object} opts Options | ||
|  |  * @param {String} opts.message Message | ||
|  |  * @param {Array} opts.choices Array of auto-complete choices objects | ||
|  |  * @param {Function} [opts.suggest] Filter function. Defaults to sort by title | ||
|  |  * @param {Number} [opts.limit=10] Max number of results to show | ||
|  |  * @param {Number} [opts.cursor=0] Cursor start position | ||
|  |  * @param {String} [opts.style='default'] Render style | ||
|  |  * @param {String} [opts.fallback] Fallback message - initial to default value | ||
|  |  * @param {String} [opts.initial] Index of the default value | ||
|  |  * @param {Stream} [opts.stdin] The Readable stream to listen to | ||
|  |  * @param {Stream} [opts.stdout] The Writable stream to write readline data to | ||
|  |  * @param {String} [opts.noMatches] The no matches found label | ||
|  |  */ | ||
|  | 
 | ||
|  | 
 | ||
|  | class AutocompletePrompt extends Prompt { | ||
|  |   constructor(opts = {}) { | ||
|  |     super(opts); | ||
|  |     this.msg = opts.message; | ||
|  |     this.suggest = opts.suggest; | ||
|  |     this.choices = opts.choices; | ||
|  |     this.initial = typeof opts.initial === 'number' ? opts.initial : getIndex(opts.choices, opts.initial); | ||
|  |     this.select = this.initial || opts.cursor || 0; | ||
|  |     this.fallback = opts.fallback || (opts.initial !== undefined ? `${figures.pointerSmall} ${getTitle(this.choices, this.initial)}` : `${figures.pointerSmall} ${opts.noMatches || 'no matches found'}`); | ||
|  |     this.suggestions = [[]]; | ||
|  |     this.page = 0; | ||
|  |     this.input = ''; | ||
|  |     this.limit = opts.limit || 10; | ||
|  |     this.cursor = 0; | ||
|  |     this.transform = style.render(opts.style); | ||
|  |     this.scale = this.transform.scale; | ||
|  |     this.render = this.render.bind(this); | ||
|  |     this.complete = this.complete.bind(this); | ||
|  |     this.clear = clear(''); | ||
|  |     this.complete(this.render); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   moveSelect(i) { | ||
|  |     this.select = i; | ||
|  | 
 | ||
|  |     if (this.suggestions[this.page].length > 0) { | ||
|  |       this.value = getVal(this.suggestions[this.page], i); | ||
|  |     } else { | ||
|  |       this.value = this.initial !== undefined ? getVal(this.choices, this.initial) : null; | ||
|  |     } | ||
|  | 
 | ||
|  |     this.fire(); | ||
|  |   } | ||
|  | 
 | ||
|  |   complete(cb) { | ||
|  |     var _this = this; | ||
|  | 
 | ||
|  |     return _asyncToGenerator(function* () { | ||
|  |       const p = _this.completing = _this.suggest(_this.input, _this.choices); | ||
|  | 
 | ||
|  |       const suggestions = yield p; | ||
|  |       if (_this.completing !== p) return; | ||
|  |       _this.suggestions = suggestions.map((s, i, arr) => ({ | ||
|  |         title: getTitle(arr, i), | ||
|  |         value: getVal(arr, i) | ||
|  |       })).reduce((arr, sug) => { | ||
|  |         if (arr[arr.length - 1].length < _this.limit) arr[arr.length - 1].push(sug);else arr.push([sug]); | ||
|  |         return arr; | ||
|  |       }, [[]]); | ||
|  |       _this.isFallback = false; | ||
|  |       _this.completing = false; | ||
|  |       if (!_this.suggestions[_this.page]) _this.page = 0; | ||
|  | 
 | ||
|  |       if (!_this.suggestions.length && _this.fallback) { | ||
|  |         const index = getIndex(_this.choices, _this.fallback); | ||
|  |         _this.suggestions = [[]]; | ||
|  |         if (index !== undefined) _this.suggestions[0].push({ | ||
|  |           title: getTitle(_this.choices, index), | ||
|  |           value: getVal(_this.choices, index) | ||
|  |         }); | ||
|  |         _this.isFallback = true; | ||
|  |       } | ||
|  | 
 | ||
|  |       const l = Math.max(suggestions.length - 1, 0); | ||
|  | 
 | ||
|  |       _this.moveSelect(Math.min(l, _this.select)); | ||
|  | 
 | ||
|  |       cb && cb(); | ||
|  |     })(); | ||
|  |   } | ||
|  | 
 | ||
|  |   reset() { | ||
|  |     this.input = ''; | ||
|  |     this.complete(() => { | ||
|  |       this.moveSelect(this.initial !== void 0 ? this.initial : 0); | ||
|  |       this.render(); | ||
|  |     }); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   abort() { | ||
|  |     this.done = this.aborted = true; | ||
|  |     this.fire(); | ||
|  |     this.render(); | ||
|  |     this.out.write('\n'); | ||
|  |     this.close(); | ||
|  |   } | ||
|  | 
 | ||
|  |   submit() { | ||
|  |     this.done = true; | ||
|  |     this.aborted = false; | ||
|  |     this.fire(); | ||
|  |     this.render(); | ||
|  |     this.out.write('\n'); | ||
|  |     this.close(); | ||
|  |   } | ||
|  | 
 | ||
|  |   _(c, key) { | ||
|  |     // TODO on ctrl+# go to page #
 | ||
|  |     let s1 = this.input.slice(0, this.cursor); | ||
|  |     let s2 = this.input.slice(this.cursor); | ||
|  |     this.input = `${s1}${c}${s2}`; | ||
|  |     this.cursor = s1.length + 1; | ||
|  |     this.complete(this.render); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   delete() { | ||
|  |     if (this.cursor === 0) return this.bell(); | ||
|  |     let s1 = this.input.slice(0, this.cursor - 1); | ||
|  |     let s2 = this.input.slice(this.cursor); | ||
|  |     this.input = `${s1}${s2}`; | ||
|  |     this.complete(this.render); | ||
|  |     this.cursor = this.cursor - 1; | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   deleteForward() { | ||
|  |     if (this.cursor * this.scale >= this.rendered.length) return this.bell(); | ||
|  |     let s1 = this.input.slice(0, this.cursor); | ||
|  |     let s2 = this.input.slice(this.cursor + 1); | ||
|  |     this.input = `${s1}${s2}`; | ||
|  |     this.complete(this.render); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   first() { | ||
|  |     this.moveSelect(0); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   last() { | ||
|  |     this.moveSelect(this.suggestions[this.page].length - 1); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   up() { | ||
|  |     if (this.select <= 0) return this.bell(); | ||
|  |     this.moveSelect(this.select - 1); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   down() { | ||
|  |     if (this.select >= this.suggestions[this.page].length - 1) return this.bell(); | ||
|  |     this.moveSelect(this.select + 1); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   next() { | ||
|  |     if (this.select === this.suggestions[this.page].length - 1) { | ||
|  |       this.page = (this.page + 1) % this.suggestions.length; | ||
|  |       this.moveSelect(0); | ||
|  |     } else this.moveSelect(this.select + 1); | ||
|  | 
 | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   nextPage() { | ||
|  |     if (this.page >= this.suggestions.length - 1) return this.bell(); | ||
|  |     this.page++; | ||
|  |     this.moveSelect(0); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   prevPage() { | ||
|  |     if (this.page <= 0) return this.bell(); | ||
|  |     this.page--; | ||
|  |     this.moveSelect(0); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   left() { | ||
|  |     if (this.cursor <= 0) return this.bell(); | ||
|  |     this.cursor = this.cursor - 1; | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   right() { | ||
|  |     if (this.cursor * this.scale >= this.rendered.length) return this.bell(); | ||
|  |     this.cursor = this.cursor + 1; | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   render() { | ||
|  |     if (this.closed) return; | ||
|  |     super.render(); | ||
|  |     if (this.lineCount) this.out.write(cursor.down(this.lineCount)); | ||
|  |     let prompt = color.bold(`${style.symbol(this.done, this.aborted)} ${this.msg} `) + `${style.delimiter(this.completing)} `; | ||
|  |     let length = strip(prompt).length; | ||
|  | 
 | ||
|  |     if (this.done && this.suggestions[this.page][this.select]) { | ||
|  |       prompt += `${this.suggestions[this.page][this.select].title}`; | ||
|  |     } else { | ||
|  |       this.rendered = `${this.transform.render(this.input)}`; | ||
|  |       length += this.rendered.length; | ||
|  |       prompt += this.rendered; | ||
|  |     } | ||
|  | 
 | ||
|  |     if (!this.done) { | ||
|  |       this.lineCount = this.suggestions[this.page].length; | ||
|  |       let suggestions = this.suggestions[this.page].reduce((acc, item, i) => acc + `\n${i === this.select ? color.cyan(item.title) : item.title}`, ''); | ||
|  | 
 | ||
|  |       if (suggestions && !this.isFallback) { | ||
|  |         prompt += suggestions; | ||
|  | 
 | ||
|  |         if (this.suggestions.length > 1) { | ||
|  |           this.lineCount++; | ||
|  |           prompt += color.blue(`\nPage ${this.page + 1}/${this.suggestions.length}`); | ||
|  |         } | ||
|  |       } else { | ||
|  |         const fallbackIndex = getIndex(this.choices, this.fallback); | ||
|  |         const fallbackTitle = fallbackIndex !== undefined ? getTitle(this.choices, fallbackIndex) : this.fallback; | ||
|  |         prompt += `\n${color.gray(fallbackTitle)}`; | ||
|  |         this.lineCount++; | ||
|  |       } | ||
|  |     } | ||
|  | 
 | ||
|  |     this.out.write(this.clear + prompt); | ||
|  |     this.clear = clear(prompt); | ||
|  | 
 | ||
|  |     if (this.lineCount && !this.done) { | ||
|  |       let pos = cursor.up(this.lineCount); | ||
|  |       pos += cursor.left + cursor.to(length); | ||
|  |       pos += cursor.move(-this.rendered.length + this.cursor * this.scale); | ||
|  |       this.out.write(pos); | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  | } | ||
|  | 
 | ||
|  | module.exports = AutocompletePrompt; |