invetory fix

Assets/Scripts/Items/Building.cs

@@ -15,7 +15,7 @@ namespace Items
 
         public void Invoke(Action<Unit> action)
         {
-            OnItemUsed += action;
+            
         }
 
 

Assets/Scripts/Items/CaptureAbility.cs

@@ -63,9 +63,18 @@ namespace Items
             unit.UseItem(this);
             HexManager.UnitCurrentCell[unit.Color].cell.PaintHex(unit.Color);
             var cell = HexManager.UnitCurrentCell[unit.Color].cell.GetNeighbor(_direction);
+            OnItemUsed?.Invoke(unit);
+
+            unit.UnitView.AnimActionDic[animName] -= DoPaint;
+            OnItemUsed = null;
+            if (cell == null)
+            {
+                return;
+            }
             cell.PaintHex(unit.Color);
             bool keepGoing = true;
-            var moveDir = _direction;
+            
+            
             itterationMove.ForEach(dir =>
             {
                 if (!keepGoing) return;
@@ -89,11 +98,7 @@ namespace Items
                 cell.PaintHex(unit.Color);
             });
             
-            OnItemUsed?.Invoke(unit);
            
-            
-            unit.UnitView.AnimActionDic[animName] -= DoPaint;
-            OnItemUsed = null;
         }
 
         public void UseAbility(Unit unit)

Assets/Scripts/Items/ItemViews/ItemView.cs

@@ -30,7 +30,7 @@ namespace Items.ItemViews
                 Despawn();
                 return;
             }
-            transform.DOMove(unit.UnitView.transform.position, 0.1f).OnComplete(() =>
+            transform.DOMove(unit.UnitView.transform.position + new Vector3(0,1,0), 0.1f).OnComplete(() =>
             {
                 unit.PickUpItem(_item);
                 Despawn();

Assets/Scripts/Items/SwitchingPlaces.cs

@@ -58,18 +58,19 @@ namespace Items
             }
 
             Debug.DrawRay(ray.origin,
-                ray.direction * hit.distance, UnityEngine.Color.red);
+                ray.direction * hit.distance, UnityEngine.Color.red, 10f);
         }
 
 
         public void UseAbility(Unit unit, Unit chosenUnit)
         {
-            if (unit == null)
+            if (chosenUnit == null || unit.isSwitched)
             {
                 DeAim();
                 return;
             }
             unit.UseItem(this);
+            unit.isSwitched = true;
             DeAim();
             OnItemUsed?.Invoke(unit);
             chosenUnit.IsBusy = true;
@@ -81,6 +82,7 @@ namespace Items
             chosenUnit.SetCell(unitCell, true);
             TimerHelper.Instance.StartTimer(() =>
             {
+                unit.isSwitched = false;
                 chosenUnit.SetCell(choseUnitCell, true, true);
                 unit.SetCell(unitCell, true);
                 chosenUnit.IsStaned = false;

Assets/Scripts/Units/Unit.cs

@@ -39,6 +39,7 @@ namespace Units
         private UnitColor _easyCaptureColor;
         
         public bool IsStaned;
+        public bool isSwitched;
         public int AttackBonus => _weapon.modifiedDamage - _weapon.damage;
 
         
@@ -424,7 +425,8 @@ namespace Units
                 _instance.transform.position);
             TimerHelper.Instance.StartTimer(() =>
             {
-                HexManager.PaintHexList(hexToPaint, UnitColor.Grey);
+                
+                HexManager.PaintHexList(hexToPaint.Where(x => x.Color == Color).ToList(), UnitColor.Grey);
 
                 Object.Destroy(_instance);
                 OnDeath?.Invoke(this);