woodpecker/server/forge/addon/logger.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
}