#!/usr/bin/env python
##	otr_windows.py
##
##
## Copyright (C) 2008 Kjell Braden <fnord@pentabarf.de>
##
## This file is part of Gajim.
##
## Gajim 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 3 only.
##
## Gajim 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.
##
## You should have received a copy of the GNU General Public License
## along with Gajim.  If not, see <http://www.gnu.org/licenses/>.
##

import gtkgui_helpers
from common import gajim

our_fp_text = """Your fingerprint:
<span weight="bold" face="monospace">%s</span>"""

their_fp_text = """Purported fingerprint for %s:
<span weight="bold" face="monospace">%s</span>"""

class ContactOtrSMPWindow:

	def gw(self, n):
		""" shorthand for self.xml.get_widget(n)"""
		return self.xml.get_widget(n)

	def __init__(self, fjid, account):
		self.fjid = fjid
		self.account = account

		self.xml = gtkgui_helpers.get_glade("contact_otr_window.glade")
		self.window = self.xml.get_widget("otr_smp_window")


		# the contact may be unknown to gajim if ContactOtrSMPWindow is created very early
		self.contact = gajim.contacts.get_contact_from_full_jid(account, fjid)
		if self.contact:
			self.window.set_title("OTR settings for %s"%
				self.contact.get_full_jid())

		self.ctx = gajim.otr_module.otrl_context_find(
			gajim.connections[self.account].otr_userstates,
			self.fjid.encode(), gajim.get_jid_from_account(self.account).encode(),
			gajim.OTR_PROTO, 1, (gajim.otr_add_appdata, self.account))[0]

		self.gw("smp_cancel_button").connect("clicked", self._on_destroy)
		self.gw("smp_ok_button").connect("clicked", self._apply)

	def show(self, response):
		# re-initialize if contact was unknown when we initially initialized
		if not self.contact:
			self.__init__(self.fjid, self.account)
		assert(self.contact) # the contact MUST be known when showing the dialog

		self.smp_running = False
		self.finished = False

		self.response = response
		if response:
			self.gw("desc_label").set_markup("<b>%s is trying to authenticate you "
				"using a secret only known to him/her and you.</b> Please enter your secret "
				"below."% self.contact.get_full_jid())
		else:
			self.gw("desc_label").set_markup("<b>You are trying to authenticate %s "
				"using a secret only known to him/her and yourself.</b> Please enter your "
				"secret below."% self.contact.get_full_jid())

		self.window.show_all()

	def _abort(self, text=None):
		self.smp_running = False
		gajim.otr_module.otrl_message_abort_smp(gajim.connections[self.account].otr_userstates,
				(gajim.otr_ui_ops, {'account':self.account}), self.ctx)
		if text:
			gajim.otr_ui_ops.gajim_log(text, self.account, self.contact.get_full_jid())

	def _finish(self, text):
		self.smp_running = False
		self.finished = True
		self.gw("smp_cancel_button").set_sensitive(False)
		self.gw("smp_ok_button").set_sensitive(True)
		self.gw("progressbar").set_fraction(1)
		gajim.otr_ui_ops.gajim_log(text, self.account, self.contact.get_full_jid())
		self.gw("desc_label").set_markup(text)
		for ctrl in gajim.interface.msg_win_mgr.get_chat_controls(self.contact.jid, self.account):
			ctrl.update_ui()
		gajim.otr_ui_ops.write_fingerprints({'account':self.account})

	def handle_tlv(self, tlvs):
		if not self.contact:
			self.__init__(self.fjid, self.account)

		if tlvs:
			nextTLV = self.ctx.smstate.nextExpected;
			tlv = gajim.otr_module.otrl_tlv_find(tlvs, gajim.otr_module.OTRL_TLV_SMP1)
			if tlv:
				if nextTLV != gajim.otr_module.OTRL_SMP_EXPECT1:
					self._abort()
				else:
					self.show(True)
					self.gw("progressbar").set_fraction(0.3)
			tlv = gajim.otr_module.otrl_tlv_find(tlvs, gajim.otr_module.OTRL_TLV_SMP2)
			if tlv:
				if nextTLV != gajim.otr_module.OTRL_SMP_EXPECT2:
					self._abort()
				else:
					self.ctx.smstate.nextExpected = gajim.otr_module.OTRL_SMP_EXPECT4;
					self.gw("progressbar").set_fraction(0.6)
			tlv = gajim.otr_module.otrl_tlv_find(tlvs, gajim.otr_module.OTRL_TLV_SMP3)
			if tlv:
				if nextTLV != gajim.otr_module.OTRL_SMP_EXPECT3:
					self._abort()
				else:
					self.ctx.smstate.nextExpected = gajim.otr_module.OTRL_SMP_EXPECT1;
					if self.ctx.active_fingerprint.trust:
						self._finish("SMP verifying succeeded")
					else:
						self._finish("SMP verifying failed")
			tlv = gajim.otr_module.otrl_tlv_find(tlvs, gajim.otr_module.OTRL_TLV_SMP4)
			if tlv:
				if nextTLV != gajim.otr_module.OTRL_SMP_EXPECT4:
					self._abort()
				else:
					self.ctx.smstate.nextExpected = gajim.otr_module.OTRL_SMP_EXPECT1;
					if self.ctx.active_fingerprint.trust:
						self._finish("SMP verifying succeeded")
					else:
						self._finish("SMP verifying failed")
			tlv = gajim.otr_module.otrl_tlv_find(tlvs, gajim.otr_module.OTRL_TLV_SMP_ABORT)
			if tlv:
				self._finish("SMP verifying aborted")

	def _on_destroy(self, widget):
		if self.smp_running:
			self._abort("user aborted SMP authentication")
		self.window.hide_all()

	def _apply(self, widget):
		if self.finished:
			self.window.hide_all()
			return
		secret = self.gw("secret_entry").get_text()
		if self.response:
			gajim.otr_module.otrl_message_respond_smp(gajim.connections[self.account].otr_userstates,
					(gajim.otr_ui_ops, {'account':self.account}), self.ctx, secret)
		else:
			gajim.otr_module.otrl_message_initiate_smp(gajim.connections[self.account].otr_userstates,
					(gajim.otr_ui_ops, {'account':self.account}), self.ctx, secret)
			self.gw("progressbar").set_fraction(0.3)
		self.smp_running = True
		widget.set_sensitive(False)


