mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
111 lines
3.1 KiB
Go
111 lines
3.1 KiB
Go
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/logrusorgru/aurora"
|
|
"github.com/olebedev/config"
|
|
"github.com/wtfutil/wtf/support"
|
|
)
|
|
|
|
const exitMessageHeader = `
|
|
____ __ ____ .___________. _______
|
|
\ \ / \ / / | || ____|
|
|
\ \/ \/ / ----| |-----| |__
|
|
\ / | | | __|
|
|
\ /\ / | | | |
|
|
\__/ \__/ |__| |__|
|
|
|
|
the personal information dashboard for your terminal
|
|
`
|
|
|
|
// DisplayExitMessage displays the onscreen exit message when the app quits
|
|
func (wtfApp *WtfApp) DisplayExitMessage() {
|
|
githubAPIKey := readGitHubAPIKey(wtfApp.config)
|
|
ghUser := support.NewGitHubUser(githubAPIKey)
|
|
|
|
exitMessageIsDisplayable := readDisplayableConfig(wtfApp.config)
|
|
|
|
wtfApp.displayExitMsg(ghUser, exitMessageIsDisplayable)
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (wtfApp *WtfApp) displayExitMsg(ghUser *support.GitHubUser, exitMessageIsDisplayable bool) string {
|
|
_ = ghUser.Load()
|
|
|
|
// If a sponsor or contributor and opt out of seeing the exit message, do not display it
|
|
if (ghUser.IsContributor || ghUser.IsSponsor) && !exitMessageIsDisplayable {
|
|
return ""
|
|
}
|
|
|
|
msgs := []string{}
|
|
|
|
msgs = append(msgs, aurora.Magenta(exitMessageHeader).String())
|
|
|
|
if ghUser.IsContributor {
|
|
msgs = append(msgs, wtfApp.contributorThankYouMessage())
|
|
}
|
|
|
|
if ghUser.IsSponsor {
|
|
msgs = append(msgs, wtfApp.sponsorThankYouMessage())
|
|
}
|
|
|
|
if !ghUser.IsContributor && !ghUser.IsSponsor {
|
|
msgs = append(msgs, wtfApp.supportRequestMessage())
|
|
}
|
|
|
|
displayMsg := strings.Join(msgs, "\n")
|
|
|
|
fmt.Println(displayMsg)
|
|
|
|
return displayMsg
|
|
}
|
|
|
|
// readDisplayableConfig figures out whether or not the exit message should be displayed
|
|
// per the user's wishes. It allows contributors and sponsors to opt out of the exit message
|
|
func readDisplayableConfig(cfg *config.Config) bool {
|
|
displayExitMsg := cfg.UBool("wtf.exitMessage.display", true)
|
|
return displayExitMsg
|
|
}
|
|
|
|
// readGitHubAPIKey attempts to find a GitHub API key somewhere in the configuration file
|
|
func readGitHubAPIKey(cfg *config.Config) string {
|
|
apiKey := cfg.UString("wtf.exitMessage.githubAPIKey", os.Getenv("WTF_GITHUB_TOKEN"))
|
|
if apiKey != "" {
|
|
return apiKey
|
|
}
|
|
|
|
moduleConfig, err := cfg.Get("wtf.mods.github")
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
return moduleConfig.UString("apiKey", "")
|
|
}
|
|
|
|
/* -------------------- Messaging -------------------- */
|
|
|
|
func (wtfApp *WtfApp) contributorThankYouMessage() string {
|
|
str := " On behalf of all the users of WTF, thank you for contributing to the source code."
|
|
str += fmt.Sprintf(" %s", aurora.Green("You rock."))
|
|
|
|
return str
|
|
}
|
|
|
|
func (wtfApp *WtfApp) sponsorThankYouMessage() string {
|
|
str := " Your sponsorship of WTF makes a difference. Thank you for sponsoring and supporting WTF."
|
|
str += fmt.Sprintf(" %s", aurora.Green("You're awesome."))
|
|
|
|
return str
|
|
}
|
|
|
|
func (wtfApp *WtfApp) supportRequestMessage() string {
|
|
str := " The development and maintenance of WTF is supported by sponsorships.\n"
|
|
str += fmt.Sprintf(" Please consider sponsoring WTF at %s\n", aurora.Green("https://github.com/sponsors/senorprogrammer"))
|
|
|
|
return str
|
|
}
|