mirror of https://github.com/edouardparis/lntop
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
139 lines
2.7 KiB
Go
139 lines
2.7 KiB
Go
package views
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/jroimartin/gocui"
|
|
|
|
"github.com/edouardparis/lntop/network"
|
|
"github.com/edouardparis/lntop/network/models"
|
|
"github.com/edouardparis/lntop/ui/color"
|
|
)
|
|
|
|
const (
|
|
CHANNELS = "channels"
|
|
CHANNELS_HEADER = "header"
|
|
)
|
|
|
|
type Channels struct {
|
|
*gocui.View
|
|
items []*models.Channel
|
|
network *network.Network
|
|
}
|
|
|
|
func (c *Channels) Set(g *gocui.Gui, x0, y0, x1, y1 int) error {
|
|
headerView, err := g.SetView(CHANNELS_HEADER, x0, y0, x1, y0+2)
|
|
if err != nil {
|
|
if err != gocui.ErrUnknownView {
|
|
return err
|
|
}
|
|
}
|
|
headerView.Frame = false
|
|
headerView.BgColor = gocui.ColorGreen
|
|
headerView.FgColor = gocui.ColorBlack | gocui.AttrBold
|
|
displayChannelsHeader(headerView)
|
|
|
|
c.View, err = g.SetView(CHANNELS, x0, y0+1, x1, y1)
|
|
if err != nil {
|
|
if err != gocui.ErrUnknownView {
|
|
return err
|
|
}
|
|
}
|
|
c.View.Frame = false
|
|
c.View.Autoscroll = true
|
|
c.View.SelBgColor = gocui.ColorCyan
|
|
c.View.SelFgColor = gocui.ColorBlack
|
|
|
|
err = c.update(context.Background())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.display()
|
|
return nil
|
|
}
|
|
|
|
func displayChannelsHeader(v *gocui.View) {
|
|
fmt.Fprintln(v, fmt.Sprintf("%-9s %12s %12s %-64s %-19s",
|
|
"status",
|
|
"local",
|
|
"capacity",
|
|
"pub_key",
|
|
"id",
|
|
))
|
|
}
|
|
|
|
func (c *Channels) Refresh(g *gocui.Gui) error {
|
|
var err error
|
|
c.View, err = g.View(CHANNELS)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = c.update(context.Background())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.display()
|
|
return nil
|
|
}
|
|
|
|
func (c *Channels) update(ctx context.Context) error {
|
|
channels, err := c.network.ListChannels(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.items = channels
|
|
return nil
|
|
}
|
|
|
|
func (c *Channels) display() {
|
|
for _, item := range c.items {
|
|
line := fmt.Sprintf("%s %s %12d %s %s",
|
|
active(item),
|
|
color.Cyan(fmt.Sprintf("%12d", item.LocalBalance)),
|
|
item.Capacity,
|
|
chartPubKey(item),
|
|
chartID(item),
|
|
)
|
|
fmt.Fprintln(c.View, line)
|
|
}
|
|
}
|
|
|
|
func NewChannels(network *network.Network) *Channels {
|
|
return &Channels{network: network}
|
|
}
|
|
|
|
func active(c *models.Channel) string {
|
|
if c.Active {
|
|
return color.Green(fmt.Sprintf("%-9s", "active"))
|
|
}
|
|
return color.Red(fmt.Sprintf("%-9s", "inactive"))
|
|
}
|
|
|
|
func chartID(c *models.Channel) string {
|
|
id := fmt.Sprintf("%-19d", c.ID)
|
|
index := int(c.LocalBalance * int64(len(id)) / c.Capacity)
|
|
|
|
var buffer bytes.Buffer
|
|
buffer.WriteString(color.Cyan(id[:index]))
|
|
buffer.WriteString(id[index:])
|
|
|
|
return buffer.String()
|
|
}
|
|
|
|
func chartPubKey(c *models.Channel) string {
|
|
pubkey := c.RemotePubKey
|
|
index := int(c.LocalBalance * int64(len(pubkey)) / c.Capacity)
|
|
|
|
var buffer bytes.Buffer
|
|
buffer.WriteString(color.Cyan(pubkey[:index]))
|
|
buffer.WriteString(pubkey[index:])
|
|
|
|
return buffer.String()
|
|
}
|