Merge pull request #31 from xentra-ai/fix/labels

fix: add more logging and support jobs

Author: xunholy

.github/workflows/build-docker.yaml

@@ -41,7 +41,7 @@ jobs:
         with:
           build-args: |
             APPLICATION_NAME=advisor
-            VERSION=0.0.1
+            VERSION=0.0.2
             SHA=${{ github.sha }}
           context: ./advisor
           file: Dockerfile

advisor/pkg/k8s/labels.go

@@ -30,7 +30,7 @@ func detectSelectorLabels(clientset *kubernetes.Clientset, origin interface{}) (
 func GetOwnerRef(clientset *kubernetes.Clientset, pod *v1.Pod) (map[string]string, error) {
 	ctx := context.TODO()
 
-	// Check if the& Pod has an owner
+	// Check if the Pod has an owner
 	if len(pod.OwnerReferences) > 0 {
 		owner := pod.OwnerReferences[0]
 
@@ -61,10 +61,17 @@ func GetOwnerRef(clientset *kubernetes.Clientset, pod *v1.Pod) (map[string]strin
 			}
 			return daemonSet.Spec.Selector.MatchLabels, nil
 
+		case "Job":
+			job, err := clientset.BatchV1().Jobs(pod.Namespace).Get(ctx, owner.Name, metav1.GetOptions{})
+			if err != nil {
+				return nil, err
+			}
+			return job.Spec.Selector.MatchLabels, nil
+
 		// Add more controller kinds here if needed
 
 		default:
-			return nil, fmt.Errorf("unknown or unsupported owner kind: %s", owner.Kind)
+			return nil, fmt.Errorf("unknown or unsupported ownerReference: %s", owner.String())
 		}
 	}
 	return pod.Labels, nil

advisor/pkg/k8s/networkpolicies.go

@@ -88,7 +88,7 @@ func transformToNetworkPolicy(podTraffic []api.PodTraffic, podDetail *api.PodDet
 			Namespace: podDetail.Namespace,
 			Labels: map[string]string{
 				"advisor.xentra.ai/managed-by": "xentra",
-				"advisor.xentra.ai/version":    "0.0.1",
+				"advisor.xentra.ai/version":    "0.0.2",
 			},
 		},
 		Spec: networkingv1.NetworkPolicySpec{