add --max, --before, --after switches to dtail and dgrep commands

1 files changed, 10 insertions, 2 deletions

diff --git a/internal/clients/args.go b/internal/clients/args.go
index 7f782f1..67d2044 100644
--- a/internal/clients/args.go
+++ b/internal/clients/args.go
@@ -6,14 +6,22 @@ import (
 	gossh "golang.org/x/crypto/ssh"
 )
 
+// LineContext is here to help filtering out only specific lines.
+type LineContext struct {
+	RegexStr      string
+	AfterContext  int
+	BeforeContext int
+	MaxCount      int
+}
+
 // Args is a helper struct to summarize common client arguments.
 type Args struct {
+	LineContext
 	Mode               omode.Mode
 	ServersStr         string
 	UserName           string
 	What               string
 	Arguments          []string
-	RegexStr           string
 	RegexInvert        bool
 	TrustAllHosts      bool
 	Discovery          string
@@ -22,5 +30,5 @@ type Args struct {
 	SSHAuthMethods     []gossh.AuthMethod
 	SSHHostKeyCallback gossh.HostKeyCallback
 	PrivateKeyPathFile string
-	Quiet            bool
+	Quiet              bool
 }