containerd/api/runtime/bootstrap/v1/helpers.go

/*
   Copyright The containerd 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 bootstrap

import (
	"fmt"
	"strconv"

	"google.golang.org/protobuf/proto"
	"google.golang.org/protobuf/types/known/anypb"
)

// LogLevelFromString converts a log level string (e.g. "debug", "info") to a
// LogLevel enum value. The accepted strings are compatible with logrus level names.
func LogLevelFromString(s string) LogLevel {
	switch s {
	case "trace":
		return LogLevel_LOG_LEVEL_TRACE
	case "debug":
		return LogLevel_LOG_LEVEL_DEBUG
	case "info":
		return LogLevel_LOG_LEVEL_INFO
	case "warn", "warning":
		return LogLevel_LOG_LEVEL_WARN
	case "error":
		return LogLevel_LOG_LEVEL_ERROR
	case "fatal":
		return LogLevel_LOG_LEVEL_FATAL
	case "panic":
		return LogLevel_LOG_LEVEL_PANIC
	default:
		if v, err := strconv.ParseInt(s, 10, 32); err == nil {
			return LogLevel(v)
		}
		return LogLevel_LOG_LEVEL_INFO
	}
}

// AddExtension adds a new extension to the BootstrapParams.
// The message is wrapped in a google.protobuf.Any with its type URL automatically set.
// If the message is already an *anypb.Any, it is used directly without double-wrapping.
func (p *BootstrapParams) AddExtension(msg proto.Message) error {
	var anyVal *anypb.Any
	if a, ok := msg.(*anypb.Any); ok {
		// Already an Any, use it directly
		anyVal = a
	} else {
		var err error
		anyVal, err = anypb.New(msg)
		if err != nil {
			return err
		}
	}

	p.Extensions = append(p.Extensions, &Extension{Value: anyVal})
	return nil
}

// FindExtension finds an extension matching the type of dst and unmarshals it.
func (p *BootstrapParams) FindExtension(dst proto.Message) (bool, error) {
	name := dst.ProtoReflect().Descriptor().FullName()

	for _, ext := range p.GetExtensions() {
		if ext.GetValue().MessageIs(dst) {
			if err := ext.GetValue().UnmarshalTo(dst); err != nil {
				return false, fmt.Errorf("failed to unmarshal extension %q: %w", name, err)
			}
			return true, nil
		}
	}

	return false, nil
}

// AddExtension adds a new extension to the BootstrapResult.
// The message is wrapped in a google.protobuf.Any with its type URL automatically set.
// If the message is already an *anypb.Any, it is used directly without double-wrapping.
func (r *BootstrapResult) AddExtension(msg proto.Message) error {
	var anyVal *anypb.Any
	if a, ok := msg.(*anypb.Any); ok {
		// Already an Any, use it directly
		anyVal = a
	} else {
		var err error
		anyVal, err = anypb.New(msg)
		if err != nil {
			return err
		}
	}

	r.Extensions = append(r.Extensions, &Extension{Value: anyVal})
	return nil
}

// FindExtension finds an extension matching the type of dst and unmarshals it.
// Extensions of any other type are ignored.
func (r *BootstrapResult) FindExtension(dst proto.Message) (bool, error) {
	name := dst.ProtoReflect().Descriptor().FullName()

	for _, ext := range r.GetExtensions() {
		if ext.GetValue().MessageIs(dst) {
			if err := ext.GetValue().UnmarshalTo(dst); err != nil {
				return false, fmt.Errorf("failed to unmarshal extension %q: %w", name, err)
			}
			return true, nil
		}
	}

	return false, nil
}