blob: 31b2703cec91064f36f6a2f09d193c8a697cf59a [file] [log] [blame]
Ned Nguyen9a7a9052019-02-05 11:04:03 -07001# -*- 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
8This service handles the creation of the portage build dependency graphs and the
9graphs mapping from portage packages to the dependency source.
10"""
11
12from __future__ import print_function
13
Ned Nguyene5d04852019-02-19 16:33:15 -070014from chromite.lib import portage_util
Alex Kleinb7cdbe62019-02-22 11:41:32 -070015from chromite.service import dependency
Ned Nguyen9a7a9052019-02-05 11:04:03 -070016
Ned Nguyene5d04852019-02-19 16:33:15 -070017
Ned Nguyen8e144bd2019-05-02 09:09:41 -060018def AugmentDepGraphProtoFromJsonMap(json_map, graph):
19 """Augment package deps from |json_map| to graph object.
Ned Nguyene5d04852019-02-19 16:33:15 -070020
21 Args:
22 json_map: the json object that stores the portage package. This is
Alex Kleinb7cdbe62019-02-22 11:41:32 -070023 generated from chromite.lib.service.dependency.GetBuildDependency()
Ned Nguyen8e144bd2019-05-02 09:09:41 -060024 graph: the proto object that represents the dependency graph (see DepGraph
25 message in chromite/api/depgraph.proto)
Ned Nguyene5d04852019-02-19 16:33:15 -070026 """
Ned Nguyene5d04852019-02-19 16:33:15 -070027 graph.build_target.name = json_map['target_board']
28
Mike Frysinger0bdbc102019-06-13 15:27:29 -040029 for data in json_map['package_deps'].values():
Ned Nguyene5d04852019-02-19 16:33:15 -070030 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 Nguyen8be10052019-02-26 08:58:45 -070043 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 Nguyene5d04852019-02-19 16:33:15 -070046 source_path = package_dep_info.dependency_source_paths.add()
47 source_path.path = path
48
Ned Nguyenb5d38722019-05-06 10:24:06 -060049
Ned Nguyen9a7a9052019-02-05 11:04:03 -070050def 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 Nguyen9a7a9052019-02-05 11:04:03 -070058
59 assert board, 'Missing build target name'
Ned Nguyene5d04852019-02-19 16:33:15 -070060
Ned Nguyen9a7a9052019-02-05 11:04:03 -070061 json_map = dependency.GetBuildDependency(board)
Ned Nguyen8e144bd2019-05-02 09:09:41 -060062 AugmentDepGraphProtoFromJsonMap(json_map, output_proto.dep_graph)