#!/usr/bin/env python3

'''Generate PIM rules for ipmi-fru-parser.
'''

import argparse
import os
import sys
import yaml
from mako.template import Template

tmpl = '''
description: >
    PIM rules for ipmi-fru-parser inventory objects.
events:

    - name: Host off at startup
      description: >
          Mark ipmi-fru-parser inventory items as cached when
          the host is off at startup.
      type: startup
      actions:
          - name: setProperty
            interface: ${cacheable_iface}
            property: ${cacheable_property}
            paths:
%for i in cacheable:
                - ${i}
%endfor
            value:
              type: ${cacheable_type}
              value: ${cacheable_cached}

    - name: Host off event
      description: >
          Mark ipmi-fru-parser inventory items as cached when
          the host goes away.
      type: match
      signatures:
          - type: signal
            interface: org.freedesktop.DBus.Properties
            path: ${off_path}
            member: PropertiesChanged
      filters:
          - name: propertyChangedTo
            interface: ${off_iface}
            property: ${off_property}
            value:
              value: ${off_value}
              type: ${off_type}
      actions:
          - name: setProperty
            interface: ${cacheable_iface}
            property: ${cacheable_property}
            paths:
%for i in cacheable:
                - ${i}
%endfor
            value:
              type: ${cacheable_type}
              value: ${cacheable_cached}

    - name: Host on at startup
      description: >
          Remove ipmi-fru-parser inventory items when the host is finished
          sending inventory items and the item is still marked as cached.
      type: startup
      filters:
          - name: propertyIs
            path: ${on_path}
            interface: ${on_iface}
            property: ${on_property}
            value:
              value: ${on_value}
              type: ${on_type}
      actions:
          - name: destroyObjects
            paths:
%for i in cacheable:
                - ${i}
%endfor
            conditions:
              - name: propertyIs
                interface: ${cacheable_iface}
                property: ${cacheable_property}
                value:
                  type: ${cacheable_type}
                  value: ${cacheable_cached}

    - name: Host on event
      description: >
          Remove ipmi-fru-parser inventory items when the host is finished
          sending inventory items and the item is still marked as cached.
      type: match
      signatures:
          - type: signal
            interface: org.freedesktop.DBus.Properties
            path: ${on_path}
            member: PropertiesChanged
      filters:
          - name: propertyChangedTo
            interface: ${on_iface}
            property: ${on_property}
            value:
              value: ${on_value}
              type: ${on_type}
      actions:
          - name: destroyObjects
            paths:
%for i in cacheable:
                - ${i}
%endfor
            conditions:
              - name: propertyIs
                interface: ${cacheable_iface}
                property: ${cacheable_property}
                value:
                  type: ${cacheable_type}
                  value: ${cacheable_cached}
'''


def get_cacheable_objs(yaml):
    cacheable = []

    for objdata in data.values():
        if not isinstance(objdata, dict):
            continue
        for path, ifaces in objdata.items():
            if not isinstance(ifaces, dict):
                continue

            if cacheable_iface_name in ifaces.keys():
                cacheable.append(path)

    return cacheable


if __name__ == '__main__':
    script_dir = os.path.dirname(os.path.realpath(__file__))

    parser = argparse.ArgumentParser(
        description='ipmi-fru-parser PIM rule generator.')
    parser.add_argument(
        '-o', '--output-dir', dest='outputdir',
        default='.', help='Output directory.')
    parser.add_argument(
        'inventory', metavar='INVENTORY', type=str,
        help='Path to inventory YAML.')

    args = parser.parse_args()

    with open(args.inventory, 'r') as fd:
        data = yaml.safe_load(fd.read())

    cacheable_iface_name = 'xyz.openbmc_project.Inventory.Decorator.Cacheable'
    target_file = os.path.join(args.outputdir, 'ipmi-fru-rules.yaml')
    cacheable = []

    if isinstance(data, dict):
        cacheable = get_cacheable_objs(data)
    if cacheable:
        with open(target_file, 'w') as out:
            out.write(
                Template(tmpl).render(
                    cacheable_iface=cacheable_iface_name,
                    cacheable_property='Cached',
                    cacheable_cached='true',
                    cacheable_type='boolean',
                    on_path='/xyz/openbmc_project/state/host0',
                    on_iface='xyz.openbmc_project.State.Boot.Progress',
                    on_property='BootProgress',
                    on_value='xyz.openbmc_project.State.Boot.Progress.ProgressStages.OSStart',
                    on_type='string',
                    off_path='/xyz/openbmc_project/state/host0',
                    off_iface='xyz.openbmc_project.State.Host',
                    off_property='CurrentHostState',
                    off_value='xyz.openbmc_project.State.Host.Off',
                    off_type='string',
                    cacheable=cacheable))


# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4