path: root/src/nodejs/unit-http/loader.js

// can only be ran as part of a --require param on the node process
if (module.parent && module.parent.id === "internal/preload") {
    const { Module } = require("module")

    if (!Module.prototype.require.__unit_loader) {
        const http = require("./http")
        const websocket = require("./websocket")

        const original = Module.prototype.require;

        Module.prototype.require = function (id) {
            switch(id) {
                case "http":
                case "unit-http":
                    return http

                case "websocket":
                case "unit-http/websocket":
                    return websocket
            }

            return original.apply(this, arguments);
        }

        Module.prototype.require.__unit_loader = true;
    }
}