OliveTin/service/internal/executor/arguments_test.go

package executor

import (
	"fmt"
	"strings"

	config "github.com/OliveTin/OliveTin/internal/config"
	"github.com/OliveTin/OliveTin/internal/entities"
	"github.com/OliveTin/OliveTin/internal/tpl"
	log "github.com/sirupsen/logrus"

	"testing"

	"github.com/stretchr/testify/assert"
)

func TestSanitizeUnsafe(t *testing.T) {
	assert.Nil(t, TypeSafetyCheck("", "_zomg_ c:/ haxxor ' bobby tables && rm -rf ", "very_dangerous_raw_string"))
}

func TestSanitizeUnimplemented(t *testing.T) {
	err := TypeSafetyCheck("", "I am a happy little argument", "greeting_type")

	assert.NotNil(t, err, "Test an argument type that does not exist")
}

func TestValidateArgumentCheckboxDefaultValues(t *testing.T) {
	arg := config.ActionArgument{
		Name: "confirm",
		Type: "checkbox",
	}
	action := config.Action{
		Title: "Test checkbox default values",
	}

	// Default checkbox values without choices should accept "1" and "0"
	err := ValidateArgument(&arg, "1", &action)
	assert.Nil(t, err, "Expected checkbox value \"1\" to be accepted without choices")

	err = ValidateArgument(&arg, "0", &action)
	assert.Nil(t, err, "Expected checkbox value \"0\" to be accepted without choices")
}

func TestMangleCheckboxValueWithChoices(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	arg := config.ActionArgument{
		Name: "confirm",
		Type: "checkbox",
		Choices: []config.ActionArgumentChoice{
			{Title: "Enabled", Value: "on"},
			{Title: "Disabled", Value: "off"},
		},
	}

	// When the incoming value matches a choice title, it should be mapped to the choice value
	out := mangleCheckboxValue(&arg, "Enabled", "Test action")
	assert.Equal(t, "on", out, "Expected checkbox title to be mangled to its value")

	out = mangleCheckboxValue(&arg, "Disabled", "Test action")
	assert.Equal(t, "off", out, "Expected checkbox title to be mangled to its value")

	// When there is no matching title, the value should be returned unchanged
	out = mangleCheckboxValue(&arg, "something-else", "Test action")
	assert.Equal(t, "something-else", out, "Expected non-matching value to be returned unchanged")
}

func TestMangleArgumentValueCheckbox(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	arg := config.ActionArgument{
		Name: "confirm",
		Type: "checkbox",
		Choices: []config.ActionArgumentChoice{
			{Title: "Yes", Value: "true-value"},
			{Title: "No", Value: "false-value"},
		},
	}

	out := MangleArgumentValue(&arg, "Yes", "Test action")
	assert.Equal(t, "true-value", out, "Expected MangleArgumentValue to delegate to mangleCheckboxValue for checkbox types")

	out = MangleArgumentValue(&arg, "No", "Test action")
	assert.Equal(t, "false-value", out)

	// For non-matching values, it should return the original value
	out = MangleArgumentValue(&arg, "maybe", "Test action")
	assert.Equal(t, "maybe", out)
}

func TestValidateArgumentCheckboxWithChoices(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	arg := config.ActionArgument{
		Name: "confirm",
		Type: "checkbox",
		Choices: []config.ActionArgumentChoice{
			{Title: "Enabled", Value: "on"},
			{Title: "Disabled", Value: "off"},
		},
	}
	action := config.Action{
		Title: "Test checkbox with choices",
	}

	// Titles should be accepted once mangled to their values
	err := ValidateArgument(&arg, "Enabled", &action)
	assert.Nil(t, err, "Expected checkbox title \"Enabled\" to be accepted after mangling to choice value")

	err = ValidateArgument(&arg, "Disabled", &action)
	assert.Nil(t, err, "Expected checkbox title \"Disabled\" to be accepted after mangling to choice value")

	// Unknown titles should be rejected because they do not match any choice value
	err = ValidateArgument(&arg, "Maybe", &action)
	assert.NotNil(t, err, "Expected unknown checkbox title to be rejected against choices")
}

func checklistTestArg() config.ActionArgument {
	return config.ActionArgument{
		Name: "directories",
		Type: "checklist",
		Choices: []config.ActionArgumentChoice{
			{Title: "Documents", Value: "documents"},
			{Title: "Photos", Value: "photos"},
			{Title: "Music", Value: "music"},
		},
	}
}

