Add proper GPLv2 headers to all BitBake files
[bitbake.git] / lib / bb / fetch / perforce.py
1 #!/usr/bin/env python
2 # ex:ts=4:sw=4:sts=4:et
3 # -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
4 """
5 BitBake 'Fetch' implementations
6
7 Classes for obtaining upstream sources for the
8 BitBake build tools.
9
10 """
11
12 # Copyright (C) 2003, 2004  Chris Larson
13 #
14 # This program is free software; you can redistribute it and/or modify
15 # it under the terms of the GNU General Public License version 2 as
16 # published by the Free Software Foundation.
17 #
18 # This program is distributed in the hope that it will be useful,
19 # but WITHOUT ANY WARRANTY; without even the implied warranty of
20 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21 # GNU General Public License for more details.
22 #
23 # You should have received a copy of the GNU General Public License along
24 # with this program; if not, write to the Free Software Foundation, Inc.,
25 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
26 #
27 # Based on functions from the base bb module, Copyright 2003 Holger Schurig
28
29 import os, re
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 Perforce(Fetch):
37     def supports(self, url, ud, d):
38         return ud.type in ['p4']
39
40     def doparse(url,d):
41         parm=[]
42         path = url.split("://")[1]
43         delim = path.find("@");
44         if delim != -1:
45             (user,pswd,host,port) = path.split('@')[0].split(":")
46             path = path.split('@')[1]
47         else:
48             (host,port) = data.getVar('P4PORT', d).split(':')
49             user = ""
50             pswd = ""
51
52         if path.find(";") != -1:
53             keys=[]
54             values=[]
55             plist = path.split(';')
56             for item in plist:
57                 if item.count('='):
58                     (key,value) = item.split('=')
59                     keys.append(key)
60                     values.append(value)
61
62             parm = dict(zip(keys,values))
63         path = "//" + path.split(';')[0]
64         host += ":%s" % (port)
65         parm["cset"] = Perforce.getcset(d, path, host, user, pswd, parm)
66
67         return host,path,user,pswd,parm
68     doparse = staticmethod(doparse)
69
70     def getcset(d, depot,host,user,pswd,parm):
71         if "cset" in parm:
72             return parm["cset"];
73         if user:
74             data.setVar('P4USER', user, d)
75         if pswd:
76             data.setVar('P4PASSWD', pswd, d)
77         if host:
78             data.setVar('P4PORT', host, d)
79
80         p4date = data.getVar("P4DATE", d, 1)
81         if "revision" in parm:
82             depot += "#%s" % (parm["revision"])
83         elif "label" in parm:
84             depot += "@%s" % (parm["label"])
85         elif p4date:
86             depot += "@%s" % (p4date)
87
88         p4cmd = data.getVar('FETCHCOMMAND_p4', d, 1)
89         bb.msg.debug(1, bb.msg.domain.Fetcher, "Running %s changes -m 1 %s" % (p4cmd, depot))
90         p4file = os.popen("%s changes -m 1 %s" % (p4cmd,depot))
91         cset = p4file.readline().strip()
92         bb.msg.debug(1, bb.msg.domain.Fetcher, "READ %s" % (cset))
93         if not cset:
94             return -1
95
96         return cset.split(' ')[1]
97     getcset = staticmethod(getcset)
98
99     def localpath(self, url, ud, d):        
100
101         (host,path,user,pswd,parm) = Perforce.doparse(url,d)
102
103         # If a label is specified, we use that as our filename
104
105         if "label" in parm:
106             ud.localfile = "%s.tar.gz" % (parm["label"])
107             return os.path.join(data.getVar("DL_DIR", d, 1), ud.localfile)
108
109         base = path
110         which = path.find('/...')
111         if which != -1:
112             base = path[:which]
113
114         if base[0] == "/":
115             base = base[1:]
116
117         cset = Perforce.getcset(d, path, host, user, pswd, parm)
118
119         ud.localfile = data.expand('%s+%s+%s.tar.gz' % (host,base.replace('/', '.'), cset), d)
120
121         return os.path.join(data.getVar("DL_DIR", d, 1), ud.localfile)
122
123     def go(self, loc, ud, d):
124         """
125         Fetch urls
126         """
127
128         # try to use the tarball stash
129         if not self.forcefetch(loc, ud, d) and Fetch.try_mirror(d, ud.localfile):
130             bb.msg.debug(1, bb.msg.domain.Fetcher, "%s already exists or was mirrored, skipping perforce checkout." % ud.localpath)
131             return
132
133         (host,depot,user,pswd,parm) = Perforce.doparse(loc, d)
134
135         if depot.find('/...') != -1:
136             path = depot[:depot.find('/...')]
137         else:
138             path = depot
139
140         if "module" in parm:
141             module = parm["module"]
142         else:
143             module = os.path.basename(path)
144
145         localdata = data.createCopy(d)
146         data.setVar('OVERRIDES', "p4:%s" % data.getVar('OVERRIDES', localdata), localdata)
147         data.update_data(localdata)
148
149         # Get the p4 command
150         if user:
151             data.setVar('P4USER', user, localdata)
152
153         if pswd:
154             data.setVar('P4PASSWD', pswd, localdata)
155
156         if host:
157             data.setVar('P4PORT', host, localdata)
158
159         p4cmd = data.getVar('FETCHCOMMAND', localdata, 1)
160
161         # create temp directory
162         bb.msg.debug(2, bb.msg.domain.Fetcher, "Fetch: creating temporary directory")
163         bb.mkdirhier(data.expand('${WORKDIR}', localdata))
164         data.setVar('TMPBASE', data.expand('${WORKDIR}/oep4.XXXXXX', localdata), localdata)
165         tmppipe = os.popen(data.getVar('MKTEMPDIRCMD', localdata, 1) or "false")
166         tmpfile = tmppipe.readline().strip()
167         if not tmpfile:
168             bb.error("Fetch: unable to create temporary directory.. make sure 'mktemp' is in the PATH.")
169             raise FetchError(module)
170
171         if "label" in parm:
172             depot = "%s@%s" % (depot,parm["label"])
173         else:
174             cset = Perforce.getcset(d, depot, host, user, pswd, parm)
175             depot = "%s@%s" % (depot,cset)
176
177         os.chdir(tmpfile)
178         bb.msg.note(1, bb.msg.domain.Fetcher, "Fetch " + loc)
179         bb.msg.note(1, bb.msg.domain.Fetcher, "%s files %s" % (p4cmd, depot))
180         p4file = os.popen("%s files %s" % (p4cmd, depot))
181
182         if not p4file:
183             bb.error("Fetch: unable to get the P4 files from %s" % (depot))
184             raise FetchError(module)
185
186         count = 0
187
188         for file in p4file: 
189             list = file.split()
190
191             if list[2] == "delete":
192                 continue
193
194             dest = list[0][len(path)+1:]
195             where = dest.find("#")
196
197             os.system("%s print -o %s/%s %s" % (p4cmd, module,dest[:where],list[0]))
198             count = count + 1
199             
200         if count == 0:
201             bb.error("Fetch:  No files gathered from the P4 fetch")
202             raise FetchError(module)
203
204         myret = os.system("tar -czf %s %s" % (ud.localpath, module))
205         if myret != 0:
206             try:
207                 os.unlink(ud.localpath)
208             except OSError:
209                 pass
210             raise FetchError(module)
211         # cleanup
212         os.system('rm -rf %s' % tmpfile)
213
214