Skip to content

Commit 0d264ec

Browse files
pavel-pythagorapavel-pythagora
committed
Add husky
1 parent faecba2 commit 0d264ec

File tree

9 files changed

+126
-108
lines changed

9 files changed

+126
-108
lines changed

‎.husky/pre-commit

+4
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
#!/usr/bin/env sh
2+
. "$(dirname -- "$0")/_/husky.sh"
3+
4+
npm run lint

‎package-lock.json

+17-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎package.json

+7-3
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,10 @@
55
"license": "ISC",
66
"description": "This repository hosts the 'code-processing' npm package, which contains a set of code processing methods for generating tests.",
77
"main": "src/index.js",
8-
"scripts": {},
8+
"scripts": {
9+
"prepare": "husky install",
10+
"lint": "npx eslint --fix --ext .js ."
11+
},
912
"repository": {
1013
"type": "git",
1114
"url": "git+https://github.com/Pythagora-io/js-code-processing.git"
@@ -46,6 +49,7 @@
4649
"eslint-config-standard": "^17.1.0",
4750
"eslint-plugin-import": "^2.28.0",
4851
"eslint-plugin-n": "^16.0.1",
49-
"eslint-plugin-promise": "^6.1.1"
52+
"eslint-plugin-promise": "^6.1.1",
53+
"husky": "^8.0.0"
5054
}
51-
}
55+
}

‎src/const/colors.js

+7-7
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
module.exports = {
2-
red: '\x1b[31m',
3-
yellow: '\x1b[33m',
4-
green: '\x1b[32m',
5-
blue: '\x1b[34m',
6-
reset: '\x1b[0m',
7-
bold: '\x1b[1m'
8-
}
2+
red: "\x1b[31m",
3+
yellow: "\x1b[33m",
4+
green: "\x1b[32m",
5+
blue: "\x1b[34m",
6+
reset: "\x1b[0m",
7+
bold: "\x1b[1m"
8+
};

‎src/const/common.js

+17-17
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,18 @@
11
module.exports = {
2-
PYTHAGORA_TESTS_DIR: 'pythagora_tests',
3-
PYTHAGORA_METADATA_DIR: '.pythagora',
4-
METADATA_FILENAME: 'metadata.json',
5-
REVIEW_DATA_FILENAME: 'review.json',
6-
EXPORT_METADATA_FILENAME: 'export.json',
7-
CONFIG_FILENAME: 'config.json',
8-
PYTHAGORA_ASYNC_STORE: 42069420,
9-
PYTHAGORA_DELIMITER: '-_-',
10-
EXPORTED_TESTS_DIR: 'pythagora_tests/exported_tests',
11-
EXPORTED_TESTS_DATA_DIR: 'pythagora_tests/exported_tests/data',
12-
SRC_TO_ROOT: '../../../',
13-
MIN_TOKENS_FOR_GPT_RESPONSE: 1640,
14-
MAX_GPT_MODEL_TOKENS: 8192,
15-
PYTHAGORA_UNIT_TESTS_VERSION: 1,
16-
PYTHAGORA_UNIT_DIR: 'pythagora_tests/unit',
17-
PYTHAGORA_API_SERVER: 'https://api.pythagora.io',
18-
}
2+
PYTHAGORA_TESTS_DIR: "pythagora_tests",
3+
PYTHAGORA_METADATA_DIR: ".pythagora",
4+
METADATA_FILENAME: "metadata.json",
5+
REVIEW_DATA_FILENAME: "review.json",
6+
EXPORT_METADATA_FILENAME: "export.json",
7+
CONFIG_FILENAME: "config.json",
8+
PYTHAGORA_ASYNC_STORE: 42069420,
9+
PYTHAGORA_DELIMITER: "-_-",
10+
EXPORTED_TESTS_DIR: "pythagora_tests/exported_tests",
11+
EXPORTED_TESTS_DATA_DIR: "pythagora_tests/exported_tests/data",
12+
SRC_TO_ROOT: "../../../",
13+
MIN_TOKENS_FOR_GPT_RESPONSE: 1640,
14+
MAX_GPT_MODEL_TOKENS: 8192,
15+
PYTHAGORA_UNIT_TESTS_VERSION: 1,
16+
PYTHAGORA_UNIT_DIR: "pythagora_tests/unit",
17+
PYTHAGORA_API_SERVER: "https://api.pythagora.io"
18+
};

‎src/index.js

+1-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ module.exports = {
1616
UnitTests,
1717
UnitTestsExpand,
1818
API
19-
}
19+
};
2020

