path: root/featureClass/featureManager.js

/// <reference types="../../CTAutocomplete" />
/// <reference lib="es2015" />
const Instant = Java.type("java.time.Instant");
import logger from "../logger";
const File = Java.type("java.io.File")
import metadata from "../metadata.js"
import soopyV2Server from "../socketConnection";
import { registerForge as registerForgeBase, unregisterForge as unregisterForgeBase} from "./forgeEvents.js"

const JSLoader = Java.type("com.chattriggers.ctjs.engine.langs.js.JSLoader")
const UrlModuleSourceProvider = Java.type("org.mozilla.javascript.commonjs.module.provider.UrlModuleSourceProvider")
const UrlModuleSourceProviderInstance = new UrlModuleSourceProvider(null, null)
const StrongCachingModuleScriptProviderClass = Java.type("org.mozilla.javascript.commonjs.module.provider.StrongCachingModuleScriptProvider")
let StrongCachingModuleScriptProvider = new StrongCachingModuleScriptProviderClass(UrlModuleSourceProviderInstance)
let CTRequire = new JSLoader.CTRequire(StrongCachingModuleScriptProvider)

let loadedModules = []

function RequireNoCache(place){
    if(loadedModules.includes(place)){
        loadedModules.push(place)
        return require(place) //performance optimisation
    }
    if(!logger.isDev) return require(place)
    
    StrongCachingModuleScriptProvider = new StrongCachingModuleScriptProviderClass(UrlModuleSourceProviderInstance)
    CTRequire = new JSLoader.CTRequire(StrongCachingModuleScriptProvider)
    return CTRequire(place)
}

