Add support for go packages in manifest files

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
TG1999 · Mar 7, 2024 · 797e20d · 797e20d
1 parent dea7760
commit 797e20d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/packageurl/utils.py b/src/packageurl/utils.py
@@ -30,12 +30,17 @@
 def get_golang_purl(go_package: str):
     """
     Return a PackageURL object given an imported ``go_package``
+    >>> get_golang_purl(go_package="github.com/gorilla/mux v1.8.1")
+    PackageURL(type="golang", namespace="github.com/gorilla", name="mux", version="v1.8.1")
     """
     if not go_package:
         return
+    version = None
+    if " " in go_package:
+        go_package , _ , version = go_package.rpartition(" ")    
     parts = go_package.split("/")
     if not parts:
         return
     name = parts[-1]
     namespace = "/".join(parts[:-1])
-    return PackageURL(type="golang", namespace=namespace, name=name)
+    return PackageURL(type="golang", namespace=namespace, name=name, version=version)
diff --git a/tests/contrib/test_utils.py b/tests/contrib/test_utils.py
@@ -66,3 +66,7 @@ def test_get_golang_purl():
     )
     assert golang_purl_1.name == "v3"
     assert golang_purl_1.namespace == "github.com/envoyproxy/go-control-plane/envoy/config/listener"
+    golang_purl_2 = get_golang_purl(go_package="github.com/grpc-ecosystem/go-grpc-middleware v1.3.0")
+    assert "pkg:golang/github.com/grpc-ecosystem/go-grpc-middleware@v1.3.0" == str(
+        golang_purl_2
+    )