gajim-plural/src/search_window.py

# -*- coding: utf-8 -*-
##	search_window.py
##
## Copyright (C) 2007 Yann Le Boulanger <asterix@lagaule.org>
##
## 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 gobject
import gtk

from common import xmpp, gajim, dataforms

import gtkgui_helpers
import dialogs
import vcard
import config
import dataforms_widget

class SearchWindow:
	def __init__(self, account, jid):
		'''Create new window.'''

		# an account object
		self.account = account
		self.jid = jid

		# retrieving widgets from xml
		self.xml = gtkgui_helpers.get_glade('search_window.glade')
		self.window = self.xml.get_widget('search_window')
		for name in ('label', 'progressbar', 'search_vbox', 'search_button',
		'add_contact_button', 'information_button'):
			self.__dict__[name] = self.xml.get_widget(name)

		# displaying the window
		self.xml.signal_autoconnect(self)
		self.window.show_all()
		self.request_form()
		self.pulse_id = gobject.timeout_add(80, self.pulse_callback)

		self.is_form = None

		# Is there a jid column in results ? if -1: no, else column number
		self.jid_column = -1

	def request_form(self):
		gajim.connections[self.account].request_search_fields(self.jid)
	
	def pulse_callback(self):
		self.progressbar.pulse()
		return True

	def on_search_window_key_press_event(self, widget, event):
		if event.keyval == gtk.keysyms.Escape:
			self.window.destroy()

	def on_search_window_destroy(self, widget):
		if self.pulse_id:
			gobject.source_remove(self.pulse_id)
		del gajim.interface.instances[self.account]['search'][self.jid]

	def on_close_button_clicked(self, button):
		self.window.destroy()

	def on_search_button_clicked(self, button):
		if self.is_form:
			self.data_form_widget.data_form.type = 'submit'
			gajim.connections[self.account].send_search_form(self.jid,
				self.data_form_widget.data_form, True)
		else:
			infos = self.data_form_widget.get_infos()
			if infos.has_key('instructions'):
				del infos['instructions']
			gajim.connections[self.account].send_search_form(self.jid, infos,
				False)

		self.search_vbox.remove(self.data_form_widget)

		self.progressbar.show()
		self.label.set_text(_('Waiting for results'))
		self.label.show()
		self.pulse_id = gobject.timeout_add(80, self.pulse_callback)
		self.search_button.hide()

	def on_add_contact_button_clicked(self, widget):
		(model, iter) = self.result_treeview.get_selection().get_selected()
		if not iter:
			return
		jid = model[iter][self.jid_column]
		dialogs.AddNewContactWindow(self.account, jid)
	
	def on_information_button_clicked(self, widget):
		(model, iter) = self.result_treeview.get_selection().get_selected()
		if not iter:
			return
		jid = model[iter][self.jid_column]
		if gajim.interface.instances[self.account]['infos'].has_key(jid):
			gajim.interface.instances[self.account]['infos'][jid].window.present()
		else:
			contact = gajim.contacts.create_contact(jid = jid, name='', groups=[],
				show='', status='', sub='', ask='', resource='', priority=0,
				keyID='', our_chatstate=None, chatstate=None)
			gajim.interface.instances[self.account]['infos'][jid] = \
				vcard.VcardWindow(contact, self.account)

	def on_form_arrived(self, form, is_form):
		if self.pulse_id:
			gobject.source_remove(self.pulse_id)
		self.progressbar.hide()
		self.label.hide()

		if is_form:
			self.is_form = True
			self.data_form_widget = dataforms_widget.DataFormWidget()
			self.dataform = dataforms.ExtendForm(node = form)
			self.data_form_widget.set_sensitive(True)
			try:
				self.data_form_widget.data_form = self.dataform
			except dataforms.Error:
				self.label.set_text(_('Error in received dataform'))
				self.label.show()
				return
			if self.data_form_widget.title:
				self.window.set_title('%s - Search - Gajim' % \
					self.data_form_widget.title)
		else:
			self.is_form = False
			self.data_form_widget = config.FakeDataForm(form)

		self.data_form_widget.show_all()
		self.search_vbox.pack_start(self.data_form_widget)

	def on_result_treeview_cursor_changed(self, treeview):
		if self.jid_column == -1:
			return
		(model, iter) = treeview.get_selection().get_selected()
		if not iter:
			return
		if model[iter][self.jid_column]:
			self.add_contact_button.set_sensitive(True)
			self.information_button.set_sensitive(True)
		else:
			self.add_contact_button.set_sensitive(False)
			self.information_button.set_sensitive(False)

	def on_result_arrived(self, form, is_form):
		if self.pulse_id:
			gobject.source_remove(self.pulse_id)
		self.progressbar.hide()
		self.label.hide()

		if not is_form:
			if not form:
				self.label.set_text(_('No result'))
				self.label.show()
				return
			# We suppose all items have the same fields
			sw = gtk.ScrolledWindow()
			sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
			self.result_treeview = gtk.TreeView()
			self.result_treeview.connect('cursor-changed',
				self.on_result_treeview_cursor_changed)
			sw.add(self.result_treeview)
			# Create model
			fieldtypes = [str]*len(form[0])
			model = gtk.ListStore(*fieldtypes)
			# Copy data to model
			for item in form:
				model.append(item.values())
			# Create columns
			counter = 0
			for field in form[0].keys():
				self.result_treeview.append_column(
					gtk.TreeViewColumn(field, gtk.CellRendererText(),
					text = counter))
				if field == 'jid':
					self.jid_column = counter
				counter += 1
			self.result_treeview.set_model(model)
			sw.show_all()
			self.search_vbox.pack_start(sw)
			if self.jid_column > -1:
				self.add_contact_button.show()
				self.information_button.show()
			return

		self.dataform = dataforms.ExtendForm(node = form)

		self.data_form_widget.set_sensitive(True)
		try:
			self.data_form_widget.data_form = self.dataform
		except dataforms.Error:
			self.label.set_text(_('Error in received dataform'))
			self.label.show()
			return

		self.result_treeview = self.data_form_widget.records_treeview
		selection = self.result_treeview.get_selection()
		selection.set_mode(gtk.SELECTION_SINGLE)
		self.result_treeview.connect('cursor-changed',
			self.on_result_treeview_cursor_changed)

		counter = 0
		for field in self.dataform.items[0].fields:
			if field.var == 'jid':
				self.jid_column = counter
				break
			counter += 1
		self.search_vbox.pack_start(self.data_form_widget)
		self.data_form_widget.show()
		if self.jid_column > -1:
			self.add_contact_button.show()
			self.information_button.show()
		if self.data_form_widget.title:
			self.window.set_title('%s - Search - Gajim' % \
				self.data_form_widget.title)