// +build routerrpc

package main

import (
	"context"
	"encoding/hex"

	"github.com/lightningnetwork/lnd/lnrpc/routerrpc"

	"github.com/urfave/cli"
)

var queryMissionControlCommand = cli.Command{
	Name:     "querymc",
	Category: "Payments",
	Usage:    "Query the internal mission control state.",
	Action:   actionDecorator(queryMissionControl),
}

func queryMissionControl(ctx *cli.Context) error {
	conn := getClientConn(ctx, false)
	defer conn.Close()

	client := routerrpc.NewRouterClient(conn)

	req := &routerrpc.QueryMissionControlRequest{}
	rpcCtx := context.Background()
	snapshot, err := client.QueryMissionControl(rpcCtx, req)
	if err != nil {
		return err
	}

	type displayNodeHistory struct {
		Pubkey               string
		LastFailTime         int64
		OtherChanSuccessProb float32
		Channels             []*routerrpc.ChannelHistory
	}

	displayResp := struct {
		Nodes []displayNodeHistory
	}{}

	for _, n := range snapshot.Nodes {
		displayResp.Nodes = append(
			displayResp.Nodes,
			displayNodeHistory{
				Pubkey:               hex.EncodeToString(n.Pubkey),
				LastFailTime:         n.LastFailTime,
				OtherChanSuccessProb: n.OtherChanSuccessProb,
				Channels:             n.Channels,
			},
		)
	}

	printJSON(displayResp)

	return nil
}