From c0fcdc52a8fff23bd0d1337f1339a94f1098508f Mon Sep 17 00:00:00 2001
From: Alan Hamlett <alan.hamlett@gmail.com>
Date: Thu, 13 Apr 2017 23:04:40 -0700
Subject: [PATCH] support  env variable

---
 plugin/wakatime.vim | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/plugin/wakatime.vim b/plugin/wakatime.vim
index 8e3b588..ae59d22 100644
--- a/plugin/wakatime.vim
+++ b/plugin/wakatime.vim
@@ -32,9 +32,13 @@ let s:VERSION = '4.0.14'
     set cpo&vim
 
     " Script Globals
+    let s:home = expand("$WAKATIME_HOME")
+    if s:home == '$WAKATIME_HOME'
+        let s:home = expand("$HOME")
+    endif
     let s:cli_location = expand("<sfile>:p:h") . '/packages/wakatime/cli.py'
-    let s:config_file = expand("$HOME/.wakatime.cfg")
-    let s:data_file = expand("$HOME/.wakatime.data")
+    let s:config_file = s:home . '/.wakatime.cfg'
+    let s:data_file = s:home . '/.wakatime.data'
     let s:config_file_already_setup = s:false
     let s:debug_mode_already_setup = s:false
     let s:is_debug_mode_on = s:false