## I need to echo out the data

global status

api = {"method": "send", "params": {"options": {"conversation_id": data, "message": {"body": input_data}}}}

#echo = status.nick_name+'\t'+input_data

#weechat.prnt_date_tags(buffer, 0, "", echo)

# And then, i need to send it

r=status.execute_api(api)

## CHECK r. Moreover, we can trigger the execution in another thread

#weechat.prnt("","Command:"+str(command))

## Maybe we can check this to see if the program is crashed

#weechat.prnt("","return_code:"+str(return_code))

weechat.prnt("","out:"+str(out))

if out == "":

return weechat.WEECHAT_RC_OK

j = json.loads(out)

if j['type'] != 'chat':

weechat.prnt_date_tags("", "","", str(j))

id = j['msg']['conversation_id']

date,body,n = handle_message(j['msg'])

global status

if id not in status.private_chans:

status.open_conv_id(j['msg'])

username = body.split('/t')

priority = "notify_private"

if len(username) == 2:

if status.nick_name in username[1]:

priority = "notify_highlight"

if weechat.buffer_get_string(status.private_chans[id], "localvar_first_message") == "":

r, f, l = status.get_last_history(id, priority)

weechat.buffer_set(status.private_chans[id], "localvar_set_first_message", str(f))

weechat.buffer_set(status.private_chans[id], "localvar_set_last_message", str(l))

return weechat.WEECHAT_RC_OK

weechat.prnt_date_tags(status.private_chans[id], date,priority, body)

if debug:

body = weechat.color('/darkgray')+"DEBUG\t"+weechat.color('/darkgray')+str(j)

weechat.prnt_date_tags(status.private_chans[id], date,"notify_none", body)

#notify_none Buffer with line is not added to hotlist.

#notify_message Buffer with line is added to hotlist with level "message".

#notify_private Buffer with line is added to hotlist with level "private".

#notify_highlight Buffer with line is added to hotlist with level "highlight".

#weechat.prnt("","err:"+str(err))

system = msg['content']['system']

body = weechat.color("/cyan")+"SYSTEM\t"

if system['systemType'] == 0:

## someone added someone to chat

addedtoteam = system['addedtoteam']

adder = addedtoteam['adder']

addee = addedtoteam['addee']

role = addedtoteam['role']

bulkAdds = addedtoteam['bulkAdds']

body = weechat.color("/cyan")+adder+" added \'"+addee+"\' to team with role "+str(role)

elif system['systemType'] == 3:

# Creation of new team

team_name = system['createteam']['team']

creator = system['createteam']['creator']

body += weechat.color("/cyan")+creator+" created \'"+team_name+"\' team"

elif system['systemType'] == 7:

# Bulk add

bulkaddtoconv = system['bulkaddtoconv']

username = bulkaddtoconv['usernames']

users = (",").join(username[:])

body = weechat.prefix("join") +users+ " have joined the channel"

elif system['systemType'] == 9:

## Creation of a new channel

new_channel = system['newchannel']

creator = new_channel['creator']

name_channel = new_channel['nameAtCreation']

body+=weechat.color("/cyan")+creator+" created \'"+name_channel+"\' channel"

else:

body+=weechat.color("red")+"type not supported. Raw message:"+str(msg)

sender = msg['sender']['username']

date = msg['sent_at']

content = msg['content']['type']

id = msg['id']

#global status

#if sender == status.nick_name:

# sender = weechat.color("lightgreen")

sender = sender+weechat.color("/lightmagenta")+" ["+str(id)+"]"

if content == 'join':

body = weechat.prefix("join")+sender+" has joined the channel"

elif content == 'system':

body = handle_system_message(msg)

elif content == 'text':

text = msg['content']['text']

msg_body = text['body'].replace('\t'," ")

## Mention username

user_mention = text['userMentions']

reply_to = text['replyTo'] if 'replyTo' in text else None

if user_mention != None:

for user in user_mention:

msg_body = msg_body.replace('@'+user['text'],weechat.color("*red")+'@'+user['text']+weechat.color("reset"))

#if user['text'] == self.nick --> PRIORITY

if reply_to != None:

msg_body = weechat.color("*darkgray") + "reply to "+str(reply_to)+"-> "+weechat.color("reset") + msg_body

body = sender+'\t'+msg_body

#elif content == 'unfurl':

# body = sender+'\t'+

#elif content == 'reaction':

# reaction = msg['content']['reaction']

elif content == 'delete':

body = sender+'\t'+weechat.color("*red")+"deleted message(s) "+str(msg['content']['delete']['messageIDs'])

elif content == 'edit':

edit = msg['content']['edit']

body = sender+'\t'+weechat.color("*red")+"edit message "+str(edit['messageID'])+" with: \'"+edit['body']+"\'"

elif content == 'metadata':

body = sender+'\t'+"Metadata: Conversation Title: "+msg['content']['metadata']['conversationTitle']

elif content == 'attachment':

