from enum import Enum

import re


class HandlerType(Enum):
    BTREE_INDEX = 'BTREE_INDEX'
    FULLTEXT_INDEX = 'FULLTEXT_INDEX'
    UNIQUE_CONSTRAINT = 'UNIQUE_CONSTRAINT'
    NOT_NULL_CONSTRAINT = 'NOT_NULL_CONSTRAINT'
    NODE_KEY_CONSTRAINT = 'NODE_KEY_CONSTRAINT'


def btree_handler(description):
    pattern = r"CREATE BTREE INDEX `\w+` FOR \(n:`(?P<label>\w+)`\) ON \(n.`(?P<property>\w+)`\) OPTIONS (?P<options>.*)"
    match = re.match(pattern, description)
    label = match.group('label')
    property = match.group('property')
    options = match.group('options')
    return f'CREATE BTREE INDEX `{label}__{property}__index` FOR (n:`{label}`) ON (n.`{property}`) OPTIONS {options}'


def fulltext_handler(description):
    return description


def unique_constraint_handler(description):
    pattern = r"CREATE CONSTRAINT `\w+` FOR \(n:`(?P<label>\w+)`\) REQUIRE \(n.`(?P<property>\w+)`\) IS UNIQUE OPTIONS (?P<options>.*)"
    match = re.match(pattern, description)
    label = match.group('label')
    property = match.group('property')
    options = match.group('options')
    return f'CREATE CONSTRAINT `{label}__{property}__isUniqueConstraint` FOR (n:`{label}`) REQUIRE (n.`{property}`) IS UNIQUE OPTIONS {options}'


def not_null_constraint_handler(description):
    pattern = r"CREATE CONSTRAINT `\w+` FOR \(n:`(?P<label>\w+)`\) REQUIRE \(n.`(?P<property>\w+)`\) IS NOT NULL;"
    match = re.match(pattern, description)
    label = match.group('label')
    property = match.group('property')
    return f'CREATE CONSTRAINT `{label}__{property}__isNotNullConstraint` FOR (n:`{label}`) REQUIRE (n.`{property}`) IS NOT NULL;'


def node_key_constraint(description):
    pattern = r"CREATE CONSTRAINT `\w+` FOR \(n:`(?P<label>\w+)`\) REQUIRE \((?P<properties>.*)\) IS NODE KEY OPTIONS (?P<options>.*)"
    match = re.match(pattern, description)
    label = match.group('label')
    properties = match.group('properties')
    properties_part = '_'.join([e.split('.')[1].strip('`') for e in match.group('properties').split(',')])
    options = match.group('options')
    return f'CREATE CONSTRAINT `{label}__{properties_part}__isUniqueConstraint` FOR (n:`{label}`) REQUIRE ({properties}) IS NODE KEY OPTIONS {options}'


rename_handlers = {
    HandlerType.BTREE_INDEX: btree_handler,
    HandlerType.FULLTEXT_INDEX: fulltext_handler,
    HandlerType.UNIQUE_CONSTRAINT: unique_constraint_handler,
    HandlerType.NOT_NULL_CONSTRAINT: not_null_constraint_handler,
    HandlerType.NODE_KEY_CONSTRAINT: node_key_constraint
}


def _get_index_type(description):
    if re.match('.*BTREE INDEX.*', description):
        return HandlerType.BTREE_INDEX
    if re.match('CREATE CONSTRAINT.*IS UNIQUE OPTIONS', description):
        return HandlerType.UNIQUE_CONSTRAINT
    if re.match('CREATE CONSTRAINT.*IS NOT NULL', description):
        return HandlerType.NOT_NULL_CONSTRAINT
    if re.match('CREATE CONSTRAINT.*IS NODE KEY OPTIONS', description):
        return HandlerType.NODE_KEY_CONSTRAINT
    if re.match('CREATE FULLTEXT INDEX.*', description):
        return HandlerType.FULLTEXT_INDEX
    raise NotImplemented


def get_indexes_description():
    with open('source_description.txt') as f:
        return f.readlines()


def save_renamed_indexes(descriptions):
    with open('new_description.txt', 'w') as f:
        f.writelines(descriptions)


def main():
    renamed_indexes = []
    for desc in get_indexes_description():
        description = desc.strip()
        index_type = _get_index_type(description)
        rename_handler = rename_handlers[index_type]
        renamed_description = rename_handler(description)
        renamed_indexes.append(renamed_description + '\n')
    save_renamed_indexes(renamed_indexes)


if __name__ == '__main__':
    main()