class FeatureManager {
    constructor(){

        this.isDev = logger.isDev

        this.messagePrefix = "&6[SOOPY V2]&7 "
        this.enabled = true //make triggers work with this context

        this.features = {};
        this.events = {}
        this.eventObjects = {}
        this.soopyEventHandlers = {}

        this.parent = undefined

        this.finishedLoading = false

        this.commandFuncs = {}

        this.lastEventId = 0

        this.customEvents = {}
        this.lastChatEventId = 0

        this.forgeEvents = {}
        this.lastForgeEventId = 0

        this.lastSoopyEventId = 0

        this.featureSettingsDataLastUpdated = false

        //PERFORMANCE RECORDING
        this.recordingPerformanceUsage = false
        this.performanceUsage = {} //{<moduleName>: {<event>: {time: 0, count: 0}}}

        this.longEventTime = 20


        this.featureMetas = {}

        this.featureSettingsData = {}

        new Thread(()=>{

            try{
                FileLib.getUrlContent("http://soopymc.my.to/api/soopyv2/ping")
            }catch(e){
                ChatLib.chat(this.messagePrefix + "&cError: Could not connect to Soopy's server. This may cause issues with features breaking but will (hopefully) be back soon.")
            }

            this.loadFeatureMetas()
            
            this.loadFeatureSettings()
            
            Object.keys(this.featureMetas).forEach((feature)=>{
                if(this.featureSettingsData[feature] && this.featureSettingsData[feature].enabled){
                    this.loadFeature(feature)
                }
            })

            this.finishedLoading = true
        }).start()

        this.registerStep(false, 30, ()=>{
            if(this.featureSettingsDataLastUpdated){
                new Thread(()=>{
                    this.saveFeatureSettings()
                }).start()
            }
        }, this)

        this.registerEvent("worldUnload", this.saveFeatureSettings, this)

        this.registerEvent("gameUnload", ()=>{
            this.saveFeatureSettings()
            this.unloadAllFeatures()

            this.enabled = false
        }, this)
        this.registerStep(true, 2, ()=>{
            if(this.reloadModuleTime!==0 && Date.now()-this.reloadModuleTime > 0){
                new Thread(()=>{
                    this.reloadModuleTime = 0
                    this.reloadingModules.forEach(m=>{
                        this.unloadFeature(m)
                    })
                    this.reloadingModules.forEach(m=>{
                        this.loadFeature(m)
                    })
                    this.reloadingModules = []
                }).start()
            }
        }, this)

        this.watches = {}
        this.addedWatches = []
        this.watchService = Java.type("java.nio.file.FileSystems").getDefault().newWatchService();
        this.reloadingModules = []
        this.reloadModuleTime = 0
        new Thread(()=>{
            while(this.enabled){
                key = this.watchService.take();
                let moduleToReload = this.watches[key]
                if(this.features[moduleToReload] && !this.reloadingModules.includes(moduleToReload)){ //if enabled && not alr in queue
                    this.reloadingModules.push(moduleToReload)
                    this.reloadModuleTime = Date.now()+5000
                }
                key.pollEvents()/*.forEach(event=>{
                    console.log(event.context().toString())
                })*/
                key.reset();
            }
        }).start()

        this.registerCommand("soopyunloadfeature", (args)=>{
            new Thread(()=>{
                this.unloadFeature(args)
            }).start()
        }, this)
        this.registerCommand("soopyloadfeature", (args)=>{
            new Thread(()=>{
                this.loadFeature(args)
            }).start()
        }, this)
        this.registerCommand("soopyreloadfeature", (args)=>{
            new Thread(()=>{
                this.unloadFeature(args)
                
                this.loadFeature(args)
            }).start()
        }, this)
        this.registerCommand("soopysetlongeventtime", (args)=>{
            this.longEventTime = parseInt(args)
        }, this)
        this.registerCommand("soopylaginformation", (args)=>{ //TODO: make this a dedicated GUI
            new Thread(()=>{
                this.recordingPerformanceUsage = true
                this.performanceUsage = {}
                ChatLib.chat(this.messagePrefix + "Recording performance impact, this will take 10 seconds to complete!")

                Thread.sleep(10000)

                let totalMsGlobal = 0
                this.recordingPerformanceUsage = false
                ChatLib.chat(this.messagePrefix + "Performance impact:")
                Object.keys(this.performanceUsage).sort((a, b)=>{
                    let totalMsA = 0
                    Object.keys(this.performanceUsage[a]).forEach((event)=>{
                        totalMsA += this.performanceUsage[a][event].time
                    })
                    let totalMsB = 0
                    Object.keys(this.performanceUsage[b]).forEach((event)=>{
                        totalMsB += this.performanceUsage[b][event].time
                    })

                    return totalMsA-totalMsB
                }).forEach((moduleName)=>{
                    let totalMs = 0
                    let totalCalls = 0
                    Object.keys(this.performanceUsage[moduleName]).forEach((event)=>{
                        totalMs += this.performanceUsage[moduleName][event].time
                        totalCalls += this.performanceUsage[moduleName][event].count
                    })

                    totalMsGlobal += totalMs

                    ChatLib.chat("&eModule: &7" + moduleName)
                    ChatLib.chat("&eTotal: &7" + totalMs.toFixed(2) + "ms (" + totalCalls + " calls)")
                    Object.keys(this.performanceUsage[moduleName]).sort((a, b)=>{return this.performanceUsage[moduleName][a].time-this.performanceUsage[moduleName][b].time}).forEach((event)=>{
                        ChatLib.chat("  &eEvent:&7 " + event + " - " + this.performanceUsage[moduleName][event].time.toFixed(2) + "ms (" + this.performanceUsage[moduleName][event].count + " calls) [" + ((this.performanceUsage[moduleName][event].time/this.performanceUsage[moduleName][event].count).toFixed(2)) + "ms avg]")
                    })
                })

                ChatLib.chat("&eTotal: &7" + totalMsGlobal.toFixed(2) + "ms")
            }).start()
        }, this)
    }

    loadFeatureSettings(){
        logger.logMessage("Loading settings", 4)

        let data = FileLib.read("soopyAddonsData", "soopyaddonsbetafeaturesdata.json")

        if(!data){
            this.loadDefaultFeatureSettings();
            return;
        }

        data = JSON.parse(data)

        this.featureSettingsData = data

        this.ensureNewSettingsExist()
    }