body = sender+"\t"+weechat.color("_lightgreen")+"sent an attachment. Use /download "+str(id)+" <output>"

#elif content == 'pin':

# body = sender+'\t'+weechat.color("_lightgreen")+"has pinned message "+str(id)

else:

body = weechat.color("*red")+str(msg)

if arg == "":

return weechat.WEECHAT_RC_ERROR

conv_id = weechat.buffer_get_string(buffer, "localvar_conversation_id")

tmp_file = tempfile.mkstemp(suffix = '.ktmp')

weechat.prnt("", str(tmp_file))

api = {"method": "download", "params": {"options": {"conversation_id": conv_id, "message_id": int(arg), "output": tmp_file[1]}}}

## CHECK r

r=status.execute_api(api)

subprocess.Popen(['xdg-open',tmp_file[1]],close_fds=True)

args = arg.split(' ')

if len(args) != 2:

return weechat.WEECHAT_RC_ERROR

conv_id = weechat.buffer_get_string(buffer, "localvar_conversation_id")

api = {"method": "download", "params": {"options": {"conversation_id": conv_id, "message_id": int(args[0]), "output": args[1]}}}

## CHECK r

r=status.execute_api(api)

args = command.split(' ')

weechat.prnt("", str(args))

if len(args) < 3:

return weechat.WEECHAT_RC_ERROR

receiver = args[1]

body = " ".join(args[2:])

api = {"method": "send", "params": {"options": {"channel": {"name": status.nick_name+','+receiver}, "message": {"body": body}}}}

r = status.execute_api(api)

## I should also search the right buffer but I receive the message so the buffer is created after

args = command.split(' ')

weechat.prnt("", str(args))

if len(args) < 3:

return weechat.WEECHAT_RC_ERROR

reply_to = int(args[1])

body = " ".join(args[2:])

conv_id = weechat.buffer_get_string(buffer, "localvar_conversation_id")

api = {"method": "send", "params": {"options": {"conversation_id": conv_id, "message": {"body": body}, "reply_to": reply_to}}}

r = status.execute_api(api)

own_lines= weechat.hdata_pointer(weechat.hdata_get("buffer"), buffer, 'own_lines')

line = weechat.hdata_pointer(weechat.hdata_get('lines'), own_lines, 'last_line')

hdata_line_data = weechat.hdata_get('line_data')

hdata_line = weechat.hdata_get('line')

d = weechat.hdata_pointer(hdata_line, line, 'data')

weechat.prnt(buffer,weechat.hdata_string(hdata_line_data, d, 'message'))

weechat.prnt(buffer,weechat.hdata_string(hdata_line_data, d, 'prefix'))

plugin = weechat.buffer_get_string(signal_data, "localvar_server")

if plugin != "KeyBase":

return weechat.WEECHAT_RC_OK

first_message = weechat.buffer_get_string(signal_data, "localvar_first_message")

conv_id = weechat.buffer_get_string(signal_data, "localvar_conversation_id")

if (first_message == ""):

weechat.prnt("", "Get history")

r, f, l = status.get_last_history(conv_id)

weechat.buffer_set(signal_data, "localvar_set_first_message", str(f))

weechat.buffer_set(signal_data, "localvar_set_last_message", str(l))

return weechat.WEECHAT_RC_OK_EAT

elif first_message != '1':

weechat.prnt("", "Retrieve others")

buffer = weechat.current_buffer()

weechat.prnt("", str(buffer))

plugin = weechat.buffer_get_string(buffer, "localvar_server")

weechat.prnt("", plugin)

if plugin != "KeyBase":

return weechat.WEECHAT_RC_OK

first_message = weechat.buffer_get_string(buffer, "localvar_first_message")

conv_id = weechat.buffer_get_string(buffer, "localvar_conversation_id")

weechat.prnt("", "first message "+first_message)

if first_message != "1":

weechat.prnt("", "retrieve message from "+str(first_message))

ids = list(range(int(first_message)-25 if int(first_message)-25 > 0 else 0,int(first_message)))

weechat.prnt("", str(ids))

status.retrieve_messages_ids(conv_id, ids)

def __init__(self, options):

self.status_name = options['server_name']

self.nick_name = options['nickname']

global debug

if options['debug'] == "true":

debug = True

else:

debug = False

self.private_chans = {}

self.private_chans_ptr = {}

self.status = weechat.buffer_new(self.status_name, "status_input_cb", "", "status_close_cb", "")

weechat.buffer_set(self.status, "localvar_set_type", "server")

weechat.buffer_set(self.status, "localvar_set_server", "keybase")

self.init_chats()

#self.get_history()

self.reader = weechat.hook_process_hashtable("keybase chat api-listen",

{"buffer_flush":"1"},0,"start_reading","")

weechat.hook_command("download", "Download an attachment", "<msg_id> <outputh_path>", "<msg_id>: ID of the message\n<output_path>: Path to store file", "", "download_message", "")

