Merge branch 'feature'

Author: HenryOwenz

internal/cloud/aws/lambda/category.go

@@ -21,6 +21,7 @@ func NewWorkflowsCategory(profile, region string) *WorkflowsCategory {
 
 	// Register operations
 	category.operations = append(category.operations, NewFunctionStatusOperation(profile, region))
+	category.operations = append(category.operations, NewLambdaExecuteOperation(profile, region))
 
 	return category
 }

internal/cloud/aws/lambda/operations.go

@@ -2,6 +2,7 @@ package lambda
 
 import (
 	"context"
+	"encoding/base64"
 	"errors"
 	"fmt"
 
@@ -152,3 +153,98 @@ func listFunctions(ctx context.Context, client *lambda.Client) ([]types.Function
 
 	return functions, nil
 }
+
+// Common errors for Lambda execution.
+var (
+	ErrInvokeFunction = errors.New("failed to invoke function")
+)
+
+// LambdaExecuteOperation represents an operation to execute a Lambda function.
+type LambdaExecuteOperation struct {
+	profile string
+	region  string
+}
+
+// NewLambdaExecuteOperation creates a new Lambda execute operation.
+func NewLambdaExecuteOperation(profile, region string) *LambdaExecuteOperation {
+	return &LambdaExecuteOperation{
+		profile: profile,
+		region:  region,
+	}
+}
+
+// Name returns the operation's name.
+func (o *LambdaExecuteOperation) Name() string {
+	return "Execute Function"
+}
+
+// Description returns the operation's description.
+func (o *LambdaExecuteOperation) Description() string {
+	return "Execute Lambda Function with JSON Payload"
+}
+
+// IsUIVisible returns whether this operation should be visible in the UI.
+func (o *LambdaExecuteOperation) IsUIVisible() bool {
+	return true
+}
+
+// ExecuteFunction executes a Lambda function with the given payload.
+func (o *LambdaExecuteOperation) ExecuteFunction(ctx context.Context, functionName string, payload string) (*cloud.LambdaExecuteResult, error) {
+	// Create a new AWS SDK client
+	client, err := getClient(ctx, o.profile, o.region)
+	if err != nil {
+		return nil, err
+	}
+
+	// Invoke the function
+	input := &lambda.InvokeInput{
+		FunctionName: aws.String(functionName),
+		Payload:      []byte(payload),
+		LogType:      types.LogTypeTail, // Include logs in the response
+	}
+
+	output, err := client.Invoke(ctx, input)
+	if err != nil {
+		return nil, fmt.Errorf("%w: %w", ErrInvokeFunction, err)
+	}
+
+	// Decode the base64-encoded logs
+	logResult := ""
+	if output.LogResult != nil {
+		decodedLogs, err := base64.StdEncoding.DecodeString(*output.LogResult)
+		if err == nil {
+			logResult = string(decodedLogs)
+		}
+	}
+
+	// Convert the payload to a string
+	payloadStr := ""
+	if output.Payload != nil {
+		payloadStr = string(output.Payload)
+	}
+
+	// Create the result
+	result := &cloud.LambdaExecuteResult{
+		StatusCode:      int(output.StatusCode),
+		ExecutedVersion: aws.ToString(output.ExecutedVersion),
+		Payload:         payloadStr,
+		LogResult:       logResult,
+	}
+
+	return result, nil
+}
+
+// Execute executes the operation with the given parameters.
+func (o *LambdaExecuteOperation) Execute(ctx context.Context, params map[string]interface{}) (interface{}, error) {
+	functionName, ok := params["functionName"].(string)
+	if !ok {
+		return nil, fmt.Errorf("function name is required")
+	}
+
+	payload, ok := params["payload"].(string)
+	if !ok {
+		return nil, fmt.Errorf("payload is required")
+	}
+
+	return o.ExecuteFunction(ctx, functionName, payload)
+}

internal/cloud/aws/provider.go

@@ -70,6 +70,14 @@ func (p *Provider) GetFunctionStatusOperation() (cloud.FunctionStatusOperation,
 	return lambda.NewFunctionStatusOperation(p.profile, p.region), nil
 }
 
+// GetLambdaExecuteOperation returns the Lambda execute operation
+func (p *Provider) GetLambdaExecuteOperation() (cloud.LambdaExecuteOperation, error) {
+	if p.profile == "" || p.region == "" {
+		return nil, ErrNotAuthenticated
+	}
+	return lambda.NewLambdaExecuteOperation(p.profile, p.region), nil
+}
+
 // GetCodePipelineManualApprovalOperation returns the CodePipeline manual approval operation
 func (p *Provider) GetCodePipelineManualApprovalOperation() (cloud.CodePipelineManualApprovalOperation, error) {
 	if p.profile == "" || p.region == "" {

internal/cloud/types.go

@@ -24,6 +24,9 @@ type Provider interface {
 	// GetFunctionStatusOperation returns the function status operation
 	GetFunctionStatusOperation() (FunctionStatusOperation, error)
 
+	// GetLambdaExecuteOperation returns the Lambda execute operation
+	GetLambdaExecuteOperation() (LambdaExecuteOperation, error)
+
 	// GetCodePipelineManualApprovalOperation returns the CodePipeline manual approval operation
 	GetCodePipelineManualApprovalOperation() (CodePipelineManualApprovalOperation, error)
 
@@ -160,6 +163,14 @@ type FunctionStatus struct {
 	LogGroup     string
 }
 
+// LambdaExecuteResult represents the result of a Lambda function execution
+type LambdaExecuteResult struct {
+	StatusCode      int
+	ExecutedVersion string
+	Payload         string
+	LogResult       string
+}
+
 // CodePipelineManualApprovalOperation represents a manual approval operation for AWS CodePipeline
 type CodePipelineManualApprovalOperation interface {
 	UIOperation
@@ -194,3 +205,11 @@ type FunctionStatusOperation interface {
 	// GetFunctionStatus returns the status of all Lambda functions
 	GetFunctionStatus(ctx context.Context) ([]FunctionStatus, error)
 }
+
+// LambdaExecuteOperation represents an operation to execute a Lambda function
+type LambdaExecuteOperation interface {
+	UIOperation
+
+	// ExecuteFunction executes a Lambda function with the given payload
+	ExecuteFunction(ctx context.Context, functionName string, payload string) (*LambdaExecuteResult, error)
+}

internal/cloudproviders/aws_wrapper.go

@@ -152,6 +152,11 @@ func (w *AWSProviderWrapper) GetStartPipelineOperation() (cloud.StartPipelineOpe
 	return w.provider.GetStartPipelineOperation()
 }
 
+// GetLambdaExecuteOperation returns the Lambda execute operation
+func (w *AWSProviderWrapper) GetLambdaExecuteOperation() (cloud.LambdaExecuteOperation, error) {
+	return w.provider.GetLambdaExecuteOperation()
+}
+
 // GetAuthenticationMethods returns the available authentication methods
 func (w *AWSProviderWrapper) GetAuthenticationMethods() []string {
 	return w.provider.GetAuthenticationMethods()

internal/ui/constants/colors.go

@@ -13,10 +13,11 @@ const (
 	ColorSubtle = "#718096" // Gray for subtle elements
 
 	// Status colors
-	ColorSuccess = "#38A169" // Subdued green for success messages (previously #00FF7F)
-	ColorError   = "#E53E3E" // Red for error messages
-	ColorWarning = "#DD6B20" // Orange for warnings
-	ColorInfo    = "#4299E1" // Brighter blue for info messages and column headers (previously #3182CE)
+	ColorSuccess   = "#38A169" // Subdued green for success messages (previously #00FF7F)
+	ColorError     = "#E53E3E" // Red for error messages
+	ColorWarning   = "#DD6B20" // Orange for warnings
+	ColorInfo      = "#4299E1" // Brighter blue for info messages and column headers (previously #3182CE)
+	ColorHighlight = "#F6E05E" // Yellow for highlighting focused elements
 
 	// Background colors
 	ColorBg    = "#1A202C" // Dark background

internal/ui/constants/dimensions.go

@@ -15,11 +15,17 @@ const (
 	TextInputCharLimit = 100
 
 	// App dimensions
-	AppMaxWidth     = 100
-	AppHeight       = 17
 	AppContentLines = 5 // Number of content lines in the UI layout
 
 	// Padding and margins
 	PaddingX = 1
 	PaddingY = 2
+
+	// Component heights
+	HeaderHeight = 2 // Height for headers/titles
+	FooterHeight = 1 // Height for footers
+
+	// Viewport dimensions
+	ViewportMarginX = 2 // Horizontal margin for viewport
+	ViewportMarginY = 3 // Vertical margin for viewport (header + footer)
 )

internal/ui/constants/keys.go

@@ -2,16 +2,19 @@ package constants
 
 // Key constants for keyboard input
 const (
-	KeyQ       = "q"
-	KeyCtrlC   = "ctrl+c"
-	KeyEnter   = "enter"
-	KeyEsc     = "esc"
-	KeyUp      = "up"
-	KeyDown    = "down"
-	KeyAltUp   = "k"
-	KeyAltDown = "j"
-	KeyAltBack = "-"
-	KeyTab     = "tab"
+	KeyQ          = "q"
+	KeyCtrlC      = "ctrl+c"
+	KeyEnter      = "enter"
+	KeyCtrlEnter  = "ctrl+enter"
+	KeyShiftEnter = "shift+enter"
+	KeyF5         = "f5"
+	KeyEsc        = "esc"
+	KeyUp         = "up"
+	KeyDown       = "down"
+	KeyAltUp      = "k"
+	KeyAltDown    = "j"
+	KeyAltBack    = "-"
+	KeyTab        = "tab"
 
 	// Vim-like navigation keys
 	KeyGotoTop         = "g"

internal/ui/constants/messages.go

@@ -11,6 +11,7 @@ const (
 	MsgLoadingFunctions  = "Loading functions..."
 	MsgStartingPipeline  = "Starting pipeline..."
 	MsgExecutingApproval = "Executing approval action..."
+	MsgExecutingLambda   = "Executing Lambda function..."
 
 	// Input placeholders
 	MsgEnterProfile          = "Enter AWS profile name..."
@@ -19,17 +20,22 @@ const (
 	MsgEnterApprovalComment  = "Enter approval comment..."
 	MsgEnterRejectionComment = "Enter rejection comment..."
 	MsgEnterCommitID         = "Enter commit ID..."
+	MsgEnterLambdaPayload    = "Enter Lambda JSON payload..."
 
 	// Success messages
 	MsgApprovalSuccess      = "Successfully approved pipeline: %s, stage: %s, action: %s"
 	MsgRejectionSuccess     = "Successfully rejected pipeline: %s, stage: %s, action: %s"
 	MsgPipelineStartSuccess = "Successfully started pipeline: %s"
+	MsgLambdaExecuteSuccess = "Successfully executed Lambda function: %s"
 
 	// Error messages
 	MsgErrorGeneric       = "Error: %s"
 	MsgErrorNoApproval    = "No approval selected"
 	MsgErrorNoPipeline    = "No pipeline selected"
 	MsgErrorNoFunction    = "No function selected"
+	MsgNoFunctionSelected = "No function selected"
+	MsgLambdaExecuteError = "Error executing Lambda function %s: %v"
 	MsgErrorEmptyCommitID = "Commit ID cannot be empty"
 	MsgErrorEmptyComment  = "Comment cannot be empty"
+	MsgErrorInvalidJSON   = "Invalid JSON payload"
 )

internal/ui/constants/titles.go

@@ -20,4 +20,6 @@ const (
 	TitleHelp            = "Help"
 	TitleFunctionStatus  = "Lambda Functions"
 	TitleFunctionDetails = "Function Details"
+	TitleLambdaExecute   = "Lambda Payload (JSON)"
+	TitleLambdaResponse  = "Lambda Response"
 )