refactor: some code refactoring

- moved the usage functions to usage.go
- removed duplicate definitions of oauth2Conf
This commit is contained in:
Dan Anglin 2024-02-23 07:49:02 +00:00
parent 70fc1acc68
commit 4cbf1d4b4d
Signed by: dananglin
GPG key ID: 0C1D44CFBEE68638
3 changed files with 96 additions and 98 deletions

View file

@ -76,7 +76,18 @@ func (c *loginCommand) Execute() error {
return fmt.Errorf("unable to register the application; %w", err) return fmt.Errorf("unable to register the application; %w", err)
} }
consentPageURL := authCodeURL(gtsClient.Authentication) oauth2Conf := oauth2.Config{
ClientID: gtsClient.Authentication.ClientID,
ClientSecret: gtsClient.Authentication.ClientSecret,
Scopes: []string{"read"},
RedirectURL: internal.RedirectUri,
Endpoint: oauth2.Endpoint{
AuthURL: gtsClient.Authentication.Instance + "/oauth/authorize",
TokenURL: gtsClient.Authentication.Instance + "/oauth/token",
},
}
consentPageURL := authCodeURL(oauth2Conf)
openLink(consentPageURL) openLink(consentPageURL)
@ -89,7 +100,7 @@ func (c *loginCommand) Execute() error {
return fmt.Errorf("failed to read access code; %w", err) return fmt.Errorf("failed to read access code; %w", err)
} }
gtsClient.Authentication, err = addAccessToken(gtsClient.Authentication, code) gtsClient.Authentication, err = addAccessToken(gtsClient.Authentication, oauth2Conf, code)
if err != nil { if err != nil {
return fmt.Errorf("unable to get the access token; %w", err) return fmt.Errorf("unable to get the access token; %w", err)
} }
@ -109,36 +120,17 @@ func (c *loginCommand) Execute() error {
return nil return nil
} }
func authCodeURL(account config.Authentication) string { func authCodeURL(oauth2Conf oauth2.Config) string {
config := oauth2.Config{ url := oauth2Conf.AuthCodeURL(
ClientID: account.ClientID, "state",
ClientSecret: account.ClientSecret, oauth2.AccessTypeOffline,
Scopes: []string{"read"}, ) + "&client_name=" + internal.ApplicationName
RedirectURL: internal.RedirectUri,
Endpoint: oauth2.Endpoint{
AuthURL: account.Instance + "/oauth/authorize",
TokenURL: account.Instance + "/oauth/token",
},
}
url := config.AuthCodeURL("state", oauth2.AccessTypeOffline) + fmt.Sprintf("&client_name=%s", internal.ApplicationName)
return url return url
} }
func addAccessToken(authentication config.Authentication, code string) (config.Authentication, error) { func addAccessToken(authentication config.Authentication, oauth2Conf oauth2.Config, code string) (config.Authentication, error) {
ouauth2Conf := oauth2.Config{ token, err := oauth2Conf.Exchange(context.Background(), code)
ClientID: authentication.ClientID,
ClientSecret: authentication.ClientSecret,
Scopes: []string{"read", "write"},
RedirectURL: internal.RedirectUri,
Endpoint: oauth2.Endpoint{
AuthURL: authentication.Instance + "/oauth/authorize",
TokenURL: authentication.Instance + "/oauth/token",
},
}
token, err := ouauth2Conf.Exchange(context.Background(), code)
if err != nil { if err != nil {
return config.Authentication{}, fmt.Errorf("unable to exchange the code for an access token; %w", err) return config.Authentication{}, fmt.Errorf("unable to exchange the code for an access token; %w", err)
} }

View file

@ -4,8 +4,6 @@ import (
"flag" "flag"
"fmt" "fmt"
"os" "os"
"slices"
"strings"
) )
type Executor interface { type Executor interface {
@ -75,71 +73,3 @@ func run() error {
return nil return nil
} }
func commandUsageFunc(name, summary string, flagset *flag.FlagSet) func() {
return func() {
var builder strings.Builder
fmt.Fprintf(
&builder,
"SUMMARY:\n %s - %s\n\nUSAGE:\n enbas %s [flags]\n\nFLAGS:",
name,
summary,
name,
)
flagset.VisitAll(func(f *flag.Flag) {
fmt.Fprintf(
&builder,
"\n -%s, --%s\n %s",
f.Name,
f.Name,
f.Usage,
)
})
builder.WriteString("\n")
w := flag.CommandLine.Output()
fmt.Fprint(w, builder.String())
}
}
func enbasUsageFunc(summaries map[string]string) func() {
cmds := make([]string, len(summaries))
ind := 0
for k := range summaries {
cmds[ind] = k
ind++
}
slices.Sort(cmds)
return func() {
var builder strings.Builder
builder.WriteString("SUMMARY:\n enbas - A GoToSocial client for the terminal.\n\n")
if binaryVersion != "" {
builder.WriteString("VERSION:\n " + binaryVersion + "\n\n")
}
builder.WriteString("USAGE:\n enbas [flags]\n enbas [command]\n\nCOMMANDS:")
for _, cmd := range cmds {
fmt.Fprintf(&builder, "\n %s\t%s", cmd, summaries[cmd])
}
builder.WriteString("\n\nFLAGS:\n -help, --help\n print the help message\n")
flag.VisitAll(func(f *flag.Flag) {
fmt.Fprintf(&builder, "\n -%s, --%s\n %s\n", f.Name, f.Name, f.Usage)
})
builder.WriteString("\nUse \"enbas [command] --help\" for more information about a command.\n")
w := flag.CommandLine.Output()
fmt.Fprint(w, builder.String())
}
}

76
cmd/enbas/usage.go Normal file
View file

@ -0,0 +1,76 @@
package main
import (
"flag"
"fmt"
"slices"
"strings"
)
func commandUsageFunc(name, summary string, flagset *flag.FlagSet) func() {
return func() {
var builder strings.Builder
fmt.Fprintf(
&builder,
"SUMMARY:\n %s - %s\n\nUSAGE:\n enbas %s [flags]\n\nFLAGS:",
name,
summary,
name,
)
flagset.VisitAll(func(f *flag.Flag) {
fmt.Fprintf(
&builder,
"\n -%s, --%s\n %s",
f.Name,
f.Name,
f.Usage,
)
})
builder.WriteString("\n")
w := flag.CommandLine.Output()
fmt.Fprint(w, builder.String())
}
}
func enbasUsageFunc(summaries map[string]string) func() {
cmds := make([]string, len(summaries))
ind := 0
for k := range summaries {
cmds[ind] = k
ind++
}
slices.Sort(cmds)
return func() {
var builder strings.Builder
builder.WriteString("SUMMARY:\n enbas - A GoToSocial client for the terminal.\n\n")
if binaryVersion != "" {
builder.WriteString("VERSION:\n " + binaryVersion + "\n\n")
}
builder.WriteString("USAGE:\n enbas [flags]\n enbas [command]\n\nCOMMANDS:")
for _, cmd := range cmds {
fmt.Fprintf(&builder, "\n %s\t%s", cmd, summaries[cmd])
}
builder.WriteString("\n\nFLAGS:\n -help, --help\n print the help message\n")
flag.VisitAll(func(f *flag.Flag) {
fmt.Fprintf(&builder, "\n -%s, --%s\n %s\n", f.Name, f.Name, f.Usage)
})
builder.WriteString("\nUse \"enbas [command] --help\" for more information about a command.\n")
w := flag.CommandLine.Output()
fmt.Fprint(w, builder.String())
}
}