added netconf save for other vrnetlabs platfroms

srl-labs · Jun 28, 2021 · 0be0302 · 0be0302
1 parent 7311207
commit 0be0302
Show file tree

Hide file tree

Showing 5 changed files with 55 additions and 0 deletions.
diff --git a/nodes/vr_csr/vr-csr.go b/nodes/vr_csr/vr-csr.go
@@ -8,6 +8,7 @@ import (
 	"context"
 	"fmt"
 
+	log "github.com/sirupsen/logrus"
 	"github.com/srl-labs/containerlab/nodes"
 	"github.com/srl-labs/containerlab/runtime"
 	"github.com/srl-labs/containerlab/types"
@@ -66,5 +67,15 @@ func (s *vrCsr) WithMgmtNet(mgmt *types.MgmtNet) {
 }
 
 func (s *vrCsr) SaveConfig(ctx context.Context, r runtime.ContainerRuntime) error {
+	err := utils.SaveCfgViaNetconf(s.cfg.LongName,
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+	)
+
+	if err != nil {
+		return err
+	}
+
+	log.Infof("saved %s running configuration to startup configuration file\n", s.cfg.ShortName)
 	return nil
 }
diff --git a/nodes/vr_veos/vr-veos.go b/nodes/vr_veos/vr-veos.go
@@ -8,6 +8,7 @@ import (
 	"context"
 	"fmt"
 
+	log "github.com/sirupsen/logrus"
 	"github.com/srl-labs/containerlab/nodes"
 	"github.com/srl-labs/containerlab/runtime"
 	"github.com/srl-labs/containerlab/types"
@@ -65,5 +66,15 @@ func (s *vrVEOS) PostDeploy(ctx context.Context, r runtime.ContainerRuntime, ns
 func (s *vrVEOS) WithMgmtNet(mgmt *types.MgmtNet) { s.mgmt = mgmt }
 
 func (s *vrVEOS) SaveConfig(ctx context.Context, r runtime.ContainerRuntime) error {
+	err := utils.SaveCfgViaNetconf(s.cfg.LongName,
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+	)
+
+	if err != nil {
+		return err
+	}
+
+	log.Infof("saved %s running configuration to startup configuration file\n", s.cfg.ShortName)
 	return nil
 }
diff --git a/nodes/vr_vmx/vr-vmx.go b/nodes/vr_vmx/vr-vmx.go
@@ -8,6 +8,7 @@ import (
 	"context"
 	"fmt"
 
+	log "github.com/sirupsen/logrus"
 	"github.com/srl-labs/containerlab/nodes"
 	"github.com/srl-labs/containerlab/runtime"
 	"github.com/srl-labs/containerlab/types"
@@ -69,5 +70,15 @@ func (s *vrVMX) PostDeploy(ctx context.Context, r runtime.ContainerRuntime, ns m
 func (s *vrVMX) WithMgmtNet(mgmt *types.MgmtNet) { s.mgmt = mgmt }
 
 func (s *vrVMX) SaveConfig(ctx context.Context, r runtime.ContainerRuntime) error {
+	err := utils.SaveCfgViaNetconf(s.cfg.LongName,
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+	)
+
+	if err != nil {
+		return err
+	}
+
+	log.Infof("saved %s running configuration to startup configuration file\n", s.cfg.ShortName)
 	return nil
 }
diff --git a/nodes/vr_xrv/vr-xrv.go b/nodes/vr_xrv/vr-xrv.go
@@ -8,6 +8,7 @@ import (
 	"context"
 	"fmt"
 
+	log "github.com/sirupsen/logrus"
 	"github.com/srl-labs/containerlab/nodes"
 	"github.com/srl-labs/containerlab/runtime"
 	"github.com/srl-labs/containerlab/types"
@@ -68,5 +69,15 @@ func (s *vrXRV) PostDeploy(ctx context.Context, r runtime.ContainerRuntime, ns m
 func (s *vrXRV) WithMgmtNet(mgmt *types.MgmtNet) { s.mgmt = mgmt }
 
 func (s *vrXRV) SaveConfig(ctx context.Context, r runtime.ContainerRuntime) error {
+	err := utils.SaveCfgViaNetconf(s.cfg.LongName,
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+	)
+
+	if err != nil {
+		return err
+	}
+
+	log.Infof("saved %s running configuration to startup configuration file\n", s.cfg.ShortName)
 	return nil
 }
diff --git a/nodes/vr_xrv9k/vr-xrv9k.go b/nodes/vr_xrv9k/vr-xrv9k.go
@@ -8,6 +8,7 @@ import (
 	"context"
 	"fmt"
 
+	log "github.com/sirupsen/logrus"
 	"github.com/srl-labs/containerlab/nodes"
 	"github.com/srl-labs/containerlab/runtime"
 	"github.com/srl-labs/containerlab/types"
@@ -71,5 +72,15 @@ func (s *vrXRV9K) PostDeploy(ctx context.Context, r runtime.ContainerRuntime, ns
 func (s *vrXRV9K) WithMgmtNet(mgmt *types.MgmtNet) { s.mgmt = mgmt }
 
 func (s *vrXRV9K) SaveConfig(ctx context.Context, r runtime.ContainerRuntime) error {
+	err := utils.SaveCfgViaNetconf(s.cfg.LongName,
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+		nodes.DefaultCredentials[s.cfg.Kind][0],
+	)
+
+	if err != nil {
+		return err
+	}
+
+	log.Infof("saved %s running configuration to startup configuration file\n", s.cfg.ShortName)
 	return nil
 }