Find README(.md) in current working dir when no arg was supplied

This commit is contained in:
Christian Muehlhaeuser 2019-11-24 05:06:01 +01:00
commit 14bc95c272
2 changed files with 17 additions and 7 deletions

View file

@ -23,9 +23,7 @@ func findGitHubREADME(s string) (*http.Response, error) {
if err != nil {
return nil, err
}
u.Host = "raw.githubusercontent.com"
readmeNames := []string{"README.md", "README"}
for _, r := range readmeNames {
v := u

22
main.go
View file

@ -16,6 +16,8 @@ import (
)
var (
readmeNames = []string{"README.md", "README"}
rootCmd = &cobra.Command{
Use: "gold SOURCE",
Short: "Render markdown on the CLI, with pizzazz!",
@ -23,7 +25,6 @@ var (
SilenceUsage: false,
RunE: execute,
}
style string
)
@ -55,15 +56,26 @@ func readerFromArg(s string) (io.ReadCloser, error) {
return resp.Body, nil
}
if len(s) == 0 {
for _, v := range readmeNames {
r, err := os.Open(v)
if err == nil {
return r, err
}
}
return nil, errors.New("missing markdown source")
}
return os.Open(s)
}
func execute(cmd *cobra.Command, args []string) error {
if len(args) != 1 {
return errors.New("missing markdown source")
var arg string
if len(args) > 0 {
arg = args[0]
}
in, err := readerFromArg(args[0])
in, err := readerFromArg(arg)
if err != nil {
return err
}