260 lines
		
	
	
		
			7.0 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
		
		
			
		
	
	
			260 lines
		
	
	
		
			7.0 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('../util'), | ||
|  |       style = _require.style, | ||
|  |       clear = _require.clear, | ||
|  |       figures = _require.figures, | ||
|  |       strip = _require.strip; | ||
|  | 
 | ||
|  | const _require2 = require('sisteransi'), | ||
|  |       erase = _require2.erase, | ||
|  |       cursor = _require2.cursor; | ||
|  | 
 | ||
|  | const _require3 = require('../dateparts'), | ||
|  |       DatePart = _require3.DatePart, | ||
|  |       Meridiem = _require3.Meridiem, | ||
|  |       Day = _require3.Day, | ||
|  |       Hours = _require3.Hours, | ||
|  |       Milliseconds = _require3.Milliseconds, | ||
|  |       Minutes = _require3.Minutes, | ||
|  |       Month = _require3.Month, | ||
|  |       Seconds = _require3.Seconds, | ||
|  |       Year = _require3.Year; | ||
|  | 
 | ||
|  | const regex = /\\(.)|"((?:\\["\\]|[^"])+)"|(D[Do]?|d{3,4}|d)|(M{1,4})|(YY(?:YY)?)|([aA])|([Hh]{1,2})|(m{1,2})|(s{1,2})|(S{1,4})|./g; | ||
|  | const regexGroups = { | ||
|  |   1: ({ | ||
|  |     token | ||
|  |   }) => token.replace(/\\(.)/g, '$1'), | ||
|  |   2: opts => new Day(opts), | ||
|  |   // Day // TODO
 | ||
|  |   3: opts => new Month(opts), | ||
|  |   // Month
 | ||
|  |   4: opts => new Year(opts), | ||
|  |   // Year
 | ||
|  |   5: opts => new Meridiem(opts), | ||
|  |   // AM/PM // TODO (special)
 | ||
|  |   6: opts => new Hours(opts), | ||
|  |   // Hours
 | ||
|  |   7: opts => new Minutes(opts), | ||
|  |   // Minutes
 | ||
|  |   8: opts => new Seconds(opts), | ||
|  |   // Seconds
 | ||
|  |   9: opts => new Milliseconds(opts) // Fractional seconds
 | ||
|  | 
 | ||
|  | }; | ||
|  | const dfltLocales = { | ||
|  |   months: 'January,February,March,April,May,June,July,August,September,October,November,December'.split(','), | ||
|  |   monthsShort: 'Jan,Feb,Mar,Apr,May,Jun,Jul,Aug,Sep,Oct,Nov,Dec'.split(','), | ||
|  |   weekdays: 'Sunday,Monday,Tuesday,Wednesday,Thursday,Friday,Saturday'.split(','), | ||
|  |   weekdaysShort: 'Sun,Mon,Tue,Wed,Thu,Fri,Sat'.split(',') | ||
|  |   /** | ||
|  |    * DatePrompt Base Element | ||
|  |    * @param {Object} opts Options | ||
|  |    * @param {String} opts.message Message | ||
|  |    * @param {Number} [opts.initial] Index of default value | ||
|  |    * @param {String} [opts.mask] The format mask | ||
|  |    * @param {object} [opts.locales] The date locales | ||
|  |    * @param {String} [opts.error] The error message shown on invalid value | ||
|  |    * @param {Function} [opts.validate] Function to validate the submitted value | ||
|  |    * @param {Stream} [opts.stdin] The Readable stream to listen to | ||
|  |    * @param {Stream} [opts.stdout] The Writable stream to write readline data to | ||
|  |    */ | ||
|  | 
 | ||
|  | }; | ||
|  | 
 | ||
|  | class DatePrompt extends Prompt { | ||
|  |   constructor(opts = {}) { | ||
|  |     super(opts); | ||
|  |     this.msg = opts.message; | ||
|  |     this.cursor = 0; | ||
|  |     this.typed = ''; | ||
|  |     this.locales = Object.assign(dfltLocales, opts.locales); | ||
|  |     this._date = opts.initial || new Date(); | ||
|  |     this.errorMsg = opts.error || 'Please Enter A Valid Value'; | ||
|  | 
 | ||
|  |     this.validator = opts.validate || (() => true); | ||
|  | 
 | ||
|  |     this.mask = opts.mask || 'YYYY-MM-DD HH:mm:ss'; | ||
|  |     this.clear = clear(''); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   get value() { | ||
|  |     return this.date; | ||
|  |   } | ||
|  | 
 | ||
|  |   get date() { | ||
|  |     return this._date; | ||
|  |   } | ||
|  | 
 | ||
|  |   set date(date) { | ||
|  |     if (date) this._date.setTime(date.getTime()); | ||
|  |   } | ||
|  | 
 | ||
|  |   set mask(mask) { | ||
|  |     let result; | ||
|  |     this.parts = []; | ||
|  | 
 | ||
|  |     while (result = regex.exec(mask)) { | ||
|  |       let match = result.shift(); | ||
|  |       let idx = result.findIndex(gr => gr != null); | ||
|  |       this.parts.push(idx in regexGroups ? regexGroups[idx]({ | ||
|  |         token: result[idx] || match, | ||
|  |         date: this.date, | ||
|  |         parts: this.parts, | ||
|  |         locales: this.locales | ||
|  |       }) : result[idx] || match); | ||
|  |     } | ||
|  | 
 | ||
|  |     let parts = this.parts.reduce((arr, i) => { | ||
|  |       if (typeof i === 'string' && typeof arr[arr.length - 1] === 'string') arr[arr.length - 1] += i;else arr.push(i); | ||
|  |       return arr; | ||
|  |     }, []); | ||
|  |     this.parts.splice(0); | ||
|  |     this.parts.push(...parts); | ||
|  |     this.reset(); | ||
|  |   } | ||
|  | 
 | ||
|  |   moveCursor(n) { | ||
|  |     this.typed = ''; | ||
|  |     this.cursor = n; | ||
|  |     this.fire(); | ||
|  |   } | ||
|  | 
 | ||
|  |   reset() { | ||
|  |     this.moveCursor(this.parts.findIndex(p => p instanceof DatePart)); | ||
|  |     this.fire(); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   abort() { | ||
|  |     this.done = this.aborted = true; | ||
|  |     this.error = false; | ||
|  |     this.fire(); | ||
|  |     this.render(); | ||
|  |     this.out.write('\n'); | ||
|  |     this.close(); | ||
|  |   } | ||
|  | 
 | ||
|  |   validate() { | ||
|  |     var _this = this; | ||
|  | 
 | ||
|  |     return _asyncToGenerator(function* () { | ||
|  |       let valid = yield _this.validator(_this.value); | ||
|  | 
 | ||
|  |       if (typeof valid === 'string') { | ||
|  |         _this.errorMsg = valid; | ||
|  |         valid = false; | ||
|  |       } | ||
|  | 
 | ||
|  |       _this.error = !valid; | ||
|  |     })(); | ||
|  |   } | ||
|  | 
 | ||
|  |   submit() { | ||
|  |     var _this2 = this; | ||
|  | 
 | ||
|  |     return _asyncToGenerator(function* () { | ||
|  |       yield _this2.validate(); | ||
|  | 
 | ||
|  |       if (_this2.error) { | ||
|  |         _this2.color = 'red'; | ||
|  | 
 | ||
|  |         _this2.fire(); | ||
|  | 
 | ||
|  |         _this2.render(); | ||
|  | 
 | ||
|  |         return; | ||
|  |       } | ||
|  | 
 | ||
|  |       _this2.done = true; | ||
|  |       _this2.aborted = false; | ||
|  | 
 | ||
|  |       _this2.fire(); | ||
|  | 
 | ||
|  |       _this2.render(); | ||
|  | 
 | ||
|  |       _this2.out.write('\n'); | ||
|  | 
 | ||
|  |       _this2.close(); | ||
|  |     })(); | ||
|  |   } | ||
|  | 
 | ||
|  |   up() { | ||
|  |     this.typed = ''; | ||
|  |     this.parts[this.cursor].up(); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   down() { | ||
|  |     this.typed = ''; | ||
|  |     this.parts[this.cursor].down(); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   left() { | ||
|  |     let prev = this.parts[this.cursor].prev(); | ||
|  |     if (prev == null) return this.bell(); | ||
|  |     this.moveCursor(this.parts.indexOf(prev)); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   right() { | ||
|  |     let next = this.parts[this.cursor].next(); | ||
|  |     if (next == null) return this.bell(); | ||
|  |     this.moveCursor(this.parts.indexOf(next)); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   next() { | ||
|  |     let next = this.parts[this.cursor].next(); | ||
|  |     this.moveCursor(next ? this.parts.indexOf(next) : this.parts.findIndex(part => part instanceof DatePart)); | ||
|  |     this.render(); | ||
|  |   } | ||
|  | 
 | ||
|  |   _(c) { | ||
|  |     if (/\d/.test(c)) { | ||
|  |       this.typed += c; | ||
|  |       this.parts[this.cursor].setTo(this.typed); | ||
|  |       this.render(); | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   render() { | ||
|  |     if (this.closed) return; | ||
|  |     if (this.firstRender) this.out.write(cursor.hide);else this.out.write(erase.lines(1)); | ||
|  |     super.render(); | ||
|  |     let clear = erase.line + (this.lines ? erase.down(this.lines) : '') + cursor.to(0); | ||
|  |     this.lines = 0; | ||
|  |     let error = ''; | ||
|  | 
 | ||
|  |     if (this.error) { | ||
|  |       let lines = this.errorMsg.split('\n'); | ||
|  |       error = lines.reduce((a, l, i) => a + `\n${i ? ` ` : figures.pointerSmall} ${color.red().italic(l)}`, ``); | ||
|  |       this.lines = lines.length; | ||
|  |     } // Print prompt
 | ||
|  | 
 | ||
|  | 
 | ||
|  |     let prompt = [style.symbol(this.done, this.aborted), color.bold(this.msg), style.delimiter(false), this.parts.reduce((arr, p, idx) => arr.concat(idx === this.cursor && !this.done ? color.cyan().underline(p.toString()) : p), []).join('')].join(' '); | ||
|  |     let position = ''; | ||
|  | 
 | ||
|  |     if (this.lines) { | ||
|  |       position += cursor.up(this.lines); | ||
|  |       position += cursor.left + cursor.to(strip(prompt).length); | ||
|  |     } | ||
|  | 
 | ||
|  |     this.out.write(clear + prompt + error + position); | ||
|  |   } | ||
|  | 
 | ||
|  | } | ||
|  | 
 | ||
|  | module.exports = DatePrompt; |