woodpecker/server/forge/addon/logger.go
qwerty287 00f0fcd416
Rework addons (use rpc) (#3268)
Co-authored-by: Anbraten <6918444+anbraten@users.noreply.github.com>
2024-04-15 10:04:21 +02:00

165 lines
3.7 KiB
Go

// Copyright 2024 Woodpecker Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package addon
import (
"bytes"
"io"
stdlog "log"
"github.com/hashicorp/go-hclog"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
)
type clientLogger struct {
logger zerolog.Logger
name string
withArgs []any
}
func convertLvl(level hclog.Level) zerolog.Level {
switch level {
case hclog.Error:
return zerolog.ErrorLevel
case hclog.Warn:
return zerolog.WarnLevel
case hclog.Info:
return zerolog.InfoLevel
case hclog.Debug:
return zerolog.DebugLevel
case hclog.Trace:
return zerolog.TraceLevel
}
return zerolog.NoLevel
}
func (c *clientLogger) applyArgs(args []any) *zerolog.Logger {
var key string
logger := c.logger.With()
args = append(args, c.withArgs)
for i, arg := range args {
switch {
case key != "":
logger.Any(key, arg)
key = ""
case i == len(args)-1:
logger.Any(hclog.MissingKey, arg)
default:
key, _ = arg.(string)
}
}
l := logger.Logger()
return &l
}
func (c *clientLogger) Log(level hclog.Level, msg string, args ...any) {
c.applyArgs(args).WithLevel(convertLvl(level)).Msg(msg)
}
func (c *clientLogger) Trace(msg string, args ...any) {
c.applyArgs(args).Trace().Msg(msg)
}
func (c *clientLogger) Debug(msg string, args ...any) {
c.applyArgs(args).Debug().Msg(msg)
}
func (c *clientLogger) Info(msg string, args ...any) {
c.applyArgs(args).Info().Msg(msg)
}
func (c *clientLogger) Warn(msg string, args ...any) {
c.applyArgs(args).Warn().Msg(msg)
}
func (c *clientLogger) Error(msg string, args ...any) {
c.applyArgs(args).Error().Msg(msg)
}
func (c *clientLogger) IsTrace() bool {
return log.Logger.GetLevel() >= zerolog.TraceLevel
}
func (c *clientLogger) IsDebug() bool {
return log.Logger.GetLevel() >= zerolog.DebugLevel
}
func (c *clientLogger) IsInfo() bool {
return log.Logger.GetLevel() >= zerolog.InfoLevel
}
func (c *clientLogger) IsWarn() bool {
return log.Logger.GetLevel() >= zerolog.WarnLevel
}
func (c *clientLogger) IsError() bool {
return log.Logger.GetLevel() >= zerolog.ErrorLevel
}
func (c *clientLogger) ImpliedArgs() []any {
return c.withArgs
}
func (c *clientLogger) With(args ...any) hclog.Logger {
return &clientLogger{
logger: c.logger,
name: c.name,
withArgs: args,
}
}
func (c *clientLogger) Name() string {
return c.name
}
func (c *clientLogger) Named(name string) hclog.Logger {
curr := c.name
if curr != "" {
curr = c.name + "."
}
return c.ResetNamed(curr + name)
}
func (c *clientLogger) ResetNamed(name string) hclog.Logger {
return &clientLogger{
logger: c.logger,
name: name,
withArgs: c.withArgs,
}
}
func (c *clientLogger) SetLevel(level hclog.Level) {
c.logger = c.logger.Level(convertLvl(level))
}
func (c *clientLogger) StandardLogger(opts *hclog.StandardLoggerOptions) *stdlog.Logger {
return stdlog.New(c.StandardWriter(opts), "", 0)
}
func (c *clientLogger) StandardWriter(*hclog.StandardLoggerOptions) io.Writer {
return ioAdapter{logger: c.logger}
}
type ioAdapter struct {
logger zerolog.Logger
}
func (i ioAdapter) Write(p []byte) (n int, err error) {
str := string(bytes.TrimRight(p, " \t\n"))
i.logger.Log().Msg(str)
return len(p), nil
}