Added way for streaming redditor's comments

Author: thecsw

reddit.go

@@ -95,6 +95,8 @@ func (c *Reddit) Comments(sort string, tdur string, limit int) ([]Comment, error
 			return nil, err
 		}
 		return comments, nil
+	case "redditor":
+		return c.getRedditorComments(c.Chain.Name, sort, tdur, limit)
 	default:
 		return nil, fmt.Errorf("'%s' type does not have an option for comments", c.Chain.Type)
 	}
@@ -252,6 +254,30 @@ func (c *Reddit) getSubredditComments(sr string, sort string, tdur string, limit
 	return ret.GetChildren(), err
 }
 
+func (c *Reddit) getRedditorComments(user string, sort string, tdur string, limit int) ([]Comment, error) {
+	target := RedditOauth + "/u/" + user + "/comments.json"
+	ans, err := c.MiraRequest("GET", target, map[string]string{
+		"sort":  sort,
+		"limit": strconv.Itoa(limit),
+		"t":     tdur,
+	})
+	ret := &CommentListing{}
+	json.Unmarshal(ans, ret)
+	return ret.GetChildren(), err
+}
+
+func (c *Reddit) getRedditorCommentsAfter(user string, sort string, last string, limit int) ([]Comment, error) {
+	target := RedditOauth + "/u/" + user + "/comments.json"
+	ans, err := c.MiraRequest("GET", target, map[string]string{
+		"sort":   sort,
+		"limit":  strconv.Itoa(limit),
+		"before": last,
+	})
+	ret := &CommentListing{}
+	json.Unmarshal(ans, ret)
+	return ret.GetChildren(), err
+}
+
 func (c *Reddit) getSubmissionComments(post_id string, sort string, tdur string, limit int) ([]Comment, []string, error) {
 	if string(post_id[1]) != "3" {
 		return nil, nil, errors.New("the passed ID36 is not a submission")
@@ -304,6 +330,8 @@ func (c *Reddit) CommentsAfter(sort string, last string, limit int) ([]Comment,
 	switch c.Chain.Type {
 	case "subreddit":
 		return c.getSubredditCommentsAfter(c.Chain.Name, sort, last, limit)
+	case "redditor":
+		return c.getRedditorCommentsAfter(c.Chain.Name, sort, last, limit)
 	default:
 		return nil, fmt.Errorf("'%s' type does not have an option for comments", c.Chain.Type)
 	}

streaming.go

@@ -31,6 +31,20 @@ func (r *Reddit) StreamCommentReplies() (<-chan Comment, chan bool) {
 // // c is the channel with all comments
 // // stop is the channel to stop the stream. Do stop <- true to stop the loop
 func (r *Reddit) StreamComments() (<-chan Comment, chan bool, error) {
+	err := r.checkType("subreddit", "redditor")
+	if err != nil {
+		return nil, nil, err
+	}
+	switch r.Chain.Type {
+	case "subreddit":
+		return r.streamSubredditComments()
+	case "redditor":
+		return r.streamRedditorComments()
+	}
+	return nil, nil, nil
+}
+
+func (r *Reddit) streamSubredditComments() (<-chan Comment, chan bool, error) {
 	c := make(chan Comment, 25)
 	stop := make(chan bool, 1)
 	anchor, err := r.Subreddit(r.Chain.Name).Comments("new", "hour", 1)
@@ -45,7 +59,36 @@ func (r *Reddit) StreamComments() (<-chan Comment, chan bool, error) {
 		for {
 			stop <- false
 			un, _ := r.Subreddit(r.Chain.Name).CommentsAfter("new", last, 25)
-			//			un, _ := r.getSubredditCommentsAfter(sr, "new", last, 25)
+			for _, v := range un {
+				c <- v
+			}
+			if len(un) > 0 {
+				last = un[0].GetId()
+			}
+			time.Sleep(r.Stream.CommentListInterval * time.Second)
+			if <-stop {
+				return
+			}
+		}
+	}()
+	return c, stop, nil
+}
+
+func (r *Reddit) streamRedditorComments() (<-chan Comment, chan bool, error) {
+	c := make(chan Comment, 25)
+	stop := make(chan bool, 1)
+	anchor, err := r.Redditor(r.Chain.Name).Comments("new", "hour", 1)
+	if err != nil {
+		return nil, nil, err
+	}
+	last := ""
+	if len(anchor) > 0 {
+		last = anchor[0].GetId()
+	}
+	go func() {
+		for {
+			stop <- false
+			un, _ := r.Redditor(r.Chain.Name).CommentsAfter("new", last, 25)
 			for _, v := range un {
 				c <- v
 			}