weechat/python/autoload/editor.py @ 3ee7f5f0db6b
Ffucking scrreens
author |
Steve Losh <steve@stevelosh.com> |
date |
Sun, 16 Dec 2018 15:23:34 -0500 |
parents |
4237c4d29d58 |
children |
7023c500dd7a |
import subprocess
import os
import tempfile
SCRIPT_NAME = 'editor'
SCRIPT_AUTHOR = 'Steve Losh <steve@stevelosh.com>'
SCRIPT_VERSION = '1.0'
SCRIPT_LICENSE = 'MIT/X11'
SCRIPT_DESC = 'Launch an external editor to compose a message'
SCRIPT_COMMAND = 'editor'
import_ok = True
EDITOR = os.environ.get('EDITOR','vim')
try:
import weechat
except ImportError:
print 'This is a weechat script, what are you doing, run it in weechat, jesus'
import_ok = False
weechat_version = 0
def get_data(suffix, initial_data):
with tempfile.NamedTemporaryFile(suffix=".%s" % suffix, mode="w+") as tf:
tf.write(initial_data)
tf.flush()
if subprocess.call([EDITOR, tf.name]) != 0:
return None
# Reopen, because most editors do atomic write-tmp+rename saves which
# fucks with Python here.
tf.file.close()
tf = file(tf.name)
return tf.read()
def editor(data, buffer, args):
suffix = args or "tmp"
line = weechat.buffer_get_string(buffer, "input")
data = get_data(suffix, line)
if data:
weechat.command(buffer, "/input delete_line")
weechat.command(buffer, data.strip())
weechat.command("", "/window refresh")
return weechat.WEECHAT_RC_OK
if __name__ == '__main__' and import_ok:
if weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION,
SCRIPT_LICENSE, SCRIPT_DESC, '', ''):
weechat_version = weechat.info_get('version_number', '') or 0
weechat.hook_command(
SCRIPT_COMMAND,
'Open $EDITOR to compose a message',
'[file-extension]',
'If an argument is given, it will be used as the extension for the temporary file.',
'',
'editor',
'')