capabilities.go

package cli

import (
	"context"
	"fmt"
	"strings"

	spb "code.fbi.h-da.de/cocsn/api/go/gosdn/southbound"
	tpb "code.fbi.h-da.de/cocsn/api/go/gosdn/transport"

	"code.fbi.h-da.de/cocsn/gosdn/nucleus"
	gpb "github.com/openconfig/gnmi/proto/gnmi"
)

// Capabilities sends a gNMI Capabilities request to the specified target
// and prints the supported models to stdout
func Capabilities(a, u, p string) error {
	opts := &tpb.TransportOption{
		Address:  a,
		Username: u,
		Password: p,
	}
	transport, err := nucleus.NewGnmiTransport(opts, nucleus.NewSBI(spb.Type_OPENCONFIG))
	if err != nil {
		return err
	}
	resp, err := transport.Capabilities(context.Background())
	if err != nil {
		return err
	}
	modelData := resp.(*gpb.CapabilityResponse).SupportedModels
	b := strings.Builder{}
	for _, elem := range modelData {
		_, err := b.WriteString(elem.Name)
		if err != nil {
			return err
		}
		_, err = b.WriteString("\n")
		if err != nil {
			return err
		}
	}
	fmt.Println(b.String())
	return nil
}