#9 - feed follows

internal/handlers/addFeedHandler.go

@@ -12,7 +12,7 @@ import (
 	"time"
 )
 
-func AddFeedHandler(s *config.State, cmd config.Command) error {
+func AddFeedHandler(s *config.State, cmd config.Command, user database.User) error {
 
 	if len(cmd.Args) < 2 {
 		log.Fatal("usage: boot-dev-blog-aggregator addfeed NAME URL")
@@ -45,6 +45,17 @@ func AddFeedHandler(s *config.State, cmd config.Command) error {
 		return err
 	}
 
+	_, err = s.DB.CreateFeedFollow(ctx, database.CreateFeedFollowParams{
+		ID:        uuid.New(),
+		CreatedAt: time.Time{},
+		UpdatedAt: time.Time{},
+		UserID:    user.ID,
+		FeedID:    f.ID,
+	})
+	if err != nil {
+		return err
+	}
+
 	fmt.Printf("%v", f)
 
 	return nil

internal/handlers/feedFollowsByUserHandler.go

@@ -0,0 +1,27 @@
+package handlers
+
+import (
+	"context"
+	"fmt"
+	"github.com/rdarius/boot-dev-blog-aggregator/internal/config"
+)
+
+func GetFeedFollowsByUserHandler(s *config.State, cmd config.Command) error {
+	ctx := context.Background()
+	user, err := s.DB.GetUser(ctx, s.Config.CurrentUserName)
+	if err != nil {
+		return err
+	}
+
+	ff, err := s.DB.GetFeedFollowsByUser(ctx, user.ID)
+
+	if err != nil {
+		return err
+	}
+
+	for _, feed := range ff {
+		fmt.Printf("* %s\n", feed.FeedName)
+	}
+
+	return nil
+}

internal/handlers/followFeedHandler.go

@@ -0,0 +1,66 @@
+package handlers
+
+import (
+	"context"
+	"fmt"
+	"github.com/google/uuid"
+	"github.com/rdarius/boot-dev-blog-aggregator/internal/config"
+	"github.com/rdarius/boot-dev-blog-aggregator/internal/database"
+	"github.com/rdarius/boot-dev-blog-aggregator/internal/rss"
+	"log"
+	"os"
+	"time"
+)
+
+func FollowFeedHandler(s *config.State, cmd config.Command) error {
+	if len(cmd.Args) < 1 {
+		log.Fatal("usage: boot-dev-blog-aggregator follow URL")
+	}
+	ctx := context.Background()
+	user, err := s.DB.GetUser(ctx, s.Config.CurrentUserName)
+	if err != nil {
+		return err
+	}
+
+	url := cmd.Args[0]
+
+	ff, err := s.DB.GetFeedByUrl(ctx, url)
+	if err != nil {
+		// no feed, get it and save it
+
+		fd, err := rss.FetchFeed(ctx, url)
+		if err != nil {
+			fmt.Printf("Failed to fetch feed: %v\n", err)
+			os.Exit(1)
+		}
+
+		res, err := s.DB.CreateFeed(ctx, database.CreateFeedParams{
+			ID:        uuid.New(),
+			CreatedAt: time.Time{},
+			UpdatedAt: time.Time{},
+			UserID:    user.ID,
+			Url:       url,
+			Name:      fd.Channel.Title,
+		})
+		if err != nil {
+			fmt.Printf("Failed to create feed: %v\n", err)
+			os.Exit(1)
+		}
+		ff = res
+	}
+
+	feed, err := s.DB.CreateFeedFollow(ctx, database.CreateFeedFollowParams{
+		ID:        uuid.New(),
+		CreatedAt: time.Time{},
+		UpdatedAt: time.Time{},
+		UserID:    user.ID,
+		FeedID:    ff.ID,
+	})
+	if err != nil {
+		return err
+	}
+
+	fmt.Printf("%v", feed)
+
+	return nil
+}