feat: add follow option

Author: dwisiswant0

common/options.go

@@ -10,11 +10,12 @@ import (
 type Options struct {
 	Concurrency int              // Set the concurrent level
 	ConfigFile  string           // Specifies the config to use
-	Stdin       bool             // Stdin specifies whether stdin input was given to the process
-	Version     bool             // Version check of teler flag
-	Input       string           // Parse log from data persistence rather than buffer stream
-	Output      *os.File         // Write log output into file
 	Configs     *parsers.Configs // Get teler configuration interface
+	Follow      bool             // Specify if the logs should be streamed
+	Input       string           // Parse log from data persistence rather than buffer stream
 	JSON        bool             // Display threats in the terminal as JSON format
+	Output      *os.File         // Write log output into file
 	RmCache     bool             // To remove all cached resources on local
+	Stdin       bool             // Stdin specifies whether stdin input was given to the process
+	Version     bool             // Version check of teler flag
 }

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible
 	github.com/goji/httpauth v0.0.0-20160601135302-2da839ab0f4d
 	github.com/kirsle/configdir v0.0.0-20170128060238-e45d2f54772f
+	github.com/kitabisa/tailpipe v1.0.0
 	github.com/logrusorgru/aurora v2.0.3+incompatible
 	github.com/panjf2000/ants/v2 v2.5.0
 	github.com/projectdiscovery/gologger v1.1.4

go.sum

@@ -161,6 +161,8 @@ github.com/kirsle/configdir v0.0.0-20170128060238-e45d2f54772f h1:dKccXx7xA56UNq
 github.com/kirsle/configdir v0.0.0-20170128060238-e45d2f54772f/go.mod h1:4rEELDSfUAlBSyUjPG0JnaNGjf13JySHFeRdD/3dLP0=
 github.com/kisielk/errcheck v1.1.0/go.mod h1:EZBBE59ingxPouuu3KfxchcWSUPOHkagtvWXihfKN4Q=
 github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=
+github.com/kitabisa/tailpipe v1.0.0 h1:/dBte3xpjDbUsHnOILJaEryX5VA65w2Dxy6JbhViwXY=
+github.com/kitabisa/tailpipe v1.0.0/go.mod h1:298TbGVS9RP6sqxtJXg7rPTAZVGBhWIWRHaF1JbborQ=
 github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
 github.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
 github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=

internal/runner/options.go

@@ -22,6 +22,9 @@ func ParseOptions() *common.Options {
 	flag.StringVar(&options.Input, "i", "", "")
 	flag.StringVar(&options.Input, "input", "", "")
 
+	flag.BoolVar(&options.Follow, "f", false, "")
+	flag.BoolVar(&options.Follow, "follow", false, "")
+
 	flag.IntVar(&options.Concurrency, "x", 20, "")
 	flag.IntVar(&options.Concurrency, "concurrent", 20, "")
 
@@ -43,6 +46,7 @@ func ParseOptions() *common.Options {
 			"Options:",
 			"  -c, --config <FILE>         teler configuration file",
 			"  -i, --input <FILE>          Analyze logs from data persistence rather than buffer stream",
+			"  -f, --follow                Specify if the logs should be streamed",
 			"  -x, --concurrent <i>        Set the concurrency level to analyze logs (default: 20)",
 			"      --rm-cache              Removes all cached resources",
 			"  -v, --version               Show current teler version",

internal/runner/runner.go

@@ -6,6 +6,7 @@ import (
 	"os"
 	"os/signal"
 
+	"github.com/kitabisa/tailpipe"
 	"github.com/logrusorgru/aurora"
 	"github.com/panjf2000/ants/v2"
 	"github.com/projectdiscovery/gologger"
@@ -22,8 +23,9 @@ import (
 // New read & pass stdin log
 func New(options *common.Options) {
 	var (
-		input *os.File
-		pass  int
+		reader *gonx.Reader
+		input  *os.File
+		pass   int
 	)
 
 	go metric(options)
@@ -83,9 +85,20 @@ func New(options *common.Options) {
 
 	config := options.Configs
 	format := removeLBR(config.Logformat)
-	buffer := gonx.NewReader(input, format)
+
+	if !options.Stdin && options.Follow {
+		f, e := tailpipe.Open(options.Input)
+		if e != nil {
+			errors.Exit(e.Error())
+		}
+
+		reader = gonx.NewReader(f, format)
+	} else {
+		reader = gonx.NewReader(input, format)
+	}
+
 	for {
-		line, err := buffer.Read()
+		line, err := reader.Read()
 		if err == io.EOF {
 			break
 		}