vim-rana/plugin/wakatime.vim

227 lines
6.3 KiB
VimL
Raw Normal View History

2013-06-25 07:57:15 +00:00
" ============================================================================
" File: wakatime.vim
" Description: Automatic time tracking for Vim.
2013-06-25 07:57:15 +00:00
" Maintainer: Wakati.Me <support@wakatime.com>
" ============================================================================
let s:VERSION = '0.2.1'
2013-06-26 04:09:52 +00:00
2013-06-25 07:57:15 +00:00
" Init {{{
" Check Vim version
if v:version < 700
echoerr "This plugin requires vim >= 7."
finish
2013-06-25 07:57:15 +00:00
endif
" Check for Python support
if !has('python')
echoerr "This plugin requires Vim to be compiled with Python support."
2013-06-25 07:57:15 +00:00
finish
endif
" Only load plugin once
if exists("g:loaded_wakatime")
finish
endif
let g:loaded_wakatime = 1
2013-06-25 07:57:15 +00:00
" Backup & Override cpoptions
let s:old_cpo = &cpo
set cpo&vim
2013-06-25 07:57:15 +00:00
" Set default away minutes
if !exists("g:wakatime_AwayMinutes")
let g:wakatime_AwayMinutes = 10
endif
" Set default action frequency in minutes
if !exists("g:wakatime_ActionFrequency")
let g:wakatime_ActionFrequency = 5
endif
" To be backwards compatible, rename config file
if filereadable(expand("$HOME/.wakatime"))
exec "silent !mv" expand("$HOME/.wakatime") expand("$HOME/.wakatime.conf")
endif
" Globals
let s:plugin_directory = expand("<sfile>:p:h") . '/'
let s:last_action = 0
let s:fresh = 1
2013-07-02 04:07:38 +00:00
" Import things python needs
python import time
python import vim
2013-06-25 07:57:15 +00:00
" }}}
2013-06-26 04:09:52 +00:00
2013-06-25 07:57:15 +00:00
" Function Definitions {{{
function! s:GetCurrentFile()
return expand("%:p")
endfunction
function! s:GetCurrentTime()
python vim.command('let current_time=%f' % time.time())
return current_time
endfunction
function! s:Api(targetFile, time, endtime, is_write, last)
let targetFile = a:targetFile
if targetFile == ''
let targetFile = a:last[2]
endif
if targetFile != ''
let cmd = ['python', s:plugin_directory . 'packages/wakatime/wakatime.py']
let cmd = cmd + ['--file', shellescape(targetFile)]
let cmd = cmd + ['--time', printf('%f', a:time)]
let cmd = cmd + ['--plugin', printf('vim-wakatime/%s', s:VERSION)]
if a:is_write
let cmd = cmd + ['--write']
endif
if a:endtime > 1
let cmd = cmd + ['--endtime', printf('%f', a:endtime)]
endif
2013-07-09 22:34:03 +00:00
"let cmd = cmd + ['--verbose']
exec 'silent !' . join(cmd, ' ') . ' &'
let time = a:time
if a:endtime > 1 && float2nr(round(time)) < float2nr(round(a:endtime))
let time = a:endtime
endif
call s:SetLastAction(time, time, targetFile)
endif
endfunction
function! s:GetLastAction()
if !filereadable(expand("$HOME/.wakatime.data"))
return [0.0, '', 0.0]
endif
let last = readfile(expand("$HOME/.wakatime.data"), '', 3)
if len(last) != 3
return [0.0, '', 0.0]
endif
return [str2float(last[0]), str2float(last[1]), last[2]]
endfunction
function! s:SetLastAction(time, last_update, targetFile)
let s:fresh = 0
call writefile([printf('%f', a:time), printf('%f', a:last_update), a:targetFile], expand("$HOME/.wakatime.data"))
endfunction
function! s:GetChar()
let c = getchar()
if c =~ '^\d\+$'
let c = nr2char(c)
endif
return c
endfunction
function! s:EnoughTimePassed(now, last)
let prev = a:last[0]
if a:now - prev > g:wakatime_ActionFrequency * 60
return 1
endif
return 0
endfunction
function! s:ShouldPromptUser(now, last)
let prev = a:last[1]
if s:fresh || prev < 1
return 0
endif
let duration = a:now - prev
if duration > g:wakatime_AwayMinutes * 60
return 1
endif
return 0
endfunction
2013-07-10 07:05:24 +00:00
function! s:Away(now, last)
let duration = a:now - a:last[1]
let units = 'seconds'
if duration > 59
let duration = round(duration / 60)
let units = 'minutes'
endif
if duration > 59
let duration = round(duration / 60)
let units = 'hours'
endif
if duration > 24
let duration = round(duration / 24)
let units = 'days'
endif
let answer = input(printf("You were away %.f %s. Add time to current file? (y/n)", duration, units))
if answer != "y"
return 1
endif
return 0
endfunction
2013-06-25 07:57:15 +00:00
" }}}
2013-06-26 04:09:52 +00:00
2013-06-25 07:57:15 +00:00
" Event Handlers {{{
function! s:normalAction()
let targetFile = s:GetCurrentFile()
let now = s:GetCurrentTime()
let last = s:GetLastAction()
if s:EnoughTimePassed(now, last) || targetFile != last[2]
if s:ShouldPromptUser(now, last)
if s:Away(now, last)
call s:Api(targetFile, now, last[0], 0, last)
else
call s:Api(targetFile, now, 0.0, 0, last)
endif
else
call s:Api(targetFile, now, last[0], 0, last)
endif
else
if now - last[1] > 5
call s:SetLastAction(last[0], now, targetFile)
endif
endif
endfunction
function! s:writeAction()
let targetFile = s:GetCurrentFile()
let now = s:GetCurrentTime()
let last = s:GetLastAction()
if s:EnoughTimePassed(now, last) || targetFile != last[2]
if s:ShouldPromptUser(now, last)
if s:Away(now, last)
call s:Api(targetFile, now, last[0], 1, last)
else
call s:Api(targetFile, now, 0.0, 1, last)
endif
else
call s:Api(targetFile, now, last[0], 1, last)
endif
else
2013-07-10 06:28:15 +00:00
call s:Api(targetFile, now, 0.0, 1, last)
endif
endfunction
2013-06-25 07:57:15 +00:00
" }}}
2013-06-26 04:09:52 +00:00
2013-06-25 07:57:15 +00:00
" Autocommand Events {{{
augroup Wakatime
autocmd!
autocmd BufEnter * call s:normalAction()
autocmd VimEnter * call s:normalAction()
autocmd BufWritePost * call s:writeAction()
autocmd CursorMoved,CursorMovedI * call s:normalAction()
augroup END
2013-06-25 07:57:15 +00:00
" }}}
2013-06-26 04:09:52 +00:00
2013-06-25 07:57:15 +00:00
" Restore cpoptions
let &cpo = s:old_cpo