ipfs-cluster/cmd/ipfs-cluster-ctl/formatters.go

package main

import (
	"encoding/json"
	"errors"
	"fmt"
	"sort"
	"strings"
	"time"

	"github.com/ipfs/ipfs-cluster/api"
	peer "github.com/libp2p/go-libp2p-peer"
)

func jsonFormatObject(resp interface{}) {
	switch resp.(type) {
	case nil:
		return
	case api.ID:
		jsonFormatPrint(resp.(api.ID).ToSerial())
	case api.GlobalPinInfo:
		jsonFormatPrint(resp.(api.GlobalPinInfo).ToSerial())
	case api.Pin:
		jsonFormatPrint(resp.(api.Pin).ToSerial())
	case api.AddedOutput:
		jsonFormatPrint(resp.(api.AddedOutput))
	case api.Version:
		jsonFormatPrint(resp.(api.Version))
	case api.Metric:
		serial := resp.(api.Metric)
		textFormatPrintMetric(&serial)
	case api.Error:
		jsonFormatPrint(resp.(api.Error))
	case []api.ID:
		r := resp.([]api.ID)
		serials := make([]api.IDSerial, len(r), len(r))
		for i, item := range r {
			serials[i] = item.ToSerial()
		}
		jsonFormatPrint(serials)

	case []api.GlobalPinInfo:
		r := resp.([]api.GlobalPinInfo)
		serials := make([]api.GlobalPinInfoSerial, len(r), len(r))
		for i, item := range r {
			serials[i] = item.ToSerial()
		}
		jsonFormatPrint(serials)
	case []api.Pin:
		r := resp.([]api.Pin)
		serials := make([]api.PinSerial, len(r), len(r))
		for i, item := range r {
			serials[i] = item.ToSerial()
		}
		jsonFormatPrint(serials)
	case []api.AddedOutput:
		serials := resp.([]api.AddedOutput)
		jsonFormatPrint(serials)
	case []api.Metric:
		serials := resp.([]api.Metric)
		jsonFormatPrint(serials)
	default:
		checkErr("", errors.New("unsupported type returned"))
	}
}

func jsonFormatPrint(obj interface{}) {
	j, err := json.MarshalIndent(obj, "", "    ")
	checkErr("generating json output", err)
	fmt.Printf("%s\n", j)
}

func textFormatObject(resp interface{}) {
	switch resp.(type) {
	case nil:
		return
	case api.ID:
		serial := resp.(api.ID).ToSerial()
		textFormatPrintIDSerial(&serial)
	case api.GlobalPinInfo:
		serial := resp.(api.GlobalPinInfo).ToSerial()
		textFormatPrintGPInfo(&serial)
	case api.Pin:
		serial := resp.(api.Pin).ToSerial()
		textFormatPrintPin(&serial)
	case api.AddedOutput:
		serial := resp.(api.AddedOutput)
		textFormatPrintAddedOutput(&serial)
	case api.Version:
		serial := resp.(api.Version)
		textFormatPrintVersion(&serial)
	case api.Error:
		serial := resp.(api.Error)
		textFormatPrintError(&serial)
	case api.Metric:
		serial := resp.(api.Metric)
		textFormatPrintMetric(&serial)
	case []api.ID:
		for _, item := range resp.([]api.ID) {
			textFormatObject(item)
		}
	case []api.GlobalPinInfo:
		for _, item := range resp.([]api.GlobalPinInfo) {
			textFormatObject(item)
		}
	case []api.Pin:
		for _, item := range resp.([]api.Pin) {
			textFormatObject(item)
		}
	case []api.AddedOutput:
		for _, item := range resp.([]api.AddedOutput) {
			textFormatObject(item)
		}
	case []api.Metric:
		for _, item := range resp.([]api.Metric) {
			textFormatObject(item)
		}
	default:
		checkErr("", errors.New("unsupported type returned"))
	}
}

