schema: refactoring and making it usable and flexible.

This commit is contained in:
fiatjaf
2025-11-23 11:27:17 -03:00
parent 8458e26291
commit b820569d93
3 changed files with 121 additions and 1656 deletions

View File

@@ -3,7 +3,10 @@ package schema
import (
_ "embed"
"fmt"
"io"
"net/http"
"net/url"
"os"
"slices"
"strconv"
"strings"
@@ -15,8 +18,26 @@ import (
"gopkg.in/yaml.v3"
)
//go:embed schema.yaml
var schemaFile []byte
const DefaultSchemaURL = "https://raw.githubusercontent.com/nostr-protocol/registry-of-kinds/refs/heads/master/schema.yaml"
func fetchSchemaFromURL(schemaURL string) (string, error) {
resp, err := http.Get(schemaURL)
if err != nil {
return "", fmt.Errorf("failed to fetch schema from URL: %w", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return "", fmt.Errorf("failed to fetch schema: HTTP %d", resp.StatusCode)
}
body, err := io.ReadAll(resp.Body)
if err != nil {
return "", fmt.Errorf("failed to read schema response: %w", err)
}
return string(body), nil
}
type Schema map[string]KindSchema
@@ -44,17 +65,32 @@ type Validator struct {
FailOnUnknown bool
}
func NewValidator(schemaData string) Validator {
func NewValidatorFromBytes(schemaData []byte) (Validator, error) {
schema := make(Schema)
if err := yaml.Unmarshal([]byte(schemaData), &schema); err != nil {
panic(fmt.Errorf("failed to parse schema.yaml: %w", err))
if err := yaml.Unmarshal(schemaData, &schema); err != nil {
return Validator{}, fmt.Errorf("failed to parse schema: %w", err)
}
return Validator{Schema: schema}, nil
}
return Validator{Schema: schema}
func NewValidatorFromSchema(sch Schema) Validator {
return Validator{Schema: sch}
}
func NewDefaultValidator() Validator {
return NewValidator(string(schemaFile))
func NewValidatorFromFile(filename string) (Validator, error) {
data, err := os.ReadFile(filename)
if err != nil {
return Validator{}, fmt.Errorf("failed to read schema file: %w", err)
}
return NewValidatorFromBytes(data)
}
func NewValidatorFromURL(schemaURL string) (Validator, error) {
schemaData, err := fetchSchemaFromURL(schemaURL)
if err != nil {
return Validator{}, err
}
return NewValidatorFromBytes([]byte(schemaData))
}
var (

File diff suppressed because it is too large Load Diff

View File

@@ -8,8 +8,9 @@ import (
"github.com/stretchr/testify/require"
)
func TestNewValidator(t *testing.T) {
v := NewValidator(string(schemaFile))
func TestNewValidatorFromURL(t *testing.T) {
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
require.NotNil(t, v.Schema)
require.False(t, v.FailOnUnknown) // default value
@@ -22,14 +23,14 @@ func TestNewValidator(t *testing.T) {
require.True(t, hasKind1111)
}
func TestNewValidatorWithInvalidYAML(t *testing.T) {
require.Panics(t, func() {
NewValidator("invalid yaml content: [[[")
})
func TestNewValidatorFromBytesWithInvalidYAML(t *testing.T) {
_, err := NewValidatorFromBytes([]byte("invalid yaml content: [[["))
require.Error(t, err)
}
func TestValidateEvent_BasicSuccess(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
// kind 1 with free content and valid p tag
evt := nostr.Event{
@@ -40,30 +41,32 @@ func TestValidateEvent_BasicSuccess(t *testing.T) {
},
}
err := v.ValidateEvent(evt)
err = v.ValidateEvent(evt)
require.NoError(t, err)
}
func TestValidateEvent_Kind0_JSONContent(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
// valid JSON content
evt := nostr.Event{
Kind: 0,
Content: `{"name":"test","about":"description"}`,
}
err := v.ValidateEvent(evt)
err = v.ValidateEvent(evt)
require.NoError(t, err)
// invalid JSON content
evt.Content = "not-json-content"
err = v.ValidateEvent(evt)
require.Error(t, err)
require.Equal(t, ErrInvalidContentJson, err)
require.Equal(t, ContentError{Err: ErrInvalidJson}, err)
}
func TestValidateEvent_UnknownKind(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
evt := nostr.Event{
Kind: nostr.Kind(39999),
@@ -71,7 +74,7 @@ func TestValidateEvent_UnknownKind(t *testing.T) {
}
// should not fail when FailOnUnknown is false (default)
err := v.ValidateEvent(evt)
err = v.ValidateEvent(evt)
require.NoError(t, err)
// should fail when FailOnUnknown is true
@@ -82,22 +85,25 @@ func TestValidateEvent_UnknownKind(t *testing.T) {
}
func TestValidateEvent_EmptyTag(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
evt := nostr.Event{
Kind: 1,
Content: "test",
Tags: nostr.Tags{
nostr.Tag{}, // empty tag
},
}
err := v.ValidateEvent(evt)
err = v.ValidateEvent(evt)
require.Error(t, err)
require.Equal(t, ErrEmptyTag, err)
}
func TestValidateNext_ID(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
tests := []struct {
name string
@@ -116,7 +122,7 @@ func TestValidateNext_ID(t *testing.T) {
},
{
name: "invalid id - not hex",
tag: nostr.Tag{"e", "zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz"},
tag: nostr.Tag{"e", "zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz"},
valid: false,
},
}
@@ -124,7 +130,7 @@ func TestValidateNext_ID(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
next := &nextSpec{Type: "id", Required: true}
err := v.validateNext(tt.tag, 1, next)
_, err := v.validateNext(tt.tag, 1, next)
if tt.valid {
require.NoError(t, err)
} else {
@@ -135,7 +141,8 @@ func TestValidateNext_ID(t *testing.T) {
}
func TestValidateNext_PubKey(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
tests := []struct {
name string
@@ -162,7 +169,7 @@ func TestValidateNext_PubKey(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
next := &nextSpec{Type: "pubkey", Required: true}
err := v.validateNext(tt.tag, 1, next)
_, err := v.validateNext(tt.tag, 1, next)
if tt.valid {
require.NoError(t, err)
} else {
@@ -173,7 +180,8 @@ func TestValidateNext_PubKey(t *testing.T) {
}
func TestValidateNext_Relay(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
tests := []struct {
name string
@@ -210,7 +218,7 @@ func TestValidateNext_Relay(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
next := &nextSpec{Type: "relay", Required: true}
err := v.validateNext(tt.tag, 1, next)
_, err := v.validateNext(tt.tag, 1, next)
if tt.valid {
require.NoError(t, err)
} else {
@@ -221,7 +229,8 @@ func TestValidateNext_Relay(t *testing.T) {
}
func TestValidateNext_Kind(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
tests := []struct {
name string
@@ -258,7 +267,7 @@ func TestValidateNext_Kind(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
next := &nextSpec{Type: "kind", Required: true}
err := v.validateNext(tt.tag, 1, next)
_, err := v.validateNext(tt.tag, 1, next)
if tt.valid {
require.NoError(t, err)
} else {
@@ -269,7 +278,8 @@ func TestValidateNext_Kind(t *testing.T) {
}
func TestValidateNext_Constrained(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
tests := []struct {
name string
@@ -294,7 +304,7 @@ func TestValidateNext_Constrained(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
next := &nextSpec{Type: "constrained", Required: true, Either: tt.allowed}
err := v.validateNext(tt.tag, 3, next)
_, err := v.validateNext(tt.tag, 3, next)
if tt.valid {
require.NoError(t, err)
} else {
@@ -305,7 +315,8 @@ func TestValidateNext_Constrained(t *testing.T) {
}
func TestValidateNext_GitCommit(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
tests := []struct {
name string
@@ -337,7 +348,7 @@ func TestValidateNext_GitCommit(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
next := &nextSpec{Type: "gitcommit", Required: true}
err := v.validateNext(tt.tag, 1, next)
_, err := v.validateNext(tt.tag, 1, next)
if tt.valid {
require.NoError(t, err)
} else {
@@ -348,7 +359,8 @@ func TestValidateNext_GitCommit(t *testing.T) {
}
func TestValidateNext_Addr(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
tests := []struct {
name string
@@ -370,7 +382,7 @@ func TestValidateNext_Addr(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
next := &nextSpec{Type: "addr", Required: true}
err := v.validateNext(tt.tag, 1, next)
_, err := v.validateNext(tt.tag, 1, next)
if tt.valid {
require.NoError(t, err)
} else {
@@ -381,71 +393,76 @@ func TestValidateNext_Addr(t *testing.T) {
}
func TestValidateNext_Free(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
// free type should accept anything
tag := nostr.Tag{"test", "any value here", "even", "multiple", "values"}
next := &nextSpec{Type: "free", Required: true}
err := v.validateNext(tag, 1, next)
_, err = v.validateNext(tag, 1, next)
require.NoError(t, err)
}
func TestValidateNext_UnknownType(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
tag := nostr.Tag{"test", "value"}
next := &nextSpec{Type: "unknown-type", Required: true}
// should not fail when FailOnUnknown is false (default)
err := v.validateNext(tag, 1, next)
_, err = v.validateNext(tag, 1, next)
require.NoError(t, err)
// should fail when FailOnUnknown is true
v.FailOnUnknown = true
err = v.validateNext(tag, 1, next)
_, err = v.validateNext(tag, 1, next)
require.Error(t, err)
require.Equal(t, ErrUnknownTagType, err)
}
func TestValidateNext_RequiredField(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
// test missing required field
tag := nostr.Tag{"test"} // only name, missing required value
next := &nextSpec{Type: "free", Required: true}
err := v.validateNext(tag, 1, next)
_, err = v.validateNext(tag, 1, next)
require.Error(t, err)
require.Contains(t, err.Error(), "missing index 1")
// test optional field
next = &nextSpec{Type: "free", Required: false}
err = v.validateNext(tag, 1, next)
_, err = v.validateNext(tag, 1, next)
require.NoError(t, err)
}
func TestValidateNext_Variadic(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
// test variadic field with multiple values
tag := nostr.Tag{"test", "value1", "value2", "value3"}
next := &nextSpec{Type: "free", Variadic: true}
err := v.validateNext(tag, 1, next)
_, err = v.validateNext(tag, 1, next)
require.NoError(t, err)
// test variadic field with single value
tag = nostr.Tag{"test", "only-one-value"}
err = v.validateNext(tag, 1, next)
_, err = v.validateNext(tag, 1, next)
require.NoError(t, err)
// test variadic field with no values (should fail if required)
tag = nostr.Tag{"test"}
next = &nextSpec{Type: "free", Variadic: true, Required: true}
err = v.validateNext(tag, 1, next)
_, err = v.validateNext(tag, 1, next)
require.Error(t, err)
}
func TestValidateEvent_Kind10002(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
// kind 10002 (relay list metadata) with valid r tags
evt := nostr.Event{
@@ -457,7 +474,7 @@ func TestValidateEvent_Kind10002(t *testing.T) {
},
}
err := v.ValidateEvent(evt)
err = v.ValidateEvent(evt)
require.NoError(t, err)
// test with invalid relay marker
@@ -474,7 +491,8 @@ func TestValidateEvent_Kind10002(t *testing.T) {
}
func TestValidateEvent_Kind1_ETag(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
// kind 1 with e tag (reply/root marker)
evt := nostr.Event{
@@ -485,7 +503,7 @@ func TestValidateEvent_Kind1_ETag(t *testing.T) {
},
}
err := v.ValidateEvent(evt)
err = v.ValidateEvent(evt)
require.NoError(t, err)
// test with invalid marker
@@ -497,7 +515,8 @@ func TestValidateEvent_Kind1_ETag(t *testing.T) {
}
func TestValidateEvent_Kind30617_RepositoryAnnouncement(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
// kind 30617 (repository announcement) with required tags
evt := nostr.Event{
@@ -515,12 +534,13 @@ func TestValidateEvent_Kind30617_RepositoryAnnouncement(t *testing.T) {
},
}
err := v.ValidateEvent(evt)
err = v.ValidateEvent(evt)
require.NoError(t, err)
}
func TestValidateEvent_MultiplePossibleTagSpecs(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
// test event with tags that could match multiple specs
// kind 1 has both "e" and "q" tags that can take different forms
@@ -533,7 +553,7 @@ func TestValidateEvent_MultiplePossibleTagSpecs(t *testing.T) {
},
}
err := v.ValidateEvent(evt)
err = v.ValidateEvent(evt)
require.NoError(t, err)
// test q tag with id format (alternative spec)
@@ -545,7 +565,8 @@ func TestValidateEvent_MultiplePossibleTagSpecs(t *testing.T) {
}
func TestSchema_ErrorMessages(t *testing.T) {
v := NewValidator(string(schemaFile))
v, err := NewValidatorFromURL(DefaultSchemaURL)
require.NoError(t, err)
tests := []struct {
name string
@@ -556,6 +577,7 @@ func TestSchema_ErrorMessages(t *testing.T) {
name: "empty tag error",
event: nostr.Event{
Kind: 1,
Content: "test",
Tags: nostr.Tags{nostr.Tag{}},
},
expError: ErrEmptyTag,