#!/usr/bin/env python3

# Copyright (C) 2022 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Generate queryview xml data for testing purposes."""

import xml.etree.ElementTree as ElementTree


def make_module(full_name,
                name,
                kind,
                variant='',
                dep_names=[],
                soong_module_type=None,
                srcs=None):
  rule = ElementTree.Element('rule', attrib={'class': kind, 'name': full_name})
  ElementTree.SubElement(
      rule, 'string', attrib={
          'name': 'soong_module_name',
          'value': name
      })
  ElementTree.SubElement(
      rule, 'string', attrib={
          'name': 'soong_module_variant',
          'value': variant
      })
  if soong_module_type:
    ElementTree.SubElement(
        rule,
        'string',
        attrib={
            'name': 'soong_module_type',
            'value': soong_module_type
        })
  for dep in dep_names:
    ElementTree.SubElement(rule, 'rule-input', attrib={'name': dep})

  if not srcs:
    return rule

  src_element = ElementTree.SubElement(rule, 'list', attrib={'name': 'srcs'})
  for src in srcs:
    ElementTree.SubElement(src_element, 'string', attrib={'value': src})

  return rule


def make_graph(modules):
  graph = ElementTree.Element('query', attrib={'version': '2'})
  graph.extend(modules)
  return graph