Resource Sharing Tax Mod: Prevent upgrading allied mex/geos. No longer disables unit sharing completely, and can buy t2 con through Unit Market

luarules/gadgets/game_disable_assist_ally.lua

@@ -32,6 +32,65 @@ local function isComplete(u)
 end
 
 
+
+-- table of all mex unitDefIDs
+local isMex = {} 
+for unitDefID, unitDef in pairs(UnitDefs) do
+	if unitDef.extractsMetal > 0 then
+		isMex[unitDefID] = true
+	end
+end
+
+local function existsNonOwnedMex(myTeam, x, y, z)
+    local units = Spring.GetUnitsInCylinder(x, z, 10)
+    for k, unitID in ipairs(units) do
+        if isMex[Spring.GetUnitDefID(unitID)] then
+            if Spring.GetUnitTeam(unitID) ~= myTeam then
+                return unitID
+            end
+        end
+    end
+    return false
+end
+
+-- table of all geo unitDefIDs
+local isGeo = {} 
+for unitDefID, unitDef in pairs(UnitDefs) do
+	if unitDef.customParams.geothermal then
+		isGeo[unitDefID] = true
+	end
+end
+
+
+local function existsNonOwnedGeo(myTeam, x, y, z)
+    local units = Spring.GetUnitsInCylinder(x, z, 10)
+    for k, unitID in ipairs(units) do
+        if isGeo[Spring.GetUnitDefID(unitID)] then
+            if Spring.GetUnitTeam(unitID) ~= myTeam then
+                return unitID
+            end
+        end
+    end
+    return false
+end
+
+function gadget:AllowUnitCreation(unitDefID, builderID, builderTeam, x, y, z)
+	-- Disallow upgrading allied mexes
+	if isMex[unitDefID] then
+		if existsNonOwnedMex(builderTeam, x, y, z) then
+			return false
+		end
+	end
+	-- Disallow upgrading allied geos
+	if isGeo[unitDefID] then
+		if existsNonOwnedGeo(builderTeam, x, y, z) then
+			return false
+		end
+	end
+	return true
+end
+
+
 function gadget:AllowCommand(unitID, unitDefID, unitTeam, cmdID, cmdParams, cmdOptions, cmdTag, synced)
 
 	-- Disallow guard commands onto labs, units that have buildOptions or can assist

luarules/gadgets/game_unit_market.lua

@@ -149,19 +149,37 @@ local function offerUnitForSale(unitID, sale_price, msgFromTeamID)
     end
 end
 
+local t2cons = {"armack", "armacv", "armaca", "armacsub", "corack", "coracv", "coraca", "coracsub"}
+function contains(table, value)
+    for _, v in pairs(table) do
+        if v == value then return true end
+    end
+    return false
+end
+
+-- Override unit sharing block from other modoptions
 local disable_unit_sharing = (
     Spring.GetModOptions().disable_unit_sharing
  or (Spring.GetModOptions().tax_resource_sharing_amount or 0) ~= 0)
 and Spring.GetModOptions().unit_market
 local saleWhitelist = {}
 
 local function tryToBuyUnit(unitID, msgFromTeamID)
+
     if not unitID or unitsForSale[unitID] == nil or unitsForSale[unitID] == 0 then return end
     local unitDefID = spGetUnitDefID(unitID)
     if not unitDefID then return end
     local unitDef = UnitDefs[unitDefID]
     if not unitDef then return end
 
+
+    -- Only allow selling t2 cons when unit sharing is disabled
+
+    --Spring.Echo("unitDefID", unitDefID, UnitDefs[unitDefID].name)
+    if(disable_unit_sharing) then
+        if not contains(t2cons, UnitDefs[unitDefID].name) then return end
+    end
+
     local old_ownerTeamID = spGetUnitTeam(unitID)
     local _, _, _, isAiTeam = spGetTeamInfo(old_ownerTeamID)
     if not spAreTeamsAllied(old_ownerTeamID, msgFromTeamID) then return end
@@ -184,6 +202,12 @@ local function tryToBuyUnit(unitID, msgFromTeamID)
     TransferUnit(unitID, msgFromTeamID)
     if msgFromTeamID ~= old_ownerTeamID and price > 0 then -- don't send resources to yourself
         ShareTeamResource(msgFromTeamID, old_ownerTeamID, "metal", price)
+        -- if tax resource sharing is on, refund tax to the seller
+        if(Spring.GetModOptions().tax_resource_sharing_amount>0) then
+            local taxRate = Spring.GetModOptions().tax_resource_sharing_amount
+            local taxAmount = price * taxRate
+            Spring.AddTeamResource(old_ownerTeamID, "metal", taxAmount)
+        end
     end
     setNotForSale(unitID)
     UnitSoldBroadcast(unitID, price, old_ownerTeamID, msgFromTeamID)