matrix/weechat-matrix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

commands: Modify the redact command to use event ids.

Browse source 
This closes #32.
This commit is contained in:
Damir Jelić 2019-01-29 15:12:37 +01:00
parent 67f3b1883a
commit 7a908e79d0
2 changed files with 87 additions and 79 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

86
matrix/commands.py
View file

@ -21,10 +21,11 @@ import re
from builtins import str
from future.moves.itertools import zip_longest
from collections import defaultdict
from functools import partial
from . import globals as G
from .colors import Formatted
from .globals import SERVERS, W, UPLOADS
from .globals import SERVERS, W, UPLOADS, SCRIPT_NAME
from .server import MatrixServer
from .utf import utf8_decode
from .utils import key_from_value, tags_from_line_data
@ -232,12 +233,10 @@ def hook_commands():
"redact",
"redact messages",
# Synopsis
('<message-number>[:"<message-part>"] [<reason>]'),
('<event-id>[:"<message-part>"] [<reason>]'),
# Description
(
"message-number: number of message to redact "
"(starting from 1 for\n"
" the last message received, counting up)\n"
"event-id: event id of the message that will be redacted\n"
" message-part: an initial part of the message (ignored, only "
"used\n"
" as visual feedback when using completion)\n"
@ -1081,47 +1080,32 @@ def matrix_kick_command_cb(data, buffer, args):
return W.WEECHAT_RC_OK
def event_id_from_line(buf, target_number):
# type: (str, int) -> str
own_lines = W.hdata_pointer(W.hdata_get("buffer"), buf, "own_lines")
if own_lines:
line = W.hdata_pointer(W.hdata_get("lines"), own_lines, "last_line")
line_number = 1
while line:
line_data = W.hdata_pointer(W.hdata_get("line"), line, "data")
if line_data:
tags = tags_from_line_data(line_data)
# Only count non redacted user messages
if (
"matrix_message" in tags
and "matrix_redacted" not in tags
and "matrix_new_redacted" not in tags
):
if line_number == target_number:
for tag in tags:
if tag.startswith("matrix_id"):
event_id = tag[10:]
return event_id
line_number += 1
line = W.hdata_move(W.hdata_get("line"), line, -1)
return ""
@utf8_decode
def matrix_redact_command_cb(data, buffer, args):
def already_redacted(line):
if SCRIPT_NAME + "_redacted" in line.tags:
return True
return False
def predicate(event_id, line):
event_tag = SCRIPT_NAME + "_id_{}".format(event_id)
tags = line.tags
if event_tag in tags:
return True
return False
for server in SERVERS.values():
if buffer in server.buffers.values():
room_buffer = server.find_room_from_ptr(buffer)
matches = re.match(r"(\d+)(:\".*\")? ?(.*)?", args)
matches = re.match(
r"^(\$[a-zA-Z0-9]+:([a-z0-9])(([a-z0-9-]{1,61})?[a-z0-9]{1})?"
r"(\.[a-z0-9](([a-z0-9-]{1,61})?[a-z0-9]{1})?)?"
r"(\.[a-zA-Z]{2,4})+)(:\".*\")? ?(.*)?$",
args
)
if not matches:
message = (
@ -1131,17 +1115,21 @@ def matrix_redact_command_cb(data, buffer, args):
W.prnt("", message)
return W.WEECHAT_RC_ERROR
line_string, _, reason = matches.groups()
line = int(line_string)
groups = matches.groups()
event_id, reason = (groups[0], groups[-1])
event_id = event_id_from_line(buffer, line)
lines = room_buffer.weechat_buffer.find_lines(
partial(predicate, event_id), max_lines=1
)
if not event_id:
message = (
"{prefix}matrix: No such message with number "
"{number} found"
).format(prefix=W.prefix("error"), number=line)
W.prnt("", message)
if not lines:
room_buffer.error(
"No such message with event id "
"{event_id} found.".format(event_id=event_id))
return W.WEECHAT_RC_OK
if already_redacted(lines[0]):
room_buffer.error("Event already redacted.")
return W.WEECHAT_RC_OK
server.room_send_redaction(room_buffer, event_id, reason)

64
matrix/completion.py
View file

@ -16,7 +16,8 @@
from __future__ import unicode_literals
from matrix.globals import SERVERS, W
from typing import List, Optional
from matrix.globals import SERVERS, W, SCRIPT_NAME
from matrix.utf import utf8_decode
from matrix.utils import tags_from_line_data
from nio import LocalProtocolError
@ -99,42 +100,61 @@ REDACTION_COMP_LEN = 50
@utf8_decode
def matrix_message_completion_cb(data, completion_item, buffer, completion):
own_lines = W.hdata_pointer(W.hdata_get("buffer"), buffer, "own_lines")
if own_lines:
line = W.hdata_pointer(W.hdata_get("lines"), own_lines, "last_line")
max_events = 500
line_number = 1
def redacted_or_not_message(tags):
# type: (List[str]) -> bool
if SCRIPT_NAME + "_redacted" in tags:
return True
if SCRIPT_NAME + "_message" not in tags:
return True
while line:
line_data = W.hdata_pointer(W.hdata_get("line"), line, "data")
return False
if line_data:
message = W.hdata_string(
W.hdata_get("line_data"), line_data, "message"
)
def event_id_from_tags(tags):
# type: (List[str]) -> Optional[str]
for tag in tags:
if tag.startswith("matrix_id"):
event_id = tag[10:]
return event_id
tags = tags_from_line_data(line_data)
return None
# Only add non redacted user messages to the completion
if (
message
and "matrix_message" in tags
and "matrix_redacted" not in tags
):
for server in SERVERS.values():
if buffer in server.buffers.values():
room_buffer = server.find_room_from_ptr(buffer)
lines = room_buffer.weechat_buffer.lines
added = 0
for line in lines:
tags = line.tags
if redacted_or_not_message(tags):
continue
event_id = event_id_from_tags(tags)
if not event_id:
continue
message = line.message
if len(message) > REDACTION_COMP_LEN + 2:
message = message[:REDACTION_COMP_LEN] + ".."
item = ('{number}:"{message}"').format(
number=line_number, message=message
item = ('{event_id}:"{message}"').format(
event_id=event_id, message=message
)
W.hook_completion_list_add(
completion, item, 0, W.WEECHAT_LIST_POS_END
)
line_number += 1
added += 1
line = W.hdata_move(W.hdata_get("line"), line, -1)
if added >= max_events:
break
return W.WEECHAT_RC_OK
return W.WEECHAT_RC_OK