203 lines
6.7 KiB
Python
203 lines
6.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
# Copyright © 2018, 2019 Damir Jelić <poljar@termina.org.uk>
|
||
|
|
#
|
||
|
|
# Permission to use, copy, modify, and/or distribute this software for
|
||
|
|
# any purpose with or without fee is hereby granted, provided that the
|
||
|
|
# above copyright notice and this permission notice appear in all copies.
|
||
|
|
#
|
||
|
|
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||
|
|
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||
|
|
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
|
||
|
|
# SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
|
||
|
|
# RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
|
||
|
|
# CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
|
||
|
|
# CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||
|
|
|
||
|
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
from . import globals as G
|
||
|
|
from .globals import SERVERS, W
|
||
|
|
from .utf import utf8_decode
|
||
|
|
|
||
|
|
|
||
|
|
@utf8_decode
|
||
|
|
def matrix_bar_item_plugin(data, item, window, buffer, extra_info):
|
||
|
|
# pylint: disable=unused-argument
|
||
|
|
for server in SERVERS.values():
|
||
|
|
if buffer in server.buffers.values() or buffer == server.server_buffer:
|
||
|
|
return "matrix{color}/{color_fg}{name}".format(
|
||
|
|
color=W.color("bar_delim"),
|
||
|
|
color_fg=W.color("bar_fg"),
|
||
|
|
name=server.name,
|
||
|
|
)
|
||
|
|
|
||
|
|
ptr_plugin = W.buffer_get_pointer(buffer, "plugin")
|
||
|
|
name = W.plugin_get_name(ptr_plugin)
|
||
|
|
|
||
|
|
return name
|
||
|
|
|
||
|
|
|
||
|
|
@utf8_decode
|
||
|
|
def matrix_bar_item_name(data, item, window, buffer, extra_info):
|
||
|
|
# pylint: disable=unused-argument
|
||
|
|
for server in SERVERS.values():
|
||
|
|
if buffer in server.buffers.values():
|
||
|
|
color = (
|
||
|
|
"status_name_ssl"
|
||
|
|
if server.ssl_context.check_hostname
|
||
|
|
else "status_name"
|
||
|
|
)
|
||
|
|
|
||
|
|
room_buffer = server.find_room_from_ptr(buffer)
|
||
|
|
room = room_buffer.room
|
||
|
|
|
||
|
|
return "{color}{name}".format(
|
||
|
|
color=W.color(color), name=room.display_name
|
||
|
|
)
|
||
|
|
|
||
|
|
if buffer == server.server_buffer:
|
||
|
|
color = (
|
||
|
|
"status_name_ssl"
|
||
|
|
if server.ssl_context.check_hostname
|
||
|
|
else "status_name"
|
||
|
|
)
|
||
|
|
|
||
|
|
return "{color}server{del_color}[{color}{name}{del_color}]".format(
|
||
|
|
color=W.color(color),
|
||
|
|
del_color=W.color("bar_delim"),
|
||
|
|
name=server.name,
|
||
|
|
)
|
||
|
|
|
||
|
|
name = W.buffer_get_string(buffer, "name")
|
||
|
|
|
||
|
|
return "{}{}".format(W.color("status_name"), name)
|
||
|
|
|
||
|
|
|
||
|
|
@utf8_decode
|
||
|
|
def matrix_bar_item_lag(data, item, window, buffer, extra_info):
|
||
|
|
# pylint: disable=unused-argument
|
||
|
|
for server in SERVERS.values():
|
||
|
|
if buffer in server.buffers.values() or buffer == server.server_buffer:
|
||
|
|
if server.lag >= G.CONFIG.network.lag_min_show:
|
||
|
|
color = W.color("irc.color.item_lag_counting")
|
||
|
|
if server.lag_done:
|
||
|
|
color = W.color("irc.color.item_lag_finished")
|
||
|
|
|
||
|
|
lag = "{0:.3f}" if round(server.lag) < 1000 else "{0:.0f}"
|
||
|
|
lag_string = "Lag: {color}{lag}{ncolor}".format(
|
||
|
|
lag=lag.format((server.lag / 1000)),
|
||
|
|
color=color,
|
||
|
|
ncolor=W.color("reset"),
|
||
|
|
)
|
||
|
|
return lag_string
|
||
|
|
return ""
|
||
|
|
|
||
|
|
return ""
|
||
|
|
|
||
|
|
|
||
|
|
@utf8_decode
|
||
|
|
def matrix_bar_item_buffer_modes(data, item, window, buffer, extra_info):
|
||
|
|
# pylint: disable=unused-argument
|
||
|
|
for server in SERVERS.values():
|
||
|
|
if buffer in server.buffers.values():
|
||
|
|
room_buffer = server.find_room_from_ptr(buffer)
|
||
|
|
room = room_buffer.room
|
||
|
|
modes = []
|
||
|
|
|
||
|
|
if room.encrypted:
|
||
|
|
modes.append(G.CONFIG.look.encrypted_room_sign)
|
||
|
|
|
||
|
|
if (server.client
|
||
|
|
and server.client.room_contains_unverified(room.room_id)):
|
||
|
|
modes.append(G.CONFIG.look.encryption_warning_sign)
|
||
|
|
|
||
|
|
if not server.connected or not server.client.logged_in:
|
||
|
|
modes.append(G.CONFIG.look.disconnect_sign)
|
||
|
|
|
||
|
|
if room_buffer.backlog_pending or server.busy:
|
||
|
|
modes.append(G.CONFIG.look.busy_sign)
|
||
|
|
|
||
|
|
return "".join(modes)
|
||
|
|
|
||
|
|
return ""
|
||
|
|
|
||
|
|
|
||
|
|
@utf8_decode
|
||
|
|
def matrix_bar_nicklist_count(data, item, window, buffer, extra_info):
|
||
|
|
# pylint: disable=unused-argument
|
||
|
|
color = W.color("status_nicklist_count")
|
||
|
|
|
||
|
|
for server in SERVERS.values():
|
||
|
|
if buffer in server.buffers.values():
|
||
|
|
room_buffer = server.find_room_from_ptr(buffer)
|
||
|
|
room = room_buffer.room
|
||
|
|
return "{}{}".format(color, room.member_count)
|
||
|
|
|
||
|
|
nicklist_enabled = bool(W.buffer_get_integer(buffer, "nicklist"))
|
||
|
|
|
||
|
|
if nicklist_enabled:
|
||
|
|
nick_count = W.buffer_get_integer(buffer, "nicklist_visible_count")
|
||
|
|
return "{}{}".format(color, nick_count)
|
||
|
|
|
||
|
|
return ""
|
||
|
|
|
||
|
|
|
||
|
|
@utf8_decode
|
||
|
|
def matrix_bar_typing_notices_cb(data, item, window, buffer, extra_info):
|
||
|
|
"""Update a status bar item showing users currently typing.
|
||
|
|
This function is called by weechat every time a buffer is switched or
|
||
|
|
W.bar_item_update(<item>) is explicitly called. The bar item shows
|
||
|
|
currently typing users for the current buffer."""
|
||
|
|
# pylint: disable=unused-argument
|
||
|
|
for server in SERVERS.values():
|
||
|
|
if buffer in server.buffers.values():
|
||
|
|
room_buffer = server.find_room_from_ptr(buffer)
|
||
|
|
room = room_buffer.room
|
||
|
|
|
||
|
|
if room.typing_users:
|
||
|
|
nicks = []
|
||
|
|
|
||
|
|
for user_id in room.typing_users:
|
||
|
|
if user_id == room.own_user_id:
|
||
|
|
continue
|
||
|
|
|
||
|
|
nick = room_buffer.displayed_nicks.get(user_id, user_id)
|
||
|
|
nicks.append(nick)
|
||
|
|
|
||
|
|
if not nicks:
|
||
|
|
return ""
|
||
|
|
|
||
|
|
msg = "{}{}".format(
|
||
|
|
G.CONFIG.look.bar_item_typing_notice_prefix,
|
||
|
|
", ".join(sorted(nicks))
|
||
|
|
)
|
||
|
|
|
||
|
|
max_len = G.CONFIG.look.max_typing_notice_item_length
|
||
|
|
if len(msg) > max_len:
|
||
|
|
msg[:max_len - 3] + "..."
|
||
|
|
|
||
|
|
return msg
|
||
|
|
|
||
|
|
return ""
|
||
|
|
|
||
|
|
return ""
|
||
|
|
|
||
|
|
|
||
|
|
def init_bar_items():
|
||
|
|
W.bar_item_new("(extra)buffer_plugin", "matrix_bar_item_plugin", "")
|
||
|
|
W.bar_item_new("(extra)buffer_name", "matrix_bar_item_name", "")
|
||
|
|
W.bar_item_new("(extra)lag", "matrix_bar_item_lag", "")
|
||
|
|
W.bar_item_new(
|
||
|
|
"(extra)buffer_nicklist_count",
|
||
|
|
"matrix_bar_nicklist_count",
|
||
|
|
""
|
||
|
|
)
|
||
|
|
W.bar_item_new(
|
||
|
|
"(extra)matrix_typing_notice",
|
||
|
|
"matrix_bar_typing_notices_cb",
|
||
|
|
""
|
||
|
|
)
|
||
|
|
W.bar_item_new("(extra)buffer_modes", "matrix_bar_item_buffer_modes", "")
|
||
|
|
W.bar_item_new("(extra)matrix_modes", "matrix_bar_item_buffer_modes", "")
|