func TestValidateArgumentChecklistSelections(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	arg := checklistTestArg()
	action := config.Action{Title: "Test checklist"}

	err := ValidateArgument(&arg, "documents", &action)
	assert.Nil(t, err)

	err = ValidateArgument(&arg, `["documents","photos"]`, &action)
	assert.Nil(t, err)

	err = ValidateArgument(&arg, `["documents","unknown"]`, &action)
	assert.NotNil(t, err)
}

func TestValidateArgumentChecklistTitleMangling(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	arg := checklistTestArg()
	action := config.Action{Title: "Test checklist title mangling"}

	err := ValidateArgument(&arg, `["Documents","Photos"]`, &action)
	assert.Nil(t, err)
}

func TestValidateArgumentChecklistEmptySelection(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	arg := checklistTestArg()
	action := config.Action{Title: "Test checklist empty"}

	err := ValidateArgument(&arg, "", &action)
	assert.Nil(t, err)

	arg.RejectNull = true
	err = ValidateArgument(&arg, "", &action)
	assert.NotNil(t, err)
}

func TestValidateArgumentChecklistWithoutChoices(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	arg := config.ActionArgument{
		Name: "directories",
		Type: "checklist",
	}
	action := config.Action{Title: "Test checklist without choices"}

	err := ValidateArgument(&arg, "documents", &action)
	assert.NotNil(t, err)
}

func TestValidateArgumentChecklistRejectsEmptySegment(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	arg := checklistTestArg()
	action := config.Action{Title: "Test checklist empty segment"}

	err := ValidateArgument(&arg, `["documents","","photos"]`, &action)
	assert.NotNil(t, err)
}

func TestMangleArgumentValueChecklist(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	arg := checklistTestArg()

	out := MangleArgumentValue(&arg, `["Documents","Music"]`, "Test action")
	assert.Equal(t, `["documents","music"]`, out)

	out = MangleArgumentValue(&arg, `["documents","photos"]`, "Test action")
	assert.Equal(t, `["documents","photos"]`, out)
}

func checklistEntityTestArg() config.ActionArgument {
	return config.ActionArgument{
		Name:   "rooms",
		Type:   "checklist",
		Entity: "room",
		Choices: []config.ActionArgumentChoice{
			{Title: "{{ room.hostname }}", Value: "{{ room.hostname }}"},
		},
	}
}

func TestValidateArgumentChecklistEntitySelections(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	entities.AddEntity("room", "0", map[string]any{"hostname": "attic"})
	entities.AddEntity("room", "1", map[string]any{"hostname": "basement"})

	arg := checklistEntityTestArg()
	action := config.Action{Title: "Test checklist entity"}

	err := ValidateArgument(&arg, "attic", &action)
	assert.Nil(t, err)

	err = ValidateArgument(&arg, `["attic","basement"]`, &action)
	assert.Nil(t, err)

	err = ValidateArgument(&arg, `["attic","unknown"]`, &action)
	assert.NotNil(t, err)
}

func TestMangleArgumentValueChecklistEntityTitles(t *testing.T) {
	log.SetLevel(log.PanicLevel)

	entities.AddEntity("room", "0", map[string]any{"hostname": "attic"})
	entities.AddEntity("room", "1", map[string]any{"hostname": "basement"})

	arg := config.ActionArgument{
		Name:   "rooms",
		Type:   "checklist",
		Entity: "room",
		Choices: []config.ActionArgumentChoice{
			{Title: "{{ room.hostname }} room", Value: "{{ room.hostname }}"},
		},
	}

	out := MangleArgumentValue(&arg, `["attic room","basement room"]`, "Test checklist entity titles")
	assert.Equal(t, `["attic","basement"]`, out)
}

func TestParseActionArgumentsChecklistEmptySelection(t *testing.T) {
	req := newExecRequest()
	req.Binding.Action = &config.Action{
		Title: "Test checklist empty selection",
		Shell: "echo 'Selected segments: {{ segments }}'",
		Arguments: []config.ActionArgument{
			{
				Name: "segments",
				Type: "checklist",
				Choices: []config.ActionArgumentChoice{
					{Value: "kitchen"},
					{Value: "bedroom"},
				},
			},
		},
	}
	req.Arguments = map[string]string{
		"segments": "",
	}

	mangleInvalidArgumentValues(req)
	out, err := parseActionArguments(req)

	assert.Nil(t, err)
	assert.Equal(t, "echo 'Selected segments: '", out)
}

func newExecRequest() *ExecutionRequest {
	return &ExecutionRequest{
		Arguments: make(map[string]string),
		Binding: &ActionBinding{
			Action: &config.Action{},
		},
	}
}

