init

cmd/kgctl/graph.go

@@ -0,0 +1,58 @@
+// Copyright 2019 the Kilo authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package main
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"github.com/squat/kilo/pkg/mesh"
+)
+
+func newGraph() *cobra.Command {
+	return &cobra.Command{
+		Use:   "graph",
+		Short: "Generates a graph of the Kilo network",
+		Long:  "",
+		RunE:  runGraph,
+	}
+}
+
+func runGraph(_ *cobra.Command, _ []string) error {
+	ns, err := opts.backend.List()
+	if err != nil {
+		return fmt.Errorf("failed to list nodes: %v", err)
+	}
+	var hostname string
+	if len(ns) != 0 {
+		hostname = ns[0].Name
+	}
+	nodes := make(map[string]*mesh.Node)
+	for _, n := range ns {
+		if n.Ready() {
+			nodes[n.Name] = n
+		}
+	}
+	t, err := mesh.NewTopology(nodes, opts.granularity, hostname, 0, []byte{}, opts.subnet)
+	if err != nil {
+		return fmt.Errorf("failed to create topology: %v", err)
+	}
+	g, err := t.Dot()
+	if err != nil {
+		return fmt.Errorf("failed to generate graph: %v", err)
+	}
+	fmt.Println(g)
+	return nil
+}