fac2bf91d9
- Add MetadataProvider interface for AI to update resource URLs - Add set_resource_url tool to AI service - Wire up metadata stores to AI service via router - Add URL discovery guidance to AI system prompt - AI can now inspect guests/containers/hosts for web services and automatically save discovered URLs to Pulse metadata Usage: Ask the AI 'Find the web URL for this container' and it will: 1. Check for listening ports and web servers 2. Get the IP address 3. Verify the URL works 4. Save it to Pulse for quick dashboard access
109 lines
3 KiB
Go
109 lines
3 KiB
Go
package ai
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
// MetadataProvider provides access to resource metadata stores
|
|
// This allows the AI to update resource URLs when it discovers web services
|
|
type MetadataProvider interface {
|
|
// SetGuestURL sets the custom URL for a Proxmox guest (VM/container)
|
|
SetGuestURL(guestID, customURL string) error
|
|
|
|
// SetDockerURL sets the custom URL for a Docker container/service
|
|
SetDockerURL(resourceID, customURL string) error
|
|
|
|
// SetHostURL sets the custom URL for a host
|
|
SetHostURL(hostID, customURL string) error
|
|
}
|
|
|
|
// SetMetadataProvider sets the metadata provider for URL updates
|
|
func (s *Service) SetMetadataProvider(mp MetadataProvider) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
s.metadataProvider = mp
|
|
log.Info().Msg("AI service: metadata provider configured for URL discovery")
|
|
}
|
|
|
|
// SetResourceURL updates the URL for a resource in Pulse
|
|
// This is called by the AI when it discovers a web service
|
|
func (s *Service) SetResourceURL(resourceType, resourceID, customURL string) error {
|
|
s.mu.RLock()
|
|
mp := s.metadataProvider
|
|
s.mu.RUnlock()
|
|
|
|
if mp == nil {
|
|
return fmt.Errorf("metadata provider not configured")
|
|
}
|
|
|
|
// Validate and normalize the URL
|
|
if customURL != "" {
|
|
// Try to parse the URL
|
|
parsedURL, err := url.Parse(customURL)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid URL format: %w", err)
|
|
}
|
|
|
|
// Ensure scheme is present
|
|
if parsedURL.Scheme == "" {
|
|
// Default to http if no scheme provided
|
|
customURL = "http://" + customURL
|
|
parsedURL, err = url.Parse(customURL)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid URL format: %w", err)
|
|
}
|
|
}
|
|
|
|
// Validate scheme
|
|
if parsedURL.Scheme != "http" && parsedURL.Scheme != "https" {
|
|
return fmt.Errorf("URL must use http:// or https:// scheme")
|
|
}
|
|
|
|
// Ensure host is present
|
|
if parsedURL.Host == "" {
|
|
return fmt.Errorf("URL must include a host")
|
|
}
|
|
}
|
|
|
|
// Route to the appropriate metadata store based on resource type
|
|
switch strings.ToLower(resourceType) {
|
|
case "guest", "vm", "container", "lxc", "qemu":
|
|
if err := mp.SetGuestURL(resourceID, customURL); err != nil {
|
|
return fmt.Errorf("failed to set guest URL: %w", err)
|
|
}
|
|
log.Info().
|
|
Str("resourceType", resourceType).
|
|
Str("resourceID", resourceID).
|
|
Str("url", customURL).
|
|
Msg("AI set guest URL")
|
|
|
|
case "docker", "docker_container", "docker_service":
|
|
if err := mp.SetDockerURL(resourceID, customURL); err != nil {
|
|
return fmt.Errorf("failed to set Docker URL: %w", err)
|
|
}
|
|
log.Info().
|
|
Str("resourceType", resourceType).
|
|
Str("resourceID", resourceID).
|
|
Str("url", customURL).
|
|
Msg("AI set Docker URL")
|
|
|
|
case "host":
|
|
if err := mp.SetHostURL(resourceID, customURL); err != nil {
|
|
return fmt.Errorf("failed to set host URL: %w", err)
|
|
}
|
|
log.Info().
|
|
Str("resourceType", resourceType).
|
|
Str("resourceID", resourceID).
|
|
Str("url", customURL).
|
|
Msg("AI set host URL")
|
|
|
|
default:
|
|
return fmt.Errorf("unknown resource type: %s (use 'guest', 'docker', or 'host')", resourceType)
|
|
}
|
|
|
|
return nil
|
|
}
|