// Copyright 2022 CloudWeGo 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 log

import (
	"github.com/cloudwego/hertz/pkg/common/hlog"
	"go.uber.org/zap/zapcore"
	"strings"

	"github.com/cloudwego/kitex/pkg/klog"
)

type Level string

const (
	LevelTrace  Level = "trace"
	LevelDebug  Level = "debug"
	LevelInfo   Level = "info"
	LevelNotice Level = "notice"
	LevelWarn   Level = "warn"
	LevelError  Level = "error"
	LevelFatal  Level = "fatal"
)

// KLogLevel return kitex log level
func (level Level) KLogLevel() klog.Level {
	l := Level(strings.ToLower(string(level)))
	switch l {
	case LevelTrace:
		return klog.LevelTrace
	case LevelDebug:
		return klog.LevelDebug
	case LevelInfo:
		return klog.LevelInfo
	case LevelNotice:
		return klog.LevelNotice
	case LevelWarn:
		return klog.LevelWarn
	case LevelError:
		return klog.LevelError
	case LevelFatal:
		return klog.LevelFatal
	default:
		return klog.LevelTrace
	}
}

func (level Level) HLogLevel() hlog.Level {
	l := Level(strings.ToLower(string(level)))
	switch l {
	case LevelTrace:
		return hlog.LevelTrace
	case LevelDebug:
		return hlog.LevelDebug
	case LevelInfo:
		return hlog.LevelInfo
	case LevelNotice:
		return hlog.LevelNotice
	case LevelWarn:
		return hlog.LevelWarn
	case LevelError:
		return hlog.LevelError
	case LevelFatal:
		return hlog.LevelFatal
	default:
		return hlog.LevelTrace
	}
}
func (level Level) ZapLevel() zapcore.Level {
	l := Level(strings.ToLower(string(level)))
	switch l {
	case LevelTrace, LevelDebug:
		return zapcore.DebugLevel
	case LevelInfo:

		return zapcore.InfoLevel
	case LevelNotice, LevelWarn:
		return zapcore.WarnLevel
	case LevelError:
		return zapcore.ErrorLevel
	case LevelFatal:
		return zapcore.FatalLevel
	default:
		return zapcore.WarnLevel
	}
}