func textFormatPrintIDSerial(obj *api.IDSerial) {
	if obj.Error != "" {
		fmt.Printf("%s | ERROR: %s\n", obj.ID, obj.Error)
		return
	}

	fmt.Printf("%s | %s | Sees %d other peers\n", obj.ID, obj.Peername, len(obj.ClusterPeers)-1)
	addrs := make(sort.StringSlice, 0, len(obj.Addresses))
	for _, a := range obj.Addresses {
		addrs = append(addrs, string(a))
	}
	addrs.Sort()
	fmt.Println("  > Addresses:")
	for _, a := range addrs {
		fmt.Printf("    - %s\n", a)
	}
	if obj.IPFS.Error != "" {
		fmt.Printf("  > IPFS ERROR: %s\n", obj.IPFS.Error)
		return
	}

	ipfsAddrs := make(sort.StringSlice, 0, len(obj.Addresses))
	for _, a := range obj.IPFS.Addresses {
		ipfsAddrs = append(ipfsAddrs, string(a))
	}
	ipfsAddrs.Sort()
	fmt.Printf("  > IPFS: %s\n", obj.IPFS.ID)
	for _, a := range ipfsAddrs {
		fmt.Printf("    - %s\n", a)
	}
}

func textFormatPrintGPInfo(obj *api.GlobalPinInfoSerial) {
	fmt.Printf("%s :\n", obj.Cid)
	peers := make(sort.StringSlice, 0, len(obj.PeerMap))
	for k := range obj.PeerMap {
		peers = append(peers, k)
	}
	peers.Sort()

	for _, k := range peers {
		v := obj.PeerMap[k]
		if len(v.PeerName) > 0 {
			fmt.Printf("    > %-15s : %s", v.PeerName, strings.ToUpper(v.Status))
		} else {
			fmt.Printf("    > %-15s : %s", k, strings.ToUpper(v.Status))
		}
		if v.Error != "" {
			fmt.Printf(": %s", v.Error)
		}
		fmt.Printf(" | %s\n", v.TS)
	}
}

func textFormatPrintPInfo(obj *api.PinInfoSerial) {
	gpinfo := api.GlobalPinInfoSerial{
		Cid: obj.Cid,
		PeerMap: map[string]api.PinInfoSerial{
			obj.Peer: *obj,
		},
	}
	textFormatPrintGPInfo(&gpinfo)
}

func textFormatPrintVersion(obj *api.Version) {
	fmt.Println(obj.Version)
}

func textFormatPrintPin(obj *api.PinSerial) {
	fmt.Printf("%s | %s | %s | ", obj.Cid, obj.Name, strings.ToUpper(obj.ToPin().Type.String()))

	if obj.ReplicationFactorMin < 0 {
		fmt.Printf("Repl. Factor: -1 | Allocations: [everywhere]")
	} else {
		var sortAlloc sort.StringSlice = obj.Allocations
		sortAlloc.Sort()
		fmt.Printf("Repl. Factor: %d--%d | Allocations: %s",
			obj.ReplicationFactorMin, obj.ReplicationFactorMax,
			sortAlloc)
	}
	var recStr string
	switch obj.MaxDepth {
	case 0:
		recStr = "Direct"
	case -1:
		recStr = "Recursive"
	default:
		recStr = fmt.Sprintf("Recursive-%d", obj.MaxDepth)
	}

	fmt.Printf(" | %s\n", recStr)
}

func textFormatPrintAddedOutput(obj *api.AddedOutput) {
	fmt.Printf("added %s %s\n", obj.Cid, obj.Name)
}

func textFormatPrintMetric(obj *api.Metric) {
	date := time.Unix(0, obj.Expire).UTC().Format(time.RFC3339)
	fmt.Printf("%s: %s | Expire: %s\n", peer.IDB58Encode(obj.Peer), obj.Value, date)
}

func textFormatPrintError(obj *api.Error) {
	fmt.Printf("An error occurred:\n")
	fmt.Printf("  Code: %d\n", obj.Code)
	fmt.Printf("  Message: %s\n", obj.Message)
}