Provide document links for referenced models

CHANGELOG.md

@@ -10,6 +10,7 @@ All notable changes to the Docker Language Server will be documented in this fil
   - textDocument/documentLink
     - add anchor resolution for all supported document links ([#348](https://github.com/docker/docker-language-server/issues/348))
     - return document links for the `file` attribute of a service object's `extends` attribute object ([#172](https://github.com/docker/docker-language-server/issues/172))
+    - provide document links for models on Docker Hub and Hugging Face ([#356](https://github.com/docker/docker-language-server/issues/356))
   - textDocument/hover
     - support hovering over referenced models ([#343](https://github.com/docker/docker-language-server/issues/343))
 

internal/compose/documentLink.go

@@ -97,6 +97,23 @@ func createObjectFileLink(u *url.URL, serviceNode *ast.MappingValueNode) *protoc
 	return nil
 }
 
+func createModelLink(serviceNode *ast.MappingValueNode) *protocol.DocumentLink {
+	if resolveAnchor(serviceNode.Key).GetToken().Value == "model" {
+		service := stringNode(serviceNode.Value)
+		if service != nil {
+			linkedText, link := extractModelLink(service.Value)
+			if linkedText != "" {
+				return &protocol.DocumentLink{
+					Range:   createRange(service.GetToken(), len(linkedText)),
+					Target:  types.CreateStringPointer(link),
+					Tooltip: types.CreateStringPointer(link),
+				}
+			}
+		}
+	}
+	return nil
+}
+
 func includedFiles(nodes []ast.Node) []*token.Token {
 	tokens := []*token.Token{}
 	for _, entry := range nodes {
@@ -201,6 +218,19 @@ func scanForLinks(u *url.URL, n *ast.MappingValueNode) []protocol.DocumentLink {
 					}
 				}
 			}
+		case "models":
+			if mappingNode, ok := resolveAnchor(n.Value).(*ast.MappingNode); ok {
+				for _, node := range mappingNode.Values {
+					if serviceAttributes, ok := resolveAnchor(node.Value).(*ast.MappingNode); ok {
+						for _, serviceAttribute := range serviceAttributes.Values {
+							link := createModelLink(serviceAttribute)
+							if link != nil {
+								links = append(links, *link)
+							}
+						}
+					}
+				}
+			}
 		}
 		return links
 	}
@@ -278,3 +308,31 @@ func extractImageLink(nodeValue string) (string, string) {
 	}
 	return nodeValue[0:idx], fmt.Sprintf("https://hub.docker.com/r/%v", nodeValue[0:idx])
 }
+
+func extractModelLink(nodeValue string) (string, string) {
+	if strings.HasPrefix(nodeValue, "hf.co") {
+		if len(nodeValue) <= 6 {
+			return "", ""
+		}
+		idx := strings.LastIndex(nodeValue, ":")
+		if idx == -1 {
+			return nodeValue, fmt.Sprintf("https://%v", nodeValue)
+		}
+		return nodeValue[0:idx], fmt.Sprintf("https://%v", nodeValue[0:idx])
+	}
+
+	idx := strings.LastIndex(nodeValue, ":")
+	if idx == -1 {
+		idx := strings.Index(nodeValue, "/")
+		if idx == -1 {
+			return nodeValue, fmt.Sprintf("https://hub.docker.com/_/%v", nodeValue)
+		}
+		return nodeValue, fmt.Sprintf("https://hub.docker.com/r/%v", nodeValue)
+	}
+
+	slashIndex := strings.Index(nodeValue, "/")
+	if slashIndex == -1 {
+		return nodeValue[0:idx], fmt.Sprintf("https://hub.docker.com/_/%v", nodeValue[0:idx])
+	}
+	return nodeValue[0:idx], fmt.Sprintf("https://hub.docker.com/r/%v", nodeValue[0:idx])
+}