list.go

package main

import (
	"fmt"

	"github.com/spf13/afero"
	"github.com/spf13/cobra"
	"github.com/srvc/fail/v4"
)

type SortKey int

const (
	_ SortKey = iota
	Updated
	Created
	Number
	Stars
	Watches
	Comments
	BestMatch
)

func sortKeyFromString(key string) (SortKey, error) {
	switch key {
	case "updated":
		return Updated, nil
	case "created":
		return Created, nil
	case "number":
		return Number, nil
	case "stars":
		return Stars, nil
	case "watches":
		return Watches, nil
	case "comments":
		return Comments, nil
	case "best_match":
		return BestMatch, nil
	}

	return 0, fail.New(fmt.Sprintf("Unknown SortKey: %s", key))
}

func NewListSubCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "list",
		Short: "list posts",
		RunE: func(cmd *cobra.Command, args []string) error {
			// Setup
			ctx := cmd.Context()
			fs := afero.NewOsFs()
			diApp, err := NewDiApp(ctx, fs)
			if err != nil {
				return fail.Wrap(err)
			}
			client := diApp.Client

			count, err := cmd.Flags().GetInt("count")
			if err != nil {
				return fail.Wrap(err)
			}
			sortKeyName, err := cmd.Flags().GetString("sort")
			if err != nil {
				return fail.Wrap(err)
			}
			sortKey, err := sortKeyFromString(sortKeyName)
			if err != nil {
				return fail.Wrap(err)
			}
			onlyTemplates, err := cmd.Flags().GetBool("only-templates")
			if err != nil {
				return fail.Wrap(err)
			}

			var posts []*Post
			if onlyTemplates {
				posts, err = client.GetTemplatePosts(cmd.Context(), count, sortKey)
				if err != nil {
					return fail.Wrap(err)
				}
			} else {
				posts, err = client.GetPosts(cmd.Context(), count, sortKey)
				if err != nil {
					return fail.Wrap(err)
				}
			}

			for _, post := range posts {
				fmt.Printf("%d: %s\n", post.Number, post.FullName)
			}
			return nil
		},
	}

	cmd.Flags().IntP("count", "c", 20, "Only print the number of posts")
	cmd.Flags().String("sort", "updated", "Sort key [updated:created:number:stars:watches:comments:best_match]")
	cmd.Flags().Bool("only-templates", false, "Show only template posts")

	cmd.AddCommand(
		newListTemplatesSubCmd(),
	)

	return cmd
}

func newListTemplatesSubCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use: "templates",
		RunE: func(cmd *cobra.Command, args []string) error {
			ctx := cmd.Context()
			fs := afero.NewOsFs()
			diApp, err := NewDiApp(ctx, fs)
			if err != nil {
				return fail.Wrap(err)
			}
			client := diApp.Client

			count, err := cmd.Flags().GetInt("count")
			if err != nil {
				return fail.Wrap(err)
			}
			sortKeyName, err := cmd.Flags().GetString("sort")
			if err != nil {
				return fail.Wrap(err)
			}
			sortKey, err := sortKeyFromString(sortKeyName)
			if err != nil {
				return fail.Wrap(err)
			}

			var posts []*Post
			posts, err = client.GetTemplatePosts(cmd.Context(), count, sortKey)
			if err != nil {
				return fail.Wrap(err)
			}

			for _, post := range posts {
				fmt.Printf("%d: %s\n", post.Number, post.FullName)
			}
			return nil
		},
	}
	cmd.Flags().IntP("count", "c", 20, "Only print the number of posts")
	cmd.Flags().String("sort", "updated", "Sort key [updated:created:number:stars:watches:comments:best_match]")
	return cmd
}