2121
// Usage example:
2222
// const Api = new API(<API ENDPOINT URL>, <PYTHAGORA API KEY>, <API KEY TYPE>);

‎src/utils/code.js

+29-34
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,7 @@ function replaceRequirePaths(code, currentPath, testFilePath) {
3131
modulePath = importPath;
3232
}
3333

34-
if (!modulePath.startsWith("./") && !modulePath.startsWith("../"))
35-
return match;
34+
if (!modulePath.startsWith("./") && !modulePath.startsWith("../")) { return match; }
3635

3736
const absoluteRequirePath = path.resolve(currentPath, modulePath);
3837

@@ -51,7 +50,6 @@ function replaceRequirePaths(code, currentPath, testFilePath) {
5150

5251
async function getAstFromFilePath(filePath) {
5352
let data = await fs.readFile(filePath, "utf8");
54-
let nodeTypesStack = [];
5553
// Remove shebang if it exists
5654
if (data.indexOf("#!") === 0) {
5755
data = "//" + data;
@@ -60,7 +58,7 @@ async function getAstFromFilePath(filePath) {
6058
const ast = babelParser.parse(data, {
6159
sourceType: "module", // Consider input as ECMAScript module
6260
locations: true,
63-
plugins: ["jsx", "objectRestSpread", "typescript"], // Enable JSX, typescript and object rest/spread syntax
61+
plugins: ["jsx", "objectRestSpread", "typescript"] // Enable JSX, typescript and object rest/spread syntax
6462
});
6563

6664
return ast;
@@ -97,14 +95,14 @@ async function getModuleTypeFromFilePath(ast) {
9795
moduleType = "CommonJS";
9896
path.stop(); // Stop traversal when a CommonJS statement is found
9997
}
100-
},
98+
}
10199
});
102100

103101
return moduleType;
104102
}
105103

106104
function collectTopRequires(node) {
107-
let requires = [];
105+
const requires = [];
108106
babelTraverse(node, {
109107
VariableDeclaration(path) {
110108
if (
@@ -117,7 +115,7 @@ function collectTopRequires(node) {
117115
},
118116
ImportDeclaration(path) {
119117
requires.push(generator(path.node).code);
120-
},
118+
}
121119
});
122120
return requires;
123121
}
@@ -147,19 +145,19 @@ function getFullPathFromRequireOrImport(importPath, filePath) {
147145
if (
148146
importPath &&
149147
(importPath.startsWith("./") || importPath.startsWith("../"))
150-
)
148+
) {
151149
importPath = path.resolve(
152150
filePath.substring(0, filePath.lastIndexOf("/")),
153151
importPath
154152
);
155-
if (importPath.lastIndexOf(".js") + ".js".length !== importPath.length)
156-
importPath += ".js";
153+
}
154+
if (importPath.lastIndexOf(".js") + ".js".length !== importPath.length) { importPath += ".js"; }
157155
return importPath;
158156
}
159157

160158
function getRelatedFunctions(node, ast, filePath, functionList) {
161-
let relatedFunctions = [];
162-
let requiresFromFile = collectTopRequires(ast);
159+
const relatedFunctions = [];
160+
const requiresFromFile = collectTopRequires(ast);
163161

164162
function processNodeRecursively(node) {
165163
if (node.type === "CallExpression") {
@@ -189,12 +187,12 @@ function getRelatedFunctions(node, ast, filePath, functionList) {
189187
requiredPath = getPathFromRequireOrImport(requiredPath);
190188
requiredPath = getFullPathFromRequireOrImport(requiredPath, filePath);
191189
}
192-
let functionFromList = functionList[requiredPath + ":" + funcName];
190+
const functionFromList = functionList[requiredPath + ":" + funcName];
193191
if (functionFromList) {
194192
relatedFunctions.push(
195193
_.extend(functionFromList, {
196194
fileName: requiredPath,
197-
importPath,
195+
importPath
198196
})
199197
);
200198
}
@@ -252,7 +250,7 @@ async function stripUnrelatedFunctions(filePath, targetFuncNames) {
252250
}
253251

254252
function processAst(ast, cb) {
255-
let nodeTypesStack = [];
253+
const nodeTypesStack = [];
256254
babelTraverse(ast, {
257255
enter(path) {
258256
nodeTypesStack.push(path.node.type);
@@ -276,7 +274,7 @@ function processAst(ast, cb) {
276274
// module.exports.funcName = function() { ... }
277275
// module.exports = function() { ... }
278276
const loc = path.node.loc.start;
279-
let funcName =
277+
const funcName =
280278
left.property.name || `anon_func_${loc.line}_${loc.column}`;
281279
return cb(funcName, path, "exportObj");
282280
}
@@ -307,9 +305,7 @@ function processAst(ast, cb) {
307305
}
308306
});
309307
}
310-
}
311-
// Handle TypeScript transpiled exports
312-
else if (
308+
} /* Handle TypeScript transpiled exports */ else if (
313309
left.type === "MemberExpression" &&
314310
left.object.name === "exports"
315311
) {
@@ -416,7 +412,7 @@ function processAst(ast, cb) {
416412
},
417413
exit(path) {
418414
nodeTypesStack.pop();
419-
},
415+
}
420416
});
421417
}
422418

