1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
package fs
import (
"context"
"fmt"
"github.com/mimecast/dtail/internal/color/brush"
"github.com/mimecast/dtail/internal/protocol"
)
// CatProcessor handles cat-style output
type CatProcessor struct {
plain bool
noColor bool
hostname string
isFirstLine bool
}
// NewCatProcessor creates a new cat processor
func NewCatProcessor(plain, noColor bool, hostname string) *CatProcessor {
return &CatProcessor{
plain: plain,
noColor: noColor,
hostname: hostname,
isFirstLine: true,
}
}
func (cp *CatProcessor) Initialize(ctx context.Context) error {
return nil
}
func (cp *CatProcessor) Cleanup() error {
return nil
}
// ProcessLine processes a single line for cat output.
// In plain mode, it preserves the original line exactly including line endings.
// In non-plain mode, it formats the line according to DTail protocol with optional colorization.
// Returns the formatted line and true (cat always outputs all lines).
func (cp *CatProcessor) ProcessLine(line []byte, lineNum int, filePath string, stats *stats, sourceID string) ([]byte, bool) {
// Update stats for matched line (cat always matches all lines)
if stats != nil {
stats.updateLineMatched()
}
// Format output to match existing behavior
if cp.plain {
// In plain mode, preserve the original line exactly as it is
// The line already includes its original line ending
result := make([]byte, len(line))
copy(result, line)
return result, true
}
// Format exactly like original basehandler.go for non-plain mode
// REMOTE|{hostname}|{TransmittedPerc}|{Count}|{SourceID}|{Content}¬
var transmittedPerc int
var count uint64
if stats != nil {
// For cat, we always transmit all matched lines, so transmittedPerc should be 100
transmittedPerc = 100
count = stats.totalLineCount()
}
// Build the protocol line
protocolLine := fmt.Sprintf("REMOTE%s%s%s%3d%s%v%s%s%s%s",
protocol.FieldDelimiter, cp.hostname, protocol.FieldDelimiter,
transmittedPerc, protocol.FieldDelimiter, count, protocol.FieldDelimiter,
sourceID, protocol.FieldDelimiter, string(line))
// Apply ANSI color formatting if not in plain mode and not noColor mode
if !cp.plain && !cp.noColor {
colorized := brush.Colorfy(protocolLine)
// Add color reset prefix for all lines except the first
var result []byte
if cp.isFirstLine {
cp.isFirstLine = false
result = make([]byte, len(colorized)+1)
copy(result, colorized)
result[len(colorized)] = '\n'
} else {
// Add color reset prefix: [39m[49m[49m[39m
colorResetPrefix := "\x1b[39m\x1b[49m\x1b[49m\x1b[39m"
result = make([]byte, len(colorResetPrefix)+len(colorized)+1)
copy(result, colorResetPrefix)
copy(result[len(colorResetPrefix):], colorized)
result[len(colorResetPrefix)+len(colorized)] = '\n'
}
return result, true
}
// No color formatting
result := make([]byte, len(protocolLine)+1)
copy(result, protocolLine)
result[len(protocolLine)] = '\n'
return result, true
}
func (cp *CatProcessor) Flush() []byte {
// Add final color reset line to match original behavior (no trailing newline)
// Only in non-plain mode with colors enabled
if !cp.plain && !cp.noColor {
return []byte("\x1b[39m\x1b[49m\x1b[49m\x1b[39m")
}
return nil
}
|
