Skip to content

Commit

Permalink
Merge pull request #27 from kubescape/command-crd
Browse files Browse the repository at this point in the history
command crd
  • Loading branch information
amirmalka authored Sep 19, 2024
2 parents 00ebf35 + 00ac53f commit d5decd5
Show file tree
Hide file tree
Showing 7 changed files with 141 additions and 1 deletion.
29 changes: 29 additions & 0 deletions pkg/command/commands.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
package command

type OperatorCommandType string

const (
OperatorCommandAppNameLabelKey string = "kubescape.io/app-name" // holds the app name label, which should execute the command (optional)
OperatorCommandNodeNameLabelKey string = "kubescape.io/node-name" // holds the node name label, app running on this node name should execute the command (optional)

// command types will be defined here
OperatorCommandTypeRuntimeResponse OperatorCommandType = "RuntimeResponse"
)

// ResponseCommand
type ResponseAction string

const (
ResponseActionKill ResponseAction = "Kill"
ResponseActionStop ResponseAction = "Stop"
ResponseActionPause ResponseAction = "Pause"
ResponseActionUnpause ResponseAction = "Unpause"
)

type ResponseCommand struct {
Namespace string `json:"namespace,omitempty"`
PodName string `json:"podName,omitempty"`
ContainerName string `json:"containerName,omitempty"`
Pid *uint32 `json:"pid,omitempty"`
Action ResponseAction `json:"action,omitempty"`
}
7 changes: 7 additions & 0 deletions pkg/command/types/api.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
package types

const (
OperatorCommandGroup string = "kubescape.io"
OperatorCommandKind string = "OperatorCommand"
OperatorCommandPlural string = "operatorcommands"
)
10 changes: 10 additions & 0 deletions pkg/command/types/v1alpha1/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
# OperatorCommand

The OperatorCommand CRD is designed to enable the execution of various actions within the cluster and reporting their status back to the backend. This CRD serves as a central mechanism for triggering and managing actions, replacing the functionality previously provided by the gateway and kollector.

How it Works

1. Creation: The backend creates a Command CRD instance, specifying the desired action and any necessary parameters for the action.
2. Synchronization: The Synchronizer, responsible for two-way communication, receives the Command CRD from the backend and saves it in the cluster.
3. Execution: The designated component in the cluster, identifies the new command via a watcher on the Kubernetes API, processes the Command CRD and performs the requested action within the cluster.
4. Status Reporting: Upon completion, the component updates the command CRD resource with the status of the action, providing information about success or failure, any relevant details, and potentially updating the Command CRD. The synchronizer, watching over the command CRD, will send it back to the backend for further processing.
16 changes: 16 additions & 0 deletions pkg/command/types/v1alpha1/api.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
package v1alpha1

import (
"github.com/kubescape/backend/pkg/command/types"
"k8s.io/apimachinery/pkg/runtime/schema"
)

const (
OperatorCommandVersion string = "v1alpha1"
)

var SchemaGroupVersionResource = schema.GroupVersionResource{
Group: types.OperatorCommandGroup,
Version: OperatorCommandVersion,
Resource: types.OperatorCommandPlural,
}
50 changes: 50 additions & 0 deletions pkg/command/types/v1alpha1/types.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
package v1alpha1

import (
"time"

"github.com/armosec/armoapi-go/identifiers"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

type OperatorCommandList struct {
metav1.TypeMeta `json:",inline"`
metav1.ListMeta `json:"metadata,omitempty"`

Items []OperatorCommand `json:"items"`
}

type OperatorCommand struct {
metav1.TypeMeta `json:",inline"`
metav1.ObjectMeta `json:"metadata,omitempty"`

Spec OperatorCommandSpec `json:"spec,omitempty"`
Status OperatorCommandStatus `json:"status,omitempty"`
}

type OperatorCommandSpec struct {
GUID string `json:"guid"` // GUID is a unique identifier for the command
CommandType string `json:"commandType"` // CommandType is the type of the command
CommandVersion string `json:"commandVersion,omitempty"` // CommandVersion is the version of the command
Designators []identifiers.PortalDesignator `json:"designators,omitempty"` // Designators are the designators for the command
Body []byte `json:"body,omitempty"` // Body is the body of the command
TTL time.Duration `json:"ttl,omitempty"` // TTL is the time to live for the command
Args map[string]interface{} `json:"args,omitempty"` // Args are the arguments for the command
CommandIndex *int `json:"commandIndex,omitempty"` // CommandIndex is the index of the command in the sequence
CommandCount *int `json:"commandCount,omitempty"` // CommandCount is the total number of commands in the sequence
}

type OperatorCommandStatus struct {
Started bool `json:"started"` // Started indicates if the command has started
StartedAt *metav1.Time `json:"startedAt,omitempty"` // StartedAt is the time at which the command was started
Completed bool `json:"completed"` // Completed indicates if the command has completed
CompletedAt *metav1.Time `json:"completedAt,omitempty"` // CompletedAt is the time at which the command was completed
Executer string `json:"executer,omitempty"` // Executer is the entity that executed the command
Error *OperatorCommandStatusError `json:"error,omitempty"` // Error is the error that occurred during the execution of the command (if any)
}

type OperatorCommandStatusError struct {
Reason string `json:"reason,omitempty"` // reason for the error (optional)
Message string `json:"message,omitempty"` // error message (optional)
ErrorCode int `json:"errorCode,omitempty"` // error code (optional)
}
28 changes: 28 additions & 0 deletions pkg/command/utils/utils.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
package utils

import (
"bytes"
"encoding/json"
)

func EncodeCommandBody[T any](data T) ([]byte, error) {
buffer := bytes.Buffer{}
encoder := json.NewEncoder(&buffer)
err := encoder.Encode(data)
if err != nil {
return nil, err
}
return buffer.Bytes(), nil
}

func DecodeCommandBody[T any](b []byte) (T, error) {
buffer := bytes.Buffer{}
buffer.Write(b)
decoder := json.NewDecoder(&buffer)
var data T
err := decoder.Decode(&data)
if err != nil {
return data, err
}
return data, nil
}
2 changes: 1 addition & 1 deletion pkg/versioncheck/versioncheck_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ func TestVersionCheckHandler_getLatestVersion(t *testing.T) {
},
want: &VersionCheckResponse{
Client: "kubescape",
ClientUpdate: "v3.0.9",
ClientUpdate: "v3.0.15",
},
wantErr: false,
},
Expand Down

0 comments on commit d5decd5

Please sign in to comment.