#!/usr/bin/env python
# Copyright 2016, Rackspace US, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This could probably be made more generic, since the biggest change per
# service is the variable mappings
import sys

VAR_MAPPINGS = {
    # Nova changes
    "nova_v21_service_name": "nova_service_name",
    "nova_v21_service_type": "nova_service_type",
    "nova_v21_service_proto": "nova_service_proto",
    "nova_v21_service_publicuri_proto": "nova_service_publicuri_proto",
    "nova_v21_service_adminuri_proto": "nova_service_adminuri_proto",
    "nova_v21_service_internaluri_proto": "nova_service_internaluri_proto",
    "nova_v21_service_port": "nova_service_port",
    "nova_v21_service_description": "nova_service_description",
    "nova_v21_service_publicuri": "nova_service_publicuri",
    "nova_v21_service_publicurl": "nova_service_publicurl",
    "nova_v21_service_adminuri": "nova_service_adminuri",
    "nova_v21_service_adminurl": "nova_service_adminurl",
    "nova_v21_service_internaluri": "nova_service_internaluri",
    "nova_v21_service_internalurl": "nova_service_internalurl",

    # Heat changes
    # These also have a different default value: "Default" -> default
    # Should this be handled in a second pass, or at the same time?
    "heat_service_project_domain_name": "heat_service_project_domain_id",
    "heat_service_user_domain_name": "heat_service_user_domain_id",

    # Galera server changes
    "galera_sst_method": "galera_wsrep_sst_method",
}


def update_variables(old_contents):
    """Replace all references to old variables.

    This includes comments and references within values for other variables.
    """
    new_contents = []

    for line in old_contents:
        words = line.split()

        for word in words:
            # Using the whitespace split above, the keys in the yaml file will
            # have a : at the end, so we need to strip that off before
            # replacing
            if word.endswith(':'):
                word = word[:-1]

            if word in VAR_MAPPINGS.keys():
                line = line.replace(word, VAR_MAPPINGS[word])

        new_contents.append(line)

    return new_contents


def main(filename):
    with open(filename, 'r') as f:
        contents = f.readlines()

    new_contents = update_variables(contents)

    with open(filename, 'w') as f:
        f.write(''.join(new_contents))

if __name__ == '__main__':
    if len(sys.argv) < 3:
        sys.exit("Filename and flag file reference required.")

    filename = sys.argv[1]
    flag_ref = sys.argv[2]
    main(filename)

    flag_file = '/etc/openstack_deploy.KILO/VARS_MIGRATED_%s' % flag_ref
    with open(flag_file, 'w') as f:
        f.write('OpenStack Kilo variables migrated.')