diff --git a/github.go b/github.go index d8a9731..fff8fde 100644 --- a/github.go +++ b/github.go @@ -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 diff --git a/main.go b/main.go index 330539e..411ab5c 100644 --- a/main.go +++ b/main.go @@ -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 }