]>
Commit | Line | Data |
---|---|---|
7dcf23b1 SH |
1 | #!/usr/bin/python |
2 | # -*- coding: utf-8 -*- | |
3 | # | |
beaa8775 | 4 | # Copyright (C) 2011-2012 Sebastien Helleu <flashcode@flashtux.org> |
7dcf23b1 SH |
5 | # |
6 | # This file is part of QWeeChat, a Qt remote GUI for WeeChat. | |
7 | # | |
8 | # QWeeChat is free software; you can redistribute it and/or modify | |
9 | # it under the terms of the GNU General Public License as published by | |
10 | # the Free Software Foundation; either version 3 of the License, or | |
11 | # (at your option) any later version. | |
12 | # | |
13 | # QWeeChat is distributed in the hope that it will be useful, | |
14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
16 | # GNU General Public License for more details. | |
17 | # | |
18 | # You should have received a copy of the GNU General Public License | |
19 | # along with QWeeChat. If not, see <http://www.gnu.org/licenses/>. | |
20 | # | |
21 | ||
22 | # | |
23 | # I/O with WeeChat/relay. | |
24 | # | |
25 | ||
26 | import struct | |
27 | import qt_compat | |
28 | QtCore = qt_compat.import_module('QtCore') | |
29 | QtNetwork = qt_compat.import_module('QtNetwork') | |
30 | ||
beaa8775 SH |
31 | _PROTO_INIT_CMD = ['init password=%(password)s,compression=gzip'] |
32 | _PROTO_SYNC_CMDS = ['(listbuffers) hdata buffer:gui_buffers(*) number,full_name,short_name,type,nicklist,title,local_variables', | |
7dcf23b1 SH |
33 | '(listlines) hdata buffer:gui_buffers(*)/own_lines/first_line(*)/data date,displayed,prefix,message', |
34 | '(nicklist) nicklist', | |
ca67c3a7 | 35 | 'sync', |
7dcf23b1 SH |
36 | ''] |
37 | ||
38 | ||
39 | class Network(QtCore.QObject): | |
40 | """I/O with WeeChat/relay.""" | |
41 | ||
42 | statusChanged = qt_compat.Signal(str, str) | |
43 | messageFromWeechat = qt_compat.Signal(QtCore.QByteArray) | |
44 | ||
45 | def __init__(self, *args): | |
67ae3204 | 46 | QtCore.QObject.__init__(*(self,) + args) |
7dcf23b1 SH |
47 | self.status_disconnected = 'disconnected' |
48 | self.status_connecting = 'connecting...' | |
49 | self.status_connected = 'connected' | |
50 | self._server = None | |
51 | self._port = None | |
77b25057 | 52 | self._ssl = None |
7dcf23b1 SH |
53 | self._password = None |
54 | self._buffer = QtCore.QByteArray() | |
77b25057 | 55 | self._socket = QtNetwork.QSslSocket() |
7dcf23b1 SH |
56 | self._socket.connected.connect(self._socket_connected) |
57 | self._socket.error.connect(self._socket_error) | |
58 | self._socket.readyRead.connect(self._socket_read) | |
59 | self._socket.disconnected.connect(self._socket_disconnected) | |
60 | ||
61 | def _socket_connected(self): | |
62 | """Slot: socket connected.""" | |
63 | self.statusChanged.emit(self.status_connected, None) | |
64 | if self._password: | |
77b25057 | 65 | self.send_to_weechat('\n'.join(_PROTO_INIT_CMD + _PROTO_SYNC_CMDS) % {'password': str(self._password)}) |
7dcf23b1 SH |
66 | |
67 | def _socket_error(self, error): | |
68 | """Slot: socket error.""" | |
69 | self.statusChanged.emit(self.status_disconnected, 'Failed, error: %s' % self._socket.errorString()) | |
70 | ||
71 | def _socket_read(self): | |
72 | """Slot: data available on socket.""" | |
77b25057 | 73 | bytes = self._socket.readAll() |
7dcf23b1 SH |
74 | self._buffer.append(bytes) |
75 | while len(self._buffer) >= 4: | |
76 | remainder = None | |
77 | length = struct.unpack('>i', self._buffer[0:4])[0] | |
78 | if len(self._buffer) < length: | |
79 | # partial message, just wait for end of message | |
80 | break | |
81 | # more than one message? | |
82 | if length < len(self._buffer): | |
83 | # save beginning of another message | |
84 | remainder = self._buffer[length:] | |
85 | self._buffer = self._buffer[0:length] | |
86 | self.messageFromWeechat.emit(self._buffer) | |
77b25057 SH |
87 | if not self.is_connected(): |
88 | return | |
7dcf23b1 SH |
89 | self._buffer.clear() |
90 | if remainder: | |
91 | self._buffer.append(remainder) | |
92 | ||
93 | def _socket_disconnected(self): | |
94 | """Slot: socket disconnected.""" | |
95 | self._server = None | |
96 | self._port = None | |
77b25057 | 97 | self._ssl = None |
7dcf23b1 SH |
98 | self._password = None |
99 | self.statusChanged.emit(self.status_disconnected, None) | |
100 | ||
101 | def is_connected(self): | |
102 | return self._socket.state() == QtNetwork.QAbstractSocket.ConnectedState | |
103 | ||
77b25057 SH |
104 | def is_ssl(self): |
105 | return self._ssl | |
106 | ||
107 | def connect_weechat(self, server, port, ssl, password): | |
7dcf23b1 SH |
108 | self._server = server |
109 | try: | |
110 | self._port = int(port) | |
111 | except: | |
112 | self._port = 0 | |
77b25057 | 113 | self._ssl = ssl |
7dcf23b1 SH |
114 | self._password = password |
115 | if self._socket.state() == QtNetwork.QAbstractSocket.ConnectedState: | |
116 | return | |
117 | if self._socket.state() != QtNetwork.QAbstractSocket.UnconnectedState: | |
118 | self._socket.abort() | |
119 | self._socket.connectToHost(self._server, self._port) | |
77b25057 SH |
120 | if self._ssl: |
121 | self._socket.ignoreSslErrors() | |
122 | self._socket.startClientEncryption() | |
7dcf23b1 SH |
123 | self.statusChanged.emit(self.status_connecting, None) |
124 | ||
125 | def disconnect_weechat(self): | |
126 | if self._socket.state() != QtNetwork.QAbstractSocket.UnconnectedState: | |
eafb7aeb | 127 | if self._socket.state() == QtNetwork.QAbstractSocket.ConnectedState: |
77b25057 | 128 | self.send_to_weechat('quit\n') |
eafb7aeb SH |
129 | self._socket.waitForBytesWritten(1000) |
130 | else: | |
7dcf23b1 SH |
131 | self.statusChanged.emit(self.status_disconnected, None) |
132 | self._socket.abort() | |
133 | ||
134 | def send_to_weechat(self, message): | |
6b520199 | 135 | self._socket.write(str(message)) |
7dcf23b1 | 136 | |
beaa8775 | 137 | def desync_weechat(self): |
77b25057 | 138 | self.send_to_weechat('desync\n') |
beaa8775 SH |
139 | |
140 | def sync_weechat(self): | |
77b25057 | 141 | self.send_to_weechat('\n'.join(_PROTO_SYNC_CMDS)) |
beaa8775 | 142 | |
7dcf23b1 SH |
143 | def status_icon(self, status): |
144 | icon = {self.status_disconnected: 'dialog-close.png', | |
145 | self.status_connecting: 'dialog-close.png', | |
146 | self.status_connected: 'dialog-ok-apply.png'} | |
147 | return icon.get(status, '') |