fetch/hg.py: Properly format the command if the user URI parameter is empty.
[bitbake.git] / lib / bb / fetch / hg.py
1 # ex:ts=4:sw=4:sts=4:et
2 # -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
3 """
4 BitBake 'Fetch' implementation for mercurial DRCS (hg).
5
6 """
7
8 # Copyright (C) 2003, 2004  Chris Larson
9 # Copyright (C) 2004        Marcin Juszkiewicz
10 # Copyright (C) 2007        Robert Schuster
11 #
12 # This program is free software; you can redistribute it and/or modify
13 # it under the terms of the GNU General Public License version 2 as
14 # published by the Free Software Foundation.
15 #
16 # This program is distributed in the hope that it will be useful,
17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19 # GNU General Public License for more details.
20 #
21 # You should have received a copy of the GNU General Public License along
22 # with this program; if not, write to the Free Software Foundation, Inc.,
23 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
24 #
25 # Based on functions from the base bb module, Copyright 2003 Holger Schurig
26
27 import os, re
28 import sys
29 import bb
30 from bb import data
31 from bb.fetch import Fetch
32 from bb.fetch import FetchError
33 from bb.fetch import MissingParameterError
34 from bb.fetch import runfetchcmd
35
36 class Hg(Fetch):
37     """Class to fetch a from mercurial repositories"""
38     def supports(self, url, ud, d):
39         """
40         Check to see if a given url can be fetched with mercurial.
41         """
42         return ud.type in ['hg']
43
44     def localpath(self, url, ud, d):
45         if not "module" in ud.parm:
46             raise MissingParameterError("hg method needs a 'module' parameter")
47
48         ud.module = ud.parm["module"]
49
50         # Create paths to mercurial checkouts
51         relpath = ud.path
52         if relpath.startswith('/'):
53             # Remove leading slash as os.path.join can't cope
54             relpath = relpath[1:]
55         ud.pkgdir = os.path.join(data.expand('${HGDIR}', d), ud.host, relpath)
56         ud.moddir = os.path.join(ud.pkgdir, ud.module)
57
58         if 'rev' in ud.parm:
59             ud.revision = ud.parm['rev']
60
61         ud.localfile = data.expand('%s_%s_%s_%s.tar.gz' % (ud.module.replace('/', '.'), ud.host, ud.path.replace('/', '.'), ud.revision), d)
62
63         return os.path.join(data.getVar("DL_DIR", d, True), ud.localfile)
64
65     def _buildhgcommand(self, ud, d, command):
66         """
67         Build up an hg commandline based on ud
68         command is "fetch", "update", "info"
69         """
70
71         basecmd = data.expand('${FETCHCMD_hg}', d)
72
73         proto = "http"
74         if "proto" in ud.parm:
75             proto = ud.parm["proto"]
76
77         host = ud.host
78         if proto == "file":
79             host = "/"
80             ud.host = "localhost"
81
82         if not ud.user:
83             hgroot = host + ud.path
84         else:
85             hgroot = ud.user + "@" + host + ud.path
86
87         if command is "info":
88             return "%s identify -i %s://%s/%s" % (basecmd, proto, hgroot, ud.module)
89
90         options = [];
91         if ud.revision:
92             options.append("-r %s" % ud.revision)
93
94         if command is "fetch":
95             cmd = "%s clone %s %s://%s/%s %s" % (basecmd, " ".join(options), proto, hgroot, ud.module, ud.module)
96         elif command is "pull":
97             # do not pass options list; limiting pull to rev causes the local
98             # repo not to contain it and immediately following "update" command
99             # will crash
100             cmd = "%s pull" % (basecmd)
101         elif command is "update":
102             cmd = "%s update -C %s" % (basecmd, " ".join(options))
103         else:
104             raise FetchError("Invalid hg command %s" % command)
105
106         return cmd
107
108     def go(self, loc, ud, d):
109         """Fetch url"""
110
111         # try to use the tarball stash
112         if Fetch.try_mirror(d, ud.localfile):
113             bb.msg.debug(1, bb.msg.domain.Fetcher, "%s already exists or was mirrored, skipping hg checkout." % ud.localpath)
114             return
115
116         bb.msg.debug(2, bb.msg.domain.Fetcher, "Fetch: checking for module directory '" + ud.moddir + "'")
117
118         if os.access(os.path.join(ud.moddir, '.hg'), os.R_OK):
119             updatecmd = self._buildhgcommand(ud, d, "pull")
120             bb.msg.note(1, bb.msg.domain.Fetcher, "Update " + loc)
121             # update sources there
122             os.chdir(ud.moddir)
123             bb.msg.debug(1, bb.msg.domain.Fetcher, "Running %s" % updatecmd)
124             runfetchcmd(updatecmd, d)
125
126             updatecmd = self._buildhgcommand(ud, d, "update")
127             bb.msg.debug(1, bb.msg.domain.Fetcher, "Running %s" % updatecmd)
128             runfetchcmd(updatecmd, d)
129         else:
130             fetchcmd = self._buildhgcommand(ud, d, "fetch")
131             bb.msg.note(1, bb.msg.domain.Fetcher, "Fetch " + loc)
132             # check out sources there
133             bb.mkdirhier(ud.pkgdir)
134             os.chdir(ud.pkgdir)
135             bb.msg.debug(1, bb.msg.domain.Fetcher, "Running %s" % fetchcmd)
136             runfetchcmd(fetchcmd, d)
137
138         os.chdir(ud.pkgdir)
139         try:
140             runfetchcmd("tar -czf %s %s" % (ud.localpath, ud.module), d)
141         except:
142             t, v, tb = sys.exc_info()
143             try:
144                 os.unlink(ud.localpath)
145             except OSError:
146                 pass
147             raise t, v, tb