1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
|
# Copyright 2010-2011 Le Coz Florent <louiz@louiz.org>
#
# This file is part of Poezio.
#
# Poezio 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 of the License.
#
# Poezio 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 Poezio. If not, see <http://www.gnu.org/licenses/>.
import logging
log = logging.getLogger(__name__)
import curses
import windows
import tabs
from sleekxmpp.xmlstream import ElementBase, ET
class PubsubNode(object):
node_type = None # unknown yet
def __init__(self, name):
self.items = []
self.name = name
class LeafNode(PubsubNode):
node_type = "leaf"
def __init__(self, name):
PubsubNode.__init__(self, name)
class CollectionNode(PubsubNode):
node_type = "collection"
def __init__(self, name):
PubsubNode.__init__(self, name)
self.subnodes = []
class PubsubItem(object):
def __init__(self, idd, content):
self.idd = idd
self.content = content
class PubsubBrowserTab(tabs.Tab):
"""
A tab containing a pubsub browser letting the user
list nodes and items, view, add and delete items, etc
"""
def __init__(self, server):
"""
Server is the name of the pubsub server, for example:
pubsub.example.com
All action done in this tab will be made on that server.
"""
tabs.Tab.__init__(self)
self.current_node = None # the subnode we are listing. None means the root
self.server = server
self.nodes = [] # the lower level of nodes
self.tab_win = windows.GlobalInfoBar()
self.upper_message = windows.Topic()
self.upper_message.set_message('Pubsub server: %s' % (self.server,))
# Node List View
node_columns = ('node', 'name',)
self.node_list_header = windows.ColumnHeaderWin(node_columns)
self.node_listview = windows.ListWin(node_columns)
# Item List View
item_columns = ('name',)
self.item_list_header = windows.ColumnHeaderWin(item_columns)
self.item_listview = windows.ListWin(item_columns)
self.default_help_message = windows.HelpText("“c”: create a node.")
self.input = self.default_help_message
self.key_func['c'] = self.command_create_node
self.key_func["KEY_DOWN"] = self.node_listview.move_cursor_down
self.key_func["KEY_UP"] = self.node_listview.move_cursor_up
self.resize()
self.get_nodes()
def resize(self):
self.upper_message.resize(1, self.width, 0, 0)
self.tab_win.resize(1, self.width, self.height-2, 0)
column_size = {'node': self.width//4,
'name': self.width//4,}
self.node_list_header.resize_columns(column_size)
self.node_list_header.resize(1, self.width//2, 1, 0)
self.node_listview.resize_columns(column_size)
self.node_listview.resize(self.height//2-2, self.width//2, 2, 0)
column_size = {'name': self.width//2,}
self.item_list_header.resize_columns(column_size)
self.item_list_header.resize(self.height//2+1, self.width//2, self.height//2, 0)
self.item_listview.resize_columns(column_size)
self.item_listview.resize(self.height//2-4, self.width//2, self.height//2+3, 0)
self.input.resize(1, self.width, self.height-1, 0)
def refresh(self):
if self.need_resize:
self.resize()
log.debug(' TAB Refresh: %s'%self.__class__.__name__)
self.upper_message.refresh()
self.node_list_header.refresh()
self.node_listview.refresh()
self.item_list_header.refresh()
self.item_listview.refresh()
self.tab_win.refresh()
self.input.refresh()
def get_name(self):
return '%s@@pubsubbrowser' % (self.server,)
def on_input(self, key):
res = self.input.do_command(key)
if res:
return True
if key in self.key_func:
return self.key_func[key]()
def get_items(self, node):
"""
Get all items in the given node
"""
items = self.core.xmpp.plugin['xep_0060'].get_items(self.server, node.name)
item_list = []
if items:
for it in items:
item_list.append(PubsubItem(it.attrib['id'], ET.tostring(it)))
node.items = item_list
log.debug('Item on node %s: %s' % (node.name, item_list))
def add_nodes(self, node_list, parent=None):
"""
Add Node objects to the list of the parent.
If parent is None, they are added to the root list.
If the current selected node is parent, we add
them directly to the node_listview
"""
log.debug('Adding nodes to %s: %s' % (node_list, parent,))
if not parent:
list_to_append = self.nodes
else:
list_to_append = parent.nodes
for node in node_list:
new_node = LeafNode(node['node'])
list_to_append.append(new_node)
self.get_items(new_node)
self.node_listview.add_lines(node_list)
def get_nodes(self, node=None):
"""
Get all subnodes of the given node. If no node is given, get
the root nodes
"""
nodes = self.core.xmpp.plugin['xep_0060'].get_nodes(self.server)
lines = [{'name': nodes[node] or '',
'node': node} for node in nodes.keys()]
self.add_nodes(lines)
def create_node(self, node_name):
if node_name:
res = self.core.xmpp.plugin['xep_0060'].create_node(self.server, node_name)
if res:
self.node_listview.add_lines([{'name': '', 'node': node_name}])
self.reset_help_message()
return True
def reset_help_message(self, txt=None):
"""
Just reset the help message when a command ends
"""
curses.curs_set(0)
self.input = self.default_help_message
return True
def command_create_node(self):
"""
Prompt for a node name and create it on Enter key
"""
curses.curs_set(1)
self.input = windows.CommandInput("[Create node]", self.reset_help_message, self.create_node, None)
self.input.resize(1, self.width, self.height-1, 0)
return True
|
