From be89259bbdd3e35160c207a91abb252a6bca7bdf Mon Sep 17 00:00:00 2001
From: boatbomber <arbitrarysoandso@gmail.com>
Date: Sun, 25 Sep 2022 17:10:30 -0400
Subject: [PATCH] Add Address and ProjectName when connected

---
 plugin/src/App/init.lua    | 8 ++++++++
 plugin/src/HeadlessAPI.lua | 2 ++
 2 files changed, 10 insertions(+)

diff --git a/plugin/src/App/init.lua b/plugin/src/App/init.lua
index c8ea22b6..9d7fae3f 100644
--- a/plugin/src/App/init.lua
+++ b/plugin/src/App/init.lua
@@ -268,6 +268,10 @@ function App:startSession(host: string?, port: string?)
 
 	serveSession:onStatusChanged(function(status, details)
 		self.headlessAPI.Connected = status == ServeSession.Status.Connected
+		if not self.headlessAPI.Connected then
+			self.headlessAPI.Address = nil
+			self.headlessAPI.ProjectName = nil
+		end
 
 		if status == ServeSession.Status.Connecting then
 			self:setPriorEndpoint(host, port)
@@ -279,6 +283,10 @@ function App:startSession(host: string?, port: string?)
 			self:addNotification("Connecting to session...")
 		elseif status == ServeSession.Status.Connected then
 			local address = string.format("%s:%s", host :: string, port :: string)
+
+			self.headlessAPI.Address = address
+			self.headlessAPI.ProjectName = details
+
 			self:setState({
 				appStatus = AppStatus.Connected,
 				projectName = details,
diff --git a/plugin/src/HeadlessAPI.lua b/plugin/src/HeadlessAPI.lua
index 0bb540a8..5c0c0ec0 100644
--- a/plugin/src/HeadlessAPI.lua
+++ b/plugin/src/HeadlessAPI.lua
@@ -4,6 +4,8 @@ function API.new(app, config, settings)
 	local Rojo = {}
 
 	Rojo.Connected = if app.serveSession then app.serveSession:getStatus() == "Connected" else false
+	Rojo.Address = nil
+	Rojo.ProjectName = nil
 	Rojo.Version = table.clone(config.version)
 	Rojo.ProtocolVersion = config.protocolVersion