Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 27 additions & 6 deletions output/output.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ import (
type Format func(results github.GithubSearchResults, writer io.Writer, options top.Options) error

func PlainOutput(results github.GithubSearchResults, writer io.Writer, options top.Options) error {
users := GithubUserList(results.Users)
users := GithubUserList(results.Users).TopPrivate(options.Amount)
fmt.Fprintln(writer, "USERS\n--------")
for i, user := range users {
fmt.Fprintf(writer, "#%+v: %+v (%+v):%+v (%+v) %+v\n", i+1, user.Name, user.Login, user.ContributionCount, user.Company, strings.Join(user.Organizations, ","))
Expand All @@ -31,7 +31,7 @@ func PlainOutput(results github.GithubSearchResults, writer io.Writer, options t
}

func CsvOutput(results github.GithubSearchResults, writer io.Writer, options top.Options) error {
users := GithubUserList(results.Users)
users := GithubUserList(results.Users).TopPrivate(options.Amount)
w := csv.NewWriter(writer)
if err := w.Write([]string{"rank", "name", "login", "contributions", "company", "organizations"}); err != nil {
return err
Expand Down Expand Up @@ -195,7 +195,13 @@ func (slice TopCommitsUsers) Len() int {
}

func (slice TopCommitsUsers) Less(i, j int) bool {
return slice[i].CommitsCount > slice[j].CommitsCount
if slice[i].CommitsCount != slice[j].CommitsCount {
return slice[i].CommitsCount > slice[j].CommitsCount
}
if slice[i].FollowerCount != slice[j].FollowerCount {
return slice[i].FollowerCount > slice[j].FollowerCount
}
return strings.ToLower(slice[i].Login) < strings.ToLower(slice[j].Login)
}

func (slice TopCommitsUsers) Swap(i, j int) {
Expand All @@ -209,7 +215,13 @@ func (slice TopPublicUsers) Len() int {
}

func (slice TopPublicUsers) Less(i, j int) bool {
return slice[i].PublicContributionCount > slice[j].PublicContributionCount
if slice[i].PublicContributionCount != slice[j].PublicContributionCount {
return slice[i].PublicContributionCount > slice[j].PublicContributionCount
}
if slice[i].FollowerCount != slice[j].FollowerCount {
return slice[i].FollowerCount > slice[j].FollowerCount
}
return strings.ToLower(slice[i].Login) < strings.ToLower(slice[j].Login)
}

func (slice TopPublicUsers) Swap(i, j int) {
Expand All @@ -223,7 +235,13 @@ func (slice TopPrivateUsers) Len() int {
}

func (slice TopPrivateUsers) Less(i, j int) bool {
return slice[i].ContributionCount > slice[j].ContributionCount
if slice[i].ContributionCount != slice[j].ContributionCount {
return slice[i].ContributionCount > slice[j].ContributionCount
}
if slice[i].FollowerCount != slice[j].FollowerCount {
return slice[i].FollowerCount > slice[j].FollowerCount
}
return strings.ToLower(slice[i].Login) < strings.ToLower(slice[j].Login)
}

func (slice TopPrivateUsers) Swap(i, j int) {
Expand All @@ -242,7 +260,10 @@ func (slice Organizations) Len() int {
}

func (slice Organizations) Less(i, j int) bool {
return slice[i].MemberCount > slice[j].MemberCount
if slice[i].MemberCount != slice[j].MemberCount {
return slice[i].MemberCount > slice[j].MemberCount
}
return strings.ToLower(slice[i].Name) < strings.ToLower(slice[j].Name)
}

func (slice Organizations) Swap(i, j int) {
Expand Down
44 changes: 44 additions & 0 deletions output/output_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
package output

import (
"bytes"
"encoding/csv"
"testing"

"most-active-github-users-counter/github"
"most-active-github-users-counter/top"
)

func TestCsvOutput_SortsByContributionsAndRespectsAmount(t *testing.T) {
results := github.GithubSearchResults{
Users: []github.User{
{Login: "a", ContributionCount: 10, FollowerCount: 999},
{Login: "b", ContributionCount: 20, FollowerCount: 1},
{Login: "c", ContributionCount: 20, FollowerCount: 5},
},
}

var buf bytes.Buffer
if err := CsvOutput(results, &buf, top.Options{Amount: 2}); err != nil {
t.Fatalf("CsvOutput error: %v", err)
}

r := csv.NewReader(bytes.NewReader(buf.Bytes()))
rows, err := r.ReadAll()
if err != nil {
t.Fatalf("failed to parse csv: %v", err)
}

// header + 2 rows (because Amount=2)
if got, want := len(rows), 3; got != want {
t.Fatalf("unexpected number of rows: got %d want %d", got, want)
}

if got, want := rows[1][2], "c"; got != want {
t.Fatalf("rank 1 login mismatch: got %q want %q", got, want)
}
if got, want := rows[2][2], "b"; got != want {
t.Fatalf("rank 2 login mismatch: got %q want %q", got, want)
}
}