mild refactoring

This commit is contained in:
Jay
2025-10-22 12:14:37 -04:00
parent 580b2e4859
commit 5055c04629
2 changed files with 95 additions and 95 deletions

112
event.go
View File

@@ -26,14 +26,14 @@ var (
Hex128Pattern = regexp.MustCompile("^[a-f0-9]{128}$") Hex128Pattern = regexp.MustCompile("^[a-f0-9]{128}$")
) )
func (event Event) Serialize() ([]byte, error) { func (e Event) Serialize() ([]byte, error) {
serialized := []interface{}{ serialized := []interface{}{
0, 0,
event.PubKey, e.PubKey,
event.CreatedAt, e.CreatedAt,
event.Kind, e.Kind,
event.Tags, e.Tags,
event.Content, e.Content,
} }
bytes, err := json.Marshal(serialized) bytes, err := json.Marshal(serialized)
@@ -43,8 +43,8 @@ func (event Event) Serialize() ([]byte, error) {
return bytes, nil return bytes, nil
} }
func (event Event) GetID() (string, error) { func (e Event) GetID() (string, error) {
bytes, err := event.Serialize() bytes, err := e.Serialize()
if err != nil { if err != nil {
return "", err return "", err
} }
@@ -52,54 +52,6 @@ func (event Event) GetID() (string, error) {
return hex.EncodeToString(hash[:]), nil return hex.EncodeToString(hash[:]), nil
} }
func (event Event) Validate() error {
if err := event.ValidateStructure(); err != nil {
return err
}
if err := event.ValidateID(); err != nil {
return err
}
return ValidateSignature(event.ID, event.Sig, event.PubKey)
}
func (event Event) ValidateStructure() error {
if !Hex64Pattern.MatchString(event.PubKey) {
return fmt.Errorf("pubkey must be 64 lowercase hex characters")
}
if !Hex64Pattern.MatchString(event.ID) {
return fmt.Errorf("id must be 64 hex characters")
}
if !Hex128Pattern.MatchString(event.Sig) {
return fmt.Errorf("signature must be 128 hex characters")
}
for _, tag := range event.Tags {
if len(tag) < 2 {
return fmt.Errorf("tags must contain at least two elements")
}
}
return nil
}
func (event Event) ValidateID() error {
computedID, err := event.GetID()
if err != nil {
return fmt.Errorf("failed to compute event id")
}
if event.ID == "" {
return fmt.Errorf("event id is empty")
}
if computedID != event.ID {
return fmt.Errorf("event id %q does not match computed id %q", event.ID, computedID)
}
return nil
}
func SignEvent(eventID, privateKeyHex string) (string, error) { func SignEvent(eventID, privateKeyHex string) (string, error) {
skBytes, err := hex.DecodeString(privateKeyHex) skBytes, err := hex.DecodeString(privateKeyHex)
if err != nil { if err != nil {
@@ -121,6 +73,54 @@ func SignEvent(eventID, privateKeyHex string) (string, error) {
return hex.EncodeToString(sig.Serialize()), nil return hex.EncodeToString(sig.Serialize()), nil
} }
func (e Event) Validate() error {
if err := e.ValidateStructure(); err != nil {
return err
}
if err := e.ValidateID(); err != nil {
return err
}
return ValidateSignature(e.ID, e.Sig, e.PubKey)
}
func (e Event) ValidateStructure() error {
if !Hex64Pattern.MatchString(e.PubKey) {
return fmt.Errorf("pubkey must be 64 lowercase hex characters")
}
if !Hex64Pattern.MatchString(e.ID) {
return fmt.Errorf("id must be 64 hex characters")
}
if !Hex128Pattern.MatchString(e.Sig) {
return fmt.Errorf("signature must be 128 hex characters")
}
for _, tag := range e.Tags {
if len(tag) < 2 {
return fmt.Errorf("tags must contain at least two elements")
}
}
return nil
}
func (e Event) ValidateID() error {
computedID, err := e.GetID()
if err != nil {
return fmt.Errorf("failed to compute event id")
}
if e.ID == "" {
return fmt.Errorf("event id is empty")
}
if computedID != e.ID {
return fmt.Errorf("event id %q does not match computed id %q", e.ID, computedID)
}
return nil
}
func ValidateSignature(eventID, eventSig, publicKeyHex string) error { func ValidateSignature(eventID, eventSig, publicKeyHex string) error {
idBytes, err := hex.DecodeString(eventID) idBytes, err := hex.DecodeString(eventID)
if err != nil { if err != nil {

View File

@@ -17,36 +17,36 @@ type Filter struct {
Extensions map[string]json.RawMessage Extensions map[string]json.RawMessage
} }
func (filter Filter) MarshalJSON() ([]byte, error) { func (f Filter) MarshalJSON() ([]byte, error) {
outputMap := make(map[string]interface{}) outputMap := make(map[string]interface{})
// Add standard fields // Add standard fields
if filter.IDs != nil { if f.IDs != nil {
outputMap["ids"] = filter.IDs outputMap["ids"] = f.IDs
} }
if filter.Authors != nil { if f.Authors != nil {
outputMap["authors"] = filter.Authors outputMap["authors"] = f.Authors
} }
if filter.Kinds != nil { if f.Kinds != nil {
outputMap["kinds"] = filter.Kinds outputMap["kinds"] = f.Kinds
} }
if filter.Since != nil { if f.Since != nil {
outputMap["since"] = *filter.Since outputMap["since"] = *f.Since
} }
if filter.Until != nil { if f.Until != nil {
outputMap["until"] = *filter.Until outputMap["until"] = *f.Until
} }
if filter.Limit != nil { if f.Limit != nil {
outputMap["limit"] = *filter.Limit outputMap["limit"] = *f.Limit
} }
// Add tags // Add tags
for key, values := range filter.Tags { for key, values := range f.Tags {
outputMap["#"+key] = values outputMap["#"+key] = values
} }
// Merge extensions // Merge extensions
for key, raw := range filter.Extensions { for key, raw := range f.Extensions {
// Disallow standard keys in extensions // Disallow standard keys in extensions
standardKeys := []string{"ids", "authors", "kinds", "since", "until", "limit"} standardKeys := []string{"ids", "authors", "kinds", "since", "until", "limit"}
found := false found := false
@@ -76,7 +76,7 @@ func (filter Filter) MarshalJSON() ([]byte, error) {
return json.Marshal(outputMap) return json.Marshal(outputMap)
} }
func (filter *Filter) UnmarshalJSON(data []byte) error { func (f *Filter) UnmarshalJSON(data []byte) error {
// Decode into raw map // Decode into raw map
raw := make(map[string]json.RawMessage) raw := make(map[string]json.RawMessage)
if err := json.Unmarshal(data, &raw); err != nil { if err := json.Unmarshal(data, &raw); err != nil {
@@ -85,21 +85,21 @@ func (filter *Filter) UnmarshalJSON(data []byte) error {
// Extract standard fields // Extract standard fields
if v, ok := raw["ids"]; ok { if v, ok := raw["ids"]; ok {
if err := json.Unmarshal(v, &filter.IDs); err != nil { if err := json.Unmarshal(v, &f.IDs); err != nil {
return err return err
} }
delete(raw, "ids") delete(raw, "ids")
} }
if v, ok := raw["authors"]; ok { if v, ok := raw["authors"]; ok {
if err := json.Unmarshal(v, &filter.Authors); err != nil { if err := json.Unmarshal(v, &f.Authors); err != nil {
return err return err
} }
delete(raw, "authors") delete(raw, "authors")
} }
if v, ok := raw["kinds"]; ok { if v, ok := raw["kinds"]; ok {
if err := json.Unmarshal(v, &filter.Kinds); err != nil { if err := json.Unmarshal(v, &f.Kinds); err != nil {
return err return err
} }
delete(raw, "kinds") delete(raw, "kinds")
@@ -107,39 +107,39 @@ func (filter *Filter) UnmarshalJSON(data []byte) error {
if v, ok := raw["since"]; ok { if v, ok := raw["since"]; ok {
if string(raw["since"]) == "null" { if string(raw["since"]) == "null" {
filter.Since = nil f.Since = nil
} else { } else {
var val int var val int
if err := json.Unmarshal(v, &val); err != nil { if err := json.Unmarshal(v, &val); err != nil {
return err return err
} }
filter.Since = &val f.Since = &val
} }
delete(raw, "since") delete(raw, "since")
} }
if v, ok := raw["until"]; ok { if v, ok := raw["until"]; ok {
if string(raw["until"]) == "null" { if string(raw["until"]) == "null" {
filter.Until = nil f.Until = nil
} else { } else {
var val int var val int
if err := json.Unmarshal(v, &val); err != nil { if err := json.Unmarshal(v, &val); err != nil {
return err return err
} }
filter.Until = &val f.Until = &val
} }
delete(raw, "until") delete(raw, "until")
} }
if v, ok := raw["limit"]; ok { if v, ok := raw["limit"]; ok {
if string(raw["limit"]) == "null" { if string(raw["limit"]) == "null" {
filter.Limit = nil f.Limit = nil
} else { } else {
var val int var val int
if err := json.Unmarshal(v, &val); err != nil { if err := json.Unmarshal(v, &val); err != nil {
return err return err
} }
filter.Limit = &val f.Limit = &val
} }
delete(raw, "limit") delete(raw, "limit")
} }
@@ -148,57 +148,57 @@ func (filter *Filter) UnmarshalJSON(data []byte) error {
for key := range raw { for key := range raw {
if strings.HasPrefix(key, "#") { if strings.HasPrefix(key, "#") {
// Leave Tags as `nil` unless tag fields exist // Leave Tags as `nil` unless tag fields exist
if filter.Tags == nil { if f.Tags == nil {
filter.Tags = make(map[string][]string) f.Tags = make(map[string][]string)
} }
tagKey := strings.TrimPrefix(key, "#") tagKey := strings.TrimPrefix(key, "#")
var tagValues []string var tagValues []string
if err := json.Unmarshal(raw[key], &tagValues); err != nil { if err := json.Unmarshal(raw[key], &tagValues); err != nil {
return err return err
} }
filter.Tags[tagKey] = tagValues f.Tags[tagKey] = tagValues
delete(raw, key) delete(raw, key)
} }
} }
// Place remaining fields in extensions // Place remaining fields in extensions
if len(raw) > 0 { if len(raw) > 0 {
filter.Extensions = raw f.Extensions = raw
} }
return nil return nil
} }
func (filter Filter) Matches(event Event) bool { func (f Filter) Matches(event Event) bool {
// Check ID // Check ID
if len(filter.IDs) > 0 { if len(f.IDs) > 0 {
if !matchesPrefix(event.ID, filter.IDs) { if !matchesPrefix(event.ID, f.IDs) {
return false return false
} }
} }
// Check Author // Check Author
if len(filter.Authors) > 0 { if len(f.Authors) > 0 {
if !matchesPrefix(event.PubKey, filter.Authors) { if !matchesPrefix(event.PubKey, f.Authors) {
return false return false
} }
} }
// Check Kind // Check Kind
if len(filter.Kinds) > 0 { if len(f.Kinds) > 0 {
if !matchesKinds(event.Kind, filter.Kinds) { if !matchesKinds(event.Kind, f.Kinds) {
return false return false
} }
} }
// Check Timestamp // Check Timestamp
if !matchesTimeRange(event.CreatedAt, filter.Since, filter.Until) { if !matchesTimeRange(event.CreatedAt, f.Since, f.Until) {
return false return false
} }
// Check Tags // Check Tags
if len(filter.Tags) > 0 { if len(f.Tags) > 0 {
if !matchesTags(event.Tags, filter.Tags) { if !matchesTags(event.Tags, f.Tags) {
return false return false
} }
} }