/
completer.js
106 lines (92 loc) · 2.85 KB
/
completer.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
'use strict';
var path = require('path');
var execFile = require('child_process').execFile;
var parseHelp = require('parse-help');
/**
* The Completer is in charge of handling `yo-complete` behavior.
* @constructor
* @param {Environment} env A yeoman environment instance
*/
var Completer = module.exports = function (env) {
this.env = env;
};
/**
* Completion event done
*
* @param {String} data Environment object as parsed by tabtab
* @param {Function} done Callback to invoke with completion results
*/
Completer.prototype.complete = function (data, done) {
if (data.last !== 'yo' && !/^-/.test(data.last)) {
return this.generator(data, done);
}
this.env.lookup(function (err) {
if (err) {
return done(err);
}
var meta = this.env.getGeneratorsMeta();
var results = Object.keys(meta).map(this.item('yo'), this);
done(null, results);
}.bind(this));
};
/**
* Generator completion event done
*
* @param {String} data Environment object as parsed by tabtab
* @param {Function} done Callback to invoke with completion results
*/
Completer.prototype.generator = function (data, done) {
var last = data.last;
var bin = path.join(__dirname, '../cli.js');
execFile('node', [bin, last, '--help'], function (err, out) {
if (err) {
return done(err);
}
var results = this.parseHelp(last, out);
done(null, results);
}.bind(this));
};
/**
* Helper to format completion results into { name, description } objects
*
* @param {String} data Environment object as parsed by tabtab
* @param {Function} done Callback to invoke with completion results
*/
Completer.prototype.item = function (desc, prefix) {
prefix = prefix || '';
return function (item) {
var name = typeof item === 'string' ? item : item.name;
desc = typeof item !== 'string' && item.description ? item.description : desc;
desc = desc.replace(/^#?\s*/g, '');
desc = desc.replace(/:/g, '->');
desc = desc.replace(/'/g, ' ');
return {
name: prefix + name,
description: desc
};
};
};
/**
* parse-help wrapper. Invokes parse-help with stdout result, returning the
* list of completion items for flags / alias.
*
* @param {String} last Last word in COMP_LINE (completed line in command line)
* @param {String} out Help output
*/
Completer.prototype.parseHelp = function (last, out) {
var help = parseHelp(out);
var alias = [];
var results = Object.keys(help.flags).map(function (key) {
var flag = help.flags[key];
if (flag.alias) {
alias.push(Object.assign({}, flag, { name: flag.alias }));
}
flag.name = key;
return flag;
}).map(this.item(last, '--'), this);
results = results.concat(alias.map(this.item(last.replace(':', '_'), '-'), this));
results = results.filter(function (r) {
return r.name !== '--help' && r.name !== '-h';
});
return results;
};