package recipe

import "testing"

func TestExecContext_WrappedCommand(t *testing.T) {
	tests := []struct {
		name string
		ctx  ExecContext
		cmd  string
		want string
	}{
		{
			name: "no context",
			ctx:  ExecContext{},
			cmd:  "apt install -y curl",
			want: "apt install -y curl",
		},
		{
			name: "workdir only",
			ctx:  ExecContext{WorkDir: "/app"},
			cmd:  "npm install",
			want: "cd '/app' && /bin/sh -c 'npm install'",
		},
		{
			name: "env only",
			ctx:  ExecContext{EnvVars: map[string]string{"PORT": "8080"}},
			cmd:  "node server.js",
			want: "PORT='8080' /bin/sh -c 'node server.js'",
		},
		{
			name: "workdir with space",
			ctx:  ExecContext{WorkDir: "/my project"},
			cmd:  "make build",
			want: "cd '/my project' && /bin/sh -c 'make build'",
		},
		{
			name: "command with single quotes",
			ctx:  ExecContext{WorkDir: "/app"},
			cmd:  "echo 'hello'",
			want: "cd '/app' && /bin/sh -c 'echo '\\''hello'\\'''",
		},
		{
			name: "env value with single quotes",
			ctx:  ExecContext{EnvVars: map[string]string{"MSG": "it's fine"}},
			cmd:  "echo $MSG",
			want: "MSG='it'\\''s fine' /bin/sh -c 'echo $MSG'",
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			got := tc.ctx.WrappedCommand(tc.cmd)
			if got != tc.want {
				t.Errorf("WrappedCommand(%q)\n  got  %q\n  want %q", tc.cmd, got, tc.want)
			}
		})
	}
}

func TestExecContext_StartCommand(t *testing.T) {
	tests := []struct {
		name string
		ctx  ExecContext
		cmd  string
		want string
	}{
		{
			name: "no context",
			ctx:  ExecContext{},
			cmd:  "python3 app.py",
			want: "nohup /bin/sh -c 'python3 app.py' >/dev/null 2>&1 &",
		},
		{
			name: "with workdir",
			ctx:  ExecContext{WorkDir: "/app"},
			cmd:  "python3 server.py",
			want: "cd '/app' && nohup /bin/sh -c 'python3 server.py' >/dev/null 2>&1 &",
		},
		{
			name: "with env",
			ctx:  ExecContext{EnvVars: map[string]string{"PORT": "9000"}},
			cmd:  "node index.js",
			want: "PORT='9000' nohup /bin/sh -c 'node index.js' >/dev/null 2>&1 &",
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			got := tc.ctx.StartCommand(tc.cmd)
			if got != tc.want {
				t.Errorf("StartCommand(%q)\n  got  %q\n  want %q", tc.cmd, got, tc.want)
			}
		})
	}
}

func TestShellescape(t *testing.T) {
	tests := []struct {
		input string
		want  string
	}{
		{"simple", "'simple'"},
		{"/path/to/dir", "'/path/to/dir'"},
		{"it's fine", "'it'\\''s fine'"},
		{"", "''"},
		{"a'b'c", "'a'\\''b'\\''c'"},
	}
	for _, tc := range tests {
		got := shellescape(tc.input)
		if got != tc.want {
			t.Errorf("shellescape(%q) = %q, want %q", tc.input, got, tc.want)
		}
	}
}