Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

command crd #27

Merged
merged 10 commits into from
Sep 19, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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
Loading