-
Notifications
You must be signed in to change notification settings - Fork 4
/
parseTestDirectory.js
163 lines (156 loc) · 5.03 KB
/
parseTestDirectory.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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
#!/usr/bin/env node
/**
* Copyright (c) 2018-present, Lmntrx Tech.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*/
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
// /!\ DO NOT MODIFY THIS FILE /!\
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
// mochaswagger-gen is installed globally on people's computers. This means
// that it is extremely difficult to have them upgrade the version and
// because there's only one global version installed, it is very prone to
// breaking changes.
//
// If you need to add a new command, please add it to the scripts/ folder.
//
// The only reason to modify this file is to add more warnings and
// troubleshooting information for the `mochaswagger-gen` command.
//
// Do not make breaking changes! We absolutely don't want to have to
// tell people to update their global version of mochaswagger-gen.
//
// Also be careful with new language features.
// This file must work on Node 0.10+.
//
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
// /!\ DO NOT MODIFY THIS FILE /!\
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
"use strict";
const path = require("path");
const glob = require("glob");
const chalk = require("chalk");
const fs = require("fs");
const ProgressBar = require("progress");
module.exports = dir => {
const root = path.resolve(dir);
const testDirName = path.basename(root);
const testFileFormats = ["/**/*.js", "/**/*.ts"];
let routes = [];
testFileFormats.forEach(testFileFormat => {
let files = glob.sync(`${testDirName}/${testFileFormat}`);
if (!files || files.length <= 0) {
console.warn(
chalk.magenta(
`${testDirName} directory does not contain files matching pattern ${testFileFormat}`
)
);
return null;
}
console.info(
chalk.cyan(`Parsing files matching pattern ${testFileFormat}`)
);
var bar = new ProgressBar(" processing [:bar] :percent :etas", {
complete: "=",
incomplete: " ",
total: files.length * 2,
renderThrottle: 0
});
files.forEach(file => {
processFile(fs.readFileSync(file).toString());
bar.tick(2);
});
});
function processFile(data) {
let blocks = data.split("it");
blocks.forEach(block => {
let lines = block.split(".");
let route = null;
lines.forEach((line, index) => {
let method = null;
if (line.includes("`") && !line.includes("$")) {
line = line.replace(/`/g, '"').trim();
}
if (line.includes("+")) {
line =
line.substring(0, line.indexOf('" +')) +
":" +
(lines[index + 1].includes("(")
? line.substring(line.indexOf('" +') + 3, line.length).trim()
: lines[index + 1].substring(0, lines[index + 1].indexOf(")"))) +
'")';
}
if (line.includes("`") && line.includes("$")) {
method = null;
} else if (line.startsWith("get")) {
method = "get";
} else if (line.startsWith("post")) {
method = "post";
} else if (line.startsWith("delete")) {
method = "delete";
} else if (line.startsWith("put")) {
method = "put";
} else if (line.startsWith("patch")) {
method = "patch";
}
if (method) {
let parameters = [];
let path = line.substring(line.indexOf("(") + 1, line.indexOf(")"));
path = path.replace(/"/g, "");
if (path.includes(":")) {
path = path.replace(/:/g, "{");
path = path.replace(/{\//g, "}");
path = path + "}";
parameters = [
{
in: "path",
required: true,
type: "string",
name: path.substring(path.indexOf("{") + 1, path.indexOf("}"))
}
];
}
if (route) {
route.method = method;
route.path = path;
route.parameters = parameters;
} else
route = {
method,
path,
parameters
};
}
});
if (
route &&
!routes.find(r => {
if (r.path === route.path && r.method === route.method) return true;
})
) {
if (route.path.startsWith("/")) routes.push(route);
else {
let d = data.replace(/\s/g, "").split(";");
let actualPath = "";
d.find(l => {
if (l.includes(route.path)) {
actualPath = l.split("=")[1].replace(/"/g, "");
return true;
}
});
route.path = actualPath;
if (
!routes.find(r => {
if (r.path === route.path && r.method === route.method)
return true;
})
)
routes.push(route);
}
}
});
}
return routes;
};