// Copyright 2022 Google LLC
//
// 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 monitoringutil implements utility functions for monitoring.
package monitoringutil

import (
	"fmt"
	"strings"

	"github.com/tink-crypto/tink-go/v2/monitoring"
	tpb "github.com/tink-crypto/tink-go/v2/proto/tink_go_proto"
)

const keytypeURLPrefix = "type.googleapis.com/google.crypto."

// DoNothingLogger is a Logger that does nothing when invoked.
type DoNothingLogger struct{}

var _ monitoring.Logger = (*DoNothingLogger)(nil)

// Log drops a log call.
func (l *DoNothingLogger) Log(uint32, int) {}

// LogFailure drops a failure call.
func (l *DoNothingLogger) LogFailure() {}

// LogKeyExport drops a key export call.
func (l *DoNothingLogger) LogKeyExport(keyID uint32) {}

func keyStatusFromProto(status tpb.KeyStatusType) (monitoring.KeyStatus, error) {
	var keyStatus monitoring.KeyStatus = 55
	switch status {
	case tpb.KeyStatusType_ENABLED:
		keyStatus = monitoring.Enabled
	case tpb.KeyStatusType_DISABLED:
		keyStatus = monitoring.Disabled
	case tpb.KeyStatusType_DESTROYED:
		keyStatus = monitoring.Destroyed
	default:
		return keyStatus, fmt.Errorf("unknown key status: %q", status)
	}
	return keyStatus, nil

}

func parseKeyTypeURL(ktu string) string {
	return strings.TrimPrefix(ktu, keytypeURLPrefix)
}

// MonitoringKeysetInfoFromKeysetInfo creates a [monitoring.KeysetInfo] from a
// [tpb.KeysetInfo] with annotations.
func MonitoringKeysetInfoFromKeysetInfo(keysetInfo *tpb.KeysetInfo, annotations map[string]string) (*monitoring.KeysetInfo, error) {
	keysetInfoEntries := make([]*monitoring.Entry, len(keysetInfo.GetKeyInfo()))
	for i, keyInfo := range keysetInfo.GetKeyInfo() {
		status, err := keyStatusFromProto(keyInfo.GetStatus())
		if err != nil {
			return nil, err
		}
		keysetInfoEntries[i] = &monitoring.Entry{
			KeyID:     keyInfo.GetKeyId(),
			Status:    status,
			KeyType:   parseKeyTypeURL(keyInfo.GetTypeUrl()),
			KeyPrefix: keyInfo.GetOutputPrefixType().String(),
		}
	}
	return &monitoring.KeysetInfo{
		Annotations:  annotations,
		PrimaryKeyID: keysetInfo.GetPrimaryKeyId(),
		Entries:      keysetInfoEntries,
	}, nil
}
