unify mirror support and make it independant of the fetcher
[bitbake.git] / lib / bb / fetch / cvs.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' implementations
5
6 Classes for obtaining upstream sources for the
7 BitBake build tools.
8
9 """
10
11 # Copyright (C) 2003, 2004  Chris Larson
12 #
13 # This program is free software; you can redistribute it and/or modify
14 # it under the terms of the GNU General Public License version 2 as
15 # published by the Free Software Foundation.
16 #
17 # This program is distributed in the hope that it will be useful,
18 # but WITHOUT ANY WARRANTY; without even the implied warranty of
19 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20 # GNU General Public License for more details.
21 #
22 # You should have received a copy of the GNU General Public License along
23 # with this program; if not, write to the Free Software Foundation, Inc.,
24 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
25 #
26 #Based on functions from the base bb module, Copyright 2003 Holger Schurig
27 #
28
29 import os
30 import bb
31 from   bb import data
32 from   bb.fetch import Fetch
33 from   bb.fetch import FetchError
34 from   bb.fetch import MissingParameterError
35
36 class Cvs(Fetch):
37     """
38     Class to fetch a module or modules from cvs repositories
39     """
40     def supports(self, url, ud, d):
41         """
42         Check to see if a given url can be fetched with cvs.
43         """
44         return ud.type in ['cvs']
45
46     def localpath(self, url, ud, d):
47         if not "module" in ud.parm:
48             raise MissingParameterError("cvs method needs a 'module' parameter")
49         ud.module = ud.parm["module"]
50
51         ud.tag = ""
52         if 'tag' in ud.parm:
53             ud.tag = ud.parm['tag']
54
55         # Override the default date in certain cases
56         if 'date' in ud.parm:
57             ud.date = ud.parm['date']
58         elif ud.tag:
59             ud.date = ""
60
61         norecurse = ''
62         if 'norecurse' in ud.parm:
63             norecurse = '_norecurse'
64
65         fullpath = ''
66         if 'fullpath' in ud.parm:
67             fullpath = '_fullpath'
68
69         ud.localfile = data.expand('%s_%s_%s_%s%s%s.tar.gz' % (ud.module.replace('/', '.'), ud.host, ud.tag, ud.date, norecurse, fullpath), d)
70
71         return os.path.join(data.getVar("DL_DIR", d, True), ud.localfile)
72
73     def forcefetch(self, url, ud, d):
74         if (ud.date == "now"):
75             return True
76         return False
77
78     def go(self, loc, ud, d):
79
80         method = "pserver"
81         if "method" in ud.parm:
82             method = ud.parm["method"]
83
84         localdir = ud.module
85         if "localdir" in ud.parm:
86             localdir = ud.parm["localdir"]
87
88         cvs_port = ""
89         if "port" in ud.parm:
90             cvs_port = ud.parm["port"]
91
92         cvs_rsh = None
93         if method == "ext":
94             if "rsh" in ud.parm:
95                 cvs_rsh = ud.parm["rsh"]
96
97         if method == "dir":
98             cvsroot = ud.path
99         else:
100             cvsroot = ":" + method
101             cvsproxyhost = data.getVar('CVS_PROXY_HOST', d, True)
102             if cvsproxyhost:
103                 cvsroot += ";proxy=" + cvsproxyhost
104             cvsproxyport = data.getVar('CVS_PROXY_PORT', d, True)
105             if cvsproxyport:
106                 cvsroot += ";proxyport=" + cvsproxyport
107             cvsroot += ":" + ud.user
108             if ud.pswd:
109                 cvsroot += ":" + ud.pswd
110             cvsroot += "@" + ud.host + ":" + cvs_port + ud.path
111
112         options = []
113         if 'norecurse' in ud.parm:
114             options.append("-l")
115         if ud.date:
116             # treat YYYYMMDDHHMM specially for CVS
117             if len(ud.date) == 12:
118                 options.append("-D \"%s %s:%s UTC\"" % (ud.date[0:8], ud.date[8:10], ud.date[10:12]))
119             else:
120                 options.append("-D \"%s UTC\"" % ud.date)
121         if ud.tag:
122             options.append("-r %s" % ud.tag)
123
124         localdata = data.createCopy(d)
125         data.setVar('OVERRIDES', "cvs:%s" % data.getVar('OVERRIDES', localdata), localdata)
126         data.update_data(localdata)
127
128         data.setVar('CVSROOT', cvsroot, localdata)
129         data.setVar('CVSCOOPTS', " ".join(options), localdata)
130         data.setVar('CVSMODULE', ud.module, localdata)
131         cvscmd = data.getVar('FETCHCOMMAND', localdata, 1)
132         cvsupdatecmd = data.getVar('UPDATECOMMAND', localdata, 1)
133
134         if cvs_rsh:
135             cvscmd = "CVS_RSH=\"%s\" %s" % (cvs_rsh, cvscmd)
136             cvsupdatecmd = "CVS_RSH=\"%s\" %s" % (cvs_rsh, cvsupdatecmd)
137
138         # create module directory
139         bb.msg.debug(2, bb.msg.domain.Fetcher, "Fetch: checking for module directory")
140         pkg = data.expand('${PN}', d)
141         pkgdir = os.path.join(data.expand('${CVSDIR}', localdata), pkg)
142         moddir = os.path.join(pkgdir,localdir)
143         if os.access(os.path.join(moddir,'CVS'), os.R_OK):
144             bb.msg.note(1, bb.msg.domain.Fetcher, "Update " + loc)
145             # update sources there
146             os.chdir(moddir)
147             myret = os.system(cvsupdatecmd)
148         else:
149             bb.msg.note(1, bb.msg.domain.Fetcher, "Fetch " + loc)
150             # check out sources there
151             bb.mkdirhier(pkgdir)
152             os.chdir(pkgdir)
153             bb.msg.debug(1, bb.msg.domain.Fetcher, "Running %s" % cvscmd)
154             myret = os.system(cvscmd)
155
156         if myret != 0 or not os.access(moddir, os.R_OK):
157             try:
158                 os.rmdir(moddir)
159             except OSError:
160                  pass
161             raise FetchError(ud.module)
162
163         # tar them up to a defined filename
164         if 'fullpath' in ud.parm:
165             os.chdir(pkgdir)
166             myret = os.system("tar -czf %s %s" % (ud.localpath, localdir))
167         else:
168             os.chdir(moddir)
169             os.chdir('..')
170             myret = os.system("tar -czf %s %s" % (ud.localpath, os.path.basename(moddir)))
171
172         if myret != 0:
173             try:
174                 os.unlink(ud.localpath)
175             except OSError:
176                 pass
177             raise FetchError(ud.module)