From d9cd83f6ca08e9ed91ca0cdafd23a5ecd671bae9 Mon Sep 17 00:00:00 2001
From: Denis Kasak <dkasak@users.noreply.github.com>
Date: Wed, 13 Feb 2019 16:31:08 +0100
Subject: Respect WEECHAT_HOME. (#6)

---
 edit.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/edit.py b/edit.py
index 65fcf42..71333dc 100644
--- a/edit.py
+++ b/edit.py
@@ -18,10 +18,15 @@ import subprocess
 import weechat
 
 
+def weechat_config_dir():
+    return os.path.expanduser(os.environ.get("WEECHAT_HOME", "~/.weechat/"))
+
+
 def edit(data, buf, args):
     editor = (weechat.config_get_plugin("editor") or
               os.environ.get("EDITOR", "vim -f"))
-    path = os.path.expanduser("~/.weechat/message.txt")
+    config_dir = weechat_config_dir()
+    path = os.path.join(config_dir, "message.txt")
     with open(path, "w+") as f:
         f.write(weechat.buffer_get_string(buf, "input"))
 
-- 
cgit v1.2.3-70-g09d2