fix contract not widening certain literal fields

Author: CapsAdmin

nattlua/analyzer/operators/index.lua

@@ -87,13 +87,15 @@ local function index_table(analyzer, self, key, raw)
 
 		if not val then return val, err end
 
+		local contract_val = val
+
 		do
 			local val = self:GetMutatedValue(key, analyzer:GetScope())
 
 			if val then
 				if val.Type == "union" then val = val:Copy(nil, true) end
 
-				if not val:GetContract() then val:SetContract(val) end
+				if not val:GetContract() then val:SetContract(contract_val) end
 
 				if val.Type == "union" then analyzer:TrackTableIndex(self, key, val) end
 
@@ -107,7 +109,11 @@ local function index_table(analyzer, self, key, raw)
 		if self:HasMutations() then
 			local tracked = self:GetMutatedValue(key, analyzer:GetScope())
 
-			if tracked then val = tracked end
+			if tracked then
+				val = tracked
+
+				if not val:GetContract() then val:SetContract(contract_val) end
+			end
 		end
 
 		if val.Type == "union" then analyzer:TrackTableIndex(self, key, val) end

test/tests/nattlua/analyzer/table.lua

@@ -62,6 +62,28 @@ test("typed table correct assignment not should error", function()
     ]])
 end)
 
+test("nested typed list field should widen to contract", function()
+	analyze([[
+        local tbl: {loadings = List<|boolean|>} = {
+            loadings = {true},
+        }
+
+        tbl.loadings[1] = false
+        attest.equal(tbl.loadings[1], false)
+    ]])
+end)
+
+test("nested typed table field should widen to contract", function()
+	analyze([[
+        local tbl: {inner = {flag = boolean}} = {
+            inner = {flag = true},
+        }
+
+        tbl.inner.flag = false
+        attest.equal(tbl.inner.flag, false)
+    ]])
+end)
+
 test("self referenced tables should be equal", function()
 	local analyzer = analyze([[
         local a = {a=true}