    saveFeatureSettings(){
        if(!this.featureSettingsDataLastUpdated) return
        
        FileLib.write("soopyAddonsData", "soopyaddonsbetafeaturesdata.json", JSON.stringify(this.featureSettingsData))
        
        this.featureSettingsDataLastUpdated = false

        logger.logMessage("Saved settings", 4)
    }

    loadDefaultFeatureSettings(){
        Object.keys(this.featureMetas).forEach((feature)=>{
            this.featureSettingsData[feature] = {
                enabled: this.featureMetas[feature].defaultEnabled,
                subSettings: {}
            }
        })
        
        this.featureSettingsDataLastUpdated= true

        logger.logMessage("Loaded default settings", 4)
    }

    ensureNewSettingsExist(){
        Object.keys(this.featureMetas).forEach((feature)=>{
            if(!this.featureSettingsData[feature]){
                this.featureSettingsData[feature] = {
                    enabled: this.featureMetas[feature].defaultEnabled,
                    subSettings: {}
                }
                this.featureSettingsDataLastUpdated= true
                logger.logMessage("Loaded default settings for " + feature, 4)
            }
        })
    }

    startCatchingEvent(event){
        if(this.eventObjects[event]) return

        //SBA compatability or something (removed)
        // if(event === "renderOverlay"){
        //     let lastPartialTick = undefined
        //     this.eventObjects[event] = register(event, (...args)=>{
        //         let pTicks = Tessellator.getPartialTicks()
        //         if(pTicks !== lastPartialTick){
        //             lastPartialTick = pTicks
        //             this.triggerEvent(event, args)
        //         }
        //     })
        // }else{
            this.eventObjects[event] = register(event, (...args)=>{
                // let start = Date.now()
                this.triggerEvent(event, args)
                // this.eventTimingData[event] = (this.eventTimingData[event] || 0)+(Date.now()-start)
            })
        //}

        logger.logMessage("Registered " + event + " event", 4)
    }

    triggerEvent(event, args){
        if(this.events[event])
        try{
            for(Event of Object.values(this.events[event])){
                if(Event.context.enabled){
                    this.startRecordingPerformance(Event.context.constructor.name, event)
                    let start = Date.now()
                    Event.func.call(Event.context, ...args)
                    let time = Date.now()-start
                    if(time > this.longEventTime){
                        logger.logMessage("Long event triggered [" + time + "ms] (" + Event.context.constructor.name + "/" + event + ")", 3)
                    }
                    this.stopRecordingPerformance(Event.context.constructor.name, event)
                }
            }
        }catch(e){
            logger.logMessage("Error in " + event + " event: " + JSON.stringify(e, undefined, 2), 2)
            
            soopyV2Server.reportError(e, "Error in " + event + " event.")
        }
    }
    triggerSoopy(event, args){
        if(this.soopyEventHandlers[event])
        try{
            for(Event of Object.values(this.soopyEventHandlers[event])){
                if(Event.context.enabled){
                    this.startRecordingPerformance(Event.context.constructor.name, event)
                    let start = Date.now()
                    Event.func.call(Event.context, ...args)
                    let time = Date.now()-start
                    if(time > this.longEventTime){
                        logger.logMessage("Long event triggered [" + time + "ms] (" + context.constructor.name + "/" + event + ")", 3)
                    }
                    this.stopRecordingPerformance(Event.context.constructor.name, event)
                }
            }
        }catch(e){
            logger.logMessage("Error in soopy " + event + " event: " + JSON.stringify(e, undefined, 2), 2)
            soopyV2Server.reportError(e, "Error in soopy " + event + " event.")
        }
    }

    stopCatchingEvent(event){
        if(!this.eventObjects[event]) return

        this.eventObjects[event].unregister()
        delete this.eventObjects[event]
        delete this.events[event]
        logger.logMessage("Unregistered " + event + " event", 4)
    }

