git.maemo.org Git - gc-dialer/blob - src/gc_views.py

   1 #!/usr/bin/python2.5
   2
   3 """
   4 DialCentral - Front end for Google's Grand Central service.
   5 Copyright (C) 2008  Mark Bergman bergman AT merctech DOT com
   6
   7 This library is free software; you can redistribute it and/or
   8 modify it under the terms of the GNU Lesser General Public
   9 License as published by the Free Software Foundation; either
  10 version 2.1 of the License, or (at your option) any later version.
  11
  12 This library is distributed in the hope that it will be useful,
  13 but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 Lesser General Public License for more details.
  16
  17 You should have received a copy of the GNU Lesser General Public
  18 License along with this library; if not, write to the Free Software
  19 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  20 """
  21
  22 from __future__ import with_statement
  23
  24 import threading
  25 import warnings
  26
  27 import gobject
  28 import gtk
  29
  30 import gtk_toolbox
  31 import null_backend
  32
  33
  34 def make_ugly(prettynumber):
  35         """
  36         function to take a phone number and strip out all non-numeric
  37         characters
  38
  39         >>> make_ugly("+012-(345)-678-90")
  40         '01234567890'
  41         """
  42         import re
  43         uglynumber = re.sub('\D', '', prettynumber)
  44         return uglynumber
  45
  46
  47 def make_pretty(phonenumber):
  48         """
  49         Function to take a phone number and return the pretty version
  50         pretty numbers:
  51                 if phonenumber begins with 0:
  52                         ...-(...)-...-....
  53                 if phonenumber begins with 1: ( for gizmo callback numbers )
  54                         1 (...)-...-....
  55                 if phonenumber is 13 digits:
  56                         (...)-...-....
  57                 if phonenumber is 10 digits:
  58                         ...-....
  59         >>> make_pretty("12")
  60         '12'
  61         >>> make_pretty("1234567")
  62         '123-4567'
  63         >>> make_pretty("2345678901")
  64         '(234)-567-8901'
  65         >>> make_pretty("12345678901")
  66         '1 (234)-567-8901'
  67         >>> make_pretty("01234567890")
  68         '+012-(345)-678-90'
  69         """
  70         if phonenumber is None or phonenumber is "":
  71                 return ""
  72
  73         phonenumber = make_ugly(phonenumber)
  74
  75         if len(phonenumber) < 3:
  76                 return phonenumber
  77
  78         if phonenumber[0] == "0":
  79                 prettynumber = ""
  80                 prettynumber += "+%s" % phonenumber[0:3]
  81                 if 3 < len(phonenumber):
  82                         prettynumber += "-(%s)" % phonenumber[3:6]
  83                         if 6 < len(phonenumber):
  84                                 prettynumber += "-%s" % phonenumber[6:9]
  85                                 if 9 < len(phonenumber):
  86                                         prettynumber += "-%s" % phonenumber[9:]
  87                 return prettynumber
  88         elif len(phonenumber) <= 7:
  89                 prettynumber = "%s-%s" % (phonenumber[0:3], phonenumber[3:])
  90         elif len(phonenumber) > 8 and phonenumber[0] == "1":
  91                 prettynumber = "1 (%s)-%s-%s" % (phonenumber[1:4], phonenumber[4:7], phonenumber[7:])
  92         elif len(phonenumber) > 7:
  93                 prettynumber = "(%s)-%s-%s" % (phonenumber[0:3], phonenumber[3:6], phonenumber[6:])
  94         return prettynumber
  95
  96
  97 class MergedAddressBook(object):
  98         """
  99         Merger of all addressbooks
 100         """
 101
 102         def __init__(self, addressbookFactories, sorter = None):
 103                 self.__addressbookFactories = addressbookFactories
 104                 self.__addressbooks = None
 105                 self.__sort_contacts = sorter if sorter is not None else self.null_sorter
 106
 107         def clear_caches(self):
 108                 self.__addressbooks = None
 109                 for factory in self.__addressbookFactories:
 110                         factory.clear_caches()
 111
 112         def get_addressbooks(self):
 113                 """
 114                 @returns Iterable of (Address Book Factory, Book Id, Book Name)
 115                 """
 116                 yield self, "", ""
 117
 118         def open_addressbook(self, bookId):
 119                 return self
 120
 121         def contact_source_short_name(self, contactId):
 122                 if self.__addressbooks is None:
 123                         return ""
 124                 bookIndex, originalId = contactId.split("-", 1)
 125                 return self.__addressbooks[int(bookIndex)].contact_source_short_name(originalId)
 126
 127         @staticmethod
 128         def factory_name():
 129                 return "All Contacts"
 130
 131         def get_contacts(self):
 132                 """
 133                 @returns Iterable of (contact id, contact name)
 134                 """
 135                 if self.__addressbooks is None:
 136                         self.__addressbooks = list(
 137                                 factory.open_addressbook(id)
 138                                 for factory in self.__addressbookFactories
 139                                 for (f, id, name) in factory.get_addressbooks()
 140                         )
 141                 contacts = (
 142                         ("-".join([str(bookIndex), contactId]), contactName)
 143                                 for (bookIndex, addressbook) in enumerate(self.__addressbooks)
 144                                         for (contactId, contactName) in addressbook.get_contacts()
 145                 )
 146                 sortedContacts = self.__sort_contacts(contacts)
 147                 return sortedContacts
 148
 149         def get_contact_details(self, contactId):
 150                 """
 151                 @returns Iterable of (Phone Type, Phone Number)
 152                 """
 153                 if self.__addressbooks is None:
 154                         return []
 155                 bookIndex, originalId = contactId.split("-", 1)
 156                 return self.__addressbooks[int(bookIndex)].get_contact_details(originalId)
 157
 158         @staticmethod
 159         def null_sorter(contacts):
 160                 """
 161                 Good for speed/low memory
 162                 """
 163                 return contacts
 164
 165         @staticmethod
 166         def basic_firtname_sorter(contacts):
 167                 """
 168                 Expects names in "First Last" format
 169                 """
 170                 contactsWithKey = [
 171                         (contactName.rsplit(" ", 1)[0], (contactId, contactName))
 172                                 for (contactId, contactName) in contacts
 173                 ]
 174                 contactsWithKey.sort()
 175                 return (contactData for (lastName, contactData) in contactsWithKey)
 176
 177         @staticmethod
 178         def basic_lastname_sorter(contacts):
 179                 """
 180                 Expects names in "First Last" format
 181                 """
 182                 contactsWithKey = [
 183                         (contactName.rsplit(" ", 1)[-1], (contactId, contactName))
 184                                 for (contactId, contactName) in contacts
 185                 ]
 186                 contactsWithKey.sort()
 187                 return (contactData for (lastName, contactData) in contactsWithKey)
 188
 189         @staticmethod
 190         def reversed_firtname_sorter(contacts):
 191                 """
 192                 Expects names in "Last, First" format
 193                 """
 194                 contactsWithKey = [
 195                         (contactName.split(", ", 1)[-1], (contactId, contactName))
 196                                 for (contactId, contactName) in contacts
 197                 ]
 198                 contactsWithKey.sort()
 199                 return (contactData for (lastName, contactData) in contactsWithKey)
 200
 201         @staticmethod
 202         def reversed_lastname_sorter(contacts):
 203                 """
 204                 Expects names in "Last, First" format
 205                 """
 206                 contactsWithKey = [
 207                         (contactName.split(", ", 1)[0], (contactId, contactName))
 208                                 for (contactId, contactName) in contacts
 209                 ]
 210                 contactsWithKey.sort()
 211                 return (contactData for (lastName, contactData) in contactsWithKey)
 212
 213         @staticmethod
 214         def guess_firstname(name):
 215                 if ", " in name:
 216                         return name.split(", ", 1)[-1]
 217                 else:
 218                         return name.rsplit(" ", 1)[0]
 219
 220         @staticmethod
 221         def guess_lastname(name):
 222                 if ", " in name:
 223                         return name.split(", ", 1)[0]
 224                 else:
 225                         return name.rsplit(" ", 1)[-1]
 226
 227         @classmethod
 228         def advanced_firstname_sorter(cls, contacts):
 229                 contactsWithKey = [
 230                         (cls.guess_firstname(contactName), (contactId, contactName))
 231                                 for (contactId, contactName) in contacts
 232                 ]
 233                 contactsWithKey.sort()
 234                 return (contactData for (lastName, contactData) in contactsWithKey)
 235
 236         @classmethod
 237         def advanced_lastname_sorter(cls, contacts):
 238                 contactsWithKey = [
 239                         (cls.guess_lastname(contactName), (contactId, contactName))
 240                                 for (contactId, contactName) in contacts
 241                 ]
 242                 contactsWithKey.sort()
 243                 return (contactData for (lastName, contactData) in contactsWithKey)
 244
 245
 246 class PhoneTypeSelector(object):
 247
 248         ACTION_CANCEL = "cancel"
 249         ACTION_SELECT = "select"
 250         ACTION_DIAL = "dial"
 251         ACTION_SEND_SMS = "sms"
 252
 253         def __init__(self, widgetTree, gcBackend):
 254                 self._gcBackend = gcBackend
 255                 self._widgetTree = widgetTree
 256
 257                 self._dialog = self._widgetTree.get_widget("phonetype_dialog")
 258                 self._smsDialog = SmsEntryDialog(self._widgetTree)
 259
 260                 self._smsButton = self._widgetTree.get_widget("sms_button")
 261                 self._smsButton.connect("clicked", self._on_phonetype_send_sms)
 262
 263                 self._dialButton = self._widgetTree.get_widget("dial_button")
 264                 self._dialButton.connect("clicked", self._on_phonetype_dial)
 265
 266                 self._selectButton = self._widgetTree.get_widget("select_button")
 267                 self._selectButton.connect("clicked", self._on_phonetype_select)
 268
 269                 self._cancelButton = self._widgetTree.get_widget("cancel_button")
 270                 self._cancelButton.connect("clicked", self._on_phonetype_cancel)
 271
 272                 self._typemodel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING)
 273                 self._typeviewselection = None
 274
 275                 self._message = self._widgetTree.get_widget("phoneSelectionMessage")
 276                 self._typeview = self._widgetTree.get_widget("phonetypes")
 277                 self._typeview.connect("row-activated", self._on_phonetype_select)
 278
 279                 self._action = self.ACTION_CANCEL
 280
 281         def run(self, contactDetails, message = "", parent = None):
 282                 self._action = self.ACTION_CANCEL
 283                 self._typemodel.clear()
 284                 self._typeview.set_model(self._typemodel)
 285
 286                 # Add the column to the treeview
 287                 textrenderer = gtk.CellRendererText()
 288                 numberColumn = gtk.TreeViewColumn("Phone Numbers", textrenderer, text=0)
 289                 self._typeview.append_column(numberColumn)
 290
 291                 textrenderer = gtk.CellRendererText()
 292                 typeColumn = gtk.TreeViewColumn("Phone Type", textrenderer, text=1)
 293                 self._typeview.append_column(typeColumn)
 294
 295                 self._typeviewselection = self._typeview.get_selection()
 296                 self._typeviewselection.set_mode(gtk.SELECTION_SINGLE)
 297
 298                 for phoneType, phoneNumber in contactDetails:
 299                         display = " - ".join((phoneNumber, phoneType))
 300                         display = phoneType
 301                         row = (phoneNumber, display)
 302                         self._typemodel.append(row)
 303
 304                 self._typeviewselection.select_iter(self._typemodel.get_iter_first())
 305                 if message:
 306                         self._message.set_markup(message)
 307                         self._message.show()
 308                 else:
 309                         self._message.set_markup("")
 310                         self._message.hide()
 311
 312                 if parent is not None:
 313                         self._dialog.set_transient_for(parent)
 314
 315                 try:
 316                         userResponse = self._dialog.run()
 317                 finally:
 318                         self._dialog.hide()
 319
 320                 if userResponse == gtk.RESPONSE_OK:
 321                         phoneNumber = self._get_number()
 322                         phoneNumber = make_ugly(phoneNumber)
 323                 else:
 324                         phoneNumber = ""
 325                 if not phoneNumber:
 326                         self._action = self.ACTION_CANCEL
 327
 328                 if self._action == self.ACTION_SEND_SMS:
 329                         smsMessage = self._smsDialog.run(phoneNumber, message, parent)
 330                         if not smsMessage:
 331                                 phoneNumber = ""
 332                                 self._action = self.ACTION_CANCEL
 333                 else:
 334                         smsMessage = ""
 335
 336                 self._typeviewselection.unselect_all()
 337                 self._typeview.remove_column(numberColumn)
 338                 self._typeview.remove_column(typeColumn)
 339                 self._typeview.set_model(None)
 340
 341                 return self._action, phoneNumber, smsMessage
 342
 343         def _get_number(self):
 344                 model, itr = self._typeviewselection.get_selected()
 345                 if not itr:
 346                         return ""
 347
 348                 phoneNumber = self._typemodel.get_value(itr, 0)
 349                 return phoneNumber
 350
 351         def _on_phonetype_dial(self, *args):
 352                 self._dialog.response(gtk.RESPONSE_OK)
 353                 self._action = self.ACTION_DIAL
 354
 355         def _on_phonetype_send_sms(self, *args):
 356                 self._dialog.response(gtk.RESPONSE_OK)
 357                 self._action = self.ACTION_SEND_SMS
 358
 359         def _on_phonetype_select(self, *args):
 360                 self._dialog.response(gtk.RESPONSE_OK)
 361                 self._action = self.ACTION_SELECT
 362
 363         def _on_phonetype_cancel(self, *args):
 364                 self._dialog.response(gtk.RESPONSE_CANCEL)
 365                 self._action = self.ACTION_CANCEL
 366
 367
 368 class SmsEntryDialog(object):
 369
 370         """
 371         @todo Add multi-SMS messages like GoogleVoice
 372         """
 373
 374         MAX_CHAR = 160
 375
 376         def __init__(self, widgetTree):
 377                 self._widgetTree = widgetTree
 378                 self._dialog = self._widgetTree.get_widget("smsDialog")
 379
 380                 self._smsButton = self._widgetTree.get_widget("sendSmsButton")
 381                 self._smsButton.connect("clicked", self._on_send)
 382
 383                 self._cancelButton = self._widgetTree.get_widget("cancelSmsButton")
 384                 self._cancelButton.connect("clicked", self._on_cancel)
 385
 386                 self._letterCountLabel = self._widgetTree.get_widget("smsLetterCount")
 387                 self._message = self._widgetTree.get_widget("smsMessage")
 388                 self._smsEntry = self._widgetTree.get_widget("smsEntry")
 389                 self._smsEntry.get_buffer().connect("changed", self._on_entry_changed)
 390
 391         def run(self, number, message = "", parent = None):
 392                 if message:
 393                         self._message.set_markup(message)
 394                         self._message.show()
 395                 else:
 396                         self._message.set_markup("")
 397                         self._message.hide()
 398                 self._smsEntry.get_buffer().set_text("")
 399                 self._update_letter_count()
 400
 401                 if parent is not None:
 402                         self._dialog.set_transient_for(parent)
 403
 404                 try:
 405                         userResponse = self._dialog.run()
 406                 finally:
 407                         self._dialog.hide()
 408
 409                 if userResponse == gtk.RESPONSE_OK:
 410                         entryBuffer = self._smsEntry.get_buffer()
 411                         enteredMessage = entryBuffer.get_text(entryBuffer.get_start_iter(), entryBuffer.get_end_iter())
 412                         enteredMessage = enteredMessage[0:self.MAX_CHAR]
 413                 else:
 414                         enteredMessage = ""
 415
 416                 return enteredMessage.strip()
 417
 418         def _update_letter_count(self, *args):
 419                 entryLength = self._smsEntry.get_buffer().get_char_count()
 420                 charsLeft = self.MAX_CHAR - entryLength
 421                 self._letterCountLabel.set_text(str(charsLeft))
 422                 if charsLeft < 0:
 423                         self._smsButton.set_sensitive(False)
 424                 else:
 425                         self._smsButton.set_sensitive(True)
 426
 427         def _on_entry_changed(self, *args):
 428                 self._update_letter_count()
 429
 430         def _on_send(self, *args):
 431                 self._dialog.response(gtk.RESPONSE_OK)
 432
 433         def _on_cancel(self, *args):
 434                 self._dialog.response(gtk.RESPONSE_CANCEL)
 435
 436
 437 class Dialpad(object):
 438
 439         def __init__(self, widgetTree, errorDisplay):
 440                 self._errorDisplay = errorDisplay
 441                 self._smsDialog = SmsEntryDialog(widgetTree)
 442
 443                 self._numberdisplay = widgetTree.get_widget("numberdisplay")
 444                 self._dialButton = widgetTree.get_widget("dial")
 445                 self._phonenumber = ""
 446                 self._prettynumber = ""
 447                 self._clearall_id = None
 448
 449                 callbackMapping = {
 450                         "on_dial_clicked": self._on_dial_clicked,
 451                         "on_sms_clicked": self._on_sms_clicked,
 452                         "on_digit_clicked": self._on_digit_clicked,
 453                         "on_clear_number": self._on_clear_number,
 454                         "on_back_clicked": self._on_backspace,
 455                         "on_back_pressed": self._on_back_pressed,
 456                         "on_back_released": self._on_back_released,
 457                 }
 458                 widgetTree.signal_autoconnect(callbackMapping)
 459
 460                 self._window = gtk_toolbox.find_parent_window(self._numberdisplay)
 461
 462         def enable(self):
 463                 self._dialButton.grab_focus()
 464
 465         def disable(self):
 466                 pass
 467
 468         def number_selected(self, action, number, message):
 469                 """
 470                 @note Actual dial function is patched in later
 471                 """
 472                 raise NotImplementedError
 473
 474         def get_number(self):
 475                 return self._phonenumber
 476
 477         def set_number(self, number):
 478                 """
 479                 Set the callback phonenumber
 480                 """
 481                 try:
 482                         self._phonenumber = make_ugly(number)
 483                         self._prettynumber = make_pretty(self._phonenumber)
 484                         self._numberdisplay.set_label("<span size='30000' weight='bold'>%s</span>" % (self._prettynumber))
 485                 except TypeError, e:
 486                         self._errorDisplay.push_exception(e)
 487
 488         def clear(self):
 489                 self.set_number("")
 490
 491         @staticmethod
 492         def name():
 493                 return "Dialpad"
 494
 495         def load_settings(self, config, section):
 496                 pass
 497
 498         def save_settings(self, config, section):
 499                 """
 500                 @note Thread Agnostic
 501                 """
 502                 pass
 503
 504         def _on_sms_clicked(self, widget):
 505                 action = PhoneTypeSelector.ACTION_SEND_SMS
 506                 phoneNumber = self.get_number()
 507
 508                 message = self._smsDialog.run(phoneNumber, "", self._window)
 509                 if not message:
 510                         phoneNumber = ""
 511                         action = PhoneTypeSelector.ACTION_CANCEL
 512
 513                 if action == PhoneTypeSelector.ACTION_CANCEL:
 514                         return
 515                 self.number_selected(action, phoneNumber, message)
 516
 517         def _on_dial_clicked(self, widget):
 518                 action = PhoneTypeSelector.ACTION_DIAL
 519                 phoneNumber = self.get_number()
 520                 message = ""
 521                 self.number_selected(action, phoneNumber, message)
 522
 523         def _on_clear_number(self, *args):
 524                 self.clear()
 525
 526         def _on_digit_clicked(self, widget):
 527                 self.set_number(self._phonenumber + widget.get_name()[-1])
 528
 529         def _on_backspace(self, widget):
 530                 self.set_number(self._phonenumber[:-1])
 531
 532         def _on_clearall(self):
 533                 self.clear()
 534                 return False
 535
 536         def _on_back_pressed(self, widget):
 537                 self._clearall_id = gobject.timeout_add(1000, self._on_clearall)
 538
 539         def _on_back_released(self, widget):
 540                 if self._clearall_id is not None:
 541                         gobject.source_remove(self._clearall_id)
 542                 self._clearall_id = None
 543
 544
 545 class AccountInfo(object):
 546
 547         def __init__(self, widgetTree, backend, errorDisplay):
 548                 self._errorDisplay = errorDisplay
 549                 self._backend = backend
 550                 self._isPopulated = False
 551
 552                 self._callbackList = gtk.ListStore(gobject.TYPE_STRING)
 553                 self._accountViewNumberDisplay = widgetTree.get_widget("gcnumber_display")
 554                 self._callbackCombo = widgetTree.get_widget("callbackcombo")
 555                 self._onCallbackentryChangedId = 0
 556
 557                 self._defaultCallback = ""
 558
 559         def enable(self):
 560                 assert self._backend.is_authed()
 561                 self._accountViewNumberDisplay.set_use_markup(True)
 562                 self.set_account_number("")
 563                 self._callbackList.clear()
 564                 self._onCallbackentryChangedId = self._callbackCombo.get_child().connect("changed", self._on_callbackentry_changed)
 565                 self.update(force=True)
 566
 567         def disable(self):
 568                 self._callbackCombo.get_child().disconnect(self._onCallbackentryChangedId)
 569
 570                 self.clear()
 571
 572                 self._callbackList.clear()
 573
 574         def get_selected_callback_number(self):
 575                 return make_ugly(self._callbackCombo.get_child().get_text())
 576
 577         def set_account_number(self, number):
 578                 """
 579                 Displays current account number
 580                 """
 581                 self._accountViewNumberDisplay.set_label("<span size='23000' weight='bold'>%s</span>" % (number))
 582
 583         def update(self, force = False):
 584                 if not force and self._isPopulated:
 585                         return
 586                 self._populate_callback_combo()
 587                 self.set_account_number(self._backend.get_account_number())
 588
 589         def clear(self):
 590                 self._callbackCombo.get_child().set_text("")
 591                 self.set_account_number("")
 592                 self._isPopulated = False
 593
 594         @staticmethod
 595         def name():
 596                 return "Account Info"
 597
 598         def load_settings(self, config, section):
 599                 self._defaultCallback = config.get(section, "callback")
 600
 601         def save_settings(self, config, section):
 602                 """
 603                 @note Thread Agnostic
 604                 """
 605                 callback = self.get_selected_callback_number()
 606                 config.set(section, "callback", callback)
 607
 608         def _populate_callback_combo(self):
 609                 self._isPopulated = True
 610                 self._callbackList.clear()
 611                 try:
 612                         callbackNumbers = self._backend.get_callback_numbers()
 613                 except StandardError, e:
 614                         self._errorDisplay.push_exception(e)
 615                         self._isPopulated = False
 616                         return
 617
 618                 for number, description in callbackNumbers.iteritems():
 619                         self._callbackList.append((make_pretty(number),))
 620
 621                 self._callbackCombo.set_model(self._callbackList)
 622                 self._callbackCombo.set_text_column(0)
 623                 #callbackNumber = self._backend.get_callback_number()
 624                 callbackNumber = self._defaultCallback
 625                 self._callbackCombo.get_child().set_text(make_pretty(callbackNumber))
 626
 627         def _set_callback_number(self, number):
 628                 try:
 629                         if not self._backend.is_valid_syntax(number):
 630                                 self._errorDisplay.push_message("%s is not a valid callback number" % number)
 631                         elif number == self._backend.get_callback_number():
 632                                 warnings.warn(
 633                                         "Callback number already is %s" % (
 634                                                 self._backend.get_callback_number(),
 635                                         ),
 636                                         UserWarning,
 637                                         2
 638                                 )
 639                         else:
 640                                 self._backend.set_callback_number(number)
 641                                 assert make_ugly(number) == make_ugly(self._backend.get_callback_number()), "Callback number should be %s but instead is %s" % (
 642                                         make_pretty(number), make_pretty(self._backend.get_callback_number())
 643                                 )
 644                                 warnings.warn(
 645                                         "Callback number set to %s" % (
 646                                                 self._backend.get_callback_number(),
 647                                         ),
 648                                         UserWarning, 2
 649                                 )
 650                 except StandardError, e:
 651                         self._errorDisplay.push_exception(e)
 652
 653         def _on_callbackentry_changed(self, *args):
 654                 text = self.get_selected_callback_number()
 655                 number = make_ugly(text)
 656                 self._set_callback_number(number)
 657
 658
 659 class RecentCallsView(object):
 660
 661         NUMBER_IDX = 0
 662         DATE_IDX = 1
 663         ACTION_IDX = 2
 664         FROM_IDX = 3
 665
 666         def __init__(self, widgetTree, backend, errorDisplay):
 667                 self._errorDisplay = errorDisplay
 668                 self._backend = backend
 669
 670                 self._isPopulated = False
 671                 self._recentmodel = gtk.ListStore(
 672                         gobject.TYPE_STRING, # number
 673                         gobject.TYPE_STRING, # date
 674                         gobject.TYPE_STRING, # action
 675                         gobject.TYPE_STRING, # from
 676                 )
 677                 self._recentview = widgetTree.get_widget("recentview")
 678                 self._recentviewselection = None
 679                 self._onRecentviewRowActivatedId = 0
 680
 681                 textrenderer = gtk.CellRendererText()
 682                 textrenderer.set_property("yalign", 0)
 683                 self._dateColumn = gtk.TreeViewColumn("Date")
 684                 self._dateColumn.pack_start(textrenderer, expand=True)
 685                 self._dateColumn.add_attribute(textrenderer, "text", self.DATE_IDX)
 686
 687                 textrenderer = gtk.CellRendererText()
 688                 textrenderer.set_property("yalign", 0)
 689                 self._actionColumn = gtk.TreeViewColumn("Action")
 690                 self._actionColumn.pack_start(textrenderer, expand=True)
 691                 self._actionColumn.add_attribute(textrenderer, "text", self.ACTION_IDX)
 692
 693                 textrenderer = gtk.CellRendererText()
 694                 textrenderer.set_property("yalign", 0)
 695                 self._fromColumn = gtk.TreeViewColumn("From")
 696                 self._fromColumn.pack_start(textrenderer, expand=True)
 697                 self._fromColumn.add_attribute(textrenderer, "text", self.FROM_IDX)
 698                 self._fromColumn.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)
 699
 700                 self._window = gtk_toolbox.find_parent_window(self._recentview)
 701                 self._phoneTypeSelector = PhoneTypeSelector(widgetTree, self._backend)
 702
 703         def enable(self):
 704                 assert self._backend.is_authed()
 705                 self._recentview.set_model(self._recentmodel)
 706
 707                 self._recentview.append_column(self._dateColumn)
 708                 self._recentview.append_column(self._actionColumn)
 709                 self._recentview.append_column(self._fromColumn)
 710                 self._recentviewselection = self._recentview.get_selection()
 711                 self._recentviewselection.set_mode(gtk.SELECTION_SINGLE)
 712
 713                 self._onRecentviewRowActivatedId = self._recentview.connect("row-activated", self._on_recentview_row_activated)
 714
 715         def disable(self):
 716                 self._recentview.disconnect(self._onRecentviewRowActivatedId)
 717
 718                 self.clear()
 719
 720                 self._recentview.remove_column(self._dateColumn)
 721                 self._recentview.remove_column(self._actionColumn)
 722                 self._recentview.remove_column(self._fromColumn)
 723                 self._recentview.set_model(None)
 724
 725         def number_selected(self, action, number, message):
 726                 """
 727                 @note Actual dial function is patched in later
 728                 """
 729                 raise NotImplementedError
 730
 731         def update(self, force = False):
 732                 if not force and self._isPopulated:
 733                         return
 734                 backgroundPopulate = threading.Thread(target=self._idly_populate_recentview)
 735                 backgroundPopulate.setDaemon(True)
 736                 backgroundPopulate.start()
 737
 738         def clear(self):
 739                 self._isPopulated = False
 740                 self._recentmodel.clear()
 741
 742         @staticmethod
 743         def name():
 744                 return "Recent Calls"
 745
 746         def load_settings(self, config, section):
 747                 pass
 748
 749         def save_settings(self, config, section):
 750                 """
 751                 @note Thread Agnostic
 752                 """
 753                 pass
 754
 755         def _idly_populate_recentview(self):
 756                 self._isPopulated = True
 757                 self._recentmodel.clear()
 758
 759                 try:
 760                         recentItems = self._backend.get_recent()
 761                 except StandardError, e:
 762                         self._errorDisplay.push_exception_with_lock(e)
 763                         self._isPopulated = False
 764                         recentItems = []
 765
 766                 for personName, phoneNumber, date, action in recentItems:
 767                         if not personName:
 768                                 personName = "Unknown"
 769                         prettyNumber = phoneNumber[2:] if phoneNumber.startswith("+1") else phoneNumber
 770                         prettyNumber = make_pretty(prettyNumber)
 771                         description = "%s - %s" % (personName, prettyNumber)
 772                         item = (phoneNumber, date, action.capitalize(), description)
 773                         with gtk_toolbox.gtk_lock():
 774                                 self._recentmodel.append(item)
 775
 776                 return False
 777
 778         def _on_recentview_row_activated(self, treeview, path, view_column):
 779                 model, itr = self._recentviewselection.get_selected()
 780                 if not itr:
 781                         return
 782
 783                 number = self._recentmodel.get_value(itr, self.NUMBER_IDX)
 784                 number = make_ugly(number)
 785                 contactPhoneNumbers = [("Phone", number)]
 786                 description = self._recentmodel.get_value(itr, self.FROM_IDX)
 787
 788                 action, phoneNumber, message = self._phoneTypeSelector.run(
 789                         contactPhoneNumbers,
 790                         message = description,
 791                         parent = self._window,
 792                 )
 793                 if action == PhoneTypeSelector.ACTION_CANCEL:
 794                         return
 795                 assert phoneNumber
 796
 797                 self.number_selected(action, phoneNumber, message)
 798                 self._recentviewselection.unselect_all()
 799
 800
 801 class MessagesView(object):
 802
 803         NUMBER_IDX = 0
 804         DATE_IDX = 1
 805         HEADER_IDX = 2
 806         MESSAGE_IDX = 3
 807
 808         def __init__(self, widgetTree, backend, errorDisplay):
 809                 self._errorDisplay = errorDisplay
 810                 self._backend = backend
 811
 812                 self._isPopulated = False
 813                 self._messagemodel = gtk.ListStore(
 814                         gobject.TYPE_STRING, # number
 815                         gobject.TYPE_STRING, # date
 816                         gobject.TYPE_STRING, # header
 817                         gobject.TYPE_STRING, # message
 818                 )
 819                 self._messageview = widgetTree.get_widget("messages_view")
 820                 self._messageviewselection = None
 821                 self._onMessageviewRowActivatedId = 0
 822
 823                 textrenderer = gtk.CellRendererText()
 824                 textrenderer.set_property("yalign", 0)
 825                 self._dateColumn = gtk.TreeViewColumn("Date")
 826                 self._dateColumn.pack_start(textrenderer, expand=True)
 827                 self._dateColumn.add_attribute(textrenderer, "markup", self.DATE_IDX)
 828
 829                 textrenderer = gtk.CellRendererText()
 830                 textrenderer.set_property("yalign", 0)
 831                 self._headerColumn = gtk.TreeViewColumn("From")
 832                 self._headerColumn.pack_start(textrenderer, expand=True)
 833                 self._headerColumn.add_attribute(textrenderer, "markup", self.HEADER_IDX)
 834
 835                 textrenderer = gtk.CellRendererText()
 836                 textrenderer.set_property("yalign", 0)
 837                 self._messageColumn = gtk.TreeViewColumn("Messages")
 838                 self._messageColumn.pack_start(textrenderer, expand=True)
 839                 self._messageColumn.add_attribute(textrenderer, "markup", self.MESSAGE_IDX)
 840                 self._messageColumn.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)
 841
 842                 self._window = gtk_toolbox.find_parent_window(self._messageview)
 843                 self._phoneTypeSelector = PhoneTypeSelector(widgetTree, self._backend)
 844
 845         def enable(self):
 846                 assert self._backend.is_authed()
 847                 self._messageview.set_model(self._messagemodel)
 848
 849                 self._messageview.append_column(self._dateColumn)
 850                 self._messageview.append_column(self._headerColumn)
 851                 self._messageview.append_column(self._messageColumn)
 852                 self._messageviewselection = self._messageview.get_selection()
 853                 self._messageviewselection.set_mode(gtk.SELECTION_SINGLE)
 854
 855                 self._onMessageviewRowActivatedId = self._messageview.connect("row-activated", self._on_messageview_row_activated)
 856
 857         def disable(self):
 858                 self._messageview.disconnect(self._onMessageviewRowActivatedId)
 859
 860                 self.clear()
 861
 862                 self._messageview.remove_column(self._dateColumn)
 863                 self._messageview.remove_column(self._headerColumn)
 864                 self._messageview.remove_column(self._messageColumn)
 865                 self._messageview.set_model(None)
 866
 867         def number_selected(self, action, number, message):
 868                 """
 869                 @note Actual dial function is patched in later
 870                 """
 871                 raise NotImplementedError
 872
 873         def update(self, force = False):
 874                 if not force and self._isPopulated:
 875                         return
 876                 backgroundPopulate = threading.Thread(target=self._idly_populate_messageview)
 877                 backgroundPopulate.setDaemon(True)
 878                 backgroundPopulate.start()
 879
 880         def clear(self):
 881                 self._isPopulated = False
 882                 self._messagemodel.clear()
 883
 884         @staticmethod
 885         def name():
 886                 return "Messages"
 887
 888         def load_settings(self, config, section):
 889                 pass
 890
 891         def save_settings(self, config, section):
 892                 """
 893                 @note Thread Agnostic
 894                 """
 895                 pass
 896
 897         def _idly_populate_messageview(self):
 898                 self._isPopulated = True
 899                 self._messagemodel.clear()
 900
 901                 try:
 902                         messageItems = self._backend.get_messages()
 903                 except StandardError, e:
 904                         self._errorDisplay.push_exception_with_lock(e)
 905                         self._isPopulated = False
 906                         messageItems = []
 907
 908                 for header, number, relativeDate, message in messageItems:
 909                         number = make_ugly(number)
 910                         row = (number, relativeDate, header, message)
 911                         with gtk_toolbox.gtk_lock():
 912                                 self._messagemodel.append(row)
 913
 914                 return False
 915
 916         def _on_messageview_row_activated(self, treeview, path, view_column):
 917                 model, itr = self._messageviewselection.get_selected()
 918                 if not itr:
 919                         return
 920
 921                 contactPhoneNumbers = [("Phone", self._messagemodel.get_value(itr, self.NUMBER_IDX))]
 922                 description = self._messagemodel.get_value(itr, self.MESSAGE_IDX)
 923
 924                 action, phoneNumber, message = self._phoneTypeSelector.run(
 925                         contactPhoneNumbers,
 926                         message = description,
 927                         parent = self._window,
 928                 )
 929                 if action == PhoneTypeSelector.ACTION_CANCEL:
 930                         return
 931                 assert phoneNumber
 932
 933                 self.number_selected(action, phoneNumber, message)
 934                 self._messageviewselection.unselect_all()
 935
 936
 937 class ContactsView(object):
 938
 939         def __init__(self, widgetTree, backend, errorDisplay):
 940                 self._errorDisplay = errorDisplay
 941                 self._backend = backend
 942
 943                 self._addressBook = None
 944                 self._addressBookFactories = [null_backend.NullAddressBook()]
 945
 946                 self._booksList = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING)
 947                 self._booksSelectionBox = widgetTree.get_widget("addressbook_combo")
 948
 949                 self._isPopulated = False
 950                 self._contactsmodel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING)
 951                 self._contactsviewselection = None
 952                 self._contactsview = widgetTree.get_widget("contactsview")
 953
 954                 self._contactColumn = gtk.TreeViewColumn("Contact")
 955                 displayContactSource = False
 956                 if displayContactSource:
 957                         textrenderer = gtk.CellRendererText()
 958                         self._contactColumn.pack_start(textrenderer, expand=False)
 959                         self._contactColumn.add_attribute(textrenderer, 'text', 0)
 960                 textrenderer = gtk.CellRendererText()
 961                 self._contactColumn.pack_start(textrenderer, expand=True)
 962                 self._contactColumn.add_attribute(textrenderer, 'text', 1)
 963                 textrenderer = gtk.CellRendererText()
 964                 self._contactColumn.pack_start(textrenderer, expand=True)
 965                 self._contactColumn.add_attribute(textrenderer, 'text', 4)
 966                 self._contactColumn.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)
 967                 self._contactColumn.set_sort_column_id(1)
 968                 self._contactColumn.set_visible(True)
 969
 970                 self._onContactsviewRowActivatedId = 0
 971                 self._onAddressbookComboChangedId = 0
 972                 self._window = gtk_toolbox.find_parent_window(self._contactsview)
 973                 self._phoneTypeSelector = PhoneTypeSelector(widgetTree, self._backend)
 974
 975         def enable(self):
 976                 assert self._backend.is_authed()
 977
 978                 self._contactsview.set_model(self._contactsmodel)
 979                 self._contactsview.append_column(self._contactColumn)
 980                 self._contactsviewselection = self._contactsview.get_selection()
 981                 self._contactsviewselection.set_mode(gtk.SELECTION_SINGLE)
 982
 983                 self._booksList.clear()
 984                 for (factoryId, bookId), (factoryName, bookName) in self.get_addressbooks():
 985                         if factoryName and bookName:
 986                                 entryName = "%s: %s" % (factoryName, bookName)
 987                         elif factoryName:
 988                                 entryName = factoryName
 989                         elif bookName:
 990                                 entryName = bookName
 991                         else:
 992                                 entryName = "Bad name (%d)" % factoryId
 993                         row = (str(factoryId), bookId, entryName)
 994                         self._booksList.append(row)
 995
 996                 self._booksSelectionBox.set_model(self._booksList)
 997                 cell = gtk.CellRendererText()
 998                 self._booksSelectionBox.pack_start(cell, True)
 999                 self._booksSelectionBox.add_attribute(cell, 'text', 2)
