diff --git a/plugin/src/Components/App.lua b/plugin/src/Components/App.lua index 7d534759..79436972 100644 --- a/plugin/src/Components/App.lua +++ b/plugin/src/Components/App.lua @@ -182,6 +182,13 @@ function App:didMount() preloadAssets() end +function App:willUnmount() + if self.currentSession ~= nil then + self.currentSession:disconnect() + self.currentSession = nil + end +end + function App:didUpdate() local connectActive = self.state.sessionStatus == SessionStatus.ConfiguringSession or self.state.sessionStatus == SessionStatus.Connected diff --git a/plugin/src/init.server.lua b/plugin/src/init.server.lua index dc547c85..b3b52bd0 100644 --- a/plugin/src/init.server.lua +++ b/plugin/src/init.server.lua @@ -4,10 +4,6 @@ end local Roact = require(script.Parent.Roact) -Roact.setGlobalConfig({ - elementTracing = true, -}) - local App = require(script.Components.App) local app = Roact.createElement(App, { @@ -16,4 +12,6 @@ local app = Roact.createElement(App, { Roact.mount(app, game:GetService("CoreGui"), "Rojo UI") --- TODO: Detect another instance of Rojo coming online and shut down this one. \ No newline at end of file +plugin.Unloading:Connect(function() + Roact.unmount(app) +end) \ No newline at end of file