From cf71c5d87d382b099313a00a96d36db2ecfb8e17 Mon Sep 17 00:00:00 2001
From: Maksym H <Maksym48@pm.me>
Date: Thu, 1 Sep 2022 12:28:05 +0300
Subject: [PATCH] Update HUD

---
 hud.lua | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/hud.lua b/hud.lua
index 271bfbf..afd4924 100644
--- a/hud.lua
+++ b/hud.lua
@@ -5,14 +5,13 @@ local S = areas.S
 areas.hud = {}
 
 local vround = vector.round
-local tconcat = table.concat
+local tconcat, tinsert = table.concat, table.insert
 
 local function update_hud(player, name, pos)
-	local areaStrings = {
-		S("Areas:")
-	}
+	local areaStrings = {}
+	local getAreasAtPos = areas:getAreasAtPos(pos)
 
-	for id, area in pairs(areas:getAreasAtPos(pos)) do
+	for id, area in pairs(getAreasAtPos) do
 		areaStrings[#areaStrings + 1] = ("%s [%u] (%s)%s%s")
 			:format(area.name, id, area.owner,
 				area.open and (" [" .. S("Open") .. "]") or "",
@@ -27,9 +26,15 @@ local function update_hud(player, name, pos)
 		areaStrings[#areaStrings + 1] = str
 	end
 
-	-- "Areas:" text has index 1
-	local areaString = #areaStrings > 1 and tconcat(areaStrings, "\n") or ""
+	if areas.invite_code then
+		areaStrings[#areaStrings + 1] = areas.invite_code
+	end
 
+	if next(getAreasAtPos) then
+		tinsert(areaStrings, 1, S("Areas:"))
+	end
+
+	local areaString = tconcat(areaStrings, "\n")
 	local hud = areas.hud[name]
 	if not hud then
 		hud = {}