nip11: support hex nip numbers as strings.

This commit is contained in:
fiatjaf
2024-11-30 08:53:08 -03:00
parent 9df2fc8d7b
commit a546203758
2 changed files with 8 additions and 11 deletions

View File

@@ -25,10 +25,7 @@ func TestAddSupportedNIP(t *testing.T) {
info.AddSupportedNIP(1)
info.AddSupportedNIP(18)
for i, v := range []int{0, 1, 2, 12, 13, 17, 18, 19, 44} {
assert.Equal(t, v, info.SupportedNIPs[i], "expected info.SupportedNIPs[%d] to equal %v, got %v",
i, v, info.SupportedNIPs)
}
assert.Contains(t, info.SupportedNIPs, 0, 1, 2, 12, 13, 17, 18, 19, 44)
}
func TestFetch(t *testing.T) {

View File

@@ -1,6 +1,8 @@
package nip11
import "slices"
import (
"slices"
)
type RelayInformationDocument struct {
URL string `json:"-"`
@@ -9,7 +11,7 @@ type RelayInformationDocument struct {
Description string `json:"description"`
PubKey string `json:"pubkey"`
Contact string `json:"contact"`
SupportedNIPs []int `json:"supported_nips"`
SupportedNIPs []any `json:"supported_nips"`
Software string `json:"software"`
Version string `json:"version"`
@@ -24,14 +26,12 @@ type RelayInformationDocument struct {
}
func (info *RelayInformationDocument) AddSupportedNIP(number int) {
idx, exists := slices.BinarySearch(info.SupportedNIPs, number)
if exists {
idx := slices.IndexFunc(info.SupportedNIPs, func(n any) bool { return n == number })
if idx != -1 {
return
}
info.SupportedNIPs = append(info.SupportedNIPs, -1)
copy(info.SupportedNIPs[idx+1:], info.SupportedNIPs[idx:])
info.SupportedNIPs[idx] = number
info.SupportedNIPs = append(info.SupportedNIPs, number)
}
type RelayLimitationDocument struct {