Improved upvotes coloring, added dates to questions and answers

Author: shadowy-pycoder

cmd/goso/cli.go

@@ -13,7 +13,11 @@ import (
 	"github.com/shadowy-pycoder/goso"
 )
 
-const app string = "goso"
+const (
+	app                  string = "goso"
+	questionCountDefault int    = 10
+	answerCountDefault   int    = 3
+)
 
 const usagePrefix string = `                                                                  
  .d88b.   .d88b.  .d8888b   .d88b.  
@@ -56,7 +60,7 @@ func root(args []string) error {
 	}
 	q, set := os.LookupEnv("GOSO_QUESTIONS")
 	if !set {
-		qn = 1
+		qn = questionCountDefault
 	} else {
 		qn, err = strconv.Atoi(q)
 		if err != nil {
@@ -68,7 +72,7 @@ func root(args []string) error {
 	}
 	a, set := os.LookupEnv("GOSO_ANSWERS")
 	if !set {
-		an = 1
+		an = answerCountDefault
 	} else {
 		an, err = strconv.Atoi(a)
 		if err != nil {

goso.go

@@ -12,6 +12,7 @@ import (
 	"slices"
 	"strconv"
 	"strings"
+	"time"
 
 	"github.com/alecthomas/chroma/v2/formatters"
 	"github.com/alecthomas/chroma/v2/lexers"
@@ -30,6 +31,10 @@ const (
 	green         string = "\033[32m"
 	yellow        string = "\033[33m"
 	magenta       string = "\033[35m"
+	questionColor string = "\033[38;5;204m"
+	answerColor   string = "\033[38;5;255m"
+	downvoted     string = "\033[38;5;160m"
+	lightgray     string = "\033[38;5;248m"
 )
 
 var (
@@ -83,7 +88,9 @@ var (
 		"<code>", green,
 		"</code>", reset,
 		"<br />", "",
+		"<br/>", "",
 		"<hr />", "",
+		"<hr/>", "",
 		"<sup>", "",
 		"</sup>", "",
 		"<sub>", "",
@@ -235,35 +242,63 @@ type Config struct {
 	Client       *http.Client
 }
 type Answer struct {
-	Score int
-	Body  string
-	Link  string
+	Author     string
+	Score      int
+	Body       string
+	Link       string
+	IsAccepted bool
+	Date       time.Time
 }
 
 func (a *Answer) String() string {
 	line := strings.Repeat("─", 80)
+	color := yellow
+	if a.IsAccepted {
+		color = green
+	} else if a.Score < 0 {
+		color = downvoted
+	}
 	return fmt.Sprintf(`
 %s
-%s[%d]%s %s
+%s[%d]%s %sAnswer from %s%s%s
+%sDate: %s
+Link: %s%s
 %s
-`, line, yellow, a.Score, reset, a.Link, line)
+`,
+		line,
+		color, a.Score, reset, answerColor, bold, a.Author, reset,
+		lightgray, a.Date.Format(time.RFC822),
+		a.Link, reset,
+		line)
 }
 
 type Result struct {
 	Title       string
 	Link        string
 	QuestionId  int
 	UpvoteCount int
+	Date        time.Time
 	Answers     []*Answer
 }
 
 func (r *Result) String() string {
 	line := strings.Repeat("─", 80)
+	color := yellow
+	if r.UpvoteCount < 0 {
+		color = downvoted
+	}
+
 	return fmt.Sprintf(`
 %s
-%s[%d]%s %s
-%s
-%s`, line, yellow, r.UpvoteCount, reset, r.Title, r.Link, line)
+%s[%d]%s %s%s%s%s
+%sDate: %s
+Link: %s%s
+%s`,
+		line,
+		color, r.UpvoteCount, reset, bold, questionColor, r.Title, reset,
+		lightgray, r.Date.Format(time.RFC822),
+		r.Link, reset,
+		line)
 }
 
 func prepareText(text string) string {
@@ -307,13 +342,15 @@ func FetchStackOverflow(conf *Config, gr *GoogleSearchResult) (map[int]*Result,
 	questions := make([]string, 0, len(gr.Items))
 	for _, item := range gr.Items {
 		var upvoteCount int
+		var dateCreated time.Time
 		if len(item.Pagemap.Question) > 0 {
 			question := item.Pagemap.Question[0]
 			answerCount, _ := strconv.Atoi(question.Answercount)
 			if answerCount == 0 {
 				continue
 			}
 			upvoteCount, _ = strconv.Atoi(question.Upvotecount)
+			dateCreated, _ = time.Parse("2006-01-02T15:04:05", question.Datecreated)
 		}
 		u, _ := netUrl.Parse(item.Link)
 		questionStr := strings.Split(u.Path, "/")[2]
@@ -324,6 +361,7 @@ func FetchStackOverflow(conf *Config, gr *GoogleSearchResult) (map[int]*Result,
 			Link:        item.Link,
 			QuestionId:  questionId,
 			UpvoteCount: upvoteCount,
+			Date:        dateCreated,
 		}
 	}
 	question_ids := strings.Join(questions, ";")
@@ -353,9 +391,12 @@ func FetchStackOverflow(conf *Config, gr *GoogleSearchResult) (map[int]*Result,
 		}
 		result.Answers = append(result.Answers,
 			&Answer{
-				Score: item.Score,
-				Body:  item.Body,
-				Link:  fmt.Sprintf("https://stackoverflow.com/a/%d", item.AnswerID),
+				Author:     item.Owner.DisplayName,
+				Score:      item.Score,
+				Body:       item.Body,
+				Link:       fmt.Sprintf("https://stackoverflow.com/a/%d", item.AnswerID),
+				IsAccepted: item.IsAccepted,
+				Date:       time.Unix(int64(item.CreationDate), 0).UTC(),
 			})
 	}
 	return results, nil