diff options
author | spf13 <[email protected]> | 2014-03-31 13:23:34 -0400 |
---|---|---|
committer | spf13 <[email protected]> | 2014-03-31 13:23:34 -0400 |
commit | e50b9d8ac1c69aa16cf45b1e4ff95b22d535adb7 (patch) | |
tree | 8c16489a5cf2631649805d547256d4427d9486d8 | |
parent | 2fa3761ec993657330d5b9ddbaaab1f58797fb61 (diff) | |
download | hugo-e50b9d8ac1c69aa16cf45b1e4ff95b22d535adb7.tar.gz hugo-e50b9d8ac1c69aa16cf45b1e4ff95b22d535adb7.zip |
Adding support for logging & verbose logging. Consolidation of error handling. Integration of jWalterWeatherman library. Fixed #137
-rw-r--r-- | commands/hugo.go | 58 | ||||
-rw-r--r-- | commands/limit_darwin.go | 21 | ||||
-rw-r--r-- | commands/server.go | 16 | ||||
-rw-r--r-- | commands/version.go | 1 | ||||
-rw-r--r-- | docs/content/overview/usage.md | 3 | ||||
-rw-r--r-- | helpers/pygments.go | 8 | ||||
-rw-r--r-- | helpers/url.go | 3 | ||||
-rw-r--r-- | hugolib/config.go | 24 | ||||
-rw-r--r-- | hugolib/metadata.go | 24 | ||||
-rw-r--r-- | hugolib/page.go | 17 | ||||
-rw-r--r-- | hugolib/shortcode.go | 11 | ||||
-rw-r--r-- | hugolib/site.go | 45 | ||||
-rw-r--r-- | hugolib/summary.go | 5 | ||||
-rw-r--r-- | main.go | 3 | ||||
-rw-r--r-- | utils/utils.go | 10 |
15 files changed, 139 insertions, 110 deletions
diff --git a/commands/hugo.go b/commands/hugo.go index 8bda329b4..9a42a0edb 100644 --- a/commands/hugo.go +++ b/commands/hugo.go @@ -15,18 +15,20 @@ package commands import ( "fmt" - "github.com/mostafah/fsync" - "github.com/spf13/cobra" - "github.com/spf13/hugo/hugolib" - "github.com/spf13/hugo/utils" - "github.com/spf13/hugo/watcher" - "github.com/spf13/nitro" "os" "path/filepath" "runtime" "strings" "sync" "time" + + "github.com/mostafah/fsync" + "github.com/spf13/cobra" + "github.com/spf13/hugo/hugolib" + "github.com/spf13/hugo/utils" + "github.com/spf13/hugo/watcher" + jww "github.com/spf13/jwalterweatherman" + "github.com/spf13/nitro" ) var Config *hugolib.Config @@ -44,8 +46,8 @@ Complete documentation is available at http://hugo.spf13.com`, } var hugoCmdV *cobra.Command -var BuildWatch, Draft, UglyUrls, Verbose bool -var Source, Destination, BaseUrl, CfgFile string +var BuildWatch, Draft, UglyUrls, Verbose, Logging, VerboseLog bool +var Source, Destination, BaseUrl, CfgFile, LogFile string func Execute() { AddCommands() @@ -67,6 +69,9 @@ func init() { HugoCmd.PersistentFlags().BoolVar(&UglyUrls, "uglyurls", false, "if true, use /filename.html instead of /filename/") HugoCmd.PersistentFlags().StringVarP(&BaseUrl, "base-url", "b", "", "hostname (and path) to the root eg. http://spf13.com/") HugoCmd.PersistentFlags().StringVar(&CfgFile, "config", "", "config file (default is path/config.yaml|json|toml)") + HugoCmd.PersistentFlags().BoolVar(&Logging, "log", false, "Enable Logging") + HugoCmd.PersistentFlags().StringVar(&LogFile, "logfile", "", "Log File path (if set, logging enabled automatically)") + HugoCmd.PersistentFlags().BoolVar(&VerboseLog, "verboselog", false, "verbose logging") HugoCmd.PersistentFlags().BoolVar(&nitro.AnalysisOn, "stepAnalysis", false, "display memory and timing of different steps of the program") HugoCmd.Flags().BoolVarP(&BuildWatch, "watch", "w", false, "watch filesystem for changes and recreate as needed") hugoCmdV = HugoCmd @@ -86,12 +91,35 @@ func InitializeConfig() { if hugoCmdV.PersistentFlags().Lookup("verbose").Changed { Config.Verbose = Verbose } + + if hugoCmdV.PersistentFlags().Lookup("logfile").Changed { + Config.LogFile = LogFile + } + if BaseUrl != "" { Config.BaseUrl = BaseUrl } if Destination != "" { Config.PublishDir = Destination } + + if VerboseLog || Logging || Config.LogFile != "" { + if Config.LogFile != "" { + jww.SetLogFile(Config.LogFile) + } else { + jww.UseTempLogFile("hugo") + } + } else { + jww.DiscardLogging() + } + + if Config.Verbose { + jww.SetStdoutThreshold(jww.LevelDebug) + } + + if VerboseLog { + jww.SetLogThreshold(jww.LevelDebug) + } } func build(watches ...bool) { @@ -103,8 +131,8 @@ func build(watches ...bool) { utils.StopOnErr(buildSite(BuildWatch || watch)) if BuildWatch { - fmt.Println("Watching for changes in", Config.GetAbsPath(Config.ContentDir)) - fmt.Println("Press ctrl+c to stop") + jww.FEEDBACK.Println("Watching for changes in", Config.GetAbsPath(Config.ContentDir)) + jww.FEEDBACK.Println("Press ctrl+c to stop") utils.CheckErr(NewWatcher(0)) } } @@ -123,7 +151,7 @@ func getDirList() []string { var a []string walker := func(path string, fi os.FileInfo, err error) error { if err != nil { - fmt.Println("Walker: ", err) + jww.ERROR.Println("Walker: ", err) return nil } @@ -151,7 +179,7 @@ func buildSite(watching ...bool) (err error) { return } site.Stats() - fmt.Printf("in %v ms\n", int(1000*time.Since(startTime).Seconds())) + jww.FEEDBACK.Printf("in %v ms\n", int(1000*time.Since(startTime).Seconds())) return nil } @@ -182,9 +210,7 @@ func NewWatcher(port int) error { for { select { case evs := <-watcher.Event: - if Verbose { - fmt.Println(evs) - } + jww.INFO.Println(evs) static_changed := false dynamic_changed := false @@ -214,7 +240,7 @@ func NewWatcher(port int) error { if static_changed { fmt.Print("Static file changed, syncing\n\n") - utils.CheckErr(copyStatic(), fmt.Sprintf("Error copying static files to %s", Config.GetAbsPath(Config.PublishDir))) + utils.StopOnErr(copyStatic(), fmt.Sprintf("Error copying static files to %s", Config.GetAbsPath(Config.PublishDir))) } if dynamic_changed { diff --git a/commands/limit_darwin.go b/commands/limit_darwin.go index 6f5c60f7f..62fc0f5bc 100644 --- a/commands/limit_darwin.go +++ b/commands/limit_darwin.go @@ -15,9 +15,10 @@ package commands import ( - "fmt" - "github.com/spf13/cobra" "syscall" + + "github.com/spf13/cobra" + jww "github.com/spf13/jwalterweatherman" ) func init() { @@ -33,22 +34,22 @@ var limit = &cobra.Command{ var rLimit syscall.Rlimit err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit) if err != nil { - fmt.Println("Error Getting Rlimit ", err) + jww.ERROR.Println("Error Getting Rlimit ", err) } - fmt.Println("Current rLimit:", rLimit) + jww.FEEDBACK.Println("Current rLimit:", rLimit) - fmt.Println("Attempting to increase limit") + jww.FEEDBACK.Println("Attempting to increase limit") rLimit.Max = 999999 rLimit.Cur = 999999 err = syscall.Setrlimit(syscall.RLIMIT_NOFILE, &rLimit) if err != nil { - fmt.Println("Error Setting rLimit ", err) + jww.ERROR.Println("Error Setting rLimit ", err) } err = syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit) if err != nil { - fmt.Println("Error Getting rLimit ", err) + jww.ERROR.Println("Error Getting rLimit ", err) } - fmt.Println("rLimit after change:", rLimit) + jww.FEEDBACK.Println("rLimit after change:", rLimit) }, } @@ -56,14 +57,14 @@ func tweakLimit() { var rLimit syscall.Rlimit err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit) if err != nil { - fmt.Println("Unable to obtain rLimit", err) + jww.ERROR.Println("Unable to obtain rLimit", err) } if rLimit.Cur < rLimit.Max { rLimit.Max = 999999 rLimit.Cur = 999999 err = syscall.Setrlimit(syscall.RLIMIT_NOFILE, &rLimit) if err != nil { - fmt.Println("Unable to increase number of open files limit", err) + jww.ERROR.Println("Unable to increase number of open files limit", err) } } } diff --git a/commands/server.go b/commands/server.go index 874703b9b..b8183817f 100644 --- a/commands/server.go +++ b/commands/server.go @@ -15,11 +15,13 @@ package commands import ( "fmt" - "github.com/spf13/cobra" "net/http" "os" "strconv" "strings" + + "github.com/spf13/cobra" + jww "github.com/spf13/jwalterweatherman" ) var serverPort int @@ -62,7 +64,7 @@ func server(cmd *cobra.Command, args []string) { // Watch runs its own server as part of the routine if serverWatch { - fmt.Println("Watching for changes in", Config.GetAbsPath(Config.ContentDir)) + jww.FEEDBACK.Println("Watching for changes in", Config.GetAbsPath(Config.ContentDir)) err := NewWatcher(serverPort) if err != nil { fmt.Println(err) @@ -73,21 +75,19 @@ func server(cmd *cobra.Command, args []string) { } func serve(port int) { - if Verbose { - fmt.Println("Serving pages from " + Config.GetAbsPath(Config.PublishDir)) - } + jww.FEEDBACK.Println("Serving pages from " + Config.GetAbsPath(Config.PublishDir)) if BaseUrl == "" { - fmt.Printf("Web Server is available at %s\n", Config.BaseUrl) + jww.FEEDBACK.Printf("Web Server is available at %s\n", Config.BaseUrl) } else { - fmt.Printf("Web Server is available at http://localhost:%v\n", port) + jww.FEEDBACK.Printf("Web Server is available at http://localhost:%v\n", port) } fmt.Println("Press ctrl+c to stop") err := http.ListenAndServe(":"+strconv.Itoa(port), http.FileServer(http.Dir(Config.GetAbsPath(Config.PublishDir)))) if err != nil { - fmt.Fprintf(os.Stderr, "Error: %s\n", err.Error()) + jww.ERROR.Printf("Error: %s\n", err.Error()) os.Exit(1) } } diff --git a/commands/version.go b/commands/version.go index 95f676a8c..ec4e5c0c8 100644 --- a/commands/version.go +++ b/commands/version.go @@ -15,6 +15,7 @@ package commands import ( "fmt" + "github.com/spf13/cobra" ) diff --git a/docs/content/overview/usage.md b/docs/content/overview/usage.md index 9cfa4e66f..cb94669f5 100644 --- a/docs/content/overview/usage.md +++ b/docs/content/overview/usage.md @@ -31,9 +31,12 @@ Make sure either hugo is in your path or provide a path to it. -D, --build-drafts=false: include content marked as draft --config="": config file (default is path/config.yaml|json|toml) -d, --destination="": filesystem path to write files to + --log=false: Enable Logging + --logfile="": Log File path (if set, logging enabled automatically) -s, --source="": filesystem path to read files relative from --uglyurls=false: if true, use /filename.html instead of /filename/ -v, --verbose=false: verbose output + --verboselog=false: verbose logging -w, --watch=false: watch filesystem for changes and recreate as needed Use "hugo help [command]" for more information about that command. diff --git a/helpers/pygments.go b/helpers/pygments.go index 28b296227..a858fba4b 100644 --- a/helpers/pygments.go +++ b/helpers/pygments.go @@ -15,16 +15,18 @@ package helpers import ( "bytes" - "log" "os/exec" "strings" + + jww "github.com/spf13/jwalterweatherman" ) func Highlight(code string, lexer string) string { var pygmentsBin = "pygmentize" if _, err := exec.LookPath(pygmentsBin); err != nil { - log.Print("Highlighting requries Pygments to be installed and in the path") + + jww.WARN.Println("Highlighting requries Pygments to be installed and in the path") return code } @@ -37,7 +39,7 @@ func Highlight(code string, lexer string) string { cmd.Stderr = &stderr if err := cmd.Run(); err != nil { - log.Print(stderr.String()) + jww.ERROR.Print(stderr.String()) return code } diff --git a/helpers/url.go b/helpers/url.go index e3428eda6..ffafe7af0 100644 --- a/helpers/url.go +++ b/helpers/url.go @@ -14,13 +14,10 @@ package helpers import ( - "fmt" "net/url" "path" ) -var _ = fmt.Println - // Similar to MakePath, but with Unicode handling // Example: // uri: Vim (text editor) diff --git a/hugolib/config.go b/hugolib/config.go index a52b0737c..c3f753cd4 100644 --- a/hugolib/config.go +++ b/hugolib/config.go @@ -16,21 +16,23 @@ package hugolib import ( "encoding/json" "fmt" - "github.com/BurntSushi/toml" - "github.com/spf13/hugo/helpers" "io/ioutil" - "launchpad.net/goyaml" "os" "path" "path/filepath" "strings" + + "github.com/BurntSushi/toml" + "github.com/spf13/hugo/helpers" + jww "github.com/spf13/jwalterweatherman" + "launchpad.net/goyaml" ) // config file items type Config struct { ContentDir, PublishDir, BaseUrl, StaticDir string Path, CacheDir, LayoutDir, DefaultLayout string - ConfigFile string + ConfigFile, LogFile string Title string Indexes map[string]string // singular, plural ProcessFilters map[string][]string @@ -50,8 +52,8 @@ func SetupConfig(cfgfile *string, path *string) *Config { c.ConfigFile = cfg if err != nil { - fmt.Printf("%v", err) - fmt.Println(" using defaults instead") + jww.ERROR.Printf("%v", err) + jww.FEEDBACK.Println("using defaults instead") } // set defaults @@ -92,19 +94,19 @@ func (c *Config) readInConfig() { switch path.Ext(c.ConfigFile) { case ".yaml": if err := goyaml.Unmarshal(file, &c); err != nil { - fmt.Printf("Error parsing config: %s", err) + jww.ERROR.Printf("Error parsing config: %s", err) os.Exit(1) } case ".json": if err := json.Unmarshal(file, &c); err != nil { - fmt.Printf("Error parsing config: %s", err) + jww.ERROR.Printf("Error parsing config: %s", err) os.Exit(1) } case ".toml": if _, err := toml.Decode(string(file), &c); err != nil { - fmt.Printf("Error parsing config: %s", err) + jww.ERROR.Printf("Error parsing config: %s", err) os.Exit(1) } } @@ -115,13 +117,13 @@ func (c *Config) setPath(p string) { if p == "" { path, err := findPath() if err != nil { - fmt.Printf("Error finding path: %s", err) + jww.ERROR.Printf("Error finding path: %s", err) } c.Path = path } else { path, err := filepath.Abs(p) if err != nil { - fmt.Printf("Error finding path: %s", err) + jww.ERROR.Printf("Error finding path: %s", err) } c.Path = path } diff --git a/hugolib/metadata.go b/hugolib/metadata.go index c1fe138fe..b37d5451f 100644 --- a/hugolib/metadata.go +++ b/hugolib/metadata.go @@ -3,9 +3,10 @@ package hugolib import ( "errors" "fmt" - "os" "strconv" "time" + + jww "github.com/spf13/jwalterweatherman" ) func interfaceToTime(i interface{}) time.Time { @@ -17,9 +18,9 @@ func interfaceToTime(i interface{}) time.Time { if e == nil { return d } - errorln("Could not parse Date/Time format:", e) + jww.ERROR.Println("Could not parse Date/Time format:", e) default: - errorln("Only Time is supported for this key") + jww.ERROR.Println("Only Time is supported for this key") } return *new(time.Time) @@ -53,11 +54,6 @@ func stringToDate(s string) (time.Time, error) { }) } -// TODO remove this and return a proper error. -func errorln(str string, a ...interface{}) { - fmt.Fprintln(os.Stderr, str, a) -} - func parseDateWith(s string, dates []string) (d time.Time, e error) { for _, dateType := range dates { if d, e = time.Parse(dateType, s); e == nil { @@ -77,7 +73,7 @@ func interfaceToBool(i interface{}) bool { } return false default: - errorln("Only Boolean values are supported for this YAML key") + jww.ERROR.Println("Only Boolean values are supported for this YAML key") } return false @@ -109,11 +105,11 @@ func interfaceToFloat64(i interface{}) float64 { if err == nil { return float64(v) } else { - errorln("Only Floats are supported for this key\nErr:", err) + jww.ERROR.Println("Only Floats are supported for this key\nErr:", err) } default: - errorln("Only Floats are supported for this key") + jww.ERROR.Println("Only Floats are supported for this key") } return 0.0 @@ -136,10 +132,10 @@ func interfaceToInt(i interface{}) int { if err == nil { return int(v) } else { - errorln("Only Ints are supported for this key\nErr:", err) + jww.ERROR.Println("Only Ints are supported for this key\nErr:", err) } default: - errorln("Only Ints are supported for this key") + jww.ERROR.Println("Only Ints are supported for this key") } return 0 @@ -154,7 +150,7 @@ func interfaceToString(i interface{}) string { case int: return strconv.FormatInt(int64(i.(int)), 10) default: - errorln(fmt.Sprintf("Only Strings are supported for this key (got type '%T'): %s", s, s)) + jww.ERROR.Println(fmt.Sprintf("Only Strings are supported for this key (got type '%T'): %s", s, s)) } return "" diff --git a/hugolib/page.go b/hugolib/page.go index 6add32e5a..8c846c465 100644 --- a/hugolib/page.go +++ b/hugolib/page.go @@ -17,19 +17,21 @@ import ( "bytes" "errors" "fmt" + "html/template" + "io" + "net/url" + "path" + "strings" + "time" + "github.com/BurntSushi/toml" "github.com/spf13/hugo/helpers" "github.com/spf13/hugo/parser" "github.com/spf13/hugo/template/bundle" + jww "github.com/spf13/jwalterweatherman" "github.com/theplant/blackfriday" - "html/template" - "io" "launchpad.net/goyaml" json "launchpad.net/rjson" - "net/url" - "path" - "strings" - "time" ) type Page struct { @@ -142,6 +144,8 @@ func newPage(filename string) *Page { File: File{FileName: filename, Extension: "html"}, Node: Node{Keywords: make([]string, 10, 30)}, Params: make(map[string]interface{})} + + jww.DEBUG.Println("Reading from", page.File.FileName) page.Date, _ = time.Parse("20060102", "20080101") page.guessSection() return &page @@ -212,6 +216,7 @@ func ReadFrom(buf io.Reader, name string) (page *Page, err error) { // Parse for metadata & body if err = p.parse(buf); err != nil { + jww.ERROR.Print(err) return } diff --git a/hugolib/shortcode.go b/hugolib/shortcode.go index bea89fe2a..92e81195e 100644 --- a/hugolib/shortcode.go +++ b/hugolib/shortcode.go @@ -15,15 +15,14 @@ package hugolib import ( "bytes" - "fmt" - "github.com/spf13/hugo/template/bundle" "html/template" "reflect" "strings" "unicode" -) -var _ = fmt.Println + "github.com/spf13/hugo/template/bundle" + jww "github.com/spf13/jwalterweatherman" +) type ShortcodeFunc func([]string) string @@ -296,8 +295,8 @@ func ShortcodeRender(tmpl *template.Template, data *ShortcodeWithPage) string { buffer := new(bytes.Buffer) err := tmpl.Execute(buffer, data) if err != nil { - fmt.Println("error processing shortcode", tmpl.Name(), "\n ERR:", err) - fmt.Println(data) + jww.ERROR.Println("error processing shortcode", tmpl.Name(), "\n ERR:", err) + jww.WARN.Println(data) } return buffer.String() } diff --git a/hugolib/site.go b/hugolib/site.go index f53d05559..7602853d5 100644 --- a/hugolib/site.go +++ b/hugolib/site.go @@ -14,21 +14,23 @@ package hugolib import ( - "bitbucket.org/pkg/inflect" "bytes" "fmt" - "github.com/spf13/hugo/helpers" - "github.com/spf13/hugo/source" - "github.com/spf13/hugo/target" - "github.com/spf13/hugo/template/bundle" - "github.com/spf13/hugo/transform" - "github.com/spf13/nitro" "html/template" "io" "os" "strings" "sync" "time" + + "bitbucket.org/pkg/inflect" + "github.com/spf13/hugo/helpers" + "github.com/spf13/hugo/source" + "github.com/spf13/hugo/target" + "github.com/spf13/hugo/template/bundle" + "github.com/spf13/hugo/transform" + jww "github.com/spf13/jwalterweatherman" + "github.com/spf13/nitro" ) var _ = transform.AbsURL @@ -104,9 +106,9 @@ func (s *Site) Build() (err error) { return } if err = s.Render(); err != nil { - fmt.Printf("Error rendering site: %s\nAvailable templates:\n", err) + jww.ERROR.Printf("Error rendering site: %s\nAvailable templates:\n", err) for _, template := range s.Tmpl.Templates() { - fmt.Printf("\t%s\n", template.Name()) + jww.ERROR.Printf("\t%s\n", template.Name()) } return } @@ -190,7 +192,7 @@ func (s *Site) Render() (err error) { func (s *Site) checkDescriptions() { for _, p := range s.Pages { if len(p.Description) < 60 { - fmt.Println(p.FileName + " ") + jww.FEEDBACK.Println(p.FileName + " ") } } } @@ -309,9 +311,7 @@ func (s *Site) BuildSiteMeta() (err error) { s.Indexes[plural].Add(idx, x) } } else { - if s.Config.Verbose { - fmt.Fprintf(os.Stderr, "Invalid %s in %s\n", plural, p.File.FileName) - } + jww.ERROR.Printf("Invalid %s in %s\n", plural, p.File.FileName) } } } @@ -533,9 +533,9 @@ func (s *Site) RenderHomePage() error { } func (s *Site) Stats() { - fmt.Printf("%d pages created \n", len(s.Pages)) + jww.FEEDBACK.Printf("%d pages created \n", len(s.Pages)) for _, pl := range s.Config.Indexes { - fmt.Printf("%d %s index created\n", len(s.Indexes[pl]), pl) + jww.FEEDBACK.Printf("%d %s index created\n", len(s.Indexes[pl]), pl) } } @@ -572,9 +572,7 @@ func (s *Site) render(d interface{}, out string, layouts ...string) (err error) layout := s.findFirstLayout(layouts...) if layout == "" { - if s.Config.Verbose { - fmt.Printf("Unable to locate layout: %s\n", layouts) - } + jww.WARN.Printf("Unable to locate layout: %s\n", layouts) return } @@ -601,7 +599,7 @@ func (s *Site) render(d interface{}, out string, layouts ...string) (err error) err = s.renderThing(d, layout, renderBuffer) if err != nil { // Behavior here should be dependent on if running in server or watch mode. - fmt.Println(fmt.Errorf("Rendering error: %v", err)) + jww.ERROR.Println(fmt.Errorf("Rendering error: %v", err)) if !s.Running() { os.Exit(-1) } @@ -631,7 +629,6 @@ func (s *Site) renderThing(d interface{}, layout string, w io.Writer) error { if s.Tmpl.Lookup(layout) == nil { return fmt.Errorf("Layout not found: %s", layout) } - //defer w.Close() return s.Tmpl.ExecuteTemplate(w, layout, d) } @@ -652,9 +649,7 @@ func (s *Site) initTarget() { func (s *Site) WritePublic(path string, reader io.Reader) (err error) { s.initTarget() - if s.Config.Verbose { - fmt.Println(path) - } + jww.DEBUG.Println("writing to", path) return s.Target.Publish(path, reader) } @@ -666,9 +661,7 @@ func (s *Site) WriteAlias(path string, permalink template.HTML) (err error) { } } - if s.Config.Verbose { - fmt.Println(path) - } + jww.DEBUG.Println("alias created at", path) return s.Alias.Publish(path, permalink) } diff --git a/hugolib/summary.go b/hugolib/summary.go index 1053e1975..3b1aa6b39 100644 --- a/hugolib/summary.go +++ b/hugolib/summary.go @@ -2,9 +2,10 @@ package hugolib import ( "bytes" - "fmt" "os/exec" "strings" + + jww "github.com/spf13/jwalterweatherman" ) var summaryLength = 70 @@ -62,7 +63,7 @@ func getRstContent(content []byte) string { var out bytes.Buffer cmd.Stdout = &out if err := cmd.Run(); err != nil { - fmt.Println(err) + jww.ERROR.Println(err) } rstLines := strings.Split(out.String(), "\n") @@ -14,8 +14,9 @@ package main import ( - "github.com/spf13/hugo/commands" "runtime" + + "github.com/spf13/hugo/commands" ) func main() { diff --git a/utils/utils.go b/utils/utils.go index a79740c9a..bb0248084 100644 --- a/utils/utils.go +++ b/utils/utils.go @@ -1,22 +1,24 @@ package utils import ( - "log" "os" + + jww "github.com/spf13/jwalterweatherman" ) func CheckErr(err error, s ...string) { if err != nil { for _, message := range s { - log.Fatalf(message) + jww.ERROR.Println(message) } - log.Fatalf("Fatal Error: %v", err) } } func StopOnErr(err error, s ...string) { if err != nil { - CheckErr(err, s...) + for _, message := range s { + jww.CRITICAL.Println(message) + } os.Exit(-1) } } |