cmd.go

package main

import (
	"context"
	"fmt"
)

type command struct {
	name string
	args []string
}

type commands struct {
	handlers map[string]func(context.Context, *state, command) error
}

func (c *commands) register(name string, f func(context.Context, *state, command) error) {
	c.handlers[name] = f
}

func (c *commands) run(ctx context.Context, s *state, cmd command) error {

	f, exists := c.handlers[cmd.name]
	if !exists {
		return fmt.Errorf("no %s command found", cmd.name)
	}

	return f(ctx, s, cmd)
}

func getCliCommands() *commands {
	// set up and register commands
	cmds := commands{
		handlers: make(map[string]func(context.Context, *state, command) error),
	}

	cmds.register("login", handlerLogin)
	cmds.register("register", handlerRegister)
	cmds.register("reset", handlerReset)
	cmds.register("users", handlerGetUsers)
	cmds.register("agg", handlerFetchFeed)
	cmds.register("addfeed", middlewareLoggedIn(handlerAddFeed))
	cmds.register("feeds", handlerListFeeds)
	cmds.register("follow", middlewareLoggedIn(handlerFollowFeed))
	cmds.register("following", middlewareLoggedIn(handlerFeedFollowsForUser))
	cmds.register("unfollow", middlewareLoggedIn(handlerDeleteFeedFollow))
	cmds.register("browse", middlewareLoggedIn(handlerListPosts))

	return &cmds
}