sourceafBedSheet::BedSheetWebMod.fan

using concurrent::ActorPool
using concurrent::AtomicRef
using web::WebMod
using afIoc::Registry
using afIoc::RegistryBuilder

** The top-level `web::WebMod` to be passed to [Wisp]`http://fantom.org/doc/wisp/index.html`. 
const class BedSheetWebMod : WebMod {
    private const static Log log := Utils.getLog(BedSheetWebMod#)

    const Str           moduleName
    const Int           port
    const [Str:Obj]     bedSheetOptions
    const [Str:Obj]?    registryOptions
    
    private const AtomicRef reg := AtomicRef()
    
    ** The 'afIoc' registry
    Registry registry {
        get { reg.val }
        set { throw Err() }
    }
    
    new make(Str moduleName, Int port, [Str:Obj] bedSheetOptions, [Str:Obj]? registryOptions := null) {
        this.moduleName         = moduleName
        this.port               = port
        this.registryOptions    = registryOptions
        this.bedSheetOptions    = bedSheetOptions
    }

    override Void onService() {
        req.mod = this
        try {
            httpPipeline := (HttpPipeline) registry.dependencyByType(HttpPipeline#)
            httpPipeline.service
        } catch (Err err) {
            // theoretically, this should have already been dealt with by our Err Pipeline Processor...
            // ...but it's handy for BedSheet development!
            errPrinter := (ErrPrinter) registry.dependencyByType(ErrPrinter#)
            Env.cur.err.printLine(errPrinter.errToStr(err))
            throw err
        }
    }

    override Void onStart() {
        log.info(BsLogMsgs.bedSheetWebModStarting(moduleName, port))

        // pod name given...
        Pod? pod
        if (!moduleName.contains("::")) {
            pod = Pod.find(moduleName, true)
            log.info(BsLogMsgs.bedSheetWebModFoundPod(pod))
        }

        // mod name given...
        Type? mod
        if (moduleName.contains("::")) {
            mod = Type.find(moduleName, true)
            log.info(BsLogMsgs.bedSheetWebModFoundType(mod))
        }

        // construct this last so logs look nicer ("...adding module IocModule")
        bob := RegistryBuilder()
        if (pod != null) {
            bob.addModulesFromDependencies(pod, true)
        }
        if (mod != null) {
            bob.addModule(BedSheetModule#)
            bob.addModule(mod)          
        }
        
        bannerText  := easterEgg("Alien-Factory BedSheet v${typeof.pod.version}, IoC v${Registry#.pod.version}")
        options     := Str:Obj["bannerText":bannerText]
        if (registryOptions != null)
            options.setAll(registryOptions)

        if (bedSheetOptions.containsKey("iocModules"))
            bob.addModules(bedSheetOptions["iocModules"])

        reg.val = bob.build(options).startup

        if (bedSheetOptions["pingProxy"] == true) {
            pingPort := (Int) bedSheetOptions["pingProxyPort"]
            destroyer := registry.autobuild(AppDestroyer#, [ActorPool(), pingPort]) as AppDestroyer
            destroyer.start
        }
    }

    override Void onStop() {
        reg := (Registry?) reg.val
        reg?.shutdown
        log.info(BsLogMsgs.bedSheetWebModStopping(moduleName))
    }
    
    private Str easterEgg(Str title) {
        quotes := loadQuotes
        if (quotes.isEmpty || (Int.random(0..8) != 2))
            return title
        return quotes[Int.random(0..<quotes.size)]
    }
    
    private Str[] loadQuotes() {
        typeof.pod.file(`/res/misc/quotes.txt`).readAllLines.exclude { it.isEmpty || it.startsWith("#")}
    }
}