feat: escape support (#3)

cmd/proxy/main.go

@@ -2,6 +2,9 @@ package main
 
 import (
 	"context"
+	"os"
+
+	"github.com/rs/zerolog"
 
 	logger2 "github.com/ft-t/browser-switcher/pkg/logger"
 )
@@ -10,6 +13,8 @@ func main() {
 	logger := logger2.GetLogger()
 	ctx := logger.WithContext(context.Background())
 
+	zerolog.Ctx(ctx).Debug().Msgf("starting proxy with arguments: %v", os.Args)
+
 	if err := run(ctx); err != nil {
 		logger.Panic().Err(err).Msg("failed to run")
 	}

cmd/proxy/win.go

@@ -11,6 +11,8 @@ import (
 
 	"github.com/rs/zerolog"
 	"golang.org/x/sys/windows/registry"
+
+	"github.com/ft-t/browser-switcher/pkg/escaper"
 )
 
 func findAppRegistrationPath(ctx context.Context) (string, error) {
@@ -39,11 +41,18 @@ func run(ctx context.Context) error {
 		return err
 	}
 
+	var escapedArgs []string
+	for _, arg := range os.Args[1:] {
+		escapedArgs = append(escapedArgs, escaper.Escape(arg))
+	}
+
 	args := slices.Concat([]string{
 		"/C",
 		"start",
 		appPath,
-	}, os.Args[1:])
+	}, escapedArgs)
+
+	zerolog.Ctx(ctx).Debug().Msgf("running command: cmd.exe %v", escapedArgs)
 
 	cmd := exec.Command(
 		"cmd.exe",

cmd/switcher/main.go

@@ -10,6 +10,7 @@ import (
 	"gopkg.in/natefinch/lumberjack.v2"
 
 	config2 "github.com/ft-t/browser-switcher/pkg/config"
+	"github.com/ft-t/browser-switcher/pkg/escaper"
 	"github.com/ft-t/browser-switcher/pkg/launcher"
 	"github.com/ft-t/browser-switcher/pkg/selector"
 	"github.com/ft-t/browser-switcher/pkg/ui"
@@ -37,8 +38,11 @@ func main() {
 	}
 
 	targetURL := os.Args[1]
+	targetURL = escaper.Unescape(targetURL)
 
-	ctx = lg.With().Str("targetURL", targetURL).Logger().WithContext(ctx)
+	ctx = lg.With().Str("rawTargetURL", os.Args[1]).
+		Str("targetURL", targetURL).
+		Logger().WithContext(ctx)
 
 	browserConfig, err := config2.ReadConfig(ctx)
 

pkg/escaper/escape.go

@@ -0,0 +1,20 @@
+package escaper
+
+import (
+	"fmt"
+	"strings"
+)
+
+var charsToEscape = []string{
+	"&",
+}
+
+func Escape(
+	input string,
+) string {
+	for _, char := range charsToEscape {
+		input = strings.ReplaceAll(input, char, fmt.Sprintf(`"%s"`, char))
+	}
+
+	return input
+}

pkg/escaper/unescape.go

@@ -0,0 +1,16 @@
+package escaper
+
+import (
+	"fmt"
+	"strings"
+)
+
+func Unescape(
+	input string,
+) string {
+	for _, char := range charsToEscape {
+		input = strings.ReplaceAll(input, fmt.Sprintf(`"%s"`, char), char)
+	}
+
+	return input
+}