client.go

package ciena

import (
	"bytes"
	"code.fbi.h-da.de/cocsn/gosdn/database"
	"code.fbi.h-da.de/cocsn/gosdn/log"
	"code.fbi.h-da.de/cocsn/gosdn/restconf/util"
	apiclient "code.fbi.h-da.de/cocsn/swagger/apis/mcp/client"
	"crypto/tls"
	"github.com/go-openapi/runtime"
	httptransport "github.com/go-openapi/runtime/client"
	"github.com/go-openapi/strfmt"
	"net/http"
)

//MCPClient handles requests to a Ciena MCP RESTCONF endpoint
type MCPClient struct {
	transport *httptransport.Runtime
	client    *apiclient.ServiceTopologyTAPI
	database  *database.Database
	buffer    *bytes.Buffer
}

func (c MCPClient) GetConfig() string {
	//TODO: Fill with life
	return "..."
}

//NewMCPClient creates a client
func NewMCPClient(endpoint, username, password string, database *database.Database) *MCPClient {
	// create the transport
	transport := httptransport.New(endpoint, "/", nil)
	transport.Transport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
	transport.Debug = true
	// create the API client, with the transport
	basicAuth := httptransport.BasicAuth(username, password)
	// authenticate client
	transport.DefaultAuthentication = basicAuth
	client := apiclient.New(transport, strfmt.Default)

	buffer := new(bytes.Buffer)
	transport.Consumers[runtime.JSONMime] = util.YANGConsumer{Data: buffer}

	return &MCPClient{
		transport: transport,
		client:    client,
		database:  database,
		buffer:    buffer,
	}
}

func (c *MCPClient) GetConnections() error {
	defer c.buffer.Reset()
	_, err := c.client.TapiConnectivityCore.GetTapiCoreContextConnection(nil)
	c.database.StoreConnections(c.buffer.String())
	log.Debug(c.buffer.Next(25))
	return err
}

func (c *MCPClient) GetLinks() error {
	defer c.buffer.Reset()
	_, err := c.client.TapiTopologyCore.GetTapiCoreContextTopologyMcpBaseTopologyLink(nil)
	c.database.StoreConnections(c.buffer.String())
	log.Debug(c.buffer.Next(25))
	return err
}

func (c *MCPClient) GetNodes() error {
	defer c.buffer.Reset()
	_, err := c.client.TapiTopologyCore.GetTapiCoreContextTopologyMcpBaseTopologyNode(nil)
	c.database.StoreNodes(c.buffer.String())
	log.Debug(c.buffer.Next(25))
	return err
}

func (c *MCPClient) GetNodeEdgePoints() error {
	defer c.buffer.Reset()
	_, err := c.client.TapiTopologyCore.GetTapiCoreContextTopologyMcpBaseTopologyNodeEdgePoint(nil)
	c.database.StoreNodeEdgePoints(c.buffer.String())
	log.Debug(c.buffer.Next(25))
	return err
}