Merge pull request #91 from kubescape/feat/instanceType

Adding instance type
kubescape · Feb 19, 2024 · cde5aed · cde5aed
2 parents a0ceab8 + a829d86
commit cde5aed
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/instanceidhandler/interface.go b/instanceidhandler/interface.go
@@ -1,6 +1,11 @@
 package instanceidhandler
 
+import "github.com/kubescape/k8s-interface/instanceidhandler/v1/helpers"
+
 type IInstanceID interface {
+	// GetInstanceType returns the type of the instance ID
+	GetInstanceType() helpers.InstanceType
+
 	GetAPIVersion() string
 	GetNamespace() string
 	GetKind() string

diff --git a/instanceidhandler/v1/containerinstance/instanceidhandler.go b/instanceidhandler/v1/containerinstance/instanceidhandler.go
@@ -17,6 +17,8 @@ const (
 	stringFormat    = helpers.PrefixApiVersion + "%s" + helpers.StringFormatSeparator + helpers.PrefixNamespace + "%s" + helpers.StringFormatSeparator + helpers.PrefixKind + "%s" + helpers.StringFormatSeparator + helpers.PrefixName + "%s" + helpers.StringFormatSeparator + prefixContainer + "%s"
 )
 
+const InstanceType helpers.InstanceType = "container"
+
 // ensure that InstanceID implements IInstanceID
 var _ instanceidhandler.IInstanceID = &InstanceID{}
 
@@ -28,6 +30,9 @@ type InstanceID struct {
 	containerName string
 }
 
+func (id *InstanceID) GetInstanceType() helpers.InstanceType {
+	return InstanceType
+}
 func (id *InstanceID) GetAPIVersion() string {
 	return id.apiVersion
 }

diff --git a/instanceidhandler/v1/helpers/keys.go b/instanceidhandler/v1/helpers/keys.go
@@ -2,6 +2,8 @@ package helpers
 
 import "github.com/kubescape/k8s-interface/k8sinterface"
 
+type InstanceType string
+
 // metadata keys
 const (
 	metadataPrefix                    = "kubescape.io"

diff --git a/instanceidhandler/v1/initcontainerinstance/instanceidhandler.go b/instanceidhandler/v1/initcontainerinstance/instanceidhandler.go
@@ -17,6 +17,8 @@ const (
 	stringFormat        = helpers.PrefixApiVersion + "%s" + helpers.StringFormatSeparator + helpers.PrefixNamespace + "%s" + helpers.StringFormatSeparator + helpers.PrefixKind + "%s" + helpers.StringFormatSeparator + helpers.PrefixName + "%s" + helpers.StringFormatSeparator + prefixInitContainer + "%s"
 )
 
+const InstanceType helpers.InstanceType = "initContainer"
+
 // ensure that InstanceID implements IInstanceID
 var _ instanceidhandler.IInstanceID = &InstanceID{}
 
@@ -28,6 +30,10 @@ type InstanceID struct {
 	initContainerName string
 }
 
+func (id *InstanceID) GetInstanceType() helpers.InstanceType {
+	return InstanceType
+}
+
 func (id *InstanceID) GetAPIVersion() string {
 	return id.apiVersion
 }