Ned Nguyen | 9a7a905 | 2019-02-05 11:04:03 -0700 | [diff] [blame] | 1 | # -*- coding: utf-8 -*- |
| 2 | # Copyright 2019 The Chromium OS Authors. All rights reserved. |
| 3 | # Use of this source code is governed by a BSD-style license that can be |
| 4 | # found in the LICENSE file. |
| 5 | |
| 6 | """Build graph dependency creation service. |
| 7 | |
| 8 | This service handles the creation of the portage build dependency graphs and the |
| 9 | graphs mapping from portage packages to the dependency source. |
| 10 | """ |
| 11 | |
| 12 | from __future__ import print_function |
| 13 | |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 14 | from chromite.lib import portage_util |
Alex Klein | b7cdbe6 | 2019-02-22 11:41:32 -0700 | [diff] [blame] | 15 | from chromite.service import dependency |
Ned Nguyen | 9a7a905 | 2019-02-05 11:04:03 -0700 | [diff] [blame] | 16 | |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 17 | |
Ned Nguyen | 8e144bd | 2019-05-02 09:09:41 -0600 | [diff] [blame] | 18 | def AugmentDepGraphProtoFromJsonMap(json_map, graph): |
| 19 | """Augment package deps from |json_map| to graph object. |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 20 | |
| 21 | Args: |
| 22 | json_map: the json object that stores the portage package. This is |
Alex Klein | b7cdbe6 | 2019-02-22 11:41:32 -0700 | [diff] [blame] | 23 | generated from chromite.lib.service.dependency.GetBuildDependency() |
Ned Nguyen | 8e144bd | 2019-05-02 09:09:41 -0600 | [diff] [blame] | 24 | graph: the proto object that represents the dependency graph (see DepGraph |
| 25 | message in chromite/api/depgraph.proto) |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 26 | """ |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 27 | graph.build_target.name = json_map['target_board'] |
| 28 | |
Mike Frysinger | 0bdbc10 | 2019-06-13 15:27:29 -0400 | [diff] [blame] | 29 | for data in json_map['package_deps'].values(): |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 30 | package_dep_info = graph.package_deps.add() |
| 31 | package_info = package_dep_info.package_info |
| 32 | package_info.package_name = data['name'] |
| 33 | package_info.category = data['category'] |
| 34 | package_info.version = data['version'] |
| 35 | for dep in data['deps']: |
| 36 | cpv = portage_util.SplitCPV(dep, strict=False) |
| 37 | dep_package = package_dep_info.dependency_packages.add() |
| 38 | dep_package.package_name = cpv.package |
| 39 | dep_package.category = cpv.category |
| 40 | if cpv.version: |
| 41 | dep_package.version = cpv.version |
| 42 | |
Ned Nguyen | 8be1005 | 2019-02-26 08:58:45 -0700 | [diff] [blame] | 43 | package_CPV = '%s/%s-%s' % ( |
| 44 | package_info.category, package_info.package_name, package_info.version) |
| 45 | for path in json_map['source_path_mapping'][package_CPV]: |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 46 | source_path = package_dep_info.dependency_source_paths.add() |
| 47 | source_path.path = path |
| 48 | |
Ned Nguyen | b5d3872 | 2019-05-06 10:24:06 -0600 | [diff] [blame] | 49 | |
Ned Nguyen | 9a7a905 | 2019-02-05 11:04:03 -0700 | [diff] [blame] | 50 | def GetBuildDependencyGraph(input_proto, output_proto): |
| 51 | """Create the build dependency graph. |
| 52 | |
| 53 | Args: |
| 54 | input_proto (GetBuildDependencyGraphRequest): The input arguments message. |
| 55 | output_proto (GetBuildDependencyGraphResponse): The empty output message. |
| 56 | """ |
| 57 | board = input_proto.build_target.name |
Ned Nguyen | 9a7a905 | 2019-02-05 11:04:03 -0700 | [diff] [blame] | 58 | |
| 59 | assert board, 'Missing build target name' |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 60 | |
Ned Nguyen | 9a7a905 | 2019-02-05 11:04:03 -0700 | [diff] [blame] | 61 | json_map = dependency.GetBuildDependency(board) |
Ned Nguyen | 8e144bd | 2019-05-02 09:09:41 -0600 | [diff] [blame] | 62 | AugmentDepGraphProtoFromJsonMap(json_map, output_proto.dep_graph) |