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; |