cmd_chat_featured_bots.go

package client

import (
	"fmt"
	"strconv"
	"strings"

	"github.com/keybase/cli"
	"github.com/adamwalz/keybase-client/go/flexibletable"
	"github.com/adamwalz/keybase-client/go/libcmdline"
	"github.com/adamwalz/keybase-client/go/libkb"
	"github.com/adamwalz/keybase-client/go/protocol/keybase1"
	"github.com/kyokomi/emoji"
	context "golang.org/x/net/context"
)

const defaultFeaturedBotsLimit = 10

type CmdChatFeaturedBots struct {
	libkb.Contextified
	limit int
	page  int
}

func NewCmdChatFeaturedBotsRunner(g *libkb.GlobalContext) *CmdChatFeaturedBots {
	return &CmdChatFeaturedBots{
		Contextified: libkb.NewContextified(g),
	}
}

func newCmdChatFeaturedBots(cl *libcmdline.CommandLine, g *libkb.GlobalContext) cli.Command {
	return cli.Command{
		Name:  "featured-bots",
		Usage: "List featured chat bots available on Keybase.",
		Action: func(c *cli.Context) {
			cl.ChooseCommand(NewCmdChatFeaturedBotsRunner(g), "featured-bots", c)
			cl.SetNoStandalone()
			cl.SetLogForward(libcmdline.LogForwardNone)
		},
		Flags: []cli.Flag{
			cli.IntFlag{
				Name:  "n, number",
				Usage: fmt.Sprintf("Number of bots to display, defaults to %d", defaultFeaturedBotsLimit),
				Value: defaultFeaturedBotsLimit,
			},
			cli.IntFlag{
				Name:  "p, page",
				Usage: "page",
			},
		},
	}
}

func (c *CmdChatFeaturedBots) Run() (err error) {
	cli, err := GetFeaturedBotsClient(c.G())
	if err != nil {
		return err
	}

	offset := c.limit * c.page
	res, err := cli.FeaturedBots(context.Background(), keybase1.FeaturedBotsArg{
		Limit:     c.limit,
		Offset:    offset,
		SkipCache: true,
	})
	if err != nil {
		return err
	}
	err = displayFeaturedBots(c.G(), res.Bots)
	return err
}

func (c *CmdChatFeaturedBots) ParseArgv(ctx *cli.Context) (err error) {
	c.limit = ctx.Int("number")
	c.page = ctx.Int("page")
	return nil
}

func (c *CmdChatFeaturedBots) GetUsage() libkb.Usage {
	return libkb.Usage{
		Config: true,
		API:    true,
	}
}

func displayFeaturedBots(g *libkb.GlobalContext, bots []keybase1.FeaturedBot) error {
	ui := g.UI.GetTerminalUI()
	if len(bots) == 0 {
		return ui.Output("Not bots found\n")
	}

	if err := ui.Output("To add a bot see `keybase chat add-bot-member --help`\n"); err != nil {
		return err
	}
	table := &flexibletable.Table{}
	for i, bot := range bots {
		err := table.Insert(flexibletable.Row{
			flexibletable.Cell{
				Frame:     [2]string{"[", "]"},
				Alignment: flexibletable.Right,
				Content:   flexibletable.SingleCell{Item: strconv.Itoa(i + 1)},
			},
			flexibletable.Cell{
				Alignment: flexibletable.Left,
				Content:   flexibletable.SingleCell{Item: bot.DisplayName()},
			},
			flexibletable.Cell{
				Alignment: flexibletable.Left,
				Content:   flexibletable.SingleCell{Item: fmt.Sprintf("by @%s", bot.Owner())},
			},
			flexibletable.Cell{
				Alignment: flexibletable.Left,
				Content:   flexibletable.SingleCell{Item: emoji.Sprint(bot.Description)},
			},
			flexibletable.Cell{
				Alignment: flexibletable.Left,
				Content:   flexibletable.SingleCell{Item: strings.ReplaceAll(emoji.Sprint(bot.ExtendedDescriptionRaw), "\n", " ") + "\n\n"},
			},
		})
		if err != nil {
			return err
		}
	}
	w, _ := ui.TerminalSize()
	if err := table.Render(ui.OutputWriter(), " ", w, []flexibletable.ColumnConstraint{
		5,                                 // visualIndex
		64,                                // displayName
		64,                                // ownerName
		256,                               // description
		flexibletable.ExpandableWrappable, // extendedDescription
	}); err != nil {
		return fmt.Errorf("rendering conversation info list view error: %v\n", err)
	}
	return nil
}