sublime-wakatime/sublime-wakatime.py

""" ==========================================================
File:        sublime-wakatime.py
Description: Automatic time tracking for Sublime Text 2 and 3.
Maintainer:  Wakati.Me <support@wakatime.com>
Website:     https://www.wakati.me/
==========================================================="""

__version__ = '0.2.1'

import time
import uuid
from os.path import expanduser, dirname, realpath, isfile
from subprocess import call, Popen

import sublime
import sublime_plugin


# Prompt user if no activity for this many minutes
AWAY_MINUTES = 5

# globals
PLUGIN_DIR = dirname(realpath(__file__))
API_CLIENT = '%s/packages/wakatime/wakatime.py' % PLUGIN_DIR
LAST_ACTION = 0
LAST_FILE = None

# To be backwards compatible, rename config file
if isfile(expanduser('~/.wakatime')):
    call([
        'mv',
        expanduser('~/.wakatime'),
        expanduser('~/.wakatime.conf')
    ])


def api(targetFile, timestamp, isWrite=False, endtime=None):
    global LAST_ACTION, LAST_FILE
    if not targetFile:
        targetFile = LAST_FILE
    if targetFile:
        cmd = ['python', API_CLIENT,
            '--file', targetFile,
            '--time', str('%f' % timestamp),
            '--plugin', 'sublime-wakatime/%s' % __version__,
        ]
        if isWrite:
            cmd.append('--write')
        if endtime:
            cmd.extend(['--endtime', str('%f' % endtime)])
        Popen(cmd)
        LAST_ACTION = timestamp
        if endtime and endtime > LAST_ACTION:
            LAST_ACTION = endtime
        LAST_FILE = targetFile


def away(now):
    if LAST_ACTION == 0:
        return False
    duration = now - LAST_ACTION
    if duration > AWAY_MINUTES * 60:
        duration = int(duration)
        units = 'seconds'
        if duration > 59:
            duration = int(duration / 60.0)
            units = 'minutes'
        if duration > 59:
            duration = int(duration / 60.0)
            units = 'hours'
        if duration > 24:
            duration = int(duration / 24.0)
            units = 'days'
        return sublime\
            .ok_cancel_dialog("You were away %d %s. Add time to current file?"\
            % (duration, units), 'Yes, log this time')


def enough_time_passed(now):
    return (now - LAST_ACTION >= 299)


class WakatimeListener(sublime_plugin.EventListener):

    def on_post_save(self, view):
        api(view.file_name(), time.time(), isWrite=True)

    def on_activated(self, view):
        now = time.time()
        targetFile = view.file_name()
        if enough_time_passed(now) or targetFile != LAST_FILE:
            if away(now):
                api(targetFile, LAST_ACTION, endtime=now)
            else:
                api(targetFile, now)

    def on_selection_modified(self, view):
        now = time.time()
        targetFile = view.file_name()
        if enough_time_passed(now) or targetFile != LAST_FILE:
            if away(now):
                api(targetFile, LAST_ACTION, endtime=now)
            else:
                api(targetFile, now)