Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 1 | """ |
| 2 | BitBake 'Fetch' implementation for bzr. |
| 3 | |
| 4 | """ |
| 5 | |
| 6 | # Copyright (C) 2007 Ross Burton |
| 7 | # Copyright (C) 2007 Richard Purdie |
| 8 | # |
| 9 | # Classes for obtaining upstream sources for the |
| 10 | # BitBake build tools. |
| 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 | import os |
| 27 | import sys |
| 28 | import logging |
| 29 | import bb |
| 30 | from bb import data |
| 31 | from bb.fetch2 import FetchMethod |
| 32 | from bb.fetch2 import FetchError |
| 33 | from bb.fetch2 import runfetchcmd |
| 34 | from bb.fetch2 import logger |
| 35 | |
| 36 | class Bzr(FetchMethod): |
| 37 | def supports(self, ud, d): |
| 38 | return ud.type in ['bzr'] |
| 39 | |
| 40 | def urldata_init(self, ud, d): |
| 41 | """ |
| 42 | init bzr specific variable within url data |
| 43 | """ |
| 44 | # Create paths to bzr checkouts |
| 45 | relpath = self._strip_leading_slashes(ud.path) |
| 46 | ud.pkgdir = os.path.join(data.expand('${BZRDIR}', d), ud.host, relpath) |
| 47 | |
| 48 | ud.setup_revisons(d) |
| 49 | |
| 50 | if not ud.revision: |
| 51 | ud.revision = self.latest_revision(ud, d) |
| 52 | |
| 53 | ud.localfile = data.expand('bzr_%s_%s_%s.tar.gz' % (ud.host, ud.path.replace('/', '.'), ud.revision), d) |
| 54 | |
| 55 | def _buildbzrcommand(self, ud, d, command): |
| 56 | """ |
| 57 | Build up an bzr commandline based on ud |
| 58 | command is "fetch", "update", "revno" |
| 59 | """ |
| 60 | |
| 61 | basecmd = data.expand('${FETCHCMD_bzr}', d) |
| 62 | |
| 63 | proto = ud.parm.get('protocol', 'http') |
| 64 | |
| 65 | bzrroot = ud.host + ud.path |
| 66 | |
| 67 | options = [] |
| 68 | |
| 69 | if command == "revno": |
| 70 | bzrcmd = "%s revno %s %s://%s" % (basecmd, " ".join(options), proto, bzrroot) |
| 71 | else: |
| 72 | if ud.revision: |
| 73 | options.append("-r %s" % ud.revision) |
| 74 | |
| 75 | if command == "fetch": |
| 76 | bzrcmd = "%s branch %s %s://%s" % (basecmd, " ".join(options), proto, bzrroot) |
| 77 | elif command == "update": |
| 78 | bzrcmd = "%s pull %s --overwrite" % (basecmd, " ".join(options)) |
| 79 | else: |
| 80 | raise FetchError("Invalid bzr command %s" % command, ud.url) |
| 81 | |
| 82 | return bzrcmd |
| 83 | |
| 84 | def download(self, ud, d): |
| 85 | """Fetch url""" |
| 86 | |
| 87 | if os.access(os.path.join(ud.pkgdir, os.path.basename(ud.pkgdir), '.bzr'), os.R_OK): |
| 88 | bzrcmd = self._buildbzrcommand(ud, d, "update") |
| 89 | logger.debug(1, "BZR Update %s", ud.url) |
| 90 | bb.fetch2.check_network_access(d, bzrcmd, ud.url) |
| 91 | os.chdir(os.path.join (ud.pkgdir, os.path.basename(ud.path))) |
| 92 | runfetchcmd(bzrcmd, d) |
| 93 | else: |
| 94 | bb.utils.remove(os.path.join(ud.pkgdir, os.path.basename(ud.pkgdir)), True) |
| 95 | bzrcmd = self._buildbzrcommand(ud, d, "fetch") |
| 96 | bb.fetch2.check_network_access(d, bzrcmd, ud.url) |
| 97 | logger.debug(1, "BZR Checkout %s", ud.url) |
| 98 | bb.utils.mkdirhier(ud.pkgdir) |
| 99 | os.chdir(ud.pkgdir) |
| 100 | logger.debug(1, "Running %s", bzrcmd) |
| 101 | runfetchcmd(bzrcmd, d) |
| 102 | |
| 103 | os.chdir(ud.pkgdir) |
| 104 | |
| 105 | scmdata = ud.parm.get("scmdata", "") |
| 106 | if scmdata == "keep": |
| 107 | tar_flags = "" |
| 108 | else: |
| 109 | tar_flags = "--exclude '.bzr' --exclude '.bzrtags'" |
| 110 | |
| 111 | # tar them up to a defined filename |
| 112 | runfetchcmd("tar %s -czf %s %s" % (tar_flags, ud.localpath, os.path.basename(ud.pkgdir)), d, cleanup = [ud.localpath]) |
| 113 | |
| 114 | def supports_srcrev(self): |
| 115 | return True |
| 116 | |
| 117 | def _revision_key(self, ud, d, name): |
| 118 | """ |
| 119 | Return a unique key for the url |
| 120 | """ |
| 121 | return "bzr:" + ud.pkgdir |
| 122 | |
| 123 | def _latest_revision(self, ud, d, name): |
| 124 | """ |
| 125 | Return the latest upstream revision number |
| 126 | """ |
| 127 | logger.debug(2, "BZR fetcher hitting network for %s", ud.url) |
| 128 | |
| 129 | bb.fetch2.check_network_access(d, self._buildbzrcommand(ud, d, "revno"), ud.url) |
| 130 | |
| 131 | output = runfetchcmd(self._buildbzrcommand(ud, d, "revno"), d, True) |
| 132 | |
| 133 | return output.strip() |
| 134 | |
| 135 | def sortable_revision(self, ud, d, name): |
| 136 | """ |
| 137 | Return a sortable revision number which in our case is the revision number |
| 138 | """ |
| 139 | |
| 140 | return False, self._build_revision(ud, d) |
| 141 | |
| 142 | def _build_revision(self, ud, d): |
| 143 | return ud.revision |