func TestArgumentValueNullable(t *testing.T) {
	req := newExecRequest()
	req.Binding.Action = &config.Action{
		Title: "Release the hounds",
		Shell: "echo 'Releasing {{ count }} hounds'",
		Arguments: []config.ActionArgument{
			{
				Name:       "count",
				Type:       "int",
				RejectNull: false,
			},
		},
	}
	req.Arguments = map[string]string{
		"count": "",
	}

	out, err := parseActionArguments(req)

	assert.Equal(t, "echo 'Releasing  hounds'", out)
	assert.Nil(t, err)

	req.Binding.Action.Arguments[0].RejectNull = true

	_, err = parseActionArguments(req)

	assert.NotNil(t, err)
}

func TestArgumentNameNumbers(t *testing.T) {
	req := newExecRequest()
	req.Binding.Action = &config.Action{
		Title: "Do some tickles",
		Shell: "echo 'Tickling {{ person1name }}'",
		Arguments: []config.ActionArgument{
			{
				Name: "person1name",
				Type: "ascii",
			},
		},
	}

	req.Arguments = map[string]string{
		"person1name": "Fred",
	}

	out, err := parseActionArguments(req)

	assert.Equal(t, "echo 'Tickling Fred'", out)
	assert.Nil(t, err)
}

func TestArgumentNotProvided(t *testing.T) {
	req := newExecRequest()
	req.Binding.Action = &config.Action{
		Title: "Do some tickles",
		Shell: "echo 'Tickling {{ personName }}'",
		Arguments: []config.ActionArgument{
			{
				Name: "person",
				Type: "ascii",
			},
		},
	}

	req.Arguments = map[string]string{}

	out, err := parseActionArguments(req)

	assert.Equal(t, "", out)
	assert.Equal(t, err.Error(), "required arg not provided: personName")
}

func TestExecArrayParsing(t *testing.T) {
	req := newExecRequest()
	req.Binding.Action = &config.Action{
		Title:     "List files",
		Exec:      []string{"ls", "-alh"},
		Arguments: []config.ActionArgument{},
	}

	req.Arguments = map[string]string{}

	out, err := parseActionExec(req.Arguments, req.Binding.Action, req.Binding.Entity)

	assert.Nil(t, err)
	assert.Equal(t, []string{"ls", "-alh"}, out)
}

func TestExecArrayWithTemplateReplacement(t *testing.T) {
	a1 := config.Action{
		Title: "List specific path",
		Exec:  []string{"ls", "-alh", "{{path}}"},
		Arguments: []config.ActionArgument{
			{
				Name: "path",
				Type: "ascii_identifier",
			},
		},
	}

	values := map[string]string{
		"path": "tmp",
	}

	out, err := parseActionExec(values, &a1, nil)

	assert.Nil(t, err)
	assert.Equal(t, []string{"ls", "-alh", "tmp"}, out)
}