1000                 self._booksSelectionBox.set_active(0)
1001
1002                 self._onContactsviewRowActivatedId = self._contactsview.connect("row-activated", self._on_contactsview_row_activated)
1003                 self._onAddressbookComboChangedId = self._booksSelectionBox.connect("changed", self._on_addressbook_combo_changed)
1004
1005         def disable(self):
1006                 self._contactsview.disconnect(self._onContactsviewRowActivatedId)
1007                 self._booksSelectionBox.disconnect(self._onAddressbookComboChangedId)
1008
1009                 self.clear()
1010
1011                 self._booksSelectionBox.clear()
1012                 self._booksSelectionBox.set_model(None)
1013                 self._contactsview.set_model(None)
1014                 self._contactsview.remove_column(self._contactColumn)
1015
1016         def number_selected(self, action, number, message):
1017                 """
1018                 @note Actual dial function is patched in later
1019                 """
1020                 raise NotImplementedError
1021
1022         def get_addressbooks(self):
1023                 """
1024                 @returns Iterable of ((Factory Id, Book Id), (Factory Name, Book Name))
1025                 """
1026                 for i, factory in enumerate(self._addressBookFactories):
1027                         for bookFactory, bookId, bookName in factory.get_addressbooks():
1028                                 yield (i, bookId), (factory.factory_name(), bookName)
1029
1030         def open_addressbook(self, bookFactoryId, bookId):
1031                 self._addressBook = self._addressBookFactories[bookFactoryId].open_addressbook(bookId)
1032                 self.update(force=True)
1033
1034         def update(self, force = False):
1035                 if not force and self._isPopulated:
1036                         return
1037                 backgroundPopulate = threading.Thread(target=self._idly_populate_contactsview)
1038                 backgroundPopulate.setDaemon(True)
1039                 backgroundPopulate.start()
1040
1041         def clear(self):
1042                 self._isPopulated = False
1043                 self._contactsmodel.clear()
1044
1045         def clear_caches(self):
1046                 for factory in self._addressBookFactories:
1047                         factory.clear_caches()
1048                 self._addressBook.clear_caches()
1049
1050         def append(self, book):
1051                 self._addressBookFactories.append(book)
1052
1053         def extend(self, books):
1054                 self._addressBookFactories.extend(books)
1055
1056         @staticmethod
1057         def name():
1058                 return "Contacts"
1059
1060         def load_settings(self, config, section):
1061                 pass
1062
1063         def save_settings(self, config, section):
1064                 """
1065                 @note Thread Agnostic
1066                 """
1067                 pass
1068
1069         def _idly_populate_contactsview(self):
1070                 self._isPopulated = True
1071                 self.clear()
1072
1073                 # completely disable updating the treeview while we populate the data
1074                 self._contactsview.freeze_child_notify()
1075                 try:
1076                         self._contactsview.set_model(None)
1077
1078                         addressBook = self._addressBook
1079                         try:
1080                                 contacts = addressBook.get_contacts()
1081                         except StandardError, e:
1082                                 contacts = []
1083                                 self._isPopulated = False
1084                                 self._errorDisplay.push_exception_with_lock(e)
1085                         for contactId, contactName in contacts:
1086                                 contactType = (addressBook.contact_source_short_name(contactId), )
1087                                 self._contactsmodel.append(contactType + (contactName, "", contactId) + ("", ))
1088
1089                         # restart the treeview data rendering
1090                         self._contactsview.set_model(self._contactsmodel)
1091                 finally:
1092                         self._contactsview.thaw_child_notify()
1093                 return False
1094
1095         def _on_addressbook_combo_changed(self, *args, **kwds):
1096                 itr = self._booksSelectionBox.get_active_iter()
1097                 if itr is None:
1098                         return
1099                 factoryId = int(self._booksList.get_value(itr, 0))
1100                 bookId = self._booksList.get_value(itr, 1)
1101                 self.open_addressbook(factoryId, bookId)
1102
1103         def _on_contactsview_row_activated(self, treeview, path, view_column):
1104                 model, itr = self._contactsviewselection.get_selected()
1105                 if not itr:
1106                         return
1107
1108                 contactId = self._contactsmodel.get_value(itr, 3)
1109                 contactName = self._contactsmodel.get_value(itr, 1)
1110                 try:
1111                         contactDetails = self._addressBook.get_contact_details(contactId)
1112                 except StandardError, e:
1113                         contactDetails = []
1114                         self._errorDisplay.push_exception(e)
1115                 contactPhoneNumbers = [phoneNumber for phoneNumber in contactDetails]
1116
1117                 if len(contactPhoneNumbers) == 0:
1118                         return
1119
1120                 action, phoneNumber, message = self._phoneTypeSelector.run(
1121                         contactPhoneNumbers,
1122                         message = contactName,
1123                         parent = self._window,
1124                 )
1125                 if action == PhoneTypeSelector.ACTION_CANCEL:
1126                         return
1127                 assert phoneNumber
1128
1129                 self.number_selected(action, phoneNumber, message)
1130                 self._contactsviewselection.unselect_all()