# Fedora Developer Shell
#
# 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; version 2 of the License.
#
# 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 Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
# Authors: Yaakov M. Nemoy <ynemoy@redhat.com>
#
from __future__ import with_statement

from os import getcwd

from base.base import log
from base.exceptions import ExecutionException
from base.dirfactory import DirFactory
from base.util import pwd

from modules.darcs import Darcs
from modules.hackage import Hackage
from modules.port import Port
from modules.sourceball import SourceBall

class HaskellPort(Port):
    def __init__(self, package=None):
        super(HaskellPort, self).__init__()
        if not package:
            package = getcwd()
        self.pkg = DirFactory(package)
        self.hackage = Hackage()

    def add_sourceball(self, sourceball):
        '''copies a tarball into the package
        
        tarball is a path to some tarball
        '''
        with pwd(self.pkg.dir):
            pkg_src = SourceBall('', sourceball)
            pkg_src.cfg['buildsystem'] = 'cabal'
            name = pkg_src.name
            self.pkg.add_source(name)
        return self.close_later(pkg_src)
    
    def add_darcs(self, url, tgt, *args):
        '''creates a darcs variant of a cabal package using darcs source

        url is a url to some darcs repo
        tgt is the local name of the darcs repo
        '''
        with pwd(self.pkg.dir):
            pkg_src = Darcs(tgt, url, *args)
            pkg_src.cfg['buildsystem'] = 'cabal'
            name = pkg_src.name
            self.pkg.add_source(name)
        return self.close_later(pkg_src)

    def install_tag(self, tag):
        '''assuming a package that supports tagging, install a specific tag

        tag is the name of the tag in the DVCS
        '''
        with pwd(self.package.dir):
            with self.package.tag(tag):
                self.install()

    def install_sourceball(self, tarball, target='home', *args):
        '''given a tarball, copy it in and install it
        '''
        pkg_src = self.add_sourceball(tarball)
        Cabal(pkg_src).install_source(target, *args)

    def add_from_hackage(self, pkg, ver):
        '''get a specific package from hackage
        
        pkg is the name of the package desired
        ver is the version wanted
        '''
        sb_loc = self.hackage.url(pkg, ver)
        return self.add_sourceball(sb_loc)

    def add_latest(self, pkg):
        '''get the latest version of a package from hackage
        
        pkg is the package desired
        '''
        ver = self.hackage.latest_version(pkg)
        return self.add_from_hackage(pkg, ver)

    def install_from_hackage(self, pkg, ver, target='home', *args):
        '''get and install a specific package from hackage
        
        pkg is the desired package
        ver is the version wanted
        target is the location to install to, either 'home' or 'root'
        '''
        sb_loc = self.hackage.url(pkg, ver)
        self.install_sourceball(sb_loc, target, *args)

    def install_latest(self, pkg, target='home', *args):
        '''get and install the latest version of a package from hackage'''
        ver = self.hackage.latest_version(pkg)
        self.install_from_hackage(pkg, ver, target, *args)

    def add_upstream(self, pkg, tgt=None, *args):
        if not tgt:
            tgt = pkg
        return self.add_darcs(self.hackage.darcs_url(pkg), tgt, *args)

    def install_upstream(self, pkg, tgt=None, target='home', *args):
        pkg_src = self.add_upstream(pkg, tgt, *args)
        Cabal(pkg_src).install_source(target)

    def close(self):
        super(HaskellPort, self).close()
        self.pkg.close()