    registerEvent(event, func, context){
        if(!this.events[event]){
            this.events[event] = []
            this.startCatchingEvent(event)
        }

        let theEvent = {
            func: func,
            context: context,
            id: this.lastEventId++,
            event: event
        }
        this.events[event].push(theEvent)

        return theEvent
    }
    registerSoopy(event, func, context){
        if(!this.soopyEventHandlers[event]){
            this.soopyEventHandlers[event] = []
        }

        let theEvent = {
            func: func,
            context: context,
            id: this.lastSoopyEventId++,
            event: event
        }
        this.soopyEventHandlers[event].push(theEvent)

        return theEvent
    }

    registerChat(criteria, func, context){
        let event = this.registerCustom("chat", func, context)

        event.trigger.setChatCriteria(criteria)

        return event
    }

    registerActionBar(criteria, func, context){

        let event = this.registerCustom("actionBar", func, context)

        event.trigger.setChatCriteria(criteria)

        return event
    }
    registerCommand(commandName, func, context){

        let event = this.registerCustom("command", func, context)

        event.trigger.setName(commandName, true)

        return event
    }
    registerStep(isFps, interval, func, context){
        let event = this.registerCustom("step", func, context)

        event.trigger[isFps?"setFps":"setDelay"](interval)

        return event
    }

    registerCustom(type, func, context){
        let id = this.lastChatEventId++

        this.customEvents[id] = {
            func: func,
            context: context,
            trigger: register(type, (...args)=>{
                try{
                    if(context.enabled){
                        this.startRecordingPerformance(context.constructor.name, type)
                        let start = Date.now()
                        func.call(context, ...(args || []))
                        let time = Date.now()-start
                        if(time > this.longEventTime){
                            logger.logMessage("Long event triggered [" + time + "ms] (" + context.constructor.name + "/" + type + ")", 3)
                        }
                        this.stopRecordingPerformance(context.constructor.name, type)
                    }
                }catch(e){
                    logger.logMessage("Error in " + type + " event: " + JSON.stringify(e, undefined, 2), 2)
                    
                    soopyV2Server.reportError(e, "Error in " + type + " event.")
                }
            }),
            id: id
        }

        return this.customEvents[id]
    }

    registerForge(event, func, context){ 
        let id = this.lastForgeEventId++

        this.forgeEvents[id] = {
            func: func,
            context: context,
            trigger: registerForgeBase(event, (...args)=>{
                try{
                    if(context.enabled){
                        this.startRecordingPerformance(context.constructor.name, event.class.name)
                        let start = Date.now()
                        func.call(context, ...(args || []))
                        let time = Date.now()-start
                        if(time > this.longEventTime){
                            console.log(this.messagePrefix + "Long (forge) event triggered (" + context.constructor.name + "/" + event.class.toString() + ")")
                        }
                        this.stopRecordingPerformance(context.constructor.name, event.class.name)
                    }
                }catch(e){
                    logger.logMessage("Error in " + event.class.toString() + " (forge) event: " + JSON.stringify(e, undefined, 2), 2)
                    
                    soopyV2Server.reportError(e, "Error in " + event.class.toString() + " (forge) event.")
                }
            }),
            id: id
        }

        return this.forgeEvents[id]
    }

    unregisterForge(event){
        unregisterForgeBase(this.forgeEvents[event.id].trigger)
        delete this.forgeEvents[event.id]
    }

    unregisterCustom(event){
        event.trigger.unregister()

        delete this.customEvents[event.id]
    }

    unregisterEvent(event){
        if(!this.events[event.event]) return

        this.events[event.event] = this.events[event.event].filter((e)=>{
            return e.id !== event.id
        })
        
        if(this.events[event.event].length === 0){
            this.stopCatchingEvent(event.event)
            delete this.events[event.event]
        }
    }

