'''

Created on Jun 9, 2014

@author: lzrak47

'''

import os

from termcolor import colored

from branch import Branch

from constants import GIT_DIR

from repository import Repository

from utils import get_all_files_in_dir, filter_by_gitignore, less_str

class Command(object):

'''

handle all commands

'''

@staticmethod

def cmd_init(workspace):

Repository.create_repository(workspace)

@staticmethod

def cmd_add(file):

if file == '.':

Repository().stage(filter_by_gitignore(get_all_files_in_dir('.', GIT_DIR)))

else:

Repository().stage([file])

@staticmethod

def cmd_rm(file, cached=False):

Repository().delete(file)

if not cached:

os.remove(file)

@staticmethod

def cmd_commit(msg):

Repository().commit(msg)

@staticmethod

def cmd_log(num, use_less=True):

res = Repository().show_log(num)

if use_less:

less_str(res)

else:

print res

@staticmethod

def cmd_status():

Repository().show_status()

@staticmethod

def cmd_branch(name, is_deleted=False):

b = Branch()

if not name:

for branch in b.get_all_branches():

print '* %s' % colored(branch, 'green') if branch == b.head_name else ' %s' % branch

elif is_deleted:

b.delete_branch(name)

else :

b.add_branch(name)

@staticmethod

def cmd_reset(commit_sha1, is_soft=False, is_hard=False):

repo = Repository()

pre_entries = dict(repo.index.entries)

repo.update_head_commit(commit_sha1)

if not is_soft:

repo.rebuild_index_from_commit(commit_sha1)

if is_hard:

repo.rebuild_working_tree(pre_entries)

@staticmethod

def cmd_checkout(branch):

b = Branch()

b.switch_branch(branch)

repo = Repository()

pre_entries = dict(repo.index.entries)

repo.rebuild_index_from_commit(repo.branch.head_commit)

repo.rebuild_working_tree(pre_entries)

@staticmethod

def cmd_diff(cached=False, use_less=True):

if cached:

res = Repository().diff_between_index_and_head_tree()

else:

res = Repository().diff_between_working_tree_and_index()

if use_less:

less_str(res)

else:

print res

@staticmethod

def cmd_push():

pass

@staticmethod

def cmd_clone():

pass