ctx.go

package schedo

import (
	"context"

	schedo "github.com/useschedo/golang-sdk"
)

// LogLevel represents the severity level of a log entry
type LogLevel string

const (
	LogLevelInfo  LogLevel = "INFO"
	LogLevelWarn  LogLevel = "WARN"
	LogLevelError LogLevel = "ERROR"
	LogLevelDebug LogLevel = "DEBUG"
)

// LogEntry represents a single log entry with its level
type LogEntry struct {
	Level LogLevel
	Line  string
}

type JobExecutionContext struct {
	ExecutionID int64
	JobCode     string
	client      *schedo.Client
	metadata    map[string]any
	logs        []LogEntry
	context.Context
}

func NewJobExecutionContext(ctx context.Context, client *schedo.Client, executionID int64, jobCode string, metadata map[string]any) *JobExecutionContext {
	return &JobExecutionContext{
		ExecutionID: executionID,
		JobCode:     jobCode,
		client:      client,
		metadata:    metadata,
		logs:        make([]LogEntry, 0),
		Context:     ctx,
	}
}

func (j *JobExecutionContext) WriteLog(line string) error {
	return j.WriteLogWithLevel(line, LogLevelInfo)
}

// WriteLogWithLevel writes a log line with the specified log level
func (j *JobExecutionContext) WriteLogWithLevel(line string, level LogLevel) error {
	j.logs = append(j.logs, LogEntry{
		Level: level,
		Line:  line,
	})
	return nil
}

// GetLogs returns all collected logs as a single string with each entry on a new line
// Format: [LEVEL] message
func (j *JobExecutionContext) GetLogs() string {
	if len(j.logs) == 0 {
		return ""
	}

	var result string
	for i, entry := range j.logs {
		if i > 0 {
			result += "\n"
		}
		result += "[" + string(entry.Level) + "] " + entry.Line
	}
	return result
}

// ClearLogs removes all collected logs
func (j *JobExecutionContext) ClearLogs() {
	j.logs = make([]LogEntry, 0)
}

func (j *JobExecutionContext) GetMetadataField(key string) (any, bool) {
	value, ok := j.metadata[key]
	return value, ok
}

func (j *JobExecutionContext) GetMetadata(key string) map[string]any {
	return j.metadata
}