mappings = {}

class SimpleMapping(object):

  def __init__(self, cmd, cli, vat):
    if cmd in mappings:
      raise Exception('{} already in cmd db!'.format(cmd))

    self.cmd = cmd
    self.cli = cli
    self.vat = vat
    mappings[cmd] = self

  def generate(self, mode, args):
    s = ''
    # simply append arguments string to right command
    if mode == 'vat':
      s = self.vat + ' ' + args
    else:
      s = self.cli + ' ' + args
    return s


class CustomMapping(SimpleMapping):

  def generate(self, mode, args):
    s = ''
    if mode == 'vat':
      s = self.vat
    else:
      s = self.cli

    args = args.split(' ')
    return s.format(*args)


class RepeatableLocators(SimpleMapping):

  def generate(self, mode, args):
    name = args[:args.index(' ')]  # first word is ls name
    locs = args[args.index(' '):]

    if mode == 'vat':
      s = self.vat
    else:
      s = self.cli

    s = s + ' ' + name + locs
    return s


SimpleMapping('lisp_state', 'one', 'one_enable_disable')
SimpleMapping('lisp_map_resolver', 'one map-resolver', 'one_add_del_map_resolver')
SimpleMapping('lisp_map_server', 'one map-server', 'one_add_del_map_server')
SimpleMapping('lisp_local_eid', 'one eid-table', 'one_add_del_local_eid')
SimpleMapping('lisp_remote_mapping', 'one remote-mapping', 'one_add_del_remote_mapping')
SimpleMapping('lisp_pitr', 'one pitr ls', 'one_pitr_set_locator_set locator-set')
SimpleMapping('lisp_adjacency', 'one adjacency', 'one_add_del_adjacency')
SimpleMapping('lisp_map_request_mode', 'one map-request mode', 'one_map_request_mode')
SimpleMapping('set_if_ip', 'set int ip address', 'sw_interface_add_del_address')
SimpleMapping('lisp_rloc_probe_state', 'one rloc-probe', 'one_rloc_probe_enable_disable')
SimpleMapping('lisp_map_register_state', 'one map-register', 'one_map_register_enable_disable')
SimpleMapping('lisp_stats', 'one statistics', 'one_stats_enable_disable')
SimpleMapping('lisp_l2_arp', 'one l2 arp', 'one_add_del_l2_arp_entry')
SimpleMapping('lisp_ndp_entry', 'one ndp', 'one_add_del_ndp_entry')
SimpleMapping('lisp_nsh_set_ls', 'one nsh-mapping', 'one_nsh_set_locator_set')

SimpleMapping('ipsec_sad_add_del_entry', 'TODO', 'ipsec_sad_add_del_entry')
SimpleMapping('ipsec_spd_add_del', 'todo', 'ipsec_spd_add_del')
SimpleMapping('ipsec_interface_add_del_spd' , 'todo', 'ipsec_interface_add_del_spd')
SimpleMapping('ipsec_spd_add_del_entry', 'todo', 'ipsec_spd_add_del_entry')

CustomMapping('lisp_eid_map_bd',
              'one eid-table map vni {0} bd {1}',
              'one_eid_table_add_del_map vni {0} bd_index {1}')
CustomMapping('lisp_eid_map_vrf',
              'one eid-table map vni {0} vrf {1}',
              'one_eid_table_add_del_map vni {0} vrf {1}')
CustomMapping('set_if_l2_bridge', 'set interface l2 bridge {0} {1}',
              'sw_interface_set_l2_bridge {0} bd_id {1}')
CustomMapping('set_if_ip_table', 'set interface ip table {0} {1}',
              'sw_interface_set_table {0} vrf {1}')
CustomMapping('lisp_locator_set_with_locator',
              'one locator-set add {0} iface {1} p {2} w {3}',
              'one_add_del_locator_set locator-set {0} iface {1} p {2} w {3}')
CustomMapping('create_host_iface',
    'create host-interface name {0}\n'
    'set int state host-{0} up\n'
    'set int ip address host-{0} {1}',

    'af_packet_create name {0}\n'
    'sw_interface_set_flags host-{0} admin-up link-up\n'
    'sw_interface_add_del_address host-{0} {1}')

CustomMapping('create_host_iface_vrf',
    'create host-interface name {0}\n'
    'set int state host-{0} up\n'
    'set interface ip table host-{0} {2}\n'
    'set int ip address host-{0} {1}',

    'af_packet_create name {0}\n'
    'sw_interface_set_flags host-{0} admin-up link-up\n'
    'sw_interface_set_table host-{0} vrf {2}\n'
    'sw_interface_add_del_address host-{0} {1}')

CustomMapping('create_host_iface_vrf_v6',
    'create host-interface name {0}\n'
    'set int state host-{0} up\n'
    'set interface ip6 table host-{0} {2}\n'
    'set int ip address host-{0} {1}',

    'af_packet_create name {0}\n'
    'sw_interface_set_flags host-{0} admin-up link-up\n'
    'sw_interface_set_table host-{0} vrf {2} ipv6\n'
    'sw_interface_add_del_address host-{0} {1}')

RepeatableLocators('lisp_ls_multiple_locs',
                   'one locator-set add',
                   'one_add_del_locator_set locator-set')