forked from brl/citadel
210 lines
7.4 KiB
Python
210 lines
7.4 KiB
Python
|
# ex:ts=4:sw=4:sts=4:et
|
||
|
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
|
||
|
"""
|
||
|
BitBake 'Fetch' implementation for perforce
|
||
|
|
||
|
"""
|
||
|
|
||
|
# Copyright (C) 2003, 2004 Chris Larson
|
||
|
# Copyright (C) 2016 Kodak Alaris, Inc.
|
||
|
#
|
||
|
# This program is free software; you can redistribute it and/or modify
|
||
|
# it under the terms of the GNU General Public License version 2 as
|
||
|
# published by the Free Software Foundation.
|
||
|
#
|
||
|
# 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, write to the Free Software Foundation, Inc.,
|
||
|
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
||
|
#
|
||
|
# Based on functions from the base bb module, Copyright 2003 Holger Schurig
|
||
|
|
||
|
import os
|
||
|
import logging
|
||
|
import bb
|
||
|
from bb.fetch2 import FetchMethod
|
||
|
from bb.fetch2 import FetchError
|
||
|
from bb.fetch2 import logger
|
||
|
from bb.fetch2 import runfetchcmd
|
||
|
|
||
|
class Perforce(FetchMethod):
|
||
|
""" Class to fetch from perforce repositories """
|
||
|
def supports(self, ud, d):
|
||
|
""" Check to see if a given url can be fetched with perforce. """
|
||
|
return ud.type in ['p4']
|
||
|
|
||
|
def urldata_init(self, ud, d):
|
||
|
"""
|
||
|
Initialize perforce specific variables within url data. If P4CONFIG is
|
||
|
provided by the env, use it. If P4PORT is specified by the recipe, use
|
||
|
its values, which may override the settings in P4CONFIG.
|
||
|
"""
|
||
|
ud.basecmd = d.getVar('FETCHCMD_p4')
|
||
|
if not ud.basecmd:
|
||
|
ud.basecmd = "/usr/bin/env p4"
|
||
|
|
||
|
ud.dldir = d.getVar('P4DIR')
|
||
|
if not ud.dldir:
|
||
|
ud.dldir = '%s/%s' % (d.getVar('DL_DIR'), 'p4')
|
||
|
|
||
|
path = ud.url.split('://')[1]
|
||
|
path = path.split(';')[0]
|
||
|
delim = path.find('@');
|
||
|
if delim != -1:
|
||
|
(ud.user, ud.pswd) = path.split('@')[0].split(':')
|
||
|
ud.path = path.split('@')[1]
|
||
|
else:
|
||
|
ud.path = path
|
||
|
|
||
|
ud.usingp4config = False
|
||
|
p4port = d.getVar('P4PORT')
|
||
|
|
||
|
if p4port:
|
||
|
logger.debug(1, 'Using recipe provided P4PORT: %s' % p4port)
|
||
|
ud.host = p4port
|
||
|
else:
|
||
|
logger.debug(1, 'Trying to use P4CONFIG to automatically set P4PORT...')
|
||
|
ud.usingp4config = True
|
||
|
p4cmd = '%s info | grep "Server address"' % ud.basecmd
|
||
|
bb.fetch2.check_network_access(d, p4cmd, ud.url)
|
||
|
ud.host = runfetchcmd(p4cmd, d, True)
|
||
|
ud.host = ud.host.split(': ')[1].strip()
|
||
|
logger.debug(1, 'Determined P4PORT to be: %s' % ud.host)
|
||
|
if not ud.host:
|
||
|
raise FetchError('Could not determine P4PORT from P4CONFIG')
|
||
|
|
||
|
if ud.path.find('/...') >= 0:
|
||
|
ud.pathisdir = True
|
||
|
else:
|
||
|
ud.pathisdir = False
|
||
|
|
||
|
cleanedpath = ud.path.replace('/...', '').replace('/', '.')
|
||
|
cleanedhost = ud.host.replace(':', '.')
|
||
|
ud.pkgdir = os.path.join(ud.dldir, cleanedhost, cleanedpath)
|
||
|
|
||
|
ud.setup_revisions(d)
|
||
|
|
||
|
ud.localfile = d.expand('%s_%s_%s.tar.gz' % (cleanedhost, cleanedpath, ud.revision))
|
||
|
|
||
|
def _buildp4command(self, ud, d, command, depot_filename=None):
|
||
|
"""
|
||
|
Build a p4 commandline. Valid commands are "changes", "print", and
|
||
|
"files". depot_filename is the full path to the file in the depot
|
||
|
including the trailing '#rev' value.
|
||
|
"""
|
||
|
p4opt = ""
|
||
|
|
||
|
if ud.user:
|
||
|
p4opt += ' -u "%s"' % (ud.user)
|
||
|
|
||
|
if ud.pswd:
|
||
|
p4opt += ' -P "%s"' % (ud.pswd)
|
||
|
|
||
|
if ud.host and not ud.usingp4config:
|
||
|
p4opt += ' -p %s' % (ud.host)
|
||
|
|
||
|
if hasattr(ud, 'revision') and ud.revision:
|
||
|
pathnrev = '%s@%s' % (ud.path, ud.revision)
|
||
|
else:
|
||
|
pathnrev = '%s' % (ud.path)
|
||
|
|
||
|
if depot_filename:
|
||
|
if ud.pathisdir: # Remove leading path to obtain filename
|
||
|
filename = depot_filename[len(ud.path)-1:]
|
||
|
else:
|
||
|
filename = depot_filename[depot_filename.rfind('/'):]
|
||
|
filename = filename[:filename.find('#')] # Remove trailing '#rev'
|
||
|
|
||
|
if command == 'changes':
|
||
|
p4cmd = '%s%s changes -m 1 //%s' % (ud.basecmd, p4opt, pathnrev)
|
||
|
elif command == 'print':
|
||
|
if depot_filename != None:
|
||
|
p4cmd = '%s%s print -o "p4/%s" "%s"' % (ud.basecmd, p4opt, filename, depot_filename)
|
||
|
else:
|
||
|
raise FetchError('No depot file name provided to p4 %s' % command, ud.url)
|
||
|
elif command == 'files':
|
||
|
p4cmd = '%s%s files //%s' % (ud.basecmd, p4opt, pathnrev)
|
||
|
else:
|
||
|
raise FetchError('Invalid p4 command %s' % command, ud.url)
|
||
|
|
||
|
return p4cmd
|
||
|
|
||
|
def _p4listfiles(self, ud, d):
|
||
|
"""
|
||
|
Return a list of the file names which are present in the depot using the
|
||
|
'p4 files' command, including trailing '#rev' file revision indicator
|
||
|
"""
|
||
|
p4cmd = self._buildp4command(ud, d, 'files')
|
||
|
bb.fetch2.check_network_access(d, p4cmd, ud.url)
|
||
|
p4fileslist = runfetchcmd(p4cmd, d, True)
|
||
|
p4fileslist = [f.rstrip() for f in p4fileslist.splitlines()]
|
||
|
|
||
|
if not p4fileslist:
|
||
|
raise FetchError('Unable to fetch listing of p4 files from %s@%s' % (ud.host, ud.path))
|
||
|
|
||
|
count = 0
|
||
|
filelist = []
|
||
|
|
||
|
for filename in p4fileslist:
|
||
|
item = filename.split(' - ')
|
||
|
lastaction = item[1].split()
|
||
|
logger.debug(1, 'File: %s Last Action: %s' % (item[0], lastaction[0]))
|
||
|
if lastaction[0] == 'delete':
|
||
|
continue
|
||
|
filelist.append(item[0])
|
||
|
|
||
|
return filelist
|
||
|
|
||
|
def download(self, ud, d):
|
||
|
""" Get the list of files, fetch each one """
|
||
|
filelist = self._p4listfiles(ud, d)
|
||
|
if not filelist:
|
||
|
raise FetchError('No files found in depot %s@%s' % (ud.host, ud.path))
|
||
|
|
||
|
bb.utils.remove(ud.pkgdir, True)
|
||
|
bb.utils.mkdirhier(ud.pkgdir)
|
||
|
|
||
|
for afile in filelist:
|
||
|
p4fetchcmd = self._buildp4command(ud, d, 'print', afile)
|
||
|
bb.fetch2.check_network_access(d, p4fetchcmd, ud.url)
|
||
|
runfetchcmd(p4fetchcmd, d, workdir=ud.pkgdir)
|
||
|
|
||
|
runfetchcmd('tar -czf %s p4' % (ud.localpath), d, cleanup=[ud.localpath], workdir=ud.pkgdir)
|
||
|
|
||
|
def clean(self, ud, d):
|
||
|
""" Cleanup p4 specific files and dirs"""
|
||
|
bb.utils.remove(ud.localpath)
|
||
|
bb.utils.remove(ud.pkgdir, True)
|
||
|
|
||
|
def supports_srcrev(self):
|
||
|
return True
|
||
|
|
||
|
def _revision_key(self, ud, d, name):
|
||
|
""" Return a unique key for the url """
|
||
|
return 'p4:%s' % ud.pkgdir
|
||
|
|
||
|
def _latest_revision(self, ud, d, name):
|
||
|
""" Return the latest upstream scm revision number """
|
||
|
p4cmd = self._buildp4command(ud, d, "changes")
|
||
|
bb.fetch2.check_network_access(d, p4cmd, ud.url)
|
||
|
tip = runfetchcmd(p4cmd, d, True)
|
||
|
|
||
|
if not tip:
|
||
|
raise FetchError('Could not determine the latest perforce changelist')
|
||
|
|
||
|
tipcset = tip.split(' ')[1]
|
||
|
logger.debug(1, 'p4 tip found to be changelist %s' % tipcset)
|
||
|
return tipcset
|
||
|
|
||
|
def sortable_revision(self, ud, d, name):
|
||
|
""" Return a sortable revision number """
|
||
|
return False, self._build_revision(ud, d)
|
||
|
|
||
|
def _build_revision(self, ud, d):
|
||
|
return ud.revision
|
||
|
|