Add feed command

cmd/feed.go

@@ -0,0 +1,101 @@
+package cmd
+
+import (
+	"database/sql"
+	"encoding/json"
+	"fmt"
+	"log"
+
+	"github.com/Jaculabilis/intake/core"
+	"github.com/spf13/cobra"
+)
+
+var feedCmd = &cobra.Command{
+	Use:   "feed",
+	Short: "Print item feeds",
+	Long: `Display the intake item feed in various formats.
+The default format is "headlines".
+
+Available formats:
+  headlines     Only item titles
+  json          Full item JSON
+  short         Item source and id
+`,
+	Run: func(cmd *cobra.Command, args []string) {
+		feed()
+	},
+}
+
+var feedFormat string
+var feedSource string
+var feedChannel string
+
+func init() {
+	rootCmd.AddCommand(feedCmd)
+
+	feedCmd.Flags().StringVarP(&feedFormat, "format", "f", "headlines", "Feed format")
+	feedCmd.Flags().StringVarP(&feedSource, "source", "s", "", "Limit to items from source")
+	feedCmd.Flags().StringVarP(&feedChannel, "channel", "c", "", "Limit to items from channel")
+	feedCmd.MarkFlagsMutuallyExclusive("source", "channel")
+}
+
+func feed() {
+	var formatter func(core.Item)
+	switch feedFormat {
+	case "headlines":
+		formatter = formatHeadline
+	case "json":
+		formatter = formatJson
+	case "short":
+		formatter = formatShort
+	default:
+		log.Fatalf("error: invalid format %s", feedFormat)
+	}
+
+	db, err := sql.Open("sqlite3", getDbPath())
+	if err != nil {
+		log.Fatalf("error: failed to open %s", dbPath)
+	}
+
+	core.InitDatabase(db)
+	core.MigrateDatabase(db)
+
+	var items []core.Item
+	if feedSource != "" {
+		items, err = core.GetActiveItemsForSource(db, feedSource)
+		if err != nil {
+			log.Fatalf("error: failed to fetch items from %s", feedSource)
+		}
+	} else if feedChannel != "" {
+		log.Fatal("error: unimplemented")
+	} else {
+		items, err = core.GetAllActiveItems(db)
+		if err != nil {
+			log.Fatal("error: failed to fetch items")
+		}
+	}
+
+	for _, item := range items {
+		formatter(item)
+	}
+}
+
+func formatHeadline(item core.Item) {
+	title := item.Title
+	if title == "" {
+		title = item.Id
+	}
+	fmt.Println(title)
+}
+
+func formatJson(item core.Item) {
+	data, err := json.Marshal(item)
+	if err != nil {
+		log.Fatalf("error: failed to serialize %s/%s: %v", item.Source, item.Id, err)
+	}
+	fmt.Println(string(data))
+}
+
+func formatShort(item core.Item) {
+	fmt.Printf("%s/%s\n", item.Source, item.Id)
+}

cmd/root.go

@@ -27,7 +27,7 @@ func init() {
 	rootCmd.SetHelpCommand(&cobra.Command{Hidden: true})
 
 	// All commands need to operate on a database
-	rootCmd.PersistentFlags().StringVarP(&dbPath, "db", "d", "", "Path to the intake sqlite database")
+	rootCmd.PersistentFlags().StringVarP(&dbPath, "db", "d", "", "Path to the intake sqlite database (default: INTAKE_DB)")
 }
 
 func getDbPath() string {

core/db.go

@@ -11,15 +11,15 @@ import (
 )
 
 type Item struct {
-	source  string
+	Source  string `json:"source"`
-	id      string
+	Id      string `json:"id"`
-	created int
+	Created int    `json:"created"`
-	active  bool
+	Active  bool   `json:"active"`
-	title   string
+	Title   string `json:"title"`
-	author  string
+	Author  string `json:"author"`
-	body    string
+	Body    string `json:"body"`
-	link    string
+	Link    string `json:"link"`
-	time    int
+	Time    int    `json:"time"`
 }
 
 //go:embed sql/*.sql
@@ -169,7 +169,34 @@ func DeactivateItem(db *sql.DB, source string, id string) (bool, error) {
 	return active, nil
 }
 
-func GetActiveItems(db *sql.DB, source string) ([]Item, error) {
+func GetAllActiveItems(db *sql.DB) ([]Item, error) {
+	rows, err := db.Query(`
+		select
+			source,
+			id,
+			created,
+			active,
+			title,
+			author,
+			body,
+			link,
+			time
+		from items
+		where active <> 0
+	`)
+	if err != nil {
+		return nil, err
+	}
+	var items []Item
+	for rows.Next() {
+		var item Item
+		rows.Scan(&item.Source, &item.Id, &item.Created, &item.Active, &item.Title, &item.Author, &item.Body, &item.Link, &item.Time)
+		items = append(items, item)
+	}
+	return items, nil
+}
+
+func GetActiveItemsForSource(db *sql.DB, source string) ([]Item, error) {
 	rows, err := db.Query(`
 		select
 			source,
@@ -192,7 +219,7 @@ func GetActiveItems(db *sql.DB, source string) ([]Item, error) {
 	var items []Item
 	for rows.Next() {
 		var item Item
-		rows.Scan(&item.source, &item.id, &item.created, &item.active, &item.title, &item.author, &item.body, &item.link, &item.time)
+		rows.Scan(&item.Source, &item.Id, &item.Created, &item.Active, &item.Title, &item.Author, &item.Body, &item.Link, &item.Time)
 		items = append(items, item)
 	}
 	return items, nil

core/db_test.go

@@ -98,14 +98,14 @@ func TestCreateSource(t *testing.T) {
 func AssertItemIs(t *testing.T, item Item, expected string) {
 	actual := fmt.Sprintf(
 		"%s/%s/%t/%s/%s/%s/%s/%d",
-		item.source,
+		item.Source,
-		item.id,
+		item.Id,
-		item.active,
+		item.Active,
-		item.title,
+		item.Title,
-		item.author,
+		item.Author,
-		item.body,
+		item.Body,
-		item.link,
+		item.Link,
-		item.time,
+		item.Time,
 	)
 	if actual != expected {
 		t.Fatalf("expected %s, got %s", expected, actual)
@@ -125,7 +125,7 @@ func TestAddItem(t *testing.T) {
 	if err := AddItem(db, "test", "two", "title", "author", "body", "link", 123456); err != nil {
 		t.Fatal(err)
 	}
-	items, err := GetActiveItems(db, "test")
+	items, err := GetActiveItemsForSource(db, "test")
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -138,7 +138,7 @@ func TestAddItem(t *testing.T) {
 	if _, err = DeactivateItem(db, "test", "one"); err != nil {
 		t.Fatal(err)
 	}
-	items, err = GetActiveItems(db, "test")
+	items, err = GetActiveItemsForSource(db, "test")
 	if err != nil {
 		t.Fatal(err)
 	}