taigrr/wtf
1
0
Fork 0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00
Code Issues Projects Releases Wiki Activity

Remove unnecessary parameter passing in WtfApp

Browse Source 
Signed-off-by: Chris Cummer <chriscummer@me.com>
This commit is contained in:
Chris Cummer 2020-10-14 12:16:48 -07:00
parent 63aa353db3
commit a746ab6975
2 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
app/exit_message.go
View File

@ -7,7 +7,6 @@ import (
"github.com/logrusorgru/aurora"
"github.com/olebedev/config"
"github.com/wtfutil/wtf/support"
)
const exitMessageHeader = `
@ -25,14 +24,14 @@ const exitMessageHeader = `
func (wtfApp *WtfApp) DisplayExitMessage() {
exitMessageIsDisplayable := readDisplayableConfig(wtfApp.config)
wtfApp.displayExitMsg(wtfApp.ghUser, exitMessageIsDisplayable)
wtfApp.displayExitMsg(exitMessageIsDisplayable)
}
/* -------------------- Unexported Functions -------------------- */
func (wtfApp *WtfApp) displayExitMsg(ghUser *support.GitHubUser, exitMessageIsDisplayable bool) string {
func (wtfApp *WtfApp) displayExitMsg(exitMessageIsDisplayable bool) string {
// If a sponsor or contributor and opt out of seeing the exit message, do not display it
if (ghUser.IsContributor || ghUser.IsSponsor) && !exitMessageIsDisplayable {
if (wtfApp.ghUser.IsContributor || wtfApp.ghUser.IsSponsor) && !exitMessageIsDisplayable {
return ""
}
@ -40,15 +39,15 @@ func (wtfApp *WtfApp) displayExitMsg(ghUser *support.GitHubUser, exitMessageIsDi
msgs = append(msgs, aurora.Magenta(exitMessageHeader).String())
if ghUser.IsContributor {
if wtfApp.ghUser.IsContributor {
msgs = append(msgs, wtfApp.contributorThankYouMessage())
}
if ghUser.IsSponsor {
if wtfApp.ghUser.IsSponsor {
msgs = append(msgs, wtfApp.sponsorThankYouMessage())
}
if !ghUser.IsContributor && !ghUser.IsSponsor {
if !wtfApp.ghUser.IsContributor && !wtfApp.ghUser.IsSponsor {
msgs = append(msgs, wtfApp.supportRequestMessage())
}

4
app/exit_message_test.go
View File

@ -52,12 +52,12 @@ func Test_displayExitMessage(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
wtfApp := WtfApp{}
ghUser := &support.GitHubUser{
wtfApp.ghUser = &support.GitHubUser{
IsContributor: tt.isContributor,
IsSponsor: tt.isSponsor,
}
actual := wtfApp.displayExitMsg(ghUser, tt.isDisplayable)
actual := wtfApp.displayExitMsg(tt.isDisplayable)
if tt.compareWith == "equals" {
assert.Equal(t, actual, tt.expected)