cloudflared-mirror/sshlog/manager.go

package sshlog

import (
	"io"
	"path/filepath"
	"time"

	"github.com/sirupsen/logrus"
)

// Manager be managing logs bruh
type Manager interface {
	NewLogger(string, *logrus.Logger) (io.WriteCloser, error)
	NewSessionLogger(string, *logrus.Logger) (io.WriteCloser, error)
}

type manager struct {
	baseDirectory string
}

// New creates a new instance of a log manager
func New(baseDirectory string) Manager {
	return &manager{
		baseDirectory: baseDirectory,
	}
}

func (m *manager) NewLogger(name string, logger *logrus.Logger) (io.WriteCloser, error) {
	return NewLogger(filepath.Join(m.baseDirectory, name), logger, time.Second, defaultFileSizeLimit)
}

func (m *manager) NewSessionLogger(name string, logger *logrus.Logger) (io.WriteCloser, error) {
	return NewSessionLogger(name, logger, time.Second, defaultFileSizeLimit)
}