import random

import goocanvas

from sugar.canvas.IconItem import IconItem
import Theme

class FriendIcon(IconItem):
	def __init__(self, friend):
		IconItem.__init__(self, icon_name='stock-buddy',
						  color=friend.get_color(), size=48,
						  x=random.random() * 1100,
						  y=random.random() * 800)
		self._friend = friend

	def get_friend(self):
		return self._friend

class FriendsGroup(goocanvas.Group):
	def __init__(self, data_model):
		goocanvas.Group.__init__(self)
		self._friend_to_child = {}

		self._theme = Theme.get_instance()
		self._theme.connect("theme-changed", self.__theme_changed_cb)

		color = self._theme.get_home_mesh_color()
		self._mesh_rect = goocanvas.Rect(width=1200, height=900,
									     line_width=0, fill_color=color)
		self.add_child(self._mesh_rect)

		color = self._theme.get_home_friends_color()
		self._friends_rect = goocanvas.Rect(x=50, y=50, width=1100, height=800,
											line_width=0, fill_color=color,
											radius_x=30, radius_y=30)
		self.add_child(self._friends_rect)

		for friend in data_model:
			self.add_friend(friend)

		data_model.connect('friend-added', self.__friend_added_cb)
		data_model.connect('friend-removed', self.__friend_removed_cb)

	def __theme_changed_cb(self, theme):
		color = self._theme.get_home_friends_color()
		self._friends_rect.set_property("fill-color", color)
		color = self._theme.get_home_mesh_color()
		self._mesh_rect.set_property("fill-color", color)

	def add_friend(self, friend):
		icon = FriendIcon(friend)
		self.add_child(icon)
		self._friend_to_child[friend] = icon

	def remove_friend(self, friend):
		icon = self._friend_to_child[friend]
		self.remove_child(self._root.find_child(icon))
		del self._friend_to_child[friend]

	def __friend_added_cb(self, data_model, friend):
		self.add_friend(friend)

	def __friend_removed_cb(self, data_model, friend):
		self.remove_friend(friend)