Basket.cs

using UnityEngine;
using System.Collections;
using System.Collections.Generic;
using SpaceTrader.Stations;
using SpaceTrader.Resources;

namespace SpaceTrader.Misc
{
	public class Basket : MonoBehaviour
	{
		public List<BasketItem> _items;

		//============================================
		// FUNCTIONS (CUSTOM)
		//============================================

		public bool Contains(Resource resource)
		{
			foreach (BasketItem i in Items)
			{
				if (i.Resource == resource)
				{
					return true;
				}
			}

			return false;
		}

		public void Add(Resource resource, int amount)
		{
			BasketItem item = Find(resource);

			if (item != null)
			{
				item.Amount += amount;
			}
			else
			{
				BasketItem basketItem = new BasketItem();
				basketItem.Resource = resource;
				basketItem.Amount = amount;
				Items.Add(basketItem);
			}
		}

		public void Remove(Resource resource, int amount)
		{
			BasketItem item = Find(resource);

			if (item != null)
			{
				// TODO: CHECK IF HAS ENOUGH, SO DOESN'T GO TO NEGATIVE NUMBERS
				//

				item.Amount -= amount;

				if (item.Amount <= 0)
					Items.Remove(item);
			}
			else
			{
				// COULDN'T FIND IN BASKET
			}
		}

		public BasketItem Find(Resource resource)
		{
			foreach (BasketItem i in Items)
			{
				if (i.Resource == resource)
				{
					return i;
				}
			}

			return null;
		}

		public int GetPriceTotal(Exporter exporter)
		{
			int total = 0;

			foreach (BasketItem item in Items)
			{
				total += item.GetPriceFromSeller(exporter) * item.Amount;
			}

			return total;
		}

		public void Clear()
		{
			Items.Clear();
		}

		//============================================
		// PROPERTIES
		//============================================

		public List<BasketItem> Items
		{
			get { return _items; }
			set { _items = value; }
		}

	}
}