class ContactOtrWindow:

	def gw(self, n):
		""" shorthand for self.xml.get_widget(n)"""
		return self.xml.get_widget(n)

	def __init__(self, contact, account, ctrl=None):
		self.contact = contact
		self.account = account
		self.ctrl = ctrl

		self.ctx = gajim.otr_module.otrl_context_find(
			gajim.connections[self.account].otr_userstates,
			self.contact.get_full_jid().encode(),
			gajim.get_jid_from_account(self.account).encode(),
			gajim.OTR_PROTO, 1, (gajim.otr_add_appdata, self.account))[0]

		self.xml = gtkgui_helpers.get_glade("contact_otr_window.glade")
		self.window = self.xml.get_widget("otr_settings_window")

		self.gw("settings_cancel_button").connect("clicked", self._on_destroy)
		self.gw("settings_ok_button").connect("clicked", self._apply)
		self.gw("otr_default_checkbutton").connect("toggled",
				self._otr_default_checkbutton_toggled)

		self.window.set_title("OTR settings for %s"%
			self.contact.get_full_jid())

		# always set the label containing our fingerprint
		self.gw("our_fp_label").set_markup(our_fp_text%
			gajim.otr_module.otrl_privkey_fingerprint(
				gajim.connections[self.account].otr_userstates,
				gajim.get_jid_from_account(self.account).encode(),
				gajim.OTR_PROTO))

		if self.ctx.msgstate != gajim.otr_module.OTRL_MSGSTATE_ENCRYPTED:
			# make the fingerprint widgets insensitive when not encrypted
			for widget in self.gw("otr_fp_vbox").get_children():
				widget.set_sensitive(False)
			# show that the fingerprint is unknown
			self.gw("their_fp_label").set_markup(
				their_fp_text%(self.contact.get_full_jid(),
				"unknown"))
			self.gw("verified_combobox").set_active(-1)
		else:
			# make the fingerprint widgets sensitive when encrypted
			for widget in self.gw("otr_fp_vbox").get_children():
				widget.set_sensitive(True)
			# show their fingerprint
			self.gw("their_fp_label").set_markup(
				their_fp_text%(self.contact.get_full_jid(),
				gajim.otr_module.otrl_privkey_hash_to_human(
					self.ctx.active_fingerprint.fingerprint)))
			# set the trust combobox
			if self.ctx.active_fingerprint.trust:
				self.gw("verified_combobox").set_active(1)
			else:
				self.gw("verified_combobox").set_active(0)

		otr_flags = gajim.config.get_per("contacts", self.contact.jid,
			"otr_flags")

		if otr_flags >= 0:
			self.gw("otr_default_checkbutton").set_active(0)
			for w in self.gw("otr_settings_vbox").get_children():
				w.set_sensitive(True)
		else:
			# per-user settings not available, using default settings
			otr_flags = gajim.config.get_per("accounts", self.account,
				"otr_flags")
			self.gw("otr_default_checkbutton").set_active(1)
			for w in self.gw("otr_settings_vbox").get_children():
				w.set_sensitive(False)

		self.gw("otr_policy_allow_v1_checkbutton").set_active(
			otr_flags & gajim.otr_module.OTRL_POLICY_ALLOW_V1)
		self.gw("otr_policy_allow_v2_checkbutton").set_active(
			otr_flags & gajim.otr_module.OTRL_POLICY_ALLOW_V2)
		self.gw("otr_policy_require_checkbutton").set_active(
			otr_flags & gajim.otr_module.OTRL_POLICY_REQUIRE_ENCRYPTION)
		self.gw("otr_policy_send_tag_checkbutton").set_active(
			otr_flags & gajim.otr_module.OTRL_POLICY_SEND_WHITESPACE_TAG)
		self.gw("otr_policy_start_on_tag_checkbutton").set_active(
			otr_flags & gajim.otr_module.OTRL_POLICY_WHITESPACE_START_AKE)
		self.gw("otr_policy_start_on_error_checkbutton").set_active(
			otr_flags & gajim.otr_module.OTRL_POLICY_ERROR_START_AKE)

		self.window.show_all()

	def _on_destroy(self, widget):
		self.window.destroy()

	def _apply(self, widget):
		# -1 when nothing is selected (ie. the connection is not encrypted)
		trust_state = self.gw("verified_combobox").get_active()
		if  trust_state == 1 and not self.ctx.active_fingerprint.trust:
			gajim.otr_module.otrl_context_set_trust(
				self.ctx.active_fingerprint, "verified")
			gajim.otr_ui_ops.write_fingerprints({'account':self.account})
		elif trust_state == 0:
			gajim.otr_module.otrl_context_set_trust(
				self.ctx.active_fingerprint, "")
			gajim.otr_ui_ops.write_fingerprints({'account':self.account})

		if not self.ctrl:
			self.ctrl = gajim.interface.msg_win_mgr.get_control(
					self.contact.jid, self.account)
		if self.ctrl:
			self.ctrl.update_ui()

		if self.gw("otr_default_checkbutton").get_active():
			# default is enabled, so remove any user-specific settings if available
			gajim.config.set_per("contacts", self.contact.jid, "otr_flags", -1)
		else:
			# build the flags using the checkboxes
			flags = 0
			flags += self.gw("otr_policy_allow_v1_checkbutton").get_active() \
				and gajim.otr_module.OTRL_POLICY_ALLOW_V1
			flags += self.gw("otr_policy_allow_v2_checkbutton").get_active() \
				and gajim.otr_module.OTRL_POLICY_ALLOW_V2
			flags += self.gw("otr_policy_require_checkbutton").get_active() \
				and gajim.otr_module.OTRL_POLICY_REQUIRE_ENCRYPTION
			flags += self.gw("otr_policy_send_tag_checkbutton").get_active() \
				and gajim.otr_module.OTRL_POLICY_SEND_WHITESPACE_TAG
			flags += self.gw("otr_policy_start_on_tag_checkbutton").get_active() \
				and gajim.otr_module.OTRL_POLICY_WHITESPACE_START_AKE
			flags += self.gw("otr_policy_start_on_error_checkbutton").get_active() \
				and gajim.otr_module.OTRL_POLICY_ERROR_START_AKE
			
			gajim.config.add_per("contacts", self.contact.jid)
			gajim.config.set_per("contacts", self.contact.jid, "otr_flags", flags)

		self._on_destroy(widget)

	def _otr_default_checkbutton_toggled(self, widget):
		for w in self.gw("otr_settings_vbox").get_children():
			w.set_sensitive(not widget.get_active())