#!/usr/bin/env python3

from dfatool.aspectc import Repo
from dfatool.codegen import MultipassDriver
from dfatool.automata import PTA
import yaml

with open('../multipass/model/driver/nrf24l01.dfa', 'r') as f:
    driver_definition = yaml.safe_load(f)
    pta = PTA.from_yaml(driver_definition)
repo = Repo('../multipass/build/repo.acp')

enum = dict()

if 'dummygen' in driver_definition and 'enum' in driver_definition['dummygen']:
    enum = driver_definition['dummygen']['enum']

drv = MultipassDriver('Nrf24l01', pta, repo.class_by_name['Nrf24l01'], enum=enum)

with open('../multipass/src/driver/dummy.cc', 'w') as f:
    f.write(drv.impl)
with open('../multipass/include/driver/dummy.h', 'w') as f:
    f.write(drv.header)