gajim-plural/plugins/gtkgui.py

#!/usr/bin/env python
##	plugins/gtkgui.py
##
## Gajim Team:
## 	- Yann Le Boulanger <asterix@crans.org>
## 	- Vincent Hanquez <tab@tuxfamily.org>
## 	- David Ferlier <krp@yazzy.org>
##
##	Copyright (C) 2003 Gajim Team
##
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published
## by the Free Software Foundation; version 2 only.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
##

import pygtk
pygtk.require('2.0')
import gtk
import gtk.glade
import gobject

class user:
	def __init__(self, *args):
		if len(args) == 0:
			self.name = ''
			self.group = ''
			self.show = ''
			self.status = ''
		elif len(args) == 4:
			self.name = args[0]
			self.group = args[1]
			self.show = args[2]
			self.status = args[3]
		elif ((len(args)) and (type (args[0]) == type (self)) and
			(self.__class__ == args[0].__class__)):
			self.name = args[0].name
			self.group = args[0].group
			self.show = args[0].show
			self.status = args[0].status
		else: raise TypeError, 'bad arguments'

class message:
	def delete_event(self, widget):
		del self.roster.tab_messages[self.jid]
		self.window.destroy()
	
	def print_conversation(self, txt, contact = None):
		txt_buffer = self.conversation.get_buffer()
		end_iter = txt_buffer.get_end_iter()
		if contact: who = 'moi'
		else: who = 'lui'
		txt_buffer.insert(end_iter, '<'+who+'> '+txt+'\n', -1)

	def on_msg_key_press_event(self, widget, event):
		if event.keyval == gtk.keysyms.Return:
			if (event.state & gtk.gdk.SHIFT_MASK):
				return 0
			txt_buffer = widget.get_buffer()
			start_iter = txt_buffer.get_start_iter()
			end_iter = txt_buffer.get_end_iter()
			txt = txt_buffer.get_text(start_iter, end_iter, 0)
			self.roster.queueOUT.put(('MSG',(self.jid, txt)))
			txt_buffer.set_text('', -1)
			self.print_conversation(txt, self.jid)
			widget.grab_focus()
			return 1
		return 0

	def __init__(self, jid, roster):
		self.jid = jid
		self.roster = roster
		self.xml = gtk.glade.XML('plugins/gtkgui.glade', 'Chat')
		self.window = self.xml.get_widget('Chat')
		self.window.set_title('Chat with ' + jid)
		self.message = self.xml.get_widget('message')
		self.conversation = self.xml.get_widget('conversation')
		self.window.show()
		self.xml.signal_connect('gtk_widget_destroy', self.delete_event)
		self.xml.signal_connect('on_msg_key_press_event', self.on_msg_key_press_event)

class roster:
	def get_icon_pixbuf(self, stock):
		return self.tree.render_icon(stock, size = gtk.ICON_SIZE_MENU, detail = None)

	def mkl_group(self):
		self.l_group = []
		for u in self.l_contact:
			if u.group in self.l_group:
				pass
			else:
				self.l_group.append(u.group)

	def mkroster(self):
		self.treestore.clear()
		for g in self.l_group:
			iter_g = self.treestore.append(None, (self.pixbufs['online'], g, 'group'))
			for c in self.l_contact:
				if c.group == g:
#					print c.status
					self.treestore.append(iter_g, (self.pixbufs[c.show], c.name, c.show))