weechat.hook_command("open", "Open (with default application) an attachment", "<msg_id>", "<msg_id>: ID of the message\n", "", "open_attachment", "")

## Hooking to classic weechat command

weechat.hook_command_run("/msg","send_new_message","")

weechat.hook_command_run("/reply", "reply_to_message", "")

weechat.hook_signal("buffer_switch", "buffer_switched", "")

weechat.hook_signal("window_scrolled", "window_scrolled", "")

weechat.hook_command("test", "", "", "", "", "test12", "")

def execute_api(self, api):

output = subprocess.check_output(['keybase', 'chat', 'api', '-m', json.dumps(api)])

#weechat.prnt("", "D "+str(output))

j = json.loads(output)

if 'error' in j:

weechat.prnt_date_tags(self.status, 0, "", weechat.prefix("error")+"[X] Error during API "+str(api))

weechat.prnt(self.status, "Debug: "+str(api))

return None

result = json.loads(output)['result']

return result

def get_last_history(self, conv_id, notify = ""):

api = {"method": "read", "params": {"options": {"conversation_id": conv_id }}}

result=self.execute_api(api)

mex = {}

for i in result['messages']:

date, body, n = handle_message(i['msg'])

mex[n] = [date, body]

od = collections.OrderedDict(sorted(mex.items()))

for n, b in od.items():

weechat.prnt_date_tags(self.private_chans[conv_id], b[0],notify ,b[1])

keys = list(od.keys())

return None, keys[0],keys[-1]

def retrieve_messages_ids(self, conv_id, ids):

api = {"method": "get", "params": {"options": {"conversation_id": conv_id, "message_ids": ids}}}

r = self.execute_api(api)

weechat.prnt("",str(r))

def retrieve_nth_page(self, conv_id, num=1000, next="", prev=""):

api = {"method": "read", "params": {"options": {"conversation_id": conv_id , "pagination":{"num":num, "next":next, "previous":prev}}}}

result = self.execute_api(api)

return result

def open_conv_id(self,msg):

conv_id = msg['conversation_id']

buff = self.create_new_buffer(msg, conv_id)

self.private_chans[conv_id] = buff

self.get_last_history(conv_id)

def init_chats(self):

api = {"method":"list"}

results=self.execute_api(api)

chats = results['conversations']

for chat in chats:

buff = self.create_new_buffer(chat, chat['id'])

self.private_chans[chat['id']] = buff

def create_new_buffer(self, msg, conv_id):

channel = msg['channel']

name = channel['name']

if channel['members_type'] == 'impteamnative':

## We don't want to replicate our username

name_splitted = name.split(',')

## We can send message to ourself

if len(name_splitted) == 1:

buff_name = name

else:

buff_name = ",".join(name_splitted[1:])

ty="private"

elif channel['members_type'] == 'team':

buff_name = channel['name']+"#"+channel['topic_name']

ty="channel"

else:

buff_name ="un::"+channel['member_type']

ty="private"

buff = weechat.buffer_new(buff_name, "private_input_cb", conv_id, "private_close_cb", conv_id)

weechat.prnt("", "buffer!"+str(buff));

weechat.buffer_set(buff, "localvar_set_type", ty)

weechat.buffer_set(buff, "localvar_set_server", self.status_name)

weechat.buffer_set(buff, "localvar_set_no_log", "1")

weechat.buffer_set(buff, "localvar_set_conversation_id", conv_id)

weechat.buffer_set(buff, "localvar_set_first_message", "")

weechat.buffer_set(buff, "localvar_set_last_message", "")

weechat.buffer_set(buff, "unread", "")

weechat.buffer_set(buff, "notify", "2")

weechat.buffer_set(buff, "nicklist", "1")

# Get channel member

api = {"method": "listmembers", "params": {"options": {"conversation_id": conv_id}}}

r=self.execute_api(api)

nicklist = ["0|Owners", "1|Admins", "2|Writers", "3|Readers", "4|Bot", "5|Restricted Bot"]

group = [None]*6

j = 0

for i in nicklist:

group[int(j)] = weechat.nicklist_add_group(buff, "", i,"weechat.color.nicklist_group", 1)

j+=1

result = r

for i in result['owners']:

weechat.nicklist_add_nick(buff, group[0] , i['username'], "red", "@", "lightgreen", 1)

for i in result['admins']:

weechat.nicklist_add_nick(buff, group[1] , i['username'], "red", "@", "lightgreen", 1)

for i in result['writers']:

weechat.nicklist_add_nick(buff, group[2] , i['username'], "red", "@", "lightgreen", 1)

for i in result['readers']:

weechat.nicklist_add_nick(buff, group[3] , i['username'], "red", "@", "lightgreen", 1)

for i in result['bots']:

weechat.nicklist_add_nick(buff, group[4] , i['username'], "red", "@", "lightgreen", 1)

for i in result['restrictedBots']:

weechat.nicklist_add_nick(buff, group[5] , i['username'], "red", "@", "lightgreen", 1)