#!/usr/bin/env python3
#
# Copyright (C) 2021 Red Hat, Inc.
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library.  If not, see
# <http://www.gnu.org/licenses/>.

# About:
# When provided with an interface file, this script outputs a new interface
# file, where all original interfaces are enclosed in an ifndef statement
# dependent on their name. All indentation inside the interface file should
# be converted to tabs before running this script.
# 
# Known issues:
# - interface already enclosed in an ifndef statement is not ignored
# - "`"  and "'" inside comments are not ignored
# - \t is always used for indentation - may result in mixture of spaces and tabs

import sys
import os
import re

if len(sys.argv) < 1:
    print(("Usage: {} <policy>.if > <output>.if").format(sys.argv[0]), file=sys.stderr)
    exit(os.EX_USAGE)

# ending index of interface
end = 0
with open(sys.argv[1], 'r') as f:
    interfaces = f.read()
    file_len = len(interfaces)

    while end < file_len:
        start = interfaces.find("interface(`", end)
        if start < 0:
            #no more interfaces
            print(interfaces[end:], end ="")
            break
        name_end = interfaces.find("'", start+11)
        name = interfaces[start+11:name_end]

        # locate the end of the interface - i.e. ending apostrophe
        reg = re.compile(r"[`']")
        # skip the ' after interface name
        i = name_end+1
        # counter for the depth (` opens a new block, while ' closes it)
        graves = 0
        while i < file_len:
            match = reg.search(interfaces, i)
            if not match:
                print("Malformed interface: {}".format(name), file=sys.stderr)
                exit(1)

            i = match.end()
            if match.group(0) == '`':
                graves += 1
            else:
                graves -= 1
                if graves == 0:
                    break

        # keep everything between the end of previous interface and start of a new one
        print(interfaces[end:start], end ="")
        # interface ends in "')" -- add 1 for the edning bracket
        end = i+1
        # print the whole interface enclosed in "ifndef"
        print("ifndef(`{}',`".format(name))
        print('\n'.join([('\t' + x) if x else x for x in interfaces[start:end].split('\n')]))
        print("')", end ="")