    unregisterSoopy(event){
        if(!this.soopyEventHandlers[event.event]) return

        this.soopyEventHandlers[event.event] = this.soopyEventHandlers[event.event].filter((e)=>{
            return e.id !== event.id
        })
        
        if(this.soopyEventHandlers[event.event].length === 0){
            delete this.events[event.event]
        }
    }

    loadFeatureMetas(){
        let featuresDir = new File("./config/ChatTriggers/modules/" + metadata.name + "/features")

        featuresDir.list().forEach((pathName)=>{
            if(pathName.includes(".")) return;

            try{
                let data = JSON.parse(FileLib.read( metadata.name + "/features/" + pathName,"metadata.json"))
                if(data === null){ 
                    return;
                }
                data.id = pathName
                this.featureMetas[pathName] = data
            }catch(e){
                console.log("Error loading feature metadata for " + pathName)
                console.log(JSON.stringify(e, undefined, 2))
            }
        })
    }

    loadFeature(feature){ //run in seperate thread so onenable can do network requests
        if(this.features[feature]) return
        
        try{
            
            let LoadedFeature = RequireNoCache("../features/" + feature + "/index.js")
            // let LoadedFeature = RequireNoCache(new File("config/ChatTriggers/modules/" + metadata.name + "/features/" + feature + "/index.js"))
            
            this.features[feature] = LoadedFeature
    
            LoadedFeature.class.setId(feature)
    
            LoadedFeature.class._onEnable(this)
    
            logger.logMessage("Loaded feature " + feature, 3)

            if(this.isDev && !this.addedWatches.includes(feature)){
                this.addedWatches.push(feature)
                let path = Java.type("java.nio.file.Paths").get("./config/ChatTriggers/modules/SoopyV2/features/" + feature + "/");
                this.watches[path.register(this.watchService, Java.type("java.nio.file.StandardWatchEventKinds").ENTRY_MODIFY)] = feature
            }
        }catch(e){
            logger.logMessage("Error loading feature " + feature, 1)
            console.log(JSON.stringify(e, undefined, 2))
            ChatLib.chat(this.messagePrefix + "Error loading feature " + feature)

            soopyV2Server.reportError(e, "Error loading feature " + feature)
        }

        return this
    }

    unloadFeature(feature){
        if(!this.features[feature]) return

        this.features[feature].class._onDisable()

        delete this.features[feature]

        logger.logMessage("Unloaded feature " + feature, 3)

        return this
    }

    unloadAllFeatures(){
        Object.keys(this.features).forEach((feature)=>{
            this.unloadFeature(feature)
        })
    }

    isFeatureLoaded(feature){
        return !!this.features[feature]
    }

    getLoadedFeatures(){
        return Object.keys(this.features)
    }

    createCustomEvent(eventId){
        logger.logMessage("Registered custom " + eventId + " event", 4)

        return {
            trigger: (...args)=>{
                this.triggerSoopy(eventId, args)
            }
        }
    }

    startRecordingPerformance(feature, event){
        if(!this.recordingPerformanceUsage) return

        if(!this.performanceUsage[feature]) this.performanceUsage[feature] = {}
        if(!this.performanceUsage[feature][event]) this.performanceUsage[feature][event] = {time: 0,count: 0}

        let instant = Instant.now()
	    let time = (instant.getEpochSecond() + (instant.getNano() / 1000000000))*1000;

        this.performanceUsage[feature][event].startTime = time
    }
    stopRecordingPerformance(feature, event){
        if(!this.recordingPerformanceUsage) return

        let instant = Instant.now()
        let time = (instant.getEpochSecond() + (instant.getNano() / 1000000000))*1000;

        this.performanceUsage[feature][event].time += time - this.performanceUsage[feature][event].startTime
        this.performanceUsage[feature][event].count++
    }
}

if(!global.soopyv2featuremanagerthing){
    global.soopyv2featuremanagerthing = new FeatureManager()
    register("gameUnload", ()=>{
        global.soopyv2featuremanagerthing = undefined
    })
}
export default global.soopyv2featuremanagerthing