forked from zhurui/management
19 lines
427 B
JavaScript
19 lines
427 B
JavaScript
|
module.exports = function (_require) {
|
||
|
|
_require = _require || require
|
||
|
|
var main = _require.main
|
||
|
|
if (main && isIISNode(main)) return handleIISNode(main)
|
||
|
|
else return main ? main.filename : process.cwd()
|
||
|
|
}
|
||
|
|
|
||
|
|
function isIISNode (main) {
|
||
|
|
return /\\iisnode\\/.test(main.filename)
|
||
|
|
}
|
||
|
|
|
||
|
|
function handleIISNode (main) {
|
||
|
|
if (!main.children.length) {
|
||
|
|
return main.filename
|
||
|
|
} else {
|
||
|
|
return main.children[0].filename
|
||
|
|
}
|
||
|
|
}
|