Виявити, якщо викликано через вимагає або безпосередньо командним рядком

Як я можу виявити, чи викликався мій файл Node.JS за допомогою SH: node path-to-fileабо JS require('path-to-file'):?

Це еквівалент Node.JS моєму попередньому запитанню в Perl: Як я можу запустити свій скрипт Perl, лише якщо він не завантажений запитом?

if (require.main === module) {
    console.log('called directly');
} else {
    console.log('required as a module');
}

Дивіться документацію для цього тут: https://nodejs.org/docs/latest/api/modules.html#modules_accessing_the_main_module

Існує ще один, трохи коротший спосіб (не окреслений у згаданих документах).

var runningAsScript = !module.parent;

Я детальніше розповів про те, як це все працює під кришкою в цій публікації блогу .

Мене трохи збентежила термінологія, що використовується в поясненнях. Тому мені довелося зробити пару швидких тестів.

Я виявив, що вони дають однакові результати:

var isCLI = !module.parent;
var isCLI = require.main === module;

А для інших розгублених людей (і безпосередньо відповісти на питання):

var isCLI = require.main === module;
var wasRequired = !isCLI;

Як і в Python, я завжди намагаюся згадати, як написати цей фрагмент кодового коду. Тому я вирішив створити для нього простий модуль. Мені потрібно було трохи розробитись, оскільки доступ до інформації модуля абонента не є прямим, але було цікаво подивитися, як це можна зробити.

Тож ідея полягає в тому, щоб зателефонувати в модуль і запитати його, чи є модуль абонента основним. Треба з'ясувати модуль функції виклику. Мій перший підхід був варіантом прийнятої відповіді:

module.exports = function () {
    return require.main === module.parent;
};

Але це не гарантовано спрацює. module.parentвказує на модуль, який завантажив нас у пам'ять, а не на той, хто нас дзвонить. Якщо саме модуль абонента завантажив цей модуль помічника в пам'ять, це добре. Але якби не так, ми безпорадні. Тому нам потрібно спробувати щось інше. Моє рішення полягало в тому, щоб генерувати стек стека і отримувати звідти ім'я модуля абонента:

module.exports = function () {
    // generate a stack trace
    const stack = (new Error()).stack;
    // the third line refers to our caller
    const stackLine = stack.split("\n")[2];
    // extract the module name from that line
    const callerModuleName = /\((.*):\d+:\d+\)$/.exec(stackLine)[1];

    return require.main.filename === callerModuleName;
};

Тепер ми можемо зробити:

if (require("./is-main-module")()) {  // notice the `()` at the end
    // do something
} else {
    // do something else
}

Або ще читабельніше:

const isMainModule = require("./is-main-module");

if (isMainModule()) {
    // do something
} else {
    // do something else
}

Неможливо забути :-)

Спробуйте це, якщо ви використовуєте модулі ES6:

if (process.mainModule.filename === __filename) {
  console.log('running as main module')
}