# -*- coding: utf-8 -*-

import logging

from .osmcsymbol import OsmcSymbol

logger = logging.getLogger(__name__)

keys = ['network', 'osmc:symbol', 'mtb:scale', 'mtb:scale:uphill']

kct_osmc_pairs = {
    'major': 'bar',
    'yes': 'bar',
    'horse': 'dot',
    'spring': 'bowl',
    'learning': 'backslash',
    'ruin': 'L',
    'interesting_object': 'turned_T',
    'peak': 'triangle',
    'local': 'corner'
}


class Relation:
    def __init__(self, row, source):
        if source == "line":
            # 0: osm_id; 1: mtb:scale; 2: mtb:scale:uphill; 3: network; 4: "osmc:symbol"
            self.id = row[0]
            self.lines = [row[0]]

            if row[1] is None:
                self.mtbScale = None
            else:
                self.mtbScale = row[1].replace("'", "")

            if row[2] is None:
                self.mtbUphill = None
            else:
                self.mtbUphill = row[2].replace("'", "")

            self.network = row[3]

            if row[4] is None:
                self.osmcSymbol = 'mtb:white:mtb_mtb'
            else:
                self.osmcSymbol = row[4]
        else:
            self.id = -row[0]
            self.lines = self.parse_members(row[1])
            self.rawTags = row[2]
            self.network = None
            self.osmcSymbol = None
            self.mtbScale = None
            self.mtbUphill = None
            self.parse_tags()

    def parse_tags(self):
        tags = dict(zip(self.rawTags[::2], self.rawTags[1::2]))
        cleanse_func = lambda x: tags[x].replace('\\', 'backslash')
        if 'network' in tags:
            self.network = tags['network']
            self.network = self.network[:3]
        if 'mtb:scale' in tags:
            self.mtbScale = cleanse_func('mtb:scale')
        if 'mtb:scale:uphill' in tags:
            self.mtbUphill = cleanse_func('mtb:scale:uphill')
        if 'osmc:symbol' in tags:
            osmc_string = cleanse_func('osmc:symbol')
            symbol = OsmcSymbol(osmc_string)
            if symbol.is_accepted():
                self.osmcSymbol = symbol.get_string_value()
            else:
                self.osmcSymbol = None
        elif self.parse_kct():
            pass
        else:
            self.osmcSymbol = None

    def parse_kct(self):
        if 'kct_red' in self.rawTags:
            color = 'red'
        elif 'kct_blue' in self.rawTags:
            color = 'blue'
        elif 'kct_green' in self.rawTags:
            color = 'green'
        elif 'kct_yellow' in self.rawTags:
            color = 'yellow'
        else:
            return False
        symbol = self.rawTags[self.rawTags.index('kct_' + color)+1]
        if symbol not in kct_osmc_pairs:
            symbol = 'yes'
        new_osmc_value = color + ':white:' + color + '_' + kct_osmc_pairs[symbol]
        self.osmcSymbol = new_osmc_value
        return True

    @staticmethod
    def parse_members(members):
        parts = []
        for member in members:
            if member.startswith('w'):
                try:
                    member_id = int(member.lstrip('w'))
                    parts.append(member_id)
                except ValueError:
                    logger.debug('Found relation member starts with "w", but it is not a way: %s' % member)
        return parts