fix #85: add --auto-headers and --custom-headers (#86)

cfg/config.go

@@ -28,7 +28,7 @@ import (
 )
 
 const (
-	Version  = "v1.5.9"
+	Version  = "v1.5.10"
 	MAXPARTS = 2
 )
 
@@ -93,6 +93,8 @@ type Config struct {
 	UseHighlight   bool
 	Interactive    bool
 	InputJSON      bool
+	AutoHeaders    bool
+	CustomHeaders  []string
 
 	SortMode        string
 	SortDescending  bool
@@ -413,6 +415,12 @@ func (conf *Config) PreparePattern(patterns []*Pattern) error {
 	return nil
 }
 
+func (conf *Config) PrepareCustomHeaders(custom string) {
+	if len(custom) > 0 {
+		conf.CustomHeaders = strings.Split(custom, ",")
+	}
+}
+
 // Parse config file.  Ignore if the file doesn't exist  but return an
 // error if it exists but fails to read or parse
 func (conf *Config) ParseConfigfile() error {