liudong
2023-05-29 340f156319b863525e50e900c58e59b86ecb3d5e
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
107
108
109
110
111
function toArr(any) {
    return any == null ? [] : Array.isArray(any) ? any : [any];
}
 
function toVal(out, key, val, opts) {
    var x, old=out[key], nxt=(
        !!~opts.string.indexOf(key) ? (val == null || val === true ? '' : String(val))
        : typeof val === 'boolean' ? val
        : !!~opts.boolean.indexOf(key) ? (val === 'false' ? false : val === 'true' || (out._.push((x = +val,x * 0 === 0) ? x : val),!!val))
        : (x = +val,x * 0 === 0) ? x : val
    );
    out[key] = old == null ? nxt : (Array.isArray(old) ? old.concat(nxt) : [old, nxt]);
}
 
module.exports = function (args, opts) {
    args = args || [];
    opts = opts || {};
 
    var k, arr, arg, name, val, out={ _:[] };
    var i=0, j=0, idx=0, len=args.length;
 
    const alibi = opts.alias !== void 0;
    const strict = opts.unknown !== void 0;
    const defaults = opts.default !== void 0;
 
    opts.alias = opts.alias || {};
    opts.string = toArr(opts.string);
    opts.boolean = toArr(opts.boolean);
 
    if (alibi) {
        for (k in opts.alias) {
            arr = opts.alias[k] = toArr(opts.alias[k]);
            for (i=0; i < arr.length; i++) {
                (opts.alias[arr[i]] = arr.concat(k)).splice(i, 1);
            }
        }
    }
 
    opts.boolean.forEach(key => {
        opts.boolean = opts.boolean.concat(opts.alias[key] = opts.alias[key] || []);
    });
 
    opts.string.forEach(key => {
        opts.string = opts.string.concat(opts.alias[key] = opts.alias[key] || []);
    });
 
    if (defaults) {
        for (k in opts.default) {
            opts.alias[k] = opts.alias[k] || [];
            (opts[typeof opts.default[k]] || []).push(k);
        }
    }
 
    const keys = strict ? Object.keys(opts.alias) : [];
 
    for (i=0; i < len; i++) {
        arg = args[i];
 
        if (arg === '--') {
            out._ = out._.concat(args.slice(++i));
            break;
        }
 
        for (j=0; j < arg.length; j++) {
            if (arg.charCodeAt(j) !== 45) break; // "-"
        }
 
        if (j === 0) {
            out._.push(arg);
        } else if (arg.substring(j, j + 3) === 'no-') {
            name = arg.substring(j + 3);
            if (strict && !~keys.indexOf(name)) {
                return opts.unknown(arg);
            }
            out[name] = false;
        } else {
            for (idx=j+1; idx < arg.length; idx++) {
                if (arg.charCodeAt(idx) === 61) break; // "="
            }
 
            name = arg.substring(j, idx);
            val = arg.substring(++idx) || (i+1 === len || (''+args[i+1]).charCodeAt(0) === 45 || args[++i]);
            arr = (j === 2 ? [name] : name);
 
            for (idx=0; idx < arr.length; idx++) {
                name = arr[idx];
                if (strict && !~keys.indexOf(name)) return opts.unknown('-'.repeat(j) + name);
                toVal(out, name, (idx + 1 < arr.length) || val, opts);
            }
        }
    }
 
    if (defaults) {
        for (k in opts.default) {
            if (out[k] === void 0) {
                out[k] = opts.default[k];
            }
        }
    }
 
    if (alibi) {
        for (k in out) {
            arr = opts.alias[k] || [];
            while (arr.length > 0) {
                out[arr.shift()] = out[k];
            }
        }
    }
 
    return out;
}