aboutsummaryrefslogtreecommitdiffstats
path: root/wee_slack.py
diff options
context:
space:
mode:
authorRyan Huber <rhuber@gmail.com>2014-10-26 21:58:41 -0700
committerRyan Huber <rhuber@gmail.com>2014-10-26 21:58:41 -0700
commita61ffda222a69170b86e66e1478de46c599bdd31 (patch)
tree60f7636179640c0ba0d19791280f7689dcbfbf95 /wee_slack.py
parentd225901e7123d2b08759cb0d5ac4685a32078cac (diff)
downloadwee-slack-a61ffda222a69170b86e66e1478de46c599bdd31.tar.gz
caved and switched to 4 space tabs
Diffstat (limited to 'wee_slack.py')
-rw-r--r--wee_slack.py1668
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