llamactl/pkg/auth/key.go

package auth

import (
	"crypto/rand"
	"encoding/hex"
	"fmt"
)

type PermissionMode string

const (
	PermissionModeAllowAll    PermissionMode = "allow_all"
	PermissionModePerInstance PermissionMode = "per_instance"
)

type APIKey struct {
	ID             int
	KeyHash        string
	Name           string
	UserID         string
	PermissionMode PermissionMode
	ExpiresAt      *int64
	CreatedAt      int64
	UpdatedAt      int64
	LastUsedAt     *int64
}

type KeyPermission struct {
	KeyID      int
	InstanceID int
	CanInfer   bool
}

// GenerateKey generates a cryptographically secure API key with the given prefix
func GenerateKey(prefix string) (string, error) {
	// Generate 32 random bytes
	bytes := make([]byte, 32)
	_, err := rand.Read(bytes)
	if err != nil {
		return "", fmt.Errorf("failed to generate random bytes: %w", err)
	}

	// Convert to hex (64 characters)
	hexStr := hex.EncodeToString(bytes)

	return fmt.Sprintf("%s-%s", prefix, hexStr), nil
}