#!/usr/bin/python

"""

A time recording tool.

Author: Eugen Sawin (sawine@me73.com)

Dependencies: libsqlite3-dev

"""

import os

import sys

import argparse

from datetime import datetime

import db

WD = "working_path"

CONFIG = {WD: str}

CONFIG_SEP = "="

DB_FILE = "cronrec.db"

config = {}

# Trying to get the user's home directory path

try: # Windows

	from win32com.shell import shellcon, shell

	HOMEDIR	= shell.SHGetFolderPath(0, shellcon.CSIDL_LOCAL_APPDATA, 0, 0)

except ImportError: # Linux (hopefully)

	HOMEDIR = os.path.expanduser("~")

CONFIG_FILE = "%s/.cronrecrc" % HOMEDIR

def read_config():

	config = {}

	with open(CONFIG_FILE, "r") as config_stream:

		config_lines = [l.split(CONFIG_SEP) for l in config_stream.readlines() 

						if CONFIG_SEP in l]

		for key, value in config_lines:

			key = key.strip().lower()

			value = value.strip()

			if key in CONFIG:

				config[key] = CONFIG[key](value)

	return config

def write_config(config):

	with open(CONFIG_FILE, "w") as config_input:

		config_input.write("\n".join([CONFIG_SEP.join((k, v)) 

							for k, v in config.iteritems() if k in CONFIG]))

def db_file():

	global config

	if WD not in config:

		print "Working directory path is not configured. \

Please use the init command."

		return None

	return config[WD] + "/" + DB_FILE

def init(args):

	global config

	last_wd = None

	if WD in config:

		last_wd = config[WD]

	config[WD] = args.working_path

	write_config(config)

	path_exists = os.path.exists(config[WD]) and os.path.isdir(config[WD])

	db_exists = path_exists and os.path.exists(db_file())\

				and os.path.isfile(db_file())

	if last_wd != config[WD]:		

		print "Changed working directory from %s to %s." % (last_wd, config[WD])

	else:

		print "Working directory remains at %s." % config[WD]

	if not path_exists:

		print "Warning: working directory %s does not exist." % config[WD]

		os.makedirs(config[WD])

		print "Working directory %s created." % config[WD]

	elif db_exists:

		print "Database file %s already exists, please delete it before \

initiating a new database at this location." % db_file()

	if not db_exists:

		db.init(db_file())	

def begin(args):

	project = None

	activity = None

	label = args.label.strip()

	if ":" in label:

		project, activity = label.split(":")

	else:

		project = label

	db.begin(db_file(), project, activity, datetime.now())	

	print "begins %s:%s" % (project, activity)

def end(args):

	project = None

	activity = None

	label = args.label.strip()

	if ":" in label:

		project, activity = label.split(":")

	else:

		project = label

	db.end(db_file(), project, activity, datetime.now())	

	print "ends %s:%s" % (project, activity)

def main():

	global config

	config = read_config()

	parser = argparse.ArgumentParser(description="Records time.")

	subs = parser.add_subparsers()

	sub_begin = subs.add_parser("init")

	sub_begin.add_argument("working_path", type=str)

	sub_begin.set_defaults(func=init)

	sub_begin = subs.add_parser("begin")

	sub_begin.add_argument("label", type=str)

	sub_begin.set_defaults(func=begin)

	sub_end = subs.add_parser("end")

	sub_end.add_argument("label", type=str)

	sub_end.set_defaults(func=end)

	args = parser.parse_args()

	args.func(args)

if __name__ == "__main__":

	main()