@@ -425,7 +421,7 @@ function getSourceCodeFromAst(ast) {
425421
}
426422

427423
function collectTestRequires(node) {
428-
let requires = [];
424+
const requires = [];
429425
babelTraverse(node, {
430426
ImportDeclaration(path) {
431427
if (
@@ -435,12 +431,11 @@ function collectTestRequires(node) {
435431
) {
436432
const requireData = {
437433
code: generator(path.node).code,
438-
functionNames: [],
434+
functionNames: []
439435
};
440436

441437
_.forEach(path.node.specifiers, (s) => {
442-
if (s.local && s.local.name)
443-
requireData.functionNames.push(s.local.name);
438+
if (s.local && s.local.name) { requireData.functionNames.push(s.local.name); }
444439
});
445440

446441
requires.push(requireData);
@@ -454,7 +449,7 @@ function collectTestRequires(node) {
454449
) {
455450
const requireData = {
456451
code: generator(path.node).code,
457-
functionNames: [],
452+
functionNames: []
458453
};
459454

460455
// In case of a CommonJS require, the function name is usually the variable identifier of the parent node
@@ -468,36 +463,36 @@ function collectTestRequires(node) {
468463

469464
requires.push(requireData);
470465
}
471-
},
466+
}
472467
});
473468
return requires;
474469
}
475470

476471
function getRelatedTestImports(ast, filePath, functionList) {
477-
let relatedCode = [];
478-
let requiresFromFile = collectTestRequires(ast);
472+
const relatedCode = [];
473+
const requiresFromFile = collectTestRequires(ast);
479474

480-
for (let fileImport in requiresFromFile) {
475+
for (const fileImport in requiresFromFile) {
481476
let requiredPath = getPathFromRequireOrImport(
482477
requiresFromFile[fileImport].code
483478
);
484479
requiredPath = getFullPathFromRequireOrImport(requiredPath, filePath);
485480

486481
_.forEach(requiresFromFile[fileImport].functionNames, (funcName) => {
487-
let functionFromList = functionList[requiredPath + ":" + funcName];
482+
const functionFromList = functionList[requiredPath + ":" + funcName];
488483
if (functionFromList) {
489484
relatedCode.push(
490485
_.extend(functionFromList, {
491-
fileName: requiredPath,
486+
fileName: requiredPath
492487
})
493488
);
494489
}
495490
});
496491
}
497492

498-
for (let relCode of relatedCode) {
493+
for (const relCode of relatedCode) {
499494
let relatedCodeImports = "";
500-
for (let func of relCode.relatedFunctions) {
495+
for (const func of relCode.relatedFunctions) {
501496
if (func.importPath) {
502497
relatedCodeImports += `${func.importPath}\n`;
503498
}
@@ -521,5 +516,5 @@ module.exports = {
521516
processAst,
522517
getModuleTypeFromFilePath,
523518
getSourceCodeFromAst,
524-
getRelatedTestImports,
519+
getRelatedTestImports
525520
};

‎src/utils/common.js

+12-12
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
const fs = require("fs");
22

33
async function checkDirectoryExists(directoryPath) {
4-
try {
5-
const stats = await fs.promises.stat(directoryPath);
6-
return stats.isDirectory();
7-
} catch (error) {
8-
if (error.code === 'ENOENT') {
9-
// Directory does not exist
10-
return false;
11-
}
12-
// Other error occurred
13-
throw error;
4+
try {
5+
const stats = await fs.promises.stat(directoryPath);
6+
return stats.isDirectory();
7+
} catch (error) {
8+
if (error.code === "ENOENT") {
9+
// Directory does not exist
10+
return false;
1411
}
12+
// Other error occurred
13+
throw error;
14+
}
1515
}
1616

1717
module.exports = {
18-
checkDirectoryExists
19-
}
18+
checkDirectoryExists
19+
};

0 commit comments

Comments
 (0)