|
9 | 9 |
|
10 | 10 | "github.com/stacklok/toolhive/pkg/config" |
11 | 11 | "github.com/stacklok/toolhive/pkg/logger" |
12 | | - "github.com/stacklok/toolhive/pkg/registry" |
13 | | - "github.com/stacklok/toolhive/pkg/registry/types" |
| 12 | + regpkg "github.com/stacklok/toolhive/pkg/registry" |
| 13 | + "github.com/stacklok/toolhive/pkg/registry/registry" |
14 | 14 | ) |
15 | 15 |
|
16 | 16 | const ( |
@@ -59,8 +59,8 @@ func getRegistryInfoWithProvider(configProvider config.Provider) (RegistryType, |
59 | 59 | } |
60 | 60 |
|
61 | 61 | // getCurrentProvider returns the current registry provider using the injected config |
62 | | -func (rr *RegistryRoutes) getCurrentProvider(w http.ResponseWriter) (registry.Provider, bool) { |
63 | | - provider, err := registry.GetDefaultProviderWithConfig(rr.configProvider) |
| 62 | +func (rr *RegistryRoutes) getCurrentProvider(w http.ResponseWriter) (regpkg.Provider, bool) { |
| 63 | + provider, err := regpkg.GetDefaultProviderWithConfig(rr.configProvider) |
64 | 64 | if err != nil { |
65 | 65 | http.Error(w, "Failed to get registry provider", http.StatusInternalServerError) |
66 | 66 | logger.Errorf("Failed to get registry provider: %v", err) |
@@ -256,7 +256,7 @@ func (rr *RegistryRoutes) updateRegistry(w http.ResponseWriter, r *http.Request) |
256 | 256 | } |
257 | 257 |
|
258 | 258 | // Reset the default provider to pick up configuration changes |
259 | | - registry.ResetDefaultProvider() |
| 259 | + regpkg.ResetDefaultProvider() |
260 | 260 |
|
261 | 261 | response := UpdateRegistryResponse{ |
262 | 262 | Message: message, |
@@ -401,8 +401,8 @@ func (rr *RegistryRoutes) listServers(w http.ResponseWriter, r *http.Request) { |
401 | 401 |
|
402 | 402 | // Build response with both container and remote servers |
403 | 403 | response := listServersResponse{ |
404 | | - Servers: make([]*types.ImageMetadata, 0, len(reg.Servers)), |
405 | | - RemoteServers: make([]*types.RemoteServerMetadata, 0, len(reg.RemoteServers)), |
| 404 | + Servers: make([]*registry.ImageMetadata, 0, len(reg.Servers)), |
| 405 | + RemoteServers: make([]*registry.RemoteServerMetadata, 0, len(reg.RemoteServers)), |
406 | 406 | } |
407 | 407 |
|
408 | 408 | // Add container servers |
@@ -460,14 +460,14 @@ func (rr *RegistryRoutes) getServer(w http.ResponseWriter, r *http.Request) { |
460 | 460 | // Build response based on server type |
461 | 461 | var response getServerResponse |
462 | 462 | if server.IsRemote() { |
463 | | - if remote, ok := server.(*types.RemoteServerMetadata); ok { |
| 463 | + if remote, ok := server.(*registry.RemoteServerMetadata); ok { |
464 | 464 | response = getServerResponse{ |
465 | 465 | RemoteServer: remote, |
466 | 466 | IsRemote: true, |
467 | 467 | } |
468 | 468 | } |
469 | 469 | } else { |
470 | | - if img, ok := server.(*types.ImageMetadata); ok { |
| 470 | + if img, ok := server.(*registry.ImageMetadata); ok { |
471 | 471 | response = getServerResponse{ |
472 | 472 | Server: img, |
473 | 473 | IsRemote: false, |
@@ -528,27 +528,27 @@ type getRegistryResponse struct { |
528 | 528 | // Source of the registry (URL, file path, or empty string for built-in) |
529 | 529 | Source string `json:"source"` |
530 | 530 | // Full registry data |
531 | | - Registry *types.Registry `json:"registry"` |
| 531 | + Registry *registry.Registry `json:"registry"` |
532 | 532 | } |
533 | 533 |
|
534 | 534 | // listServersResponse represents the response for listing servers in a registry |
535 | 535 | // |
536 | 536 | // @Description Response containing a list of servers |
537 | 537 | type listServersResponse struct { |
538 | 538 | // List of container servers in the registry |
539 | | - Servers []*types.ImageMetadata `json:"servers"` |
| 539 | + Servers []*registry.ImageMetadata `json:"servers"` |
540 | 540 | // List of remote servers in the registry (if any) |
541 | | - RemoteServers []*types.RemoteServerMetadata `json:"remote_servers,omitempty"` |
| 541 | + RemoteServers []*registry.RemoteServerMetadata `json:"remote_servers,omitempty"` |
542 | 542 | } |
543 | 543 |
|
544 | 544 | // getServerResponse represents the response for getting a server from a registry |
545 | 545 | // |
546 | 546 | // @Description Response containing server details |
547 | 547 | type getServerResponse struct { |
548 | 548 | // Container server details (if it's a container server) |
549 | | - Server *types.ImageMetadata `json:"server,omitempty"` |
| 549 | + Server *registry.ImageMetadata `json:"server,omitempty"` |
550 | 550 | // Remote server details (if it's a remote server) |
551 | | - RemoteServer *types.RemoteServerMetadata `json:"remote_server,omitempty"` |
| 551 | + RemoteServer *registry.RemoteServerMetadata `json:"remote_server,omitempty"` |
552 | 552 | // Indicates if this is a remote server |
553 | 553 | IsRemote bool `json:"is_remote"` |
554 | 554 | } |
|
0 commit comments