Skip to content

Commit d06cdcb

Browse files
authored
Renamed ServerRegistry to UpstreamRegistry (#2607)
renamed ServerRegistry to UpstreamRegistry Signed-off-by: Daniele Martinoli <dmartino@redhat.com>
1 parent 8ce0337 commit d06cdcb

File tree

3 files changed

+20
-20
lines changed

3 files changed

+20
-20
lines changed

pkg/registry/converters/registry_converters.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,20 +9,20 @@ import (
99
"github.com/stacklok/toolhive/pkg/registry/types"
1010
)
1111

12-
// NewServerRegistryFromUpstream creates a ServerRegistry from upstream ServerJSON array.
12+
// NewUpstreamRegistryFromUpstreamServers creates a UpstreamRegistry from upstream ServerJSON array.
1313
// This is used when ingesting data from upstream MCP Registry API endpoints.
14-
func NewServerRegistryFromUpstream(servers []upstreamv0.ServerJSON) *types.ServerRegistry {
15-
return &types.ServerRegistry{
14+
func NewUpstreamRegistryFromUpstreamServers(servers []upstreamv0.ServerJSON) *types.UpstreamRegistry {
15+
return &types.UpstreamRegistry{
1616
Version: "1.0.0",
1717
LastUpdated: time.Now().Format(time.RFC3339),
1818
Servers: servers,
1919
}
2020
}
2121

22-
// NewServerRegistryFromToolhive creates a ServerRegistry from ToolHive Registry.
22+
// NewUpstreamRegistryFromToolhiveRegistry creates a UpstreamRegistry from ToolHive Registry.
2323
// This converts ToolHive format to upstream ServerJSON using the converters package.
2424
// Used when ingesting data from ToolHive-format sources (Git, File, API).
25-
func NewServerRegistryFromToolhive(toolhiveReg *types.Registry) (*types.ServerRegistry, error) {
25+
func NewUpstreamRegistryFromToolhiveRegistry(toolhiveReg *types.Registry) (*types.UpstreamRegistry, error) {
2626
if toolhiveReg == nil {
2727
return nil, fmt.Errorf("toolhive registry cannot be nil")
2828
}
@@ -47,7 +47,7 @@ func NewServerRegistryFromToolhive(toolhiveReg *types.Registry) (*types.ServerRe
4747
servers = append(servers, *serverJSON)
4848
}
4949

50-
return &types.ServerRegistry{
50+
return &types.UpstreamRegistry{
5151
Version: toolhiveReg.Version,
5252
LastUpdated: toolhiveReg.LastUpdated,
5353
Servers: servers,

pkg/registry/converters/registry_converters_test.go

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -12,14 +12,14 @@ import (
1212
"github.com/stacklok/toolhive/pkg/registry/types"
1313
)
1414

15-
func TestNewServerRegistryFromToolhive(t *testing.T) {
15+
func TestNewUpstreamRegistryFromToolhiveRegistry(t *testing.T) {
1616
t.Parallel()
1717

1818
tests := []struct {
1919
name string
2020
toolhiveReg *types.Registry
2121
expectError bool
22-
validate func(*testing.T, *types.ServerRegistry)
22+
validate func(*testing.T, *types.UpstreamRegistry)
2323
}{
2424
{
2525
name: "successful conversion with container servers",
@@ -42,7 +42,7 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
4242
RemoteServers: make(map[string]*types.RemoteServerMetadata),
4343
},
4444
expectError: false,
45-
validate: func(t *testing.T, sr *types.ServerRegistry) {
45+
validate: func(t *testing.T, sr *types.UpstreamRegistry) {
4646
t.Helper()
4747
assert.Equal(t, "1.0.0", sr.Version)
4848
assert.Equal(t, "2024-01-01T00:00:00Z", sr.LastUpdated)
@@ -72,7 +72,7 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
7272
},
7373
},
7474
expectError: false,
75-
validate: func(t *testing.T, sr *types.ServerRegistry) {
75+
validate: func(t *testing.T, sr *types.UpstreamRegistry) {
7676
t.Helper()
7777
assert.Len(t, sr.Servers, 1)
7878
assert.Contains(t, sr.Servers[0].Name, "remote-server")
@@ -87,7 +87,7 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
8787
RemoteServers: make(map[string]*types.RemoteServerMetadata),
8888
},
8989
expectError: false,
90-
validate: func(t *testing.T, sr *types.ServerRegistry) {
90+
validate: func(t *testing.T, sr *types.UpstreamRegistry) {
9191
t.Helper()
9292
assert.Empty(t, sr.Servers)
9393
},
@@ -98,7 +98,7 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
9898
t.Run(tt.name, func(t *testing.T) {
9999
t.Parallel()
100100

101-
result, err := NewServerRegistryFromToolhive(tt.toolhiveReg)
101+
result, err := NewUpstreamRegistryFromToolhiveRegistry(tt.toolhiveReg)
102102

103103
if tt.expectError {
104104
assert.Error(t, err)
@@ -114,13 +114,13 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
114114
}
115115
}
116116

117-
func TestNewServerRegistryFromUpstream(t *testing.T) {
117+
func TestNewUpstreamRegistryFromUpstreamServers(t *testing.T) {
118118
t.Parallel()
119119

120120
tests := []struct {
121121
name string
122122
servers []upstreamv0.ServerJSON
123-
validate func(*testing.T, *types.ServerRegistry)
123+
validate func(*testing.T, *types.UpstreamRegistry)
124124
}{
125125
{
126126
name: "create from upstream servers",
@@ -139,7 +139,7 @@ func TestNewServerRegistryFromUpstream(t *testing.T) {
139139
},
140140
},
141141
},
142-
validate: func(t *testing.T, sr *types.ServerRegistry) {
142+
validate: func(t *testing.T, sr *types.UpstreamRegistry) {
143143
t.Helper()
144144
assert.Equal(t, "1.0.0", sr.Version)
145145
assert.NotEmpty(t, sr.LastUpdated)
@@ -150,7 +150,7 @@ func TestNewServerRegistryFromUpstream(t *testing.T) {
150150
{
151151
name: "create from empty slice",
152152
servers: []upstreamv0.ServerJSON{},
153-
validate: func(t *testing.T, sr *types.ServerRegistry) {
153+
validate: func(t *testing.T, sr *types.UpstreamRegistry) {
154154
t.Helper()
155155
assert.Empty(t, sr.Servers)
156156
},
@@ -161,7 +161,7 @@ func TestNewServerRegistryFromUpstream(t *testing.T) {
161161
t.Run(tt.name, func(t *testing.T) {
162162
t.Parallel()
163163

164-
result := NewServerRegistryFromUpstream(tt.servers)
164+
result := NewUpstreamRegistryFromUpstreamServers(tt.servers)
165165

166166
assert.NotNil(t, result)
167167
if tt.validate != nil {
@@ -183,7 +183,7 @@ func TestNewServerRegistryFromUpstream_DefaultValues(t *testing.T) {
183183
},
184184
}
185185

186-
result := NewServerRegistryFromUpstream(servers)
186+
result := NewUpstreamRegistryFromUpstreamServers(servers)
187187

188188
// Verify defaults
189189
assert.Equal(t, "1.0.0", result.Version)

pkg/registry/types/upstream_registry.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,10 @@ import (
44
upstreamv0 "github.com/modelcontextprotocol/registry/pkg/api/v0"
55
)
66

7-
// ServerRegistry is the unified internal registry format.
7+
// UpstreamRegistry is the unified internal registry format.
88
// It stores servers in upstream ServerJSON format while maintaining
99
// ToolHive-compatible metadata fields for backward compatibility.
10-
type ServerRegistry struct {
10+
type UpstreamRegistry struct {
1111
// Version is the schema version (ToolHive compatibility)
1212
Version string `json:"version"`
1313

0 commit comments

Comments
 (0)