123
86
        self.logger = logger
 
125
 
        self._update_timer_callback_tag = None
 
126
 
        self._update_timer_callback_lock = 0
 
127
 
        self.last_checker_failed = False
 
129
88
        # The widget shown normally
 
130
 
        self._text_widget = urwid.Text("")
 
 
89
        self._text_widget = urwid.Text(u"")
 
131
90
        # The widget shown when we have focus
 
132
 
        self._focus_text_widget = urwid.Text("")
 
 
91
        self._focus_text_widget = urwid.Text(u"")
 
133
92
        super(MandosClientWidget, self).__init__(
 
134
93
            update_hook=update_hook, delete_hook=delete_hook,
 
137
96
        self.opened = False
 
139
 
        last_checked_ok = isoformat_to_datetime(self.properties
 
141
 
        if last_checked_ok is None:
 
142
 
            self.last_checker_failed = True
 
144
 
            self.last_checker_failed = ((datetime.datetime.utcnow()
 
151
 
        if self.last_checker_failed:
 
152
 
            self.using_timer(True)
 
154
 
        if self.need_approval:
 
155
 
            self.using_timer(True)
 
157
 
        self.proxy.connect_to_signal("CheckerCompleted",
 
 
97
        self.proxy.connect_to_signal(u"CheckerCompleted",
 
158
98
                                     self.checker_completed,
 
160
100
                                     byte_arrays=True)
 
161
 
        self.proxy.connect_to_signal("CheckerStarted",
 
 
101
        self.proxy.connect_to_signal(u"CheckerStarted",
 
162
102
                                     self.checker_started,
 
163
103
                                     client_interface,
 
164
104
                                     byte_arrays=True)
 
165
 
        self.proxy.connect_to_signal("GotSecret",
 
 
105
        self.proxy.connect_to_signal(u"GotSecret",
 
167
107
                                     client_interface,
 
168
108
                                     byte_arrays=True)
 
169
 
        self.proxy.connect_to_signal("NeedApproval",
 
 
109
        self.proxy.connect_to_signal(u"NeedApproval",
 
170
110
                                     self.need_approval,
 
171
111
                                     client_interface,
 
172
112
                                     byte_arrays=True)
 
173
 
        self.proxy.connect_to_signal("Rejected",
 
 
113
        self.proxy.connect_to_signal(u"Rejected",
 
175
115
                                     client_interface,
 
176
116
                                     byte_arrays=True)
 
178
 
    def property_changed(self, property=None, value=None):
 
179
 
        super(self, MandosClientWidget).property_changed(property,
 
181
 
        if property == "ApprovalPending":
 
182
 
            using_timer(bool(value))
 
184
 
    def using_timer(self, flag):
 
185
 
        """Call this method with True or False when timer should be
 
186
 
        activated or deactivated.
 
188
 
        old = self._update_timer_callback_lock
 
190
 
            self._update_timer_callback_lock += 1
 
192
 
            self._update_timer_callback_lock -= 1
 
193
 
        if old == 0 and self._update_timer_callback_lock:
 
194
 
            self._update_timer_callback_tag = (gobject.timeout_add
 
197
 
        elif old and self._update_timer_callback_lock == 0:
 
198
 
            gobject.source_remove(self._update_timer_callback_tag)
 
199
 
            self._update_timer_callback_tag = None
 
201
118
    def checker_completed(self, exitstatus, condition, command):
 
202
119
        if exitstatus == 0:
 
203
 
            if self.last_checker_failed:
 
204
 
                self.last_checker_failed = False
 
205
 
                self.using_timer(False)
 
206
 
            #self.logger('Checker for client %s (command "%s")'
 
208
 
            #            % (self.properties["Name"], command))
 
 
120
            #self.logger(u'Checker for client %s (command "%s")'
 
 
122
            #            % (self.properties[u"name"], command))
 
212
 
        if not self.last_checker_failed:
 
213
 
            self.last_checker_failed = True
 
214
 
            self.using_timer(True)
 
215
124
        if os.WIFEXITED(condition):
 
216
 
            self.logger('Checker for client %s (command "%s")'
 
217
 
                        ' failed with exit code %s'
 
218
 
                        % (self.properties["Name"], command,
 
 
125
            self.logger(u'Checker for client %s (command "%s")'
 
 
126
                        u' failed with exit code %s'
 
 
127
                        % (self.properties[u"name"], command,
 
219
128
                           os.WEXITSTATUS(condition)))
 
220
 
        elif os.WIFSIGNALED(condition):
 
221
 
            self.logger('Checker for client %s (command "%s")'
 
222
 
                        ' was killed by signal %s'
 
223
 
                        % (self.properties["Name"], command,
 
 
130
        if os.WIFSIGNALED(condition):
 
 
131
            self.logger(u'Checker for client %s (command "%s")'
 
 
132
                        u' was killed by signal %s'
 
 
133
                        % (self.properties[u"name"], command,
 
224
134
                           os.WTERMSIG(condition)))
 
225
 
        elif os.WCOREDUMP(condition):
 
226
 
            self.logger('Checker for client %s (command "%s")'
 
228
 
                        % (self.properties["Name"], command))
 
230
 
            self.logger('Checker for client %s completed'
 
 
136
        if os.WCOREDUMP(condition):
 
 
137
            self.logger(u'Checker for client %s (command "%s")'
 
 
139
                        % (self.properties[u"name"], command))
 
 
140
        self.logger(u'Checker for client %s completed mysteriously')
 
234
142
    def checker_started(self, command):
 
235
 
        #self.logger('Client %s started checker "%s"'
 
236
 
        #            % (self.properties["Name"], unicode(command)))
 
 
143
        #self.logger(u'Client %s started checker "%s"'
 
 
144
        #            % (self.properties[u"name"], unicode(command)))
 
239
147
    def got_secret(self):
 
240
 
        self.last_checker_failed = False
 
241
 
        self.logger('Client %s received its secret'
 
242
 
                    % self.properties["Name"])
 
 
148
        self.logger(u'Client %s received its secret'
 
 
149
                    % self.properties[u"name"])
 
244
151
    def need_approval(self, timeout, default):
 
246
 
            message = 'Client %s needs approval within %s seconds'
 
 
153
            message = u'Client %s needs approval within %s seconds'
 
248
 
            message = 'Client %s will get its secret in %s seconds'
 
 
155
            message = u'Client %s will get its secret in %s seconds'
 
249
156
        self.logger(message
 
250
 
                    % (self.properties["Name"], timeout/1000))
 
251
 
        self.using_timer(True)
 
 
157
                    % (self.properties[u"name"], timeout/1000))
 
253
159
    def rejected(self, reason):
 
254
 
        self.logger('Client %s was rejected; reason: %s'
 
255
 
                    % (self.properties["Name"], reason))
 
 
160
        self.logger(u'Client %s was rejected; reason: %s'
 
 
161
                    % (self.properties[u"name"], reason))
 
257
163
    def selectable(self):
 
258
164
        """Make this a "selectable" widget.
 
259
165
        This overrides the method from urwid.FlowWidget."""
 
262
 
    def rows(self, maxcolrow, focus=False):
 
 
168
    def rows(self, (maxcol,), focus=False):
 
263
169
        """How many rows this widget will occupy might depend on
 
264
170
        whether we have focus or not.
 
265
171
        This overrides the method from urwid.FlowWidget"""
 
266
 
        return self.current_widget(focus).rows(maxcolrow, focus=focus)
 
 
172
        return self.current_widget(focus).rows((maxcol,), focus=focus)
 
268
174
    def current_widget(self, focus=False):
 
269
175
        if focus or self.opened:
 
 
273
179
    def update(self):
 
274
180
        "Called when what is visible on the screen should be updated."
 
275
181
        # How to add standout mode to a style
 
276
 
        with_standout = { "normal": "standout",
 
277
 
                          "bold": "bold-standout",
 
279
 
                              "underline-blink-standout",
 
280
 
                          "bold-underline-blink":
 
281
 
                              "bold-underline-blink-standout",
 
 
182
        with_standout = { u"normal": u"standout",
 
 
183
                          u"bold": u"bold-standout",
 
 
185
                              u"underline-blink-standout",
 
 
186
                          u"bold-underline-blink":
 
 
187
                              u"bold-underline-blink-standout",
 
284
190
        # Rebuild focus and non-focus widgets using current properties
 
286
192
        # Base part of a client. Name!
 
288
 
                      % {"name": self.properties["Name"]})
 
289
 
        if not self.properties["Enabled"]:
 
291
 
        elif self.properties["ApprovalPending"]:
 
292
 
            timeout = datetime.timedelta(milliseconds
 
295
 
            last_approval_request = isoformat_to_datetime(
 
296
 
                self.properties["LastApprovalRequest"])
 
297
 
            if last_approval_request is not None:
 
298
 
                timer = timeout - (datetime.datetime.utcnow()
 
299
 
                                   - last_approval_request)
 
301
 
                timer = datetime.timedelta()
 
302
 
            if self.properties["ApprovedByDefault"]:
 
303
 
                message = "Approval in %s. (d)eny?"
 
305
 
                message = "Denial in %s. (a)pprove?"
 
306
 
            message = message % unicode(timer).rsplit(".", 1)[0]
 
307
 
        elif self.last_checker_failed:
 
308
 
            timeout = datetime.timedelta(milliseconds
 
311
 
            last_ok = isoformat_to_datetime(
 
312
 
                max((self.properties["LastCheckedOK"]
 
313
 
                     or self.properties["Created"]),
 
314
 
                    self.properties["LastEnabled"]))
 
315
 
            timer = timeout - (datetime.datetime.utcnow() - last_ok)
 
316
 
            message = ('A checker has failed! Time until client'
 
318
 
                           % unicode(timer).rsplit(".", 1)[0])
 
 
193
        self._text = (u'%(name)s: '
 
 
194
                      % {u"name": self.properties[u"name"]})
 
 
196
        if self.properties[u"approved_pending"]:
 
 
197
            if self.properties[u"approved_by_default"]:
 
 
198
                self._text += u"Connection established to client. (d)eny?"
 
 
200
                self._text += u"Seeks approval to send secret. (a)pprove?"
 
321
 
        self._text = "%s%s" % (base, message)
 
 
202
            self._text += (u'%(enabled)s'
 
 
205
                                if self.properties[u"enabled"]
 
323
207
        if not urwid.supports_unicode():
 
324
208
            self._text = self._text.encode("ascii", "replace")
 
325
 
        textlist = [("normal", self._text)]
 
 
209
        textlist = [(u"normal", self._text)]
 
326
210
        self._text_widget.set_text(textlist)
 
327
211
        self._focus_text_widget.set_text([(with_standout[text[0]],
 
 
336
220
        if self.update_hook is not None:
 
337
221
            self.update_hook()
 
339
 
    def update_timer(self):
 
342
 
        return True             # Keep calling this
 
344
223
    def delete(self):
 
345
 
        if self._update_timer_callback_tag is not None:
 
346
 
            gobject.source_remove(self._update_timer_callback_tag)
 
347
 
            self._update_timer_callback_tag = None
 
348
224
        if self.delete_hook is not None:
 
349
225
            self.delete_hook(self)
 
351
 
    def render(self, maxcolrow, focus=False):
 
 
227
    def render(self, (maxcol,), focus=False):
 
352
228
        """Render differently if we have focus.
 
353
229
        This overrides the method from urwid.FlowWidget"""
 
354
 
        return self.current_widget(focus).render(maxcolrow,
 
 
230
        return self.current_widget(focus).render((maxcol,),
 
357
 
    def keypress(self, maxcolrow, key):
 
 
233
    def keypress(self, (maxcol,), key):
 
359
235
        This overrides the method from urwid.FlowWidget"""
 
361
237
            self.proxy.Enable(dbus_interface = client_interface)
 
363
239
            self.proxy.Disable(dbus_interface = client_interface)
 
365
241
            self.proxy.Approve(dbus.Boolean(True, variant_level=1),
 
366
242
                               dbus_interface = client_interface)
 
368
244
            self.proxy.Approve(dbus.Boolean(False, variant_level=1),
 
369
245
                                  dbus_interface = client_interface)
 
370
 
        elif key == "R" or key == "_" or key == "ctrl k":
 
 
246
        elif key == u"r" or key == u"_" or key == u"ctrl k":
 
371
247
            self.server_proxy_object.RemoveClient(self.proxy
 
374
250
            self.proxy.StartChecker(dbus_interface = client_interface)
 
376
252
            self.proxy.StopChecker(dbus_interface = client_interface)
 
378
254
            self.proxy.CheckedOK(dbus_interface = client_interface)
 
380
 
#         elif key == "p" or key == "=":
 
 
256
#         elif key == u"p" or key == "=":
 
381
257
#             self.proxy.pause()
 
382
 
#         elif key == "u" or key == ":":
 
 
258
#         elif key == u"u" or key == ":":
 
383
259
#             self.proxy.unpause()
 
 
260
#         elif key == u"RET":
 
 
263
#            self.proxy.Approve(True)
 
 
265
#            self.proxy.Approve(False)
 
 
420
300
        self.screen = urwid.curses_display.Screen()
 
422
302
        self.screen.register_palette((
 
424
 
                 "default", "default", None),
 
426
 
                 "default", "default", "bold"),
 
428
 
                 "default", "default", "underline"),
 
430
 
                 "default", "default", "standout"),
 
431
 
                ("bold-underline-blink",
 
432
 
                 "default", "default", ("bold", "underline")),
 
434
 
                 "default", "default", ("bold", "standout")),
 
435
 
                ("underline-blink-standout",
 
436
 
                 "default", "default", ("underline", "standout")),
 
437
 
                ("bold-underline-blink-standout",
 
438
 
                 "default", "default", ("bold", "underline",
 
 
304
                 u"default", u"default", None),
 
 
306
                 u"default", u"default", u"bold"),
 
 
308
                 u"default", u"default", u"underline"),
 
 
310
                 u"default", u"default", u"standout"),
 
 
311
                (u"bold-underline-blink",
 
 
312
                 u"default", u"default", (u"bold", u"underline")),
 
 
314
                 u"default", u"default", (u"bold", u"standout")),
 
 
315
                (u"underline-blink-standout",
 
 
316
                 u"default", u"default", (u"underline", u"standout")),
 
 
317
                (u"bold-underline-blink-standout",
 
 
318
                 u"default", u"default", (u"bold", u"underline",
 
442
322
        if urwid.supports_unicode():
 
443
 
            self.divider = "─" # \u2500
 
444
 
            #self.divider = "━" # \u2501
 
 
323
            self.divider = u"─" # \u2500
 
 
324
            #self.divider = u"━" # \u2501
 
446
 
            #self.divider = "-" # \u002d
 
447
 
            self.divider = "_" # \u005f
 
 
326
            #self.divider = u"-" # \u002d
 
 
327
            self.divider = u"_" # \u005f
 
449
329
        self.screen.start()
 
 
550
427
            and len(self.log) > self.max_log_length):
 
551
428
            del self.log[0:len(self.log)-self.max_log_length-1]
 
552
429
        self.logbox.set_focus(len(self.logbox.body.contents),
 
 
430
                              coming_from=u"above")
 
556
433
    def toggle_log_display(self):
 
557
434
        """Toggle visibility of the log buffer."""
 
558
435
        self.log_visible = not self.log_visible
 
560
 
        #self.log_message("Log visibility changed to: "
 
561
 
        #                 + unicode(self.log_visible))
 
 
437
        self.log_message(u"Log visibility changed to: "
 
 
438
                         + unicode(self.log_visible))
 
563
440
    def change_log_display(self):
 
564
441
        """Change type of log display.
 
565
442
        Currently, this toggles wrapping of text lines."""
 
566
 
        if self.log_wrap == "clip":
 
567
 
            self.log_wrap = "any"
 
 
443
        if self.log_wrap == u"clip":
 
 
444
            self.log_wrap = u"any"
 
569
 
            self.log_wrap = "clip"
 
 
446
            self.log_wrap = u"clip"
 
570
447
        for textwidget in self.log:
 
571
448
            textwidget.set_wrap_mode(self.log_wrap)
 
572
 
        #self.log_message("Wrap mode: " + self.log_wrap)
 
 
449
        self.log_message(u"Wrap mode: " + self.log_wrap)
 
574
451
    def find_and_remove_client(self, path, name):
 
575
452
        """Find an client from its object path and remove it.
 
 
655
532
            except KeyError:    # :-)
 
658
 
            if key == "q" or key == "Q":
 
 
535
            if key == u"q" or key == u"Q":
 
661
 
            elif key == "window resize":
 
 
538
            elif key == u"window resize":
 
662
539
                self.size = self.screen.get_cols_rows()
 
664
 
            elif key == "\f":  # Ctrl-L
 
 
541
            elif key == u"\f":  # Ctrl-L
 
666
 
            elif key == "l" or key == "D":
 
 
543
            elif key == u"l" or key == u"D":
 
667
544
                self.toggle_log_display()
 
669
 
            elif key == "w" or key == "i":
 
 
546
            elif key == u"w" or key == u"i":
 
670
547
                self.change_log_display()
 
672
 
            elif key == "?" or key == "f1" or key == "esc":
 
 
549
            elif key == u"?" or key == u"f1" or key == u"esc":
 
673
550
                if not self.log_visible:
 
674
551
                    self.log_visible = True
 
676
 
                self.log_message_raw(("bold",
 
680
 
                                            "l: Log window toggle",
 
681
 
                                            "TAB: Switch window",
 
683
 
                self.log_message_raw(("bold",
 
689
 
                                             "s: Start new checker",
 
 
553
                self.log_message_raw((u"bold",
 
 
557
                                            u"l: Log window toggle",
 
 
558
                                            u"TAB: Switch window",
 
 
560
                self.log_message_raw((u"bold",
 
 
566
                                             u"s: Start new checker",
 
696
573
                if self.topwidget.get_focus() is self.logbox:
 
697
574
                    self.topwidget.set_focus(0)
 
699
576
                    self.topwidget.set_focus(self.logbox)
 
701
 
            #elif (key == "end" or key == "meta >" or key == "G"
 
 
578
            #elif (key == u"end" or key == u"meta >" or key == u"G"
 
703
580
            #    pass            # xxx end-of-buffer
 
704
 
            #elif (key == "home" or key == "meta <" or key == "g"
 
 
581
            #elif (key == u"home" or key == u"meta <" or key == u"g"
 
706
583
            #    pass            # xxx beginning-of-buffer
 
707
 
            #elif key == "ctrl e" or key == "$":
 
 
584
            #elif key == u"ctrl e" or key == u"$":
 
708
585
            #    pass            # xxx move-end-of-line
 
709
 
            #elif key == "ctrl a" or key == "^":
 
 
586
            #elif key == u"ctrl a" or key == u"^":
 
710
587
            #    pass            # xxx move-beginning-of-line
 
711
 
            #elif key == "ctrl b" or key == "meta (" or key == "h":
 
 
588
            #elif key == u"ctrl b" or key == u"meta (" or key == u"h":
 
712
589
            #    pass            # xxx left
 
713
 
            #elif key == "ctrl f" or key == "meta )" or key == "l":
 
 
590
            #elif key == u"ctrl f" or key == u"meta )" or key == u"l":
 
714
591
            #    pass            # xxx right
 
716
593
            #    pass            # scroll up log
 
718
595
            #    pass            # scroll down log
 
719
596
            elif self.topwidget.selectable():
 
720
597
                self.topwidget.keypress(self.size, key)