diff options
author | Ryan Huber <rhuber@gmail.com> | 2014-10-26 21:58:41 -0700 |
---|---|---|
committer | Ryan Huber <rhuber@gmail.com> | 2014-10-26 21:58:41 -0700 |
commit | a61ffda222a69170b86e66e1478de46c599bdd31 (patch) | |
tree | 60f7636179640c0ba0d19791280f7689dcbfbf95 | |
parent | d225901e7123d2b08759cb0d5ac4685a32078cac (diff) | |
download | wee-slack-a61ffda222a69170b86e66e1478de46c599bdd31.tar.gz |
caved and switched to 4 space tabs
-rw-r--r-- | wee_slack.py | 1668 |
1 files changed, 835 insertions, 833 deletions
diff --git a/wee_slack.py b/wee_slack.py index 5c7e8be..ab1fb41 100644 --- a/wee_slack.py +++ b/wee_slack.py @@ -10,11 +10,11 @@ from websocket import create_connection import weechat as w -SCRIPT_NAME = "slack_extension" -SCRIPT_AUTHOR = "Ryan Huber <rhuber@gmail.com>" -SCRIPT_VERSION = "0.7" +SCRIPT_NAME = "slack_extension" +SCRIPT_AUTHOR = "Ryan Huber <rhuber@gmail.com>" +SCRIPT_VERSION = "0.97" SCRIPT_LICENSE = "MIT" -SCRIPT_DESC = "Extends weechat for typing notification/search/etc on slack.com" +SCRIPT_DESC = "Extends weechat for typing notification/search/etc on slack.com" BACKLOG_SIZE = 200 @@ -40,989 +40,991 @@ SLACK_API_TRANSLATOR = { } def dbg(message, fout=False): - if slack_debug != None: - message = "DEBUG: " + str(message) - if fout: - file('/tmp/debug.log','a+').writelines(message+'\n') - else: - w.prnt(slack_debug, message) + if slack_debug != None: + message = "DEBUG: " + str(message) + if fout: + file('/tmp/debug.log','a+').writelines(message+'\n') + else: + w.prnt(slack_debug, message) #hilarious, i know class Meta(list): - def __init__(self, attribute, search_list): - self.attribute = attribute - self.search_list = search_list - def __str__(self): - string = '' - for each in self.search_list.get_all(self.attribute): - string += str(each) - return string - def __repr__(self): - self.search_list.get_all(self.attribute) - def __getitem__(self, index): - channels = self.get_all() - return channels[index] - def __iter__(self): - channels = self.get_all() - for channel in channels: - yield channel - def get_all(self): - items = [] - items += self.search_list.get_all(self.attribute) - return items - def find(self, name): - items = self.search_list.find_deep(name, self.attribute) - items = [x for x in items if x != None] - if len(items) == 1: - return items[0] - elif len(items) == 0: - pass - else: - dbg("probably something bad happened with meta items: %s" % items) - return items - raise AmbiguousProblemError - def find_by_class(self, class_name): - items = self.search_list.find_by_class_deep(class_name, self.attribute) - return items + def __init__(self, attribute, search_list): + self.attribute = attribute + self.search_list = search_list + def __str__(self): + string = '' + for each in self.search_list.get_all(self.attribute): + string += str(each) + return string + def __repr__(self): + self.search_list.get_all(self.attribute) + def __getitem__(self, index): + channels = self.get_all() + return channels[index] + def __iter__(self): + channels = self.get_all() + for channel in channels: + yield channel + def get_all(self): + items = [] + items += self.search_list.get_all(self.attribute) + return items + def find(self, name): + items = self.search_list.find_deep(name, self.attribute) + items = [x for x in items if x != None] + if len(items) == 1: + return items[0] + elif len(items) == 0: + pass + else: + dbg("probably something bad happened with meta items: %s" % items) + return items + raise AmbiguousProblemError + def find_by_class(self, class_name): + items = self.search_list.find_by_class_deep(class_name, self.attribute) + return items class SearchList(list): - def find(self, name): - items = [] - for child in self: - if child.__class__ == self.__class__: - items += child.find(name) - else: - if child == name: - items.append(child) - if len(items) == 1: - return items[0] - elif items != []: - return items - def find_deep(self, name, attribute): - items = [] - for child in self: - if child.__class__ == self.__class__: - if items != None: - items += child.find_deep(name, attribute) - elif dir(child).count('find') == 1: - if items != None: - items.append(child.find(name, attribute)) - if items != []: - return items - def get_all(self, attribute): - items = [] - for child in self: - if child.__class__ == self.__class__: - items += child.get_all(attribute) - else: - items += (eval("child."+attribute)) - return items - def find_by_class(self, class_name): - items = [] - for child in self: - if child.__class__ == class_name: - items.append(child) - return items - def find_by_class_deep(self, class_name, attribute): - items = [] - for child in self: - if child.__class__ == self.__class__: - items += child.find_by_class_deep(class_name, attribute) - else: - items += (eval('child.'+attribute).find_by_class(class_name)) - return items + def find(self, name): + items = [] + for child in self: + if child.__class__ == self.__class__: + items += child.find(name) + else: + if child == name: + items.append(child) + if len(items) == 1: + return items[0] + elif items != []: + return items + def find_deep(self, name, attribute): + items = [] + for child in self: + if child.__class__ == self.__class__: + if items != None: + items += child.find_deep(name, attribute) + elif dir(child).count('find') == 1: + if items != None: + items.append(child.find(name, attribute)) + if items != []: + return items + def get_all(self, attribute): + items = [] + for child in self: + if child.__class__ == self.__class__: + items += child.get_all(attribute) + else: + items += (eval("child."+attribute)) + return items + def find_by_class(self, class_name): + items = [] + for child in self: + if child.__class__ == class_name: + items.append(child) + return items + def find_by_class_deep(self, class_name, attribute): + items = [] + for child in self: + if child.__class__ == self.__class__: + items += child.find_by_class_deep(class_name, attribute) + else: + items += (eval('child.'+attribute).find_by_class(class_name)) + return items class SlackServer(object): - def __init__(self, token): - self.nick = None - self.name = None - self.domain = None - self.login_data = None - self.buffer = None - self.token = token - self.ws = None - self.ws_hook = None - self.users = SearchList() - self.channels = SearchList() - self.connected = False - self.pingcounter = 0 - - self.identifier = None - self.connect_to_slack() - def __eq__(self, compare_str): - if compare_str == self.identifier or compare_str == self.token or compare_str == self.buffer: - return True - else: - return False - def __str__(self): - return "%s" % (self.identifier) - def __repr__(self): - return "%s" % (self.identifier) - def find(self, name, attribute): - attribute = eval("self."+attribute) - return attribute.find(name) - def ping(self): - if self.pingcounter > 999: - self.pingcounter = 0 - request = {"type":"ping","id":self.pingcounter} - try: - self.ws.send(json.dumps(request)) - except: - self.connected = False - self.pingcounter += 1 - def connect_to_slack(self): - t = time.time() - async_slack_api_request("slack.com", self.token, "rtm.start", {"ts":t}) - def connected_to_slack(self, login_data): - if login_data["ok"] == True: - self.domain = login_data["team"]["domain"] + ".slack.com" - dbg("connected to %s" % self.domain) - self.identifier = self.domain - self.nick = login_data["self"]["name"] - self.create_local_buffer() - - if self.create_slack_websocket(login_data): - w.hook_timer(1000 * 3, 0, 0, "slack_ping_cb", self.domain) - if len(self.users) and 0 or len(self.channels) == 0: - self.create_slack_mappings(login_data) - - self.connected = True - - return True - else: - w.prnt("", "\n!! slack.com login error: " + login_data["error"] + "\n Please check your API token with \"/set plugins.var.python.slack_extension.slack_api_token\"\n\n ") - self.connected = False - def create_local_buffer(self): - if not w.buffer_search("", self.domain): - self.buffer = w.buffer_new(self.domain, "input", "", "", "") - w.buffer_set(self.buffer, "nicklist", "1") - def create_slack_websocket(self, data): - web_socket_url = data['url'] - try: - self.ws = create_connection(web_socket_url) - self.ws.sock.setblocking(0) - self.ws_hook = w.hook_fd(self.ws.sock._sock.fileno(), 1, 0, 0, "slack_websocket_cb", self.identifier) - return True - except: - return False - def create_slack_mappings(self, data): - - for item in data["users"]: - self.users.append(User(self, item["name"], item["id"], item["presence"])) - - for item in data["channels"]: - if not item.has_key("last_read"): - item["last_read"] = 0 - self.channels.append(Channel(self, item["name"], item["id"], item["is_member"], item["last_read"], "#")) - for item in data["groups"]: - if not item.has_key("last_read"): - item["last_read"] = 0 - self.channels.append(GroupChannel(self, item["name"], item["id"], item["is_open"], item["last_read"], "#")) - for item in data["ims"]: - if not item.has_key("last_read"): - item["last_read"] = 0 - name = self.users.find(item["user"]).name - self.channels.append(DmChannel(self, name, item["id"], item["is_open"], item["last_read"])) - - for item in self.channels: - item.get_history() - def buffer_prnt(self, message='no message', user="SYSTEM", backlog=False): - message = message.encode('ascii', 'ignore') - if backlog == True: - tags = "no_highlight,notify_none,logger_backlog_end" - else: - tags = "" - if self.buffer: - w.prnt_date_tags(self.buffer, 0, tags, "%s\t%s" % (user, message)) - else: - pass - #w.prnt("", "%s\t%s" % (user, message)) + def __init__(self, token): + self.nick = None + self.name = None + self.domain = None + self.login_data = None + self.buffer = None + self.token = token + self.ws = None + self.ws_hook = None + self.users = SearchList() + self.channels = SearchList() + self.connected = False + self.pingcounter = 0 + + self.identifier = None + self.connect_to_slack() + def __eq__(self, compare_str): + if compare_str == self.identifier or compare_str == self.token or compare_str == self.buffer: + return True + else: + return False + def __str__(self): + return "%s" % (self.identifier) + def __repr__(self): + return "%s" % (self.identifier) + def find(self, name, attribute): + attribute = eval("self."+attribute) + return attribute.find(name) + def ping(self): + if self.pingcounter > 999: + self.pingcounter = 0 + request = {"type":"ping","id":self.pingcounter} + try: + self.ws.send(json.dumps(request)) + except: + self.connected = False + self.pingcounter += 1 + def connect_to_slack(self): + t = time.time() + async_slack_api_request("slack.com", self.token, "rtm.start", {"ts":t}) + def connected_to_slack(self, login_data): + if login_data["ok"] == True: + self.domain = login_data["team"]["domain"] + ".slack.com" + dbg("connected to %s" % self.domain) + self.identifier = self.domain + self.nick = login_data["self"]["name"] + self.create_local_buffer() + + if self.create_slack_websocket(login_data): + w.hook_timer(1000 * 3, 0, 0, "slack_ping_cb", self.domain) + if len(self.users) and 0 or len(self.channels) == 0: + self.create_slack_mappings(login_data) + + self.connected = True + + return True + else: + w.prnt("", "\n!! slack.com login error: " + login_data["error"] + "\n Please check your API token with \"/set plugins.var.python.slack_extension.slack_api_token\"\n\n ") + self.connected = False + def create_local_buffer(self): + if not w.buffer_search("", self.domain): + self.buffer = w.buffer_new(self.domain, "input", "", "", "") + w.buffer_set(self.buffer, "nicklist", "1") + def create_slack_websocket(self, data): + web_socket_url = data['url'] + try: + self.ws = create_connection(web_socket_url) + self.ws.sock.setblocking(0) + self.ws_hook = w.hook_fd(self.ws.sock._sock.fileno(), 1, 0, 0, "slack_websocket_cb", self.identifier) + return True + except: + return False + def create_slack_mappings(self, data): + + for item in data["users"]: + self.users.append(User(self, item["name"], item["id"], item["presence"])) + + for item in data["channels"]: + if not item.has_key("last_read"): + item["last_read"] = 0 + self.channels.append(Channel(self, item["name"], item["id"], item["is_member"], item["last_read"], "#")) + for item in data["groups"]: + if not item.has_key("last_read"): + item["last_read"] = 0 + self.channels.append(GroupChannel(self, item["name"], item["id"], item["is_open"], item["last_read"], "#")) + for item in data["ims"]: + if not item.has_key("last_read"): + item["last_read"] = 0 + name = self.users.find(item["user"]).name + self.channels.append(DmChannel(self, name, item["id"], item["is_open"], item["last_read"])) + + for item in self.channels: + item.get_history() + def buffer_prnt(self, message='no message', user="SYSTEM", backlog=False): + message = message.encode('ascii', 'ignore') + if backlog == True: + tags = "no_highlight,notify_none,logger_backlog_end" + else: + tags = "" + if self.buffer: + w.prnt_date_tags(self.buffer, 0, tags, "%s\t%s" % (user, message)) + else: + pass + #w.prnt("", "%s\t%s" % (user, message)) class SlackThing(object): - def __init__(self, name, identifier): - self.name = name - self.identifier = identifier - self.channel_buffer = None - def __str__(self): - return "Name: %s Id: %s CB: %s" % (self.name, self.identifier, self.channel_buffer) - def __repr__(self): - return "Name: %s Id: %s CB: %s" % (self.name, self.identifier, self.channel_buffer) - -def input(b,c,data): - channel = channels.find(c) - channel.send_message(data) - channel.buffer_prnt(channel.server.nick, data) - return w.WEECHAT_RC_ERROR + def __init__(self, name, identifier): + self.name = name + self.identifier = identifier + self.channel_buffer = None + def __str__(self): + return "Name: %s Id: %s CB: %s" % (self.name, self.identifier, self.channel_buffer) + def __repr__(self): + return "Name: %s Id: %s CB: %s" % (self.name, self.identifier, self.channel_buffer) + +def input(b, c, data): + channel = channels.find(c) + channel.send_message(data) + channel.buffer_prnt(channel.server.nick, data) + return w.WEECHAT_RC_ERROR class Channel(SlackThing): - def __init__(self, server, name, identifier, active, last_read=0, prepend_name=""): - super(Channel, self).__init__(name, identifier) - self.type = "channel" - self.server = server - self.name = prepend_name + self.name - self.typing = {} - self.active = active - self.last_read = float(last_read) - if active: - self.create_buffer() - self.attach_buffer() - def __eq__(self, compare_str): - if compare_str == self.fullname() or compare_str == self.name or compare_str == self.identifier or compare_str == self.name[1:] or (compare_str == self.channel_buffer and self.channel_buffer != None): - return True - else: - return False - def __str__(self): - return "Name: %s Id: %s Buffer: %s Active: %s" % (self.name, self.identifier, self.channel_buffer, self.active) - def __repr__(self): - return "Name: %s Id: %s Buffer: %s Active: %s" % (self.name, self.identifier, self.channel_buffer, self.active) - def create_buffer(self): - channel_buffer = w.buffer_search("", "%s.%s" % (self.server.domain, self.name)) - if channel_buffer: - self.channel_buffer = channel_buffer - else: - self.channel_buffer = w.buffer_new("%s.%s" % (self.server.domain, self.name), "input", self.name, "", "") - w.buffer_set(self.channel_buffer, "short_name", 'loading..') - def attach_buffer(self): - channel_buffer = w.buffer_search("", "%s.%s" % (self.server.domain, self.name)) - if channel_buffer != main_weechat_buffer: - self.channel_buffer = channel_buffer -# w.buffer_set(self.channel_buffer, "highlight_words", self.server.nick) - else: - self.channel_buffer = None - def detach_buffer(self): - if self.channel_buffer != None: - w.buffer_close(self.channel_buffer) - self.channel_buffer = None - def fullname(self): - return "%s.%s" % (self.server.domain, self.name) - def set_active(self): - self.active = True - def set_inactive(self): - self.active = False - def set_typing(self, user): - self.typing[user] = time.time() - def send_message(self, message): - request = {"type":"message","channel":self.identifier, "text": message} - self.server.ws.send(json.dumps(request)) - def open(self, update_remote=True): - self.create_buffer() - self.active = True - self.get_history() - if update_remote: - t = time.time() - async_slack_api_request(self.server.domain, self.server.token, SLACK_API_TRANSLATOR[self.type]["join"], {"name":self.name.lstrip("#"),"ts":t}) - def close(self, update_remote=True): - if self.active == True: - self.active = False - self.detach_buffer() - if update_remote: - t = time.time() - async_slack_api_request(self.server.domain, self.server.token, SLACK_API_TRANSLATOR[self.type]["leave"], {"channel":self.identifier,"ts":t}) - def closed(self): - self.channel_buffer = None - self.close() - def unset_typing(self, user): - try: - del self.typing[user] - except: - pass - def is_someone_typing(self): - for user in self.typing.keys(): - if self.typing[user] + 4 > time.time(): - return True - return False - def get_typing_list(self): - typing = [] - for user in self.typing.keys(): - if self.typing[user] + 4 > time.time(): - typing.append(user) - return typing - def mark_read(self, update_remote=True): - t = time.time() - - if self.channel_buffer: - w.buffer_set(self.channel_buffer, "unread", "") - if update_remote: - self.last_read = time.time() - self.set_read_marker(self.last_read) - def set_read_marker(self, time): - async_slack_api_request(self.server.domain, self.server.token, SLACK_API_TRANSLATOR[self.type]["mark"], {"channel":self.identifier,"ts":time}) - def rename(self, name=None, fmt=None): - if self.channel_buffer: - if name: - new_name = name - elif fmt: - new_name = fmt % (self.name[1:]) - else: - new_name = self.name - #w.buffer_set(self.weechat_buffer, "short_name", new_name) - w.buffer_set(self.channel_buffer, "short_name", new_name) - def buffer_prnt(self, user='unknown user', message='no message', time=0, backlog=False): - set_read_marker = False - time = float(time) - message = message.encode('ascii', 'ignore') - if backlog == True or (time != 0 and self.last_read > time): - tags = "no_highlight,notify_none,logger_backlog_end" - set_read_marker = True - elif message.find(self.server.nick) > -1: - tags = "notify_highlight" - else: - tags = "notify_message" - time = int(float(time)) - if self.channel_buffer: - w.prnt_date_tags(self.channel_buffer, time, tags, "%s\t%s" % (user, message)) - if set_read_marker: - self.mark_read(False) - else: - dbg("failed to print something..") - def get_history(self): - if self.active: - t = time.time() - async_slack_api_request(self.server.domain, self.server.token, SLACK_API_TRANSLATOR[self.type]["history"], {"channel":self.identifier,"ts":t, "count":BACKLOG_SIZE}) + def __init__(self, server, name, identifier, active, last_read=0, prepend_name=""): + super(Channel, self).__init__(name, identifier) + self.type = "channel" + self.server = server + self.name = prepend_name + self.name + self.typing = {} + self.active = active + self.last_read = float(last_read) + if active: + self.create_buffer() + self.attach_buffer() + def __eq__(self, compare_str): + if compare_str == self.fullname() or compare_str == self.name or compare_str == self.identifier or compare_str == self.name[1:] or (compare_str == self.channel_buffer and self.channel_buffer != None): + return True + else: + return False + def __str__(self): + return "Name: %s Id: %s Buffer: %s Active: %s" % (self.name, self.identifier, self.channel_buffer, self.active) + def __repr__(self): + return "Name: %s Id: %s Buffer: %s Active: %s" % (self.name, self.identifier, self.channel_buffer, self.active) + def create_buffer(self): + channel_buffer = w.buffer_search("", "%s.%s" % (self.server.domain, self.name)) + if channel_buffer: + self.channel_buffer = channel_buffer + else: + self.channel_buffer = w.buffer_new("%s.%s" % (self.server.domain, self.name), "input", self.name, "", "") + w.buffer_set(self.channel_buffer, "short_name", 'loading..') + def attach_buffer(self): + channel_buffer = w.buffer_search("", "%s.%s" % (self.server.domain, self.name)) + if channel_buffer != main_weechat_buffer: + self.channel_buffer = channel_buffer +# w.buffer_set(self.channel_buffer, "highlight_words", self.server.nick) + else: + self.channel_buffer = None + def detach_buffer(self): + if self.channel_buffer != None: + w.buffer_close(self.channel_buffer) + self.channel_buffer = None + def fullname(self): + return "%s.%s" % (self.server.domain, self.name) + def set_active(self): + self.active = True + def set_inactive(self): + self.active = False + def set_typing(self, user): + self.typing[user] = time.time() + def send_message(self, message): + request = {"type":"message","channel":self.identifier, "text": message} + self.server.ws.send(json.dumps(request)) + def open(self, update_remote=True): + self.create_buffer() + self.active = True + self.get_history() + if update_remote: + t = time.time() + async_slack_api_request(self.server.domain, self.server.token, SLACK_API_TRANSLATOR[self.type]["join"], {"name":self.name.lstrip("#"),"ts":t}) + def close(self, update_remote=True): + if self.active == True: + self.active = False + self.detach_buffer() + if update_remote: + t = time.time() + async_slack_api_request(self.server.domain, self.server.token, SLACK_API_TRANSLATOR[self.type]["leave"], {"channel":self.identifier,"ts":t}) + def closed(self): + self.channel_buffer = None + self.close() + def unset_typing(self, user): + try: + del self.typing[user] + except: + pass + def is_someone_typing(self): + for user in self.typing.keys(): + if self.typing[user] + 4 > time.time(): + return True + return False + def get_typing_list(self): + typing = [] + for user in self.typing.keys(): + if self.typing[user] + 4 > time.time(): + typing.append(user) + return typing + def mark_read(self, update_remote=True): + t = time.time() + + if self.channel_buffer: + w.buffer_set(self.channel_buffer, "unread", "") + if update_remote: + self.last_read = time.time() + self.set_read_marker(self.last_read) + def set_read_marker(self, time): + async_slack_api_request(self.server.domain, self.server.token, SLACK_API_TRANSLATOR[self.type]["mark"], {"channel":self.identifier,"ts":time}) + def rename(self, name=None, fmt=None): + if self.channel_buffer: + if name: + new_name = name + elif fmt: + new_name = fmt % (self.name[1:]) + else: + new_name = self.name + #w.buffer_set(self.weechat_buffer, "short_name", new_name) + w.buffer_set(self.channel_buffer, "short_name", new_name) + def buffer_prnt(self, user='unknown user', message='no message', time=0, backlog=False): + set_read_marker = False + time = float(time) + message = message.encode('ascii', 'ignore') + if backlog == True or (time != 0 and self.last_read > time): + tags = "no_highlight,notify_none,logger_backlog_end" + set_read_marker = True + elif message.find(self.server.nick) > -1: + tags = "notify_highlight" + else: + tags = "notify_message" + time = int(float(time)) + if self.channel_buffer: + w.prnt_date_tags(self.channel_buffer, time, tags, "%s\t%s" % (user, message)) + if set_read_marker: + self.mark_read(False) + else: + dbg("failed to print something..") + def get_history(self): + if self.active: + t = time.time() + async_slack_api_request(self.server.domain, self.server.token, SLACK_API_TRANSLATOR[self.type]["history"], {"channel":self.identifier,"ts":t, "count":BACKLOG_SIZE}) class GroupChannel(Channel): - def __init__(self, server, name, identifier, active, last_read=0, prepend_name=""): - super(GroupChannel, self).__init__(server, name, identifier, active, last_read, prepend_name) - self.type = "group" + def __init__(self, server, name, identifier, active, last_read=0, prepend_name=""): + super(GroupChannel, self).__init__(server, name, identifier, active, last_read, prepend_name) + self.type = "group" class DmChannel(Channel): - def __init__(self, server, name, identifier, active, last_read=0, prepend_name=""): - super(DmChannel, self).__init__(server, name, identifier, active, last_read, prepend_name) - self.type = "im" - def rename(self, name=None, fmt=None): - color = w.info_get('irc_nick_color', self.name) - if self.channel_buffer: - if name: - new_name = name - elif fmt: - new_name = fmt % (self.name) - else: - new_name = self.name - w.buffer_set(self.channel_buffer, "short_name", color + new_name) - def buffer_prnt(self, user='unknown user', message='no message', time=0, backlog=False): - set_read_marker = False - time = float(time) - message = message.encode('ascii', 'ignore') - if backlog == True or (time != 0 and self.last_read > time): - tags = "no_highlight,notify_none,logger_backlog_end" - set_read_marker = True - elif user == self.server.nick: - tags = "" - elif message.find(self.server.nick) > -1: - tags = "notify_highlight" - else: - tags = "notify_private,notify_message" - time = int(float(time)) - if self.channel_buffer: - w.prnt_date_tags(self.channel_buffer, time, tags, "%s\t%s" % (user, message)) - if set_read_marker: - self.mark_read(False) - else: - dbg("failed to print something..") - pass + def __init__(self, server, name, identifier, active, last_read=0, prepend_name=""): + super(DmChannel, self).__init__(server, name, identifier, active, last_read, prepend_name) + self.type = "im" + def rename(self, name=None, fmt=None): + color = w.info_get('irc_nick_color', self.name) + if self.channel_buffer: + if name: + new_name = name + elif fmt: + new_name = fmt % (self.name) + else: + new_name = self.name + w.buffer_set(self.channel_buffer, "short_name", color + new_name) + def buffer_prnt(self, user='unknown user', message='no message', time=0, backlog=False): + set_read_marker = False + time = float(time) + message = message.encode('ascii', 'ignore') + if backlog == True or (time != 0 and self.last_read > time): + tags = "no_highlight,notify_none,logger_backlog_end" + set_read_marker = True + elif user == self.server.nick: + tags = "" + elif message.find(self.server.nick) > -1: + tags = "notify_highlight" + else: + tags = "notify_private,notify_message" + time = int(float(time)) + if self.channel_buffer: + w.prnt_date_tags(self.channel_buffer, time, tags, "%s\t%s" % (user, message)) + if set_read_marker: + self.mark_read(False) + else: + dbg("failed to print something..") + pass class User(SlackThing): - def __init__(self, server, name, identifier, presence="away"): - super(User, self).__init__(name, identifier) - self.channel_buffer = w.info_get("irc_buffer", "%s.%s" % (domain, self.name)) - self.presence = presence - self.server = server - if self.presence == 'away': - self.nicklist_pointer = w.nicklist_add_nick(server.buffer, "", self.name, w.info_get('irc_nick_color_name', self.name), " ", "", 0) - else: - self.nicklist_pointer = w.nicklist_add_nick(server.buffer, "", self.name, w.info_get('irc_nick_color_name', self.name), "+", "", 1) -# w.nicklist_add_nick(server.buffer, "", self.colorized_name(), "", "", "", 1) - def __eq__(self, compare_str): - if compare_str == self.name or compare_str == self.identifier: - return True - else: - return False - def set_active(self): - self.presence = "active" - w.nicklist_nick_set(self.server.buffer, self.nicklist_pointer, "prefix", "+") - w.nicklist_nick_set(self.server.buffer, self.nicklist_pointer, "visible", "1") - def set_inactive(self): - self.presence = "away" - w.nicklist_nick_set(self.server.buffer, self.nicklist_pointer, "prefix", " ") - w.nicklist_nick_set(self.server.buffer, self.nicklist_pointer, "visible", "0") - def colorized_name(self): - color = w.info_get('irc_nick_color', self.name) - def_color = w.color('default') - return color+self.name+def_color - def open(self): - t = time.time() + 1 - #reply = async_slack_api_request("im.open", {"channel":self.identifier,"ts":t}) - async_slack_api_request(self.server.domain, self.server.token, "im.open", {"user":self.identifier,"ts":t}) + def __init__(self, server, name, identifier, presence="away"): + super(User, self).__init__(name, identifier) + self.channel_buffer = w.info_get("irc_buffer", "%s.%s" % (domain, self.name)) + self.presence = presence + self.server = server + if self.presence == 'away': + self.nicklist_pointer = w.nicklist_add_nick(server.buffer, "", self.name, w.info_get('irc_nick_color_name', self.name), " ", "", 0) + else: + self.nicklist_pointer = w.nicklist_add_nick(server.buffer, "", self.name, w.info_get('irc_nick_color_name', self.name), "+", "", 1) +# w.nicklist_add_nick(server.buffer, "", self.colorized_name(), "", "", "", 1) + def __eq__(self, compare_str): + if compare_str == self.name or compare_str == self.identifier: + return True + else: + return False + def set_active(self): + self.presence = "active" + w.nicklist_nick_set(self.server.buffer, self.nicklist_pointer, "prefix", "+") + w.nicklist_nick_set(self.server.buffer, self.nicklist_pointer, "visible", "1") + def set_inactive(self): + self.presence = "away" + w.nicklist_nick_set(self.server.buffer, self.nicklist_pointer, "prefix", " ") + w.nicklist_nick_set(self.server.buffer, self.nicklist_pointer, "visible", "0") + def colorized_name(self): + color = w.info_get('irc_nick_color', self.name) + def_color = w.color('default') + return color+self.name+def_color + def open(self): + t = time.time() + 1 + #reply = async_slack_api_request("im.open", {"channel":self.identifier,"ts":t}) + async_slack_api_request(self.server.domain, self.server.token, "im.open", {"user":self.identifier,"ts":t}) def slack_command_cb(data, current_buffer, args): - a = args.split(' ',1) - if len(a) > 1: - function_name, args = a[0], " ".join(a[1:]) - else: - function_name, args = a[0], None -# try: - cmds[function_name](current_buffer, args) -# except KeyError: -# w.prnt("", "Command not found or exception: "+function_name) - return w.WEECHAT_RC_OK + a = args.split(' ',1) + if len(a) > 1: + function_name, args = a[0], " ".join(a[1:]) + else: + function_name, args = a[0], None +# try: + cmds[function_name](current_buffer, args) +# except KeyError: +# w.prnt("", "Command not found or exception: "+function_name) + return w.WEECHAT_RC_OK def command_talk(current_buffer, args): - servers.find(current_domain_name()).users.find(args).open() + servers.find(current_domain_name()).users.find(args).open() def command_join(current_buffer, args): - servers.find(current_domain_name()).channels.find(args).open() + servers.find(current_domain_name()).channels.find(args).open() def command_channels(current_buffer, args): - server = servers.find(current_domain_name()) - for channel in server.channels: - line = "%-25s %s %s" % (channel.name, channel.identifier, channel.active) - server.buffer_prnt(line) + server = servers.find(current_domain_name()) + for channel in server.channels: + line = "%-25s %s %s" % (channel.name, channel.identifier, channel.active) + server.buffer_prnt(line) def command_users(current_buffer, args): - server = servers.find(current_domain_name()) - for user in server.users: - line = "%-40s %s %s" % (user.colorized_name(), user.identifier, user.presence) - server.buffer_prnt(line) + server = servers.find(current_domain_name()) + for user in server.users: + line = "%-40s %s %s" % (user.colorized_name(), user.identifier, user.presence) + server.buffer_prnt(line) def command_setallreadmarkers(current_buffer, args): - if args: - for channel in channels: - channel.set_read_marker(args) + if args: + for channel in channels: + channel.set_read_marker(args) def command_changetoken(current_buffer, args): - w.config_set_plugin('slack_api_token', args) + w.config_set_plugin('slack_api_token', args) def command_test(current_buffer, args): - if slack_buffer: - w.prnt(slack_buffer,"worked!") + if slack_buffer: + w.prnt(slack_buffer,"worked!") def command_away(current_buffer, args): - server = servers.find(current_domain_name()) - async_slack_api_request(server.domain, server.token, 'presence.set', {"presence":"away"}) + server = servers.find(current_domain_name()) + async_slack_api_request(server.domain, server.token, 'presence.set', {"presence":"away"}) def command_back(current_buffer, args): - server = servers.find(current_domain_name()) - async_slack_api_request(server.domain, server.token, 'presence.set', {"presence":"active"}) + server = servers.find(current_domain_name()) + async_slack_api_request(server.domain, server.token, 'presence.set', {"presence":"active"}) def command_markread(current_buffer, args): - #refactor this - one liner i think - channel = current_buffer_name(short=True) - domain = current_domain_name() - if servers.find(domain).channels.find(channel): - servers.find(domain).channels.find(channel).mark_read() + #refactor this - one liner i think + channel = current_buffer_name(short=True) + domain = current_domain_name() + if servers.find(domain).channels.find(channel): + servers.find(domain).channels.find(channel).mark_read() def command_neveraway(current_buffer, args): - global never_away - if never_away == True: - never_away = False - dbg("unset never_away") - else: - never_away = True - dbg("set never_away") + global never_away + if never_away == True: + never_away = False + dbg("unset never_away") + else: + never_away = True + dbg("set never_away") def command_printvar(current_buffer, args): - w.prnt("", str(eval(args))) + w.prnt("", str(eval(args))) def command_p(current_buffer, args): - w.prnt("", str(eval(args))) + w.prnt("", str(eval(args))) def command_debug(current_buffer, args): - create_slack_debug_buffer() + create_slack_debug_buffer() def command_debugstring(current_buffer, args): - global debug_string - if args == '': - debug_string = None - else: - debug_string = args + global debug_string + if args == '': + debug_string = None + else: + debug_string = args def command_search(current_buffer, args): - if not slack_buffer: - create_slack_buffer() - w.buffer_set(slack_buffer, "display", "1") - query = args - w.prnt(slack_buffer,"\nSearched for: %s\n\n" % (query)) - reply = slack_api_request('search.messages', {"query":query}).read() - data = json.loads(reply) - for message in data['messages']['matches']: - message["text"] = message["text"].encode('ascii', 'ignore') - formatted_message = "%s / %s:\t%s" % (message["channel"]["name"], message['username'], message['text']) - w.prnt(slack_buffer,str(formatted_message)) + pass +# if not slack_buffer: +# create_slack_buffer() +# w.buffer_set(slack_buffer, "display", "1") +# query = args +# w.prnt(slack_buffer,"\nSearched for: %s\n\n" % (query)) +# reply = slack_api_request('search.messages', {"query":query}).read() +# data = json.loads(reply) +# for message in data['messages']['matches']: +# message["text"] = message["text"].encode('ascii', 'ignore') +# formatted_message = "%s / %s:\t%s" % (message["channel"]["name"], message['username'], message['text']) +# w.prnt(slack_buffer,str(formatted_message)) def command_nick(current_buffer, args): - urllib.urlopen("https://%s/account/settings" % (domain)) - browser.select_form(nr=0) - browser.form['username'] = args - reply = browser.submit() + pass +# urllib.urlopen("https://%s/account/settings" % (domain)) +# browser.select_form(nr=0) +# browser.form['username'] = args +# reply = browser.submit() ### Websocket handling methods def slack_websocket_cb(data, fd): - server = data - try: - data = servers.find(server).ws.recv() - message_json = json.loads(data) - #this magic attaches json that helps find the right dest - message_json['myserver'] = server - except: - return w.WEECHAT_RC_OK - dbg(message_json) - #dispatch here - if message_json.has_key("type"): - function_name = message_json["type"] - else: - function_name = "unknown" - try: - proc[function_name](message_json) - except KeyError: - pass - if function_name: - dbg("Function not implemented: %s\n%s" % (function_name, message_json)) + server = data + try: + data = servers.find(server).ws.recv() + message_json = json.loads(data) + #this magic attaches json that helps find the right dest + message_json['myserver'] = server + except: + return w.WEECHAT_RC_OK + dbg(message_json) + #dispatch here + if message_json.has_key("type"): + function_name = message_json["type"] else: - dbg("Function not implemented\n%s" % (message_json)) - w.bar_item_update("slack_typing_notice") - return w.WEECHAT_RC_OK + function_name = "unknown" + try: + proc[function_name](message_json) + except KeyError: + pass + if function_name: + dbg("Function not implemented: %s\n%s" % (function_name, message_json)) + else: + dbg("Function not implemented\n%s" % (message_json)) + w.bar_item_update("slack_typing_notice") + return w.WEECHAT_RC_OK def process_pong(message_json): - pass + pass def process_presence_change(message_json): - buffer_name = "%s.%s" % (domain, message_json["user"]) - buf_ptr = w.buffer_search("",buffer_name) - if message_json["presence"] == 'active': - users.find(message_json["user"]).set_active() - else: - users.find(message_json["user"]).set_inactive() + buffer_name = "%s.%s" % (domain, message_json["user"]) + buf_ptr = w.buffer_search("",buffer_name) + if message_json["presence"] == 'active': + users.find(message_json["user"]).set_active() + else: + users.find(message_json["user"]).set_inactive() def process_channel_marked(message_json): - channel = channels.find(message_json["channel"]) - channel.mark_read(False) - if not legacy_mode: - w.buffer_set(channel.channel_buffer, "hotlist", "-1") + channel = channels.find(message_json["channel"]) + channel.mark_read(False) + if not legacy_mode: + w.buffer_set(channel.channel_buffer, "hotlist", "-1") def process_group_marked(message_json): - channel = channels.find(message_json["channel"]) - channel.mark_read(False) - if not legacy_mode: - w.buffer_set(channel.channel_buffer, "hotlist", "-1") + channel = channels.find(message_json["channel"]) + channel.mark_read(False) + if not legacy_mode: + w.buffer_set(channel.channel_buffer, "hotlist", "-1") def process_im_marked(message_json): - channel = channels.find(message_json["channel"]) - channel.mark_read(False) - if not legacy_mode: - w.buffer_set(channel.channel_buffer, "hotlist", "-1") + channel = channels.find(message_json["channel"]) + channel.mark_read(False) + if not legacy_mode: + w.buffer_set(channel.channel_buffer, "hotlist", "-1") def process_channel_created(message_json): - server = servers.find(message_json["myserver"]) - item = message_json["channel"] - server.channels.append(Channel(server, item["name"], item["id"], False, 0, "#")) - w.prnt(server.buffer, "New channel created: %s" % item["name"]) + server = servers.find(message_json["myserver"]) + item = message_json["channel"] + server.channels.append(Channel(server, item["name"], item["id"], False, 0, "#")) + w.prnt(server.buffer, "New channel created: %s" % item["name"]) def process_channel_left(message_json): - server = servers.find(message_json["myserver"]) - server.channels.find(message_json["channel"]).close(False) + server = servers.find(message_json["myserver"]) + server.channels.find(message_json["channel"]).close(False) def process_channel_joined(message_json): - server = servers.find(message_json["myserver"]) - server.channels.find(message_json["channel"]["id"]).open(False) + server = servers.find(message_json["myserver"]) + server.channels.find(message_json["channel"]["id"]).open(False) def process_group_left(message_json): - server = servers.find(message_json["myserver"]) - server.channels.find(message_json["channel"]).close(False) + server = servers.find(message_json["myserver"]) + server.channels.find(message_json["channel"]).close(False) def process_group_joined(message_json): - server = servers.find(message_json["myserver"]) - if server.channels.find(message_json["channel"]["name"]): - server.channels.find(message_json["channel"]["name"]).open(False) - else: - item = message_json["channel"] - server.channels.append(GroupChannel(server, item["name"], item["id"], item["is_open"], item["last_read"], "#")) + server = servers.find(message_json["myserver"]) + if server.channels.find(message_json["channel"]["name"]): + server.channels.find(message_json["channel"]["name"]).open(False) + else: + item = message_json["channel"] + server.channels.append(GroupChannel(server, item["name"], item["id"], item["is_open"], item["last_read"], "#")) def process_im_close(message_json): - server = servers.find(message_json["myserver"]) - server.channels.find(message_json["channel"]).close(False) + server = servers.find(message_json["myserver"]) + server.channels.find(message_json["channel"]).close(False) def process_im_open(message_json): - server = servers.find(message_json["myserver"]) - server.channels.find(message_json["channel"]).open(False) + server = servers.find(message_json["myserver"]) + server.channels.find(message_json["channel"]).open(False) def process_user_typing(message_json): - server = servers.find(message_json["myserver"]) - server.channels.find(message_json["channel"]).set_typing(server.users.find(message_json["user"]).name) + server = servers.find(message_json["myserver"]) + server.channels.find(message_json["channel"]).set_typing(server.users.find(message_json["user"]).name) #todo: does this work? def process_error(message_json): - connected = False + connected = False def process_message(message_json): - server = servers.find(message_json["myserver"]) - - mark_silly_channels_read(message_json["channel"]) - channel = message_json["channel"] - -# if message_json.has_key("subtype"): -# return - time = message_json["ts"] - #this handles edits - try: - if message_json.has_key("message"): - message_json["text"] = "Edited: " + message_json["message"]["text"] - if message_json.has_key("user"): - message_json["user"] = message_json["message"]["user"] - elif message_json.has_key("username"): - message_json["user"] = message_json["message"]["username"] - except: - dbg(message_json) + server = servers.find(message_json["myserver"]) - if message_json.has_key("user") and message_json.has_key("text"): - #below prevents typing notification from disapearing if the server sends an unfurled message - channel = server.channels.find(message_json["channel"]) - channel.unset_typing(server.users.find(message_json["user"]).name) - user = server.users.find(message_json["user"]) - if user.name != channel.server.nick: - user = user.colorized_name() - else: - user = user.name - channel.buffer_prnt(user,message_json["text"], time) - else: - if message_json.has_key("attachments"): - if message_json.has_key("username"): - name = message_json["username"] - for message in message_json["attachments"]: - if message.has_key("service_name"): - name = message["service_name"] - try: - server.channels.find(channel).buffer_prnt("-%s-" % name,str(message["fallback"]), time) - except: - server.channels.find(channel).buffer_prnt('unknown user',str(message_json), time) + mark_silly_channels_read(message_json["channel"]) + channel = message_json["channel"] + +# if message_json.has_key("subtype"): +# return + time = message_json["ts"] + #this handles edits + try: + if message_json.has_key("message"): + message_json["text"] = "Edited: " + message_json["message"]["text"] + if message_json.has_key("user"): + message_json["user"] = message_json["message"]["user"] + elif message_json.has_key("username"): + message_json["user"] = message_json["message"]["username"] + except: + dbg(message_json) + + if message_json.has_key("user") and message_json.has_key("text"): + #below prevents typing notification from disapearing if the server sends an unfurled message + channel = server.channels.find(message_json["channel"]) + channel.unset_typing(server.users.find(message_json["user"]).name) + user = server.users.find(message_json["user"]) + if user.name != channel.server.nick: + user = user.colorized_name() + else: + user = user.name + channel.buffer_prnt(user,message_json["text"], time) else: - server.channels.find(channel).buffer_prnt('unknown user',str(message_json), time) + if message_json.has_key("attachments"): + if message_json.has_key("username"): + name = message_json["username"] + for message in message_json["attachments"]: + if message.has_key("service_name"): + name = message["service_name"] + try: + server.channels.find(channel).buffer_prnt("-%s-" % name,str(message["fallback"]), time) + except: + server.channels.find(channel).buffer_prnt('unknown user',str(message_json), time) + else: + server.channels.find(channel).buffer_prnt('unknown user',str(message_json), time) ### END Websocket handling methods def typing_bar_item_cb(data, buffer, args): - typers = [x for x in channels.get_all() if x.is_someone_typing() == True] - if len(typers) > 0: - direct_typers = [] - channel_typers = [] - for dm in channels.find_by_class(DmChannel): - direct_typers.extend(dm.get_typing_list()) - direct_typers = ["D/" + x for x in direct_typers] - current_channel = current_buffer_name(short=True) - channel = channels.find(current_channel) - if channel and channel.__class__ != DmChannel: - channel_typers = channels.find(current_channel).get_typing_list() - typing_here = ", ".join(channel_typers + direct_typers) - if len(typing_here) > 0: - color = w.color('yellow') - return color + "typing: " + typing_here - return "" + typers = [x for x in channels.get_all() if x.is_someone_typing() == True] + if len(typers) > 0: + direct_typers = [] + channel_typers = [] + for dm in channels.find_by_class(DmChannel): + direct_typers.extend(dm.get_typing_list()) + direct_typers = ["D/" + x for x in direct_typers] + current_channel = current_buffer_name(short=True) + channel = channels.find(current_channel) + if channel and channel.__class__ != DmChannel: + channel_typers = channels.find(current_channel).get_typing_list() + typing_here = ", ".join(channel_typers + direct_typers) + if len(typing_here) > 0: + color = w.color('yellow') + return color + "typing: " + typing_here + return "" def typing_update_cb(data, remaining_calls): - w.bar_item_update("slack_typing_notice") - return w.WEECHAT_RC_OK + w.bar_item_update("slack_typing_notice") + return w.WEECHAT_RC_OK def buffer_list_update_cb(data, remaining_calls): - for channel in channels.find_by_class(Channel): - if channel.is_someone_typing() == True: - channel.rename(fmt=">%s") - else: - channel.rename() - for channel in channels.find_by_class(GroupChannel): - if channel.is_someone_typing() == True: - channel.rename(fmt=">%s") - else: - channel.rename() - for channel in channels.find_by_class(DmChannel): - if channel.server.users.find(channel.name).presence == "active": - channel.rename(fmt="+%s") - else: - channel.rename(fmt=" %s") - pass - return w.WEECHAT_RC_OK + for channel in channels.find_by_class(Channel): + if channel.is_someone_typing() == True: + channel.rename(fmt=">%s") + else: + channel.rename() + for channel in channels.find_by_class(GroupChannel): + if channel.is_someone_typing() == True: + channel.rename(fmt=">%s") + else: + channel.rename() + for channel in channels.find_by_class(DmChannel): + if channel.server.users.find(channel.name).presence == "active": + channel.rename(fmt="+%s") + else: + channel.rename(fmt=" %s") + pass + return w.WEECHAT_RC_OK def hotlist_cache_update_cb(data, remaining_calls): - #this keeps the hotlist dupe up to date for the buffer switch, but is prob technically a race condition. (meh) - global hotlist - prev_hotlist = hotlist - hotlist = w.infolist_get("hotlist", "", "") - w.infolist_free(prev_hotlist) - return w.WEECHAT_RC_OK + #this keeps the hotlist dupe up to date for the buffer switch, but is prob technically a race condition. (meh) + global hotlist + prev_hotlist = hotlist + hotlist = w.infolist_get("hotlist", "", "") + w.infolist_free(prev_hotlist) + return w.WEECHAT_RC_OK def buffer_closing_cb(signal, sig_type, data): - if channels.find(data): - channels.find(data).closed() - return w.WEECHAT_RC_OK + if channels.find(data): + channels.find(data).closed() + return w.WEECHAT_RC_OK def buffer_switch_cb(signal, sig_type, data): - global previous_buffer, hotlist - if channels.find(previous_buffer): - channels.find(previous_buffer).mark_read() + global previous_buffer, hotlist + if channels.find(previous_buffer): + channels.find(previous_buffer).mark_read() - channel_name = current_buffer_name() - previous_buffer = data - return w.WEECHAT_RC_OK + channel_name = current_buffer_name() + previous_buffer = data + return w.WEECHAT_RC_OK def typing_notification_cb(signal, sig_type, data): - global typing_timer - now = time.time() - if typing_timer + 4 < now: - channel = channels.find(current_buffer_name()) - if channel: - identifier = channel.identifier - request = {"type":"typing","channel":identifier} - channel.server.ws.send(json.dumps(request)) - typing_timer = now - return w.WEECHAT_RC_OK + global typing_timer + now = time.time() + if typing_timer + 4 < now: + channel = channels.find(current_buffer_name()) + if channel: + identifier = channel.identifier + request = {"type":"typing","channel":identifier} + channel.server.ws.send(json.dumps(request)) + typing_timer = now + return w.WEECHAT_RC_OK #NOTE: figured i'd do this because they do def slack_ping_cb(data, remaining): - servers.find(data).ping() - return w.WEECHAT_RC_OK + servers.find(data).ping() + return w.WEECHAT_RC_OK def slack_connection_persistence_cb(data, remaining_calls): - for server in servers: - if not server.connected: - w.prnt("", "Disconnected from slack, trying to reconnect..") - if server.ws_hook != None: - w.unhook(server.ws_hook) - server.connect_to_slack() - return w.WEECHAT_RC_OK + for server in servers: + if not server.connected: + w.prnt("", "Disconnected from slack, trying to reconnect..") + if server.ws_hook != None: + w.unhook(server.ws_hook) + server.connect_to_slack() + return w.WEECHAT_RC_OK def slack_never_away_cb(data, remaining): - global never_away - if never_away == True: - for server in servers: - identifier = server.channels.find("slackbot").identifier - request = {"type":"typing","channel":identifier} - #request = {"type":"typing","channel":"slackbot"} - server.ws.send(json.dumps(request)) - return w.WEECHAT_RC_OK + global never_away + if never_away == True: + for server in servers: + identifier = server.channels.find("slackbot").identifier + request = {"type":"typing","channel":identifier} + #request = {"type":"typing","channel":"slackbot"} + server.ws.send(json.dumps(request)) + return w.WEECHAT_RC_OK ### Slack specific requests #NOTE: switched to async/curl because sync slowed down the UI def async_slack_api_request(domain, token, request, post_data, priority=False): - t = time.time() - post_elements = pickle.dumps([domain, token, request, post_data]) - request += "?t=%s" % t - post_data["token"] = token - post_data = urllib.urlencode(post_data) - post = {"post": "1", "postfields": post_data} - url = 'https://%s/api/%s' % (domain, request) - queue_item = ['url:%s' % (url), post, 20000, 'url_processor_cb', post_elements] - if priority != True: - queue.append(QueueItem(queue_item, 'do_url', 'url_processor_cb')) - else: - queue.insert(0, QueueItem(queue_item, 'do_url', 'url_processor_cb')) + t = time.time() + post_elements = pickle.dumps([domain, token, request, post_data]) + request += "?t=%s" % t + post_data["token"] = token + post_data = urllib.urlencode(post_data) + post = {"post": "1", "postfields": post_data} + url = 'https://%s/api/%s' % (domain, request) + queue_item = ['url:%s' % (url), post, 20000, 'url_processor_cb', post_elements] + if priority != True: + queue.append(QueueItem(queue_item, 'do_url', 'url_processor_cb')) + else: + queue.insert(0, QueueItem(queue_item, 'do_url', 'url_processor_cb')) queue = [] async_queue_lock=False class QueueItem(object): - def __init__(self, data, method, callback_method=None): - self.method = method - self.callback_method = callback_method - self.data = data - def __getitem__(self, index): - return self.data[index] - def __str__(self): - return str(self.data) - def __repr__(self): - return str(self.data) + def __init__(self, data, method, callback_method=None): + self.method = method + self.callback_method = callback_method + self.data = data + def __getitem__(self, index): + return self.data[index] + def __str__(self): + return str(self.data) + def __repr__(self): + return str(self.data) def do_url(item): - try: - query = urlparse.parse_qs(item[1]["postfields"]) - if query.has_key("channel") and item[0].find('history') > -1: - channel = query["channel"][0] - channel = channels.find(channel) - channel.server.buffer_prnt("downloading channel history for %s" % (channel.name), backlog=True) - except: - pass - command = 'curl --data "%s" %s' % (item[1]["postfields"], item[0][4:]) - w.hook_process(command, 10000, item[3], item[4]) + try: + query = urlparse.parse_qs(item[1]["postfields"]) + if query.has_key("channel") and item[0].find('history') > -1: + channel = query["channel"][0] + channel = channels.find(channel) + channel.server.buffer_prnt("downloading channel history for %s" % (channel.name), backlog=True) + except: + pass + command = 'curl --data "%s" %s' % (item[1]["postfields"], item[0][4:]) + w.hook_process(command, 10000, item[3], item[4]) def async_queue_cb(data, remaining_calls): - global async_queue_lock - if async_queue_lock == False: - async_queue_lock=True - if len(queue) > 0: - item = queue.pop(0) - method = eval(item.method) - method(item) - async_queue_lock=False - else: - async_queue_lock=False - return w.WEECHAT_RC_OK + global async_queue_lock + if async_queue_lock == False: + async_queue_lock=True + if len(queue) > 0: + item = queue.pop(0) + method = eval(item.method) + method(item) + async_queue_lock=False + else: + async_queue_lock=False + return w.WEECHAT_RC_OK #funny, right? big_data = {} def url_processor_cb(data, command, return_code, out, err): - data=pickle.loads(data) - global async_queue_lock, big_data - identifier = sha.sha(str(data) + command).hexdigest() - if not big_data.has_key(identifier): - big_data[identifier] = '' - big_data[identifier] += out - if return_code == 0: - async_queue_lock=False - try: - my_json = json.loads(big_data[identifier]) - except: -# if big_data[identifier] != '': - async_queue_lock=False - dbg("curl failed, doing again...") - async_slack_api_request(*data, priority=True) - pass - my_json = False - del big_data[identifier] - - if my_json: - if data[2] == 'rtm.start': - servers.find(data[1]).connected_to_slack(my_json) - else: - query = data[3] - if query.has_key("channel"): - channel = query["channel"] - token = data[1] - if my_json.has_key("messages"): - messages = my_json["messages"].reverse() - for message in my_json["messages"]: - message["myserver"] = servers.find(token).domain - message["channel"] = servers.find(token).channels.find(channel) - process_message(message) - - return w.WEECHAT_RC_OK + data=pickle.loads(data) + global async_queue_lock, big_data + identifier = sha.sha(str(data) + command).hexdigest() + if not big_data.has_key(identifier): + big_data[identifier] = '' + big_data[identifier] += out + if return_code == 0: + async_queue_lock=False + try: + my_json = json.loads(big_data[identifier]) + except: +# if big_data[identifier] != '': + async_queue_lock=False + dbg("curl failed, doing again...") + async_slack_api_request(*data, priority=True) + pass + my_json = False + del big_data[identifier] + + if my_json: + if data[2] == 'rtm.start': + servers.find(data[1]).connected_to_slack(my_json) + else: + query = data[3] + if query.has_key("channel"): + channel = query["channel"] + token = data[1] + if my_json.has_key("messages"): + messages = my_json["messages"].reverse() + for message in my_json["messages"]: + message["myserver"] = servers.find(token).domain + message["channel"] = servers.find(token).channels.find(channel) + process_message(message) + + return w.WEECHAT_RC_OK def mark_silly_channels_read(channel): - if channel in channels_always_marked_read: - if channels.find("channel"): - channels.find("channel").mark_read() + if channel in channels_always_marked_read: + if channels.find("channel"): + channels.find("channel").mark_read() ### END Slack specific requests ### Utility Methods def current_domain_name(): - buffer = w.current_buffer() - if servers.find(buffer): - return servers.find(buffer).domain - else: - #number = w.buffer_get_integer(buffer, "number") - name = w.buffer_get_string(buffer, "name") - name = ".".join(name.split(".")[:-1]) - return name + buffer = w.current_buffer() + if servers.find(buffer): + return servers.find(buffer).domain + else: + #number = w.buffer_get_integer(buffer, "number") + name = w.buffer_get_string(buffer, "name") + name = ".".join(name.split(".")[:-1]) + return name def current_buffer_name(short=False): - buffer = w.current_buffer() - #number = w.buffer_get_integer(buffer, "number") - name = w.buffer_get_string(buffer, "name") - if short: - try: - name = name.split('.')[-1] - except: - pass - return name + buffer = w.current_buffer() + #number = w.buffer_get_integer(buffer, "number") + name = w.buffer_get_string(buffer, "name") + if short: + try: + name = name.split('.')[-1] + except: + pass + return name def closed_slack_buffer_cb(data, buffer): - global slack_buffer - slack_buffer = None - return w.WEECHAT_RC_OK + global slack_buffer + slack_buffer = None + return w.WEECHAT_RC_OK def create_slack_buffer(): - global slack_buffer - slack_buffer = w.buffer_new("slack", "", "", "closed_slack_buffer_cb", "") - w.buffer_set(slack_buffer, "notify", "0") - #w.buffer_set(slack_buffer, "display", "1") - return w.WEECHAT_RC_OK + global slack_buffer + slack_buffer = w.buffer_new("slack", "", "", "closed_slack_buffer_cb", "") + w.buffer_set(slack_buffer, "notify", "0") + #w.buffer_set(slack_buffer, "display", "1") + return w.WEECHAT_RC_OK def closed_slack_debug_buffer_cb(data, buffer): - global slack_debug - slack_debug = None - return w.WEECHAT_RC_OK + global slack_debug + slack_debug = None + return w.WEECHAT_RC_OK def create_slack_debug_buffer(): - global slack_debug, debug_string - if slack_debug != None: - w.buffer_set(slack_debug, "display", "1") - else: - debug_string = None - slack_debug = w.buffer_new("slack-debug", "", "", "closed_slack_debug_buffer_cb", "") - w.buffer_set(slack_debug, "notify", "0") -# w.buffer_set(slack_debug, "display", "1") + global slack_debug, debug_string + if slack_debug != None: + w.buffer_set(slack_debug, "display", "1") + else: + debug_string = None + slack_debug = w.buffer_new("slack-debug", "", "", "closed_slack_debug_buffer_cb", "") + w.buffer_set(slack_debug, "notify", "0") +# w.buffer_set(slack_debug, "display", "1") ### END Utility Methods # Main if __name__ == "__main__": - if w.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE, - SCRIPT_DESC, "", ""): - - if not w.config_get_plugin('slack_api_token'): - w.config_set_plugin('slack_api_token', "INSERT VALID KEY HERE!") - if not w.config_get_plugin('channels_always_marked_read'): - w.config_set_plugin('channels_always_marked_read', "") - if not w.config_get_plugin('debug_mode'): - w.config_set_plugin('debug_mode', "") - - version = w.info_get("version_number", "") or 0 - if int(version) >= 0x00040400: - legacy_mode = False - else: - legacy_mode = True - - ### Global constants - - DIRECT_MESSAGE = '*direct*' - - ### End global constants - - ### Global var section - slack_api_token = w.config_get_plugin("slack_api_token") - channels_always_marked_read = [x.strip() for x in w.config_get_plugin("channels_always_marked_read").split(',')] - - slack_debug = None - debug_mode = w.config_get_plugin("debug_mode").lower() - if debug_mode != '' or debug_mode != 'false': - create_slack_debug_buffer() - - cmds = {k[8:]: v for k, v in globals().items() if k.startswith("command_")} - proc = {k[8:]: v for k, v in globals().items() if k.startswith("process_")} - - typing_timer = time.time() - domain = None - previous_buffer = None - slack_buffer = None - name = None - never_away = False - hotlist = w.infolist_get("hotlist", "", "") - main_weechat_buffer = w.info_get("irc_buffer", "%s.%s" % (domain, "DOESNOTEXIST!@#$")) - - ### End global var section - - #channels = SearchList() - servers = SearchList() - for token in slack_api_token.split(','): - servers.append(SlackServer(token)) - channels = Meta('channels', servers) - users = Meta('users', servers) - - - w.hook_timer(10, 0, 0, "async_queue_cb", "") - w.hook_timer(6000, 0, 0, "slack_connection_persistence_cb", "") - - ### attach to the weechat hooks we need - w.hook_timer(1000, 0, 0, "typing_update_cb", "") - w.hook_timer(1000, 0, 0, "buffer_list_update_cb", "") - w.hook_timer(1000, 0, 0, "hotlist_cache_update_cb", "") - w.hook_timer(1000 * 3, 0, 0, "slack_never_away_cb", "") - w.hook_timer(1000 * 60* 29, 0, 0, "slack_never_away_cb", "") - w.hook_signal('buffer_closing', "buffer_closing_cb", "") - w.hook_signal('buffer_switch', "buffer_switch_cb", "") - w.hook_signal('window_switch', "buffer_switch_cb", "") - w.hook_signal('input_text_changed', "typing_notification_cb", "") - w.hook_command('slack','Plugin to allow typing notification and sync of read markers for slack.com', 'stuff', 'stuff2', '|'.join(cmds.keys()), 'slack_command_cb', '') - w.bar_item_new('slack_typing_notice', 'typing_bar_item_cb', '') - ### END attach to the weechat hooks we need + if w.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE, + SCRIPT_DESC, "", ""): + + if not w.config_get_plugin('slack_api_token'): + w.config_set_plugin('slack_api_token', "INSERT VALID KEY HERE!") + if not w.config_get_plugin('channels_always_marked_read'): + w.config_set_plugin('channels_always_marked_read', "") + if not w.config_get_plugin('debug_mode'): + w.config_set_plugin('debug_mode', "") + + version = w.info_get("version_number", "") or 0 + if int(version) >= 0x00040400: + legacy_mode = False + else: + legacy_mode = True + + ### Global constants + + DIRECT_MESSAGE = '*direct*' + + ### End global constants + + ### Global var section + slack_api_token = w.config_get_plugin("slack_api_token") + channels_always_marked_read = [x.strip() for x in w.config_get_plugin("channels_always_marked_read").split(',')] + + slack_debug = None + debug_mode = w.config_get_plugin("debug_mode").lower() + if debug_mode != '' or debug_mode != 'false': + create_slack_debug_buffer() + + cmds = {k[8:]: v for k, v in globals().items() if k.startswith("command_")} + proc = {k[8:]: v for k, v in globals().items() if k.startswith("process_")} + + typing_timer = time.time() + domain = None + previous_buffer = None + slack_buffer = None + name = None + never_away = False + hotlist = w.infolist_get("hotlist", "", "") + main_weechat_buffer = w.info_get("irc_buffer", "%s.%s" % (domain, "DOESNOTEXIST!@#$")) + + ### End global var section + + #channels = SearchList() + servers = SearchList() + for token in slack_api_token.split(','): + servers.append(SlackServer(token)) + channels = Meta('channels', servers) + users = Meta('users', servers) + + + w.hook_timer(10, 0, 0, "async_queue_cb", "") + w.hook_timer(6000, 0, 0, "slack_connection_persistence_cb", "") + + ### attach to the weechat hooks we need + w.hook_timer(1000, 0, 0, "typing_update_cb", "") + w.hook_timer(1000, 0, 0, "buffer_list_update_cb", "") + w.hook_timer(1000, 0, 0, "hotlist_cache_update_cb", "") + w.hook_timer(1000 * 3, 0, 0, "slack_never_away_cb", "") + w.hook_timer(1000 * 60* 29, 0, 0, "slack_never_away_cb", "") + w.hook_signal('buffer_closing', "buffer_closing_cb", "") + w.hook_signal('buffer_switch', "buffer_switch_cb", "") + w.hook_signal('window_switch', "buffer_switch_cb", "") + w.hook_signal('input_text_changed', "typing_notification_cb", "") + w.hook_command('slack','Plugin to allow typing notification and sync of read markers for slack.com', 'stuff', 'stuff2', '|'.join(cmds.keys()), 'slack_command_cb', '') + w.bar_item_new('slack_typing_notice', 'typing_bar_item_cb', '') + ### END attach to the weechat hooks we need |