# Dirty script for importing terraform state into terragrunt state

import json

with open("data.txt", "r") as fin:
    data = json.load(fin)['values']['root_module']['resources']

for i in data:
    addr = i['address']
    id = i['values']['id']


    if '"' in addr:
        addr = addr.replace('"', '\\"')
        addr = '"' + addr + '"'

    if "iam" in addr:
        _keys = set(i['values'].keys())
        resource = list(_keys - set(['etag','id','member','role','condition']))[0]

        print('terragrunt import {} "{} {} {}"'.format(
            addr,
            i['values'][resource],
            i['values']['role'], 
            i['values']['member']
            ))
        
    else:
        continue
        print("terragrunt import {} {}".format(addr, id))