Add support for interface role mapping

Change-Id: I88a01914304f5e32840810c03646d9cda1ae7fff
Signed-off-by: Feng Pan <fpan@redhat.com>

2 files changed, 63 insertions, 4 deletions

diff --git a/manifests/honeycomb.pp b/manifests/honeycomb.pp
index 6ea88b5..25739f2 100644
--- a/manifests/honeycomb.pp
+++ b/manifests/honeycomb.pp
@@ -39,6 +39,14 @@
 #   (optional) Node ID for binding VPP to Opendaylight
 #   Defaults to $::fqdn
 #
+# [*interface_role_map*]
+#   (optional) List of interface role mapping in the format
+#              of <VPP interface name>:<role name>
+#   Example:
+#      [ 'GigabitEthernet0/5/0:public-interface',
+#        'GigabitEthernet0/6/0:tenant-interface' ]
+#   Defaults to undef
+#
 class fdio::honeycomb (
   $opendaylight_ip       = '',
   $opendaylight_port     = '8081',
@@ -50,8 +58,11 @@ class fdio::honeycomb (
   $password              = 'admin',
   $bind_ip               = '127.0.0.1',
   $node_id               = $::fqdn,
+  $interface_role_map    = [],
 ) {
 
+  validate_array($interface_role_map)
+
   include ::fdio
 
   package { 'honeycomb':
@@ -91,15 +102,23 @@ class fdio::honeycomb (
       tries     => 5,
       try_sleep => 30,
       path      => '/usr/sbin:/usr/bin:/sbin:/bin',
-      notify    => Service['honeycomb'],
-    }
-
+      require   => Service['honeycomb'],
+    }->
     exec { 'Check VPP was mounted into ODL operational DS':
       command   => "curl --fail -u ${opendaylight_username}:${opendaylight_password} ${oper_mount_url} | grep ${node_id}",
       tries     => 5,
       try_sleep => 30,
       path      => '/usr/sbin:/usr/bin:/sbin:/bin',
-      subscribe => Service['honeycomb'],
+    }
+
+    if !empty($interface_role_map) {
+      configure_role_mappings { $interface_role_map:
+        honeycomb_username => $user,
+        honeycomb_password => $password,
+        honeycomb_url      => "http://${bind_ip}:${rest_port}",
+        require            => Service['honeycomb'],
+        before             => Exec['VPP Mount into ODL'],
+      }
     }
   }
 }
diff --git a/manifests/honeycomb/configure_role_mappings.pp b/manifests/honeycomb/configure_role_mappings.pp
new file mode 100644
index 0000000..826f6e6
--- /dev/null
+++ b/manifests/honeycomb/configure_role_mappings.pp
@@ -0,0 +1,40 @@
+# == Define: fdio::honeycomb::configure_role_mappings
+#
+# Defined type to configure interface role mapping in OpenDaylight
+#
+# === Parameters:
+# [*honeycomb_username*]
+#   User name for Honeycomb.
+#
+# [*honeycomb_password*]
+#   Password for Honeycomb.
+#
+# [*honeycomb_url*]
+#   Honeycomb restconf binding URL.
+#
+# [*interface_role_mapping*]
+#   List of interface role mapping in the format
+#              of <VPP interface name>:<role name>
+#   Example:
+#      [ 'GigabitEthernet0/5/0:public-interface',
+#        'GigabitEthernet0/6/0:tenant-interface' ]
+#
+define fdio::honeycomb::configure_role_mappings (
+  $honeycomb_username,
+  $honeycomb_password,
+  $honeycomb_url,
+  $interface_role_mapping = $title,
+) {
+  $mapping = split($interface_role_mapping, ':')
+  $vpp_int = regsubst($mapping[0], '/', '%2F', 'G')
+  $role_name = $mapping[1]
+  $config_url = "${honeycomb_url}/restconf/config/ietf-interfaces:interfaces/interface/${vpp_int}"
+
+  exec { "Register interface ${mapping[0]} with role ${role_name}":
+    command   => "curl -XPOST --fail -H 'Content-Type: application/json' -u ${honeycomb_username}:${honeycomb_password} ${config_url} -d \"{'description': '${role_name}'}\"",
+    tries     => 5,
+    try_sleep => 30,
+    path      => '/usr/sbin:/usr/bin:/sbin:/bin',
+    unless    => "curl -u ${honeycomb_username}:${honeycomb_password} ${config_url} | grep '${role_name}'",
+  }
+}