plugins/upload: Use embed directly instead of populating input field

Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
This commit is contained in:
Maxime “pep” Buquet 2020-04-12 16:45:21 +02:00
parent a72152c462
commit 57553abc0a
Signed by: pep
GPG key ID: DEDA74AEECA9D0F2
2 changed files with 25 additions and 11 deletions

View file

@ -28,14 +28,13 @@ class Plugin(BasePlugin):
help='Embed an image url into the contact\'s client',
usage='<image_url>')
def embed_image_url(self, args):
def embed_image_url(self, url):
tab = self.api.current_tab()
message = self.core.xmpp.make_message(tab.jid)
message['body'] = args
message['oob']['url'] = args
if isinstance(tab, tabs.MucTab):
message['type'] = 'groupchat'
else:
message['body'] = url
message['oob']['url'] = url
message['type'] = 'groupchat'
if not isinstance(tab, tabs.MucTab):
message['type'] = 'chat'
tab.add_message(
message['body'],
@ -46,3 +45,6 @@ class Plugin(BasePlugin):
typ=1,
)
message.send()
# TODO: Fix refreshing. The following doesn't work.
tab.refresh()
self.core.tab_win.refresh()

View file

@ -16,6 +16,9 @@ This plugin adds a command to the chat tabs.
"""
from typing import Optional
import asyncio
import traceback
from os.path import expanduser
@ -30,7 +33,11 @@ from poezio import tabs
class Plugin(BasePlugin):
dependencies = {'embed'}
def init(self):
self.embed = self.refs['embed']
if not self.core.xmpp['xep_0363']:
raise Exception('slixmpp XEP-0363 plugin failed to load')
for _class in (tabs.PrivateTab, tabs.StaticConversationTab, tabs.DynamicConversationTab, tabs.MucTab):
@ -43,18 +50,23 @@ class Plugin(BasePlugin):
short='Upload a file',
completion=self.completion_filename)
async def async_upload(self, filename):
async def upload(self, filename) -> Optional[str]:
try:
url = await self.core.xmpp['xep_0363'].upload_file(filename)
except UploadServiceNotFound:
self.api.information('HTTP Upload service not found.', 'Error')
return
return None
except Exception:
exception = traceback.format_exc()
self.api.information('Failed to upload file: %s' % exception,
'Error')
return
self.core.insert_input_text(url)
return None
return url
async def send_upload(self, filename):
url = await self.upload(filename)
if url is not None:
self.embed.embed_image_url(url)
@command_args_parser.quoted(1)
def command_upload(self, args):
@ -63,7 +75,7 @@ class Plugin(BasePlugin):
return
filename, = args
filename = expanduser(filename)
asyncio.ensure_future(self.async_upload(filename))
asyncio.ensure_future(self.send_upload(filename))
@staticmethod
def completion_filename(the_input):