package dgraph

import (
	"fmt"

	"next.orly.dev/pkg/crypto/keys"
)

// Relay identity methods
// We use the marker system to store the relay's private key

const relayIdentityMarkerKey = "relay_identity_secret"

// GetRelayIdentitySecret retrieves the relay's identity secret key
func (d *D) GetRelayIdentitySecret() (skb []byte, err error) {
	return d.GetMarker(relayIdentityMarkerKey)
}

// SetRelayIdentitySecret sets the relay's identity secret key
func (d *D) SetRelayIdentitySecret(skb []byte) error {
	return d.SetMarker(relayIdentityMarkerKey, skb)
}

// GetOrCreateRelayIdentitySecret retrieves or creates the relay identity
func (d *D) GetOrCreateRelayIdentitySecret() (skb []byte, err error) {
	skb, err = d.GetRelayIdentitySecret()
	if err == nil {
		return skb, nil
	}

	// Generate new identity
	skb, err = keys.GenerateSecretKey()
	if err != nil {
		return nil, fmt.Errorf("failed to generate identity: %w", err)
	}

	// Store it
	if err = d.SetRelayIdentitySecret(skb); err != nil {
		return nil, fmt.Errorf("failed to store identity: %w", err)
	}

	d.Logger.Infof("generated new relay identity")
	return skb, nil
}