sig-auth/keydirectory/in_memory_directory.go

package keydirectory

import (
	"context"
	"crypto"
	"errors"

	"github.com/common-fate/httpsig/verifier"
)

type inMemoryDirectory struct {
	records map[string]keyEntry
}

func CreateMemoryDirectory() inMemoryDirectory {
	return inMemoryDirectory{
		records: map[string]keyEntry{},
	}
}

func (dir inMemoryDirectory) GetKey(ctx context.Context, keyId string, _ string) (verifier.Algorithm, error) {
	entry, ok := dir.records[keyId]

	if !ok {
		return nil, errors.New("key not found in directory")
	}

	return entry.toAlg()
}

func (dir inMemoryDirectory) RegisterKey(key crypto.PublicKey, userId string) (string, error) {
	keyId, err := generateKeyId()

	if err != nil {
		return "", err
	}

	dir.records[keyId] = keyEntry{
		PublicKey: key,
		UserId:    userId,
	}

	return keyId, nil
}