func TestCheckShellArgumentSafetyWithURL(t *testing.T) {
	a1 := config.Action{
		Title: "Download file",
		Shell: "curl {{url}}",
		Arguments: []config.ActionArgument{
			{
				Name: "url",
				Type: "url",
			},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.NotNil(t, err)
	assert.Contains(t, err.Error(), "unsafe argument type 'url' cannot be used with Shell execution")
	assert.Contains(t, err.Error(), "https://docs.olivetin.app/action_execution/shellvsexec.html")
}

func TestCheckShellArgumentSafetyWithEmail(t *testing.T) {
	a1 := config.Action{
		Title: "Send email",
		Shell: "sendmail {{email}}",
		Arguments: []config.ActionArgument{
			{
				Name: "email",
				Type: "email",
			},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.NotNil(t, err)
	assert.Contains(t, err.Error(), "unsafe argument type 'email' cannot be used with Shell execution")
}

func TestCheckShellArgumentSafetyWithExec(t *testing.T) {
	a1 := config.Action{
		Title: "Download file",
		Exec:  []string{"curl", "{{url}}"},
		Arguments: []config.ActionArgument{
			{
				Name: "url",
				Type: "url",
			},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.Nil(t, err)
}

func TestCheckShellArgumentSafetyWithSafeTypes(t *testing.T) {
	a1 := config.Action{
		Title: "List files",
		Shell: "ls {{path}}",
		Arguments: []config.ActionArgument{
			{
				Name: "path",
				Type: "ascii_identifier",
			},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.Nil(t, err)
}

func TestCheckShellArgumentSafetyWithPassword(t *testing.T) {
	a1 := config.Action{
		Title: "Auth with password",
		Shell: "somecommand --password '{{password}}'",
		Arguments: []config.ActionArgument{
			{
				Name: "password",
				Type: "password",
			},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.NotNil(t, err)
	assert.Contains(t, err.Error(), "unsafe argument type 'password' cannot be used with Shell execution")
	assert.Contains(t, err.Error(), "https://docs.olivetin.app/action_execution/shellvsexec.html")
}

func TestCheckShellArgumentSafetyWithPasswordAndExec(t *testing.T) {
	a1 := config.Action{
		Title: "Auth with password via exec",
		Exec:  []string{"somecommand", "--password", "{{password}}"},
		Arguments: []config.ActionArgument{
			{
				Name: "password",
				Type: "password",
			},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.Nil(t, err)
}

func TestCheckShellArgumentSafetyWithHTML(t *testing.T) {
	a1 := config.Action{
		Title: "HTML shell",
		Shell: "echo {{ body }}",
		Arguments: []config.ActionArgument{
			{Name: "body", Type: "html"},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.NotNil(t, err)
	assert.Contains(t, err.Error(), "unsafe argument type 'html'")
}

func TestCheckShellArgumentSafetyWithConfirmation(t *testing.T) {
	a1 := config.Action{
		Title: "Confirm shell",
		Shell: "echo ok",
		Arguments: []config.ActionArgument{
			{Name: "agree", Type: "confirmation"},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.NotNil(t, err)
	assert.Contains(t, err.Error(), "unsafe argument type 'confirmation'")
}

func TestCheckShellArgumentSafetyWithChoicelessCheckbox(t *testing.T) {
	a1 := config.Action{
		Title: "Checkbox shell",
		Shell: "echo {{ flag }}",
		Arguments: []config.ActionArgument{
			{Name: "flag", Type: "checkbox"},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.NotNil(t, err)
	assert.Contains(t, err.Error(), "unsafe argument type 'checkbox'")
}

func TestCheckShellArgumentSafetyWithCustomRegex(t *testing.T) {
	a1 := config.Action{
		Title: "Regex shell",
		Shell: "curl {{ host }}",
		Arguments: []config.ActionArgument{
			{Name: "host", Type: "regex:[a-zA-Z0-9.-]+"},
		},
	}

	err := checkShellArgumentSafety(&a1)
	assert.NotNil(t, err)
	assert.Contains(t, err.Error(), "unsafe argument type 'regex:[a-zA-Z0-9.-]+'")
}

func TestTypeSafetyCheckUrl(t *testing.T) {
	assert.Nil(t, TypeSafetyCheck("test1", "http://google.com", "url"), "Test URL: google.com")
	assert.Nil(t, TypeSafetyCheck("test2", "http://technowax.net:80?foo=bar", "url"), "Test URL: technowax.net with query arguments")
	assert.Nil(t, TypeSafetyCheck("test3", "http://localhost:80?foo=bar", "url"), "Test URL: localhost with query arguments")
	assert.Nil(t, TypeSafetyCheck("test7", "https://example.com/path", "url"), "Test URL: https scheme")
	assert.NotNil(t, TypeSafetyCheck("test4", "http://lo  host:80", "url"), "Test a badly formed URL")
	assert.NotNil(t, TypeSafetyCheck("test5", "12345", "url"), "Test a badly formed URL")
	assert.NotNil(t, TypeSafetyCheck("test6", "_!23;", "url"), "Test a badly formed URL")
	assert.NotNil(t, TypeSafetyCheck("test8", "file:///etc/passwd", "url"), "file:// scheme must be rejected")
	assert.NotNil(t, TypeSafetyCheck("test9", "gopher://example.com", "url"), "gopher:// scheme must be rejected")
}

func TestTypeSafetyCheckRegex(t *testing.T) {
	tests := []struct {
		name     string
		field    string
		pattern  string
		value    string
		hasError bool
	}{
		{
			name:     "Issue #578 - Domain",
			field:    "domain",
			pattern:  "regex:^(?:[a-zA-Z0-9-]{1,63}.)+[a-zA-Z]{2,63}$",
			value:    "immich.example.dev",
			hasError: false,
		},
		{
			name:     "Don't allow numbers in username",
			field:    "Username",
			pattern:  "regex:^[a-zA-Z]$",
			value:    "James1234",
			hasError: true,
		},
		{
			name:     "GHSA-gvxq - reject partial regex match",
			field:    "host",
			pattern:  "regex:[a-zA-Z0-9.-]+",
			value:    "example.com; id",
			hasError: true,
		},
		{
			name:     "reject alternation bypass when pattern looks anchored",
			field:    "host",
			pattern:  "regex:^safe$|bad",
			value:    "xxxbad",
			hasError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := typeSafetyCheckRegex(tt.field, tt.value, tt.pattern)

			if tt.hasError {
				assert.NotNil(t, err, "Expected error for value %s with pattern %s, but got no error", tt.value, tt.pattern)
			} else {
				assert.Nil(t, err, "Expected no error for value %s with pattern %s, but got error: %v", tt.value, tt.pattern, err)
			}
		})
	}
}

func TestRedactShellCommand(t *testing.T) {
	cmd := "echo 'The password for Fred is toomanysecrets'"

	args := []config.ActionArgument{
		{
			Name: "personName",
			Type: "ascii",
		},
		{
			Name: "password",
			Type: "password",
		},
	}

	values := map[string]string{
		"personName": "Fred",
		"password":   "toomanysecrets",
	}

	res := redactShellCommand(cmd, args, values)

	assert.Equal(t, "echo 'The password for Fred is <redacted>'", res, "Redacted shell command should mask the password argument")

	// Test with empty password
	values["password"] = ""
	res = redactShellCommand(cmd, args, values)
	assert.Equal(t, cmd, res, "Empty password should not change the command")

	// Test with missing password argument
	delete(values, "password")
	res = redactShellCommand(cmd, args, values)
	assert.Equal(t, cmd, res, "Missing password argument should not change the command")
}

func TestTypeSafetyCheckEmail(t *testing.T) {
	tests := []struct {
		name     string
		field    string
		value    string
		hasError bool
	}{
		{"Valid simple email", "email", "user@example.com", false},
		{"Valid email with subdomain", "email", "user@mail.example.com", false},
		{"Valid email with plus", "email", "user+test@example.com", false},
		{"Valid email with dash", "email", "user-name@example.com", false},
		{"Valid email with numbers", "email", "user123@example123.com", false},
		{"Invalid email no @", "email", "userexample.com", true},
		{"Invalid email no domain", "email", "user@", true},
		{"Invalid email no user", "email", "@example.com", true},
		{"Invalid email spaces", "email", "user name@example.com", true},
		{"Invalid email double @", "email", "user@@example.com", true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := TypeSafetyCheck(tt.field, tt.value, "email")
			if tt.hasError {
				assert.NotNil(t, err, "Expected error for value '%s'", tt.value)
			} else {
				assert.Nil(t, err, "Expected no error for value '%s', but got: %v", tt.value, err)
			}
		})
	}
}

func TestTypeSafetyCheckDatetime(t *testing.T) {
	tests := []struct {
		name     string
		field    string
		value    string
		hasError bool
	}{
		{"Valid datetime", "datetime", "2023-12-25T15:30:45", false},
		{"Valid datetime morning", "datetime", "2023-01-01T00:00:00", false},
		{"Valid datetime evening", "datetime", "2023-12-31T23:59:59", false},
		{"Invalid format missing T", "datetime", "2023-12-25 15:30:45", true},
		{"Invalid format missing seconds", "datetime", "2023-12-25T15:30", true},
		{"Invalid date", "datetime", "2023-13-25T15:30:45", true},
		{"Invalid time", "datetime", "2023-12-25T25:30:45", true},
		{"Random string", "datetime", "not-a-date", true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := TypeSafetyCheck(tt.field, tt.value, "datetime")
			if tt.hasError {
				assert.NotNil(t, err, "Expected error for value '%s'", tt.value)
			} else {
				assert.Nil(t, err, "Expected no error for value '%s', but got: %v", tt.value, err)
			}
		})
	}
}

func TestTypeSafetyCheckRawStringMultiline(t *testing.T) {
	tests := []struct {
		name  string
		field string
		value string
	}{
		{"Simple string", "content", "hello world"},
		{"Multiline string", "content", "line1\nline2\nline3"},
		{"String with special chars", "content", "!@#$%^&*()"},
		{"Unicode string", "content", "héllo wörld 🌍"},
		{"Very long string", "content", strings.Repeat("a", 1000)},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := TypeSafetyCheck(tt.field, tt.value, "raw_string_multiline")
			assert.Nil(t, err, "raw_string_multiline should accept any value")
		})
	}
}

func TestTypeSafetyCheckUnicodeIdentifier(t *testing.T) {
	tests := []struct {
		name         string
		field        string
		value        string
		expectsError bool
	}{
		{"Valid unicode identifier", "name", "hello_world", false},
		{"Valid with numbers", "name", "test123", false},
		{"Valid with dots", "name", "file.txt", false},
		{"Valid with underscores", "name", "my_file_name", false},
		{"Invalid with special chars", "name", "hello@world", true},
		{"Invalid with brackets", "name", "hello[world]", true},
		{"Invalid with spaces", "name", "hello world", true},
		{"Invalid with path separators", "name", "path/to/file", true},
		{"Invalid with backslashes", "name", "path\\to\\file", true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := TypeSafetyCheck(tt.field, tt.value, "unicode_identifier")
			validateTypeSafetyResult(t, tt.value, tt.expectsError, err)
		})
	}
}

func validateTypeSafetyResult(t *testing.T, value string, expectsError bool, err error) {
	if expectsError {
		assertErrorExpected(t, value, err)
	} else {
		assertNoErrorExpected(t, value, err)
	}
}

func assertErrorExpected(t *testing.T, value string, err error) {
	if err == nil {
		t.Errorf("Expected error for value '%s', but got none", value)
	} else {
		t.Logf("Received expected error for value '%s': %v", value, err)
	}
}

func assertNoErrorExpected(t *testing.T, value string, err error) {
	if err != nil {
		t.Errorf("Expected no error for value '%s', but got: %v", value, err)
	} else {
		t.Logf("No error for valid value '%s' as expected", value)
	}
}

func TestTypeSafetyCheckAsciiIdentifier(t *testing.T) {
	tests := []struct {
		name     string
		field    string
		value    string
		hasError bool
	}{
		{"Valid identifier", "name", "hello_world", false},
		{"Valid with numbers", "name", "test123", false},
		{"Valid with dots", "name", "file.txt", false},
		{"Valid with dashes", "name", "my-file", false},
		{"Valid with underscores", "name", "my_file", false},
		{"Invalid with spaces", "name", "hello world", true},
		{"Invalid with special chars", "name", "hello@world", true},
		{"Invalid unicode", "name", "héllo", true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := TypeSafetyCheck(tt.field, tt.value, "ascii_identifier")
			if tt.hasError {
				assert.NotNil(t, err, "Expected error for value '%s'", tt.value)
			} else {
				assert.Nil(t, err, "Expected no error for value '%s', but got: %v", tt.value, err)
			}
		})
	}
}

func TestTypeSafetyCheckDnsName(t *testing.T) {
	tests := []struct {
		name     string
		value    string
		hasError bool
	}{
		{"Short name", "webserver", false},
		{"Localhost", "localhost", false},
		{"Simple domain", "example.com", false},
		{"Host with subdomain", "webserver.example.com", false},
		{"Deep subdomain", "a.b.c.example.co.uk", false},
		{"Label starting with digit", "1host.example.com", false},
		{"Trailing dot", "example.com.", false},
		{"Punycode IDN", "xn--bcher-kva.example", false},
		{"Underscore", "my_host.example.com", true},
		{"Space", "example .com", true},
		{"Leading hyphen label", "-host.example.com", true},
		{"Trailing hyphen label", "host-.example.com", true},
		{"Empty label", "example..com", true},
		{"IP address", "192.168.1.1", true},
		{"All numeric TLD", "example.123", true},
		{"All numeric short name", "12345", true},
		{"Special chars", "exam!ple.com", true},
		{"Unicode label", "bücher.example.com", true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := TypeSafetyCheck("host", tt.value, "dnsname")
			if tt.hasError {
				assert.NotNil(t, err, "Expected error for value '%s'", tt.value)
			} else {
				assert.Nil(t, err, "Expected no error for value '%s', but got: %v", tt.value, err)
			}
		})
	}
}

func TestTypeSafetyCheckShellSafeIdentifier(t *testing.T) {
	tests := []struct {
		name     string
		value    string
		hasError bool
	}{
		{"Simple username", "alice123", false},
		{"Email username", "alice@example.com", false},
		{"Plus addressing", "alice+test@example.com", false},
		{"Hyphen underscore dot", "alice-test_user.example", false},
		{"Invalid space", "alice example", true},
		{"Invalid shell substitution", "$(whoami)", true},
		{"Invalid backtick", "`whoami`", true},
		{"Invalid semicolon", "alice;id", true},
		{"Invalid ampersand", "alice&id", true},
		{"Invalid pipe", "alice|id", true},
		{"Invalid quote", "alice'example", true},
		{"Invalid slash", "alice/example", true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := TypeSafetyCheck("username", tt.value, "shell_safe_identifier")
			if tt.hasError {
				assert.NotNil(t, err, "Expected error for value '%s'", tt.value)
			} else {
				assert.Nil(t, err, "Expected no error for value '%s', but got: %v", tt.value, err)
			}
		})
	}
}

func TestTypeSafetyCheckAsciiSentence(t *testing.T) {
	tests := []struct {
		name     string
		field    string
		value    string
		hasError bool
	}{
		{"Valid sentence", "text", "Hello world", false},
		{"Valid with numbers", "text", "Test 123", false},
		{"Valid with commas", "text", "Hello, world", false},
		{"Valid with periods", "text", "Hello world.", false},
		{"Valid with multiple spaces", "text", "Hello  world", false},
		{"Invalid with special chars", "text", "Hello@world", true},
		{"Invalid with parentheses", "text", "Hello (world)", true},
		{"Invalid unicode", "text", "Héllo world", true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := TypeSafetyCheck(tt.field, tt.value, "ascii_sentence")
			if tt.hasError {
				assert.NotNil(t, err, "Expected error for value '%s'", tt.value)
			} else {
				assert.Nil(t, err, "Expected no error for value '%s', but got: %v", tt.value, err)
			}
		})
	}
}

func TestTypecheckActionArgumentEmptyName(t *testing.T) {
	arg := config.ActionArgument{
		Name: "",
		Type: "ascii",
	}
	action := config.Action{Title: "Test"}

	err := typecheckActionArgument(&arg, "test", &action)
	assert.NotNil(t, err)
	assert.Contains(t, err.Error(), "argument name cannot be empty")
}

func TestTypecheckActionArgumentConfirmation(t *testing.T) {
	arg := config.ActionArgument{
		Name: "confirm",
		Type: "confirmation",
	}
	action := config.Action{Title: "Test"}

	err := typecheckActionArgument(&arg, "any_value", &action)
	assert.Nil(t, err, "Confirmation type should always pass validation")
}

func TestTypecheckActionArgumentHtmlWithoutName(t *testing.T) {
	action := config.Action{
		Title: "Delete old backups",
		Shell: "rm -rf /opt/oliveTinOldBackups/ && sleep 5",
		Arguments: []config.ActionArgument{
			{Type: "html", Title: "Description"},
			{Type: "confirmation", Title: "Are you sure?!"},
		},
	}

	err := validateArguments(map[string]string{}, &action)
	assert.NoError(t, err)
}

func TestParseCommandForReplacements(t *testing.T) {
	tests := []struct {
		name           string
		shellCommand   string
		values         map[string]string
		expectedOutput string
		expectError    bool
		errorContains  string
	}{
		{
			name:           "Simple replacement",
			shellCommand:   "echo {{ name }}",
			values:         map[string]string{"name": "John"},
			expectedOutput: "echo John",
			expectError:    false,
		},
		{
			name:           "Multiple replacements",
			shellCommand:   "echo {{ first }} {{ last }}",
			values:         map[string]string{"first": "John", "last": "Doe"},
			expectedOutput: "echo John Doe",
			expectError:    false,
		},
		{
			name:           "Replacement with spaces in template",
			shellCommand:   "echo {{  name  }}",
			values:         map[string]string{"name": "John"},
			expectedOutput: "echo John",
			expectError:    false,
		},
		{
			name:           "Missing argument",
			shellCommand:   "echo {{ missing }}",
			values:         map[string]string{},
			expectedOutput: "",
			expectError:    true,
			errorContains:  "required arg not provided: missing",
		},
		{
			name:           "No replacements needed",
			shellCommand:   "echo hello",
			values:         map[string]string{},
			expectedOutput: "echo hello",
			expectError:    false,
		},
		{
			name:           "Multiple same argument",
			shellCommand:   "echo {{ name }} says hello {{ name }}",
			values:         map[string]string{"name": "Alice"},
			expectedOutput: "echo Alice says hello Alice",
			expectError:    false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			output, err := tpl.ParseTemplateWithActionContext(tt.shellCommand, nil, tt.values)

			if tt.expectError {
				assert.NotNil(t, err, "Expected error but got none")
				if tt.errorContains != "" {
					assert.Contains(t, err.Error(), tt.errorContains)
				}
			} else {
				assert.Nil(t, err, "Expected no error but got: %v", err)
				assert.Equal(t, tt.expectedOutput, output)
			}
		})
	}
}

func TestArgumentChoicesValidation(t *testing.T) {
	tests := []struct {
		name        string
		req         *ExecutionRequest
		expectError bool
		description string
	}{
		{
			name: "Valid choice",
			req: &ExecutionRequest{
				Binding: &ActionBinding{
					Action: &config.Action{
						Title: "Test choices",
						Shell: "echo {{ option }}",
						Arguments: []config.ActionArgument{
							{
								Name: "option",
								Type: "ascii",
								Choices: []config.ActionArgumentChoice{
									{Value: "option1", Title: "Option 1"},
									{Value: "option2", Title: "Option 2"},
								},
							},
						},
					},
				},
				Arguments: map[string]string{"option": "option1"},
			},
			expectError: false,
			description: "Should accept valid choice",
		},
		{
			name: "Invalid choice",
			req: &ExecutionRequest{
				Binding: &ActionBinding{
					Action: &config.Action{
						Title: "Test choices",
						Shell: "echo {{ option }}",
						Arguments: []config.ActionArgument{
							{
								Name: "option",
								Type: "ascii",
								Choices: []config.ActionArgumentChoice{
									{Value: "option1", Title: "Option 1"},
									{Value: "option2", Title: "Option 2"},
								},
							},
						},
					},
				},
				Arguments: map[string]string{"option": "invalid_option"},
			},
			expectError: true,
			description: "Should reject invalid choice",
		},
		{
			name: "Invalid choice",
			req: &ExecutionRequest{
				Binding: &ActionBinding{
					Action: &config.Action{
						Title: "Test choices",
						Shell: "echo {{ option }}",
						Arguments: []config.ActionArgument{
							{
								Name: "option",
								Type: "ascii",
								Choices: []config.ActionArgumentChoice{
									{Value: "option1", Title: "Option 1"},
									{Value: "option2", Title: "Option 2"},
								},
							},
						},
					},
				},
				Arguments: map[string]string{"option": "option1"},
			},
			expectError: false,
			description: "Should accept valid choice",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			_, err := parseActionArguments(tt.req)

			if tt.expectError {
				assert.NotNil(t, err, tt.description)
				assert.Contains(t, err.Error(), "predefined choices")
			} else {
				assert.Nil(t, err, tt.description)
			}
		})
	}
}

func TestTypeSafetyCheckVeryDangerousRawString(t *testing.T) {
	// This type should allow anything without validation
	tests := []string{
		"normal text",
		"_zomg_ c:/ haxxor ' bobby tables && rm -rf /",
		"$(rm -rf /)",
		"; DROP TABLE users; --",
		"../../../../etc/passwd",
		"",
		"unicode: 你好世界",
		"emojis: 🔥💀☠️",
	}

	for _, value := range tests {
		t.Run(fmt.Sprintf("Value: %s", value), func(t *testing.T) {
			err := TypeSafetyCheck("test", value, "very_dangerous_raw_string")
			assert.Nil(t, err, "very_dangerous_raw_string should accept any value including: %s", value)
		})
	}
}

func TestParseActionArgumentsWithEntityPrefix(t *testing.T) {
	req := newExecRequest()
	req.Binding.Action = &config.Action{
		Title: "Test entity prefix",
		Shell: "echo 'Processing {{ name }} for entity'",
		Arguments: []config.ActionArgument{
			{Name: "name", Type: "ascii"},
		},
	}

	req.Arguments = map[string]string{
		"name": "testuser",
	}

	req.Binding.Entity = &entities.Entity{
		Title: "entity_123",
	}

	// Test with entity prefix
	output, err := parseActionArguments(req)
	assert.Nil(t, err)
	assert.Contains(t, output, "testuser")
}

func TestComplexRegexPatterns(t *testing.T) {
	tests := []struct {
		name     string
		pattern  string
		value    string
		hasError bool
	}{
		{
			name:     "Phone number pattern",
			pattern:  "regex:^\\+?[1-9]\\d{1,14}$",
			value:    "+1234567890",
			hasError: false,
		},
		{
			name:     "Invalid phone number",
			pattern:  "regex:^\\+?[1-9]\\d{1,14}$",
			value:    "123abc",
			hasError: true,
		},
		{
			name:     "Semantic version pattern",
			pattern:  "regex:^(0|[1-9]\\d*)\\.(0|[1-9]\\d*)\\.(0|[1-9]\\d*)$",
			value:    "1.2.3",
			hasError: false,
		},
		{
			name:     "Invalid semantic version",
			pattern:  "regex:^(0|[1-9]\\d*)\\.(0|[1-9]\\d*)\\.(0|[1-9]\\d*)$",
			value:    "1.2",
			hasError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := typeSafetyCheckRegex("test", tt.value, tt.pattern)
			if tt.hasError {
				assert.NotNil(t, err)
			} else {
				assert.Nil(t, err)
			}
		})
	}
}