mirror of
https://github.com/wneessen/logranger.git
synced 2024-10-06 09:05:57 +02:00
Winni Neessen
a20b6c64c6
Module references within 'go.mod', 'main.go', and various plugin files have been changed from 'github.com/wneessen/logranger' to 'src.neessen.cloud/wneessen/logranger'. This update is essential to align with the new source location and ensure code and plugins function as expected. The Dockerfile’s commands have also been updated accordingly.
106 lines
2.7 KiB
Go
106 lines
2.7 KiB
Go
// SPDX-FileCopyrightText: 2023 Winni Neessen <wn@neessen.dev>
|
|
//
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package file
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/wneessen/go-parsesyslog"
|
|
|
|
"src.neessen.cloud/wneessen/logranger/plugins/actions"
|
|
"src.neessen.cloud/wneessen/logranger/template"
|
|
)
|
|
|
|
// File represents a file action that can be performed on a log message.
|
|
type File struct {
|
|
Enabled bool
|
|
FilePath string
|
|
OutputTemplate string
|
|
Overwrite bool
|
|
}
|
|
|
|
// Config satisfies the plugins.Action interface for the File type
|
|
// It updates the configuration of the File action based on the provided
|
|
// configuration map.
|
|
//
|
|
// It expects the configuration map to have a key "file" which contains a submap
|
|
// with the following keys:
|
|
// - "output_filepath" (string): Specifies the file path where the output will be written.
|
|
// - "output_template" (string): Specifies the template to use for formatting the output.
|
|
// - "overwrite" (bool, optional): If true, the file will be overwritten instead of appended to.
|
|
//
|
|
// If any of the required configuration parameters are missing or invalid, an error
|
|
// is returned.
|
|
func (f *File) Config(cm map[string]any) error {
|
|
if cm["file"] == nil {
|
|
return nil
|
|
}
|
|
c, ok := cm["file"].(map[string]any)
|
|
if !ok {
|
|
return fmt.Errorf("missing configuration for file action")
|
|
}
|
|
f.Enabled = true
|
|
|
|
fp, ok := c["output_filepath"].(string)
|
|
if !ok || fp == "" {
|
|
return fmt.Errorf("no output_filename configured for file action")
|
|
}
|
|
f.FilePath = fp
|
|
|
|
ot, ok := c["output_template"].(string)
|
|
if !ok || ot == "" {
|
|
return fmt.Errorf("not output_template configured for file action")
|
|
}
|
|
f.OutputTemplate = ot
|
|
|
|
if ow, ok := c["overwrite"].(bool); ok && ow {
|
|
f.Overwrite = true
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Process satisfies the plugins.Action interface for the File type
|
|
// It takes in the log message (lm), match groups (mg), and configuration map (cm).
|
|
func (f *File) Process(lm parsesyslog.LogMsg, mg []string) error {
|
|
if !f.Enabled {
|
|
return nil
|
|
}
|
|
|
|
of := os.O_APPEND | os.O_CREATE | os.O_WRONLY
|
|
if f.Overwrite {
|
|
of = os.O_TRUNC | os.O_CREATE | os.O_WRONLY
|
|
}
|
|
|
|
fh, err := os.OpenFile(f.FilePath, of, 0o600)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open file for writing in file action: %w", err)
|
|
}
|
|
defer func() {
|
|
_ = fh.Close()
|
|
}()
|
|
|
|
t, err := template.Compile(lm, mg, f.OutputTemplate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = fh.WriteString(t)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to write log message to file %q: %w",
|
|
f.FilePath, err)
|
|
}
|
|
if err = fh.Sync(); err != nil {
|
|
return fmt.Errorf("failed to sync memory to file %q: %w",
|
|
f.FilePath, err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// init registers the "file" action with the Actions map.
|
|
func init() {
|
|
actions.Add("file", &File{})
|
|
}
|