#!/usr/bin/env python
"""Mercurial Version Control Module/Wrapper"""

#   Xnt -- A Wrapper Build Tool
#   Copyright (C) 2013  Kenny Ballou

#   This program is free software: you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.

#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU General Public License for more details.

#   You should have received a copy of the GNU General Public License
#   along with this program.  If not, see <http://www.gnu.org/licenses/>.

import os
import xnt.tasks
import xnt.vcs

def hgclone(url, dest=None, rev=None, branch=None):
    """Clone a Mercurial Repository

    :param url: URI of repository to clone
    :param dest: Directory or name of cloned repository
    :param rev: Revision to clone
    :param branch: Branch to clone
    """
    assert xnt.tasks.which("hg")
    command = ["hg", "clone"]
    if rev:
        command.append("--rev")
        command.append(rev)
    command = xnt.vcs.clone_options(command, url, branch, dest)
    xnt.tasks.call(command)

def hgfetch(path, source='default'):
    """Pull and Update an already cloned Mercurial Repository

    :param path: Directory to the repository for which to pull changes
    :param source: Repository's upstream source
    """
    assert xnt.tasks.which("hg")
    command = ["hg", "pull", "-u", source]
    cwd = os.getcwd()
    os.chdir(path)
    xnt.tasks.call(command)
    os.chdir(cwd)