Brad Bishop | c342db3 | 2019-05-15 21:57:59 -0400 | [diff] [blame] | 1 | # |
| 2 | # SPDX-License-Identifier: GPL-2.0-only |
| 3 | # |
| 4 | |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 5 | import codecs |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 6 | import os |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 7 | |
| 8 | def packaged(pkg, d): |
| 9 | return os.access(get_subpkgedata_fn(pkg, d) + '.packaged', os.R_OK) |
| 10 | |
| 11 | def read_pkgdatafile(fn): |
| 12 | pkgdata = {} |
| 13 | |
| 14 | def decode(str): |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 15 | c = codecs.getdecoder("unicode_escape") |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 16 | return c(str)[0] |
| 17 | |
| 18 | if os.access(fn, os.R_OK): |
| 19 | import re |
Brad Bishop | 64c979e | 2019-11-04 13:55:29 -0500 | [diff] [blame] | 20 | with open(fn, 'r') as f: |
| 21 | lines = f.readlines() |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 22 | r = re.compile("([^:]+):\s*(.*)") |
| 23 | for l in lines: |
| 24 | m = r.match(l) |
| 25 | if m: |
| 26 | pkgdata[m.group(1)] = decode(m.group(2)) |
| 27 | |
| 28 | return pkgdata |
| 29 | |
| 30 | def get_subpkgedata_fn(pkg, d): |
| 31 | return d.expand('${PKGDATA_DIR}/runtime/%s' % pkg) |
| 32 | |
| 33 | def has_subpkgdata(pkg, d): |
| 34 | return os.access(get_subpkgedata_fn(pkg, d), os.R_OK) |
| 35 | |
| 36 | def read_subpkgdata(pkg, d): |
| 37 | return read_pkgdatafile(get_subpkgedata_fn(pkg, d)) |
| 38 | |
| 39 | def has_pkgdata(pn, d): |
| 40 | fn = d.expand('${PKGDATA_DIR}/%s' % pn) |
| 41 | return os.access(fn, os.R_OK) |
| 42 | |
| 43 | def read_pkgdata(pn, d): |
| 44 | fn = d.expand('${PKGDATA_DIR}/%s' % pn) |
| 45 | return read_pkgdatafile(fn) |
| 46 | |
| 47 | # |
| 48 | # Collapse FOO_pkg variables into FOO |
| 49 | # |
| 50 | def read_subpkgdata_dict(pkg, d): |
| 51 | ret = {} |
| 52 | subd = read_pkgdatafile(get_subpkgedata_fn(pkg, d)) |
| 53 | for var in subd: |
| 54 | newvar = var.replace("_" + pkg, "") |
| 55 | if newvar == var and var + "_" + pkg in subd: |
| 56 | continue |
| 57 | ret[newvar] = subd[var] |
| 58 | return ret |
| 59 | |
| 60 | def _pkgmap(d): |
| 61 | """Return a dictionary mapping package to recipe name.""" |
| 62 | |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 63 | pkgdatadir = d.getVar("PKGDATA_DIR") |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 64 | |
| 65 | pkgmap = {} |
| 66 | try: |
| 67 | files = os.listdir(pkgdatadir) |
| 68 | except OSError: |
| 69 | bb.warn("No files in %s?" % pkgdatadir) |
| 70 | files = [] |
| 71 | |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 72 | for pn in [f for f in files if not os.path.isdir(os.path.join(pkgdatadir, f))]: |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 73 | try: |
| 74 | pkgdata = read_pkgdatafile(os.path.join(pkgdatadir, pn)) |
| 75 | except OSError: |
| 76 | continue |
| 77 | |
| 78 | packages = pkgdata.get("PACKAGES") or "" |
| 79 | for pkg in packages.split(): |
| 80 | pkgmap[pkg] = pn |
| 81 | |
| 82 | return pkgmap |
| 83 | |
| 84 | def pkgmap(d): |
| 85 | """Return a dictionary mapping package to recipe name. |
| 86 | Cache the mapping in the metadata""" |
| 87 | |
| 88 | pkgmap_data = d.getVar("__pkgmap_data", False) |
| 89 | if pkgmap_data is None: |
| 90 | pkgmap_data = _pkgmap(d) |
| 91 | d.setVar("__pkgmap_data", pkgmap_data) |
| 92 | |
| 93 | return pkgmap_data |
| 94 | |
| 95 | def recipename(pkg, d): |
| 96 | """Return the recipe name for the given binary package name.""" |
| 97 | |
| 98 | return pkgmap(d).get(pkg) |