#					if c.status == 'Online':
#						self.treestore.append(iter_g, (self.pixbufs['Online'], c.name, 'Online'))
#					elif c.status == None:
#						self.treestore.append(iter_g, (self.pixbufs['away'], c.name, 'away'))

	def mkroster2(self, tab):
		self.l_contact = []
		for jid in tab.keys():
			user1 = user(jid, 'general', tab[jid]["Show"], tab[jid]["Status"])
			self.l_contact.append(user1)
		self.mkl_group()
		self.mkroster()
		self.tree.collapse_row((0,3))
	
	def update_iter(self, widget, path, iter, data):
		val = self.treestore.get_value(iter, 1)
		if val == data[0]:
			self.treestore.set_value(iter, 0, self.pixbufs[data[1]])
	
	def chg_status(self, jid, show, status):
		for u in self.l_contact:
			if u.name == jid:
				u.show = show
				u.status = status
				self.treestore.foreach(self.update_iter, (jid, show))
	
	def mk_menu_c(self, event):
		self.menu_c = gtk.Menu()
		item = gtk.MenuItem("user1")
		self.menu_c.append(item)
		item = gtk.MenuItem("user2")
		self.menu_c.append(item)
		item = gtk.MenuItem("user3")
		self.menu_c.append(item)
		self.menu_c.popup(None, None, None, event.button, event.time)
		self.menu_c.show_all()

	def mk_menu_g(self, event):
		self.menu_c = gtk.Menu()
		item = gtk.MenuItem("grp1")
		self.menu_c.append(item)
		item = gtk.MenuItem("grp2")
		self.menu_c.append(item)
		item = gtk.MenuItem("grp3")
		self.menu_c.append(item)
		self.menu_c.popup(None, None, None, event.button, event.time)
		self.menu_c.show_all()
	
	def on_treeview_event(self, widget, event):
		if (event.button == 3) & (event.type == gtk.gdk.BUTTON_PRESS):
			try:
				path, column, x, y = self.tree.get_path_at_pos(int(event.x), int(event.y))
				
			except TypeError:
				return
			iter = self.treestore.get_iter(path)
			data = self.treestore.get_value(iter, 2)
			if data == 'group':
				self.mk_menu_g(event)
			else:
				self.mk_menu_c(event)
			return gtk.TRUE
		return gtk.FALSE

	def on_status_changed(self, widget):
		self.queueOUT.put(('STATUS',widget.name))
	
	def on_quit(self, widget):
		self.queueOUT.put(('QUIT',''))
		gtk.mainquit()

	def on_row_activated(self, widget, path, col):
		iter = self.treestore.get_iter(path)
		jid = self.treestore.get_value(iter, 1)
		if self.tab_messages.has_key(jid):
			#NE FONCTIONNE PAS !
			self.tab_messages[jid].window.grab_focus()
		else:
			self.tab_messages[jid]=message(jid, self)
		
	def __init__(self, queueOUT):
		#initialisation des variables
		# FIXME : handle no file ...
		self.xml = gtk.glade.XML('plugins/gtkgui.glade', 'Gajim')
		self.tree = self.xml.get_widget('treeview')
		self.treestore = gtk.TreeStore(gtk.gdk.Pixbuf, str, str)
		add_pixbuf = self.get_icon_pixbuf(gtk.STOCK_ADD)
		remove_pixbuf = self.get_icon_pixbuf(gtk.STOCK_REMOVE)
		self.pixbufs = {"online":add_pixbuf, "away":remove_pixbuf, "xa":remove_pixbuf, "dnd":remove_pixbuf, "offline":remove_pixbuf}
		self.tree.set_model(self.treestore)
		self.queueOUT = queueOUT
		self.optionmenu = self.xml.get_widget('optionmenu')
		self.optionmenu.set_history(6)
		self.tab_messages = {}
		#colonnes
		self.col = gtk.TreeViewColumn()
		render_pixbuf = gtk.CellRendererPixbuf()
		self.col.pack_start(render_pixbuf, expand = False)
		self.col.add_attribute(render_pixbuf, 'pixbuf', 0)
		render_text = gtk.CellRendererText()
		self.col.pack_start(render_text, expand = True)
		self.col.add_attribute(render_text, 'text', 1)
		self.tree.append_column(self.col)
		#signales
		self.xml.signal_connect('gtk_main_quit', self.on_quit)
		self.xml.signal_connect('on_quit_activate', self.on_quit)
		self.xml.signal_connect('on_treeview_event', self.on_treeview_event)
		self.xml.signal_connect('on_status_changed', self.on_status_changed)
		self.xml.signal_connect('on_row_activated', self.on_row_activated)
#		self.mk_menu_c()

class plugin:
	def read_queue(self):
		while self.queueIN.empty() == 0:
			ev = self.queueIN.get()
			print ev
			if ev[0] == 'ROSTER':
				self.r.mkroster2(ev[1])
			elif ev[0] == 'NOTIFY':
				self.r.chg_status(ev[1][0], ev[1][1], ev[1][2])
			elif ev[0] == 'MSG':
				if not self.r.tab_messages.has_key(ev[1][0]):
					self.r.tab_messages[ev[1][0]] = message(ev[1][0], self.r)
				self.r.tab_messages[ev[1][0]].print_conversation(ev[1][1])
		return 1

	def __init__(self, quIN, quOUT):
		gtk.threads_init()
		gtk.threads_enter()
		self.queueIN = quIN
		self.r = roster(quOUT)
		self.time = gtk.timeout_add(200, self.read_queue)
		gtk.main()
		gtk.threads_leave()

if __name__ == "__main__":
	plugin(None, None)

print "plugin gui loaded"