feat: Add --preserve-path arg

cmd/all/all.go

@@ -9,14 +9,15 @@ import (
 )
 
 var FlagTarget string
+var PreservePath bool
 
 var Cmd = &cobra.Command{
 	Use:     "all",
 	Short:   "Redirect all to ...",
 	Aliases: []string{"s"},
 	Args:    cobra.NoArgs,
 	Run: func(c *cobra.Command, args []string) {
-		err := server.Server(FlagTarget)
+		err := server.Server(FlagTarget, PreservePath)
 		if err != nil {
 			log.Fatal(err)
 		}
@@ -33,4 +34,11 @@ func init() {
 		"Redirect target (eg.: https://google.com)",
 	)
 	Cmd.MarkPersistentFlagRequired("target")
+	Cmd.PersistentFlags().BoolVarP(
+		&PreservePath,
+		"preserve-path",
+		"p",
+		false,
+		"Preserve path",
+	)
 }

pkg/server/server.go

@@ -5,8 +5,11 @@ import (
 	"net/http"
 )
 
-func Server(target string) error {
+func Server(target string, preservePath bool) error {
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		if preservePath {
+			target = fmt.Sprintf("%s%s", target, r.URL.Path)
+		}
 		http.Redirect(w, r, target, http.StatusTemporaryRedirect)
 	})