maint: changes

internal/app/file.go

@@ -1,19 +0,0 @@
-package app
-
-import (
-	"errors"
-	"flag"
-	"strings"
-)
-
-func GetFilePath() (string, error) {
-	fp := ""
-
-	for _, v := range flag.Args() {
-		if !strings.HasPrefix(v, "-") {
-			fp = v
-			return fp, nil
-		}
-	}
-	return "", errors.New("not enough arguments")
-}

internal/app/routes.go

@@ -1,6 +1,7 @@
 package app
 
 import (
+	"flag"
 	"lash"
 	"lash/internal/errx"
 	"lash/internal/handlers"
@@ -11,26 +12,29 @@ import (
 func New(ctx *lash.LashContext) http.Handler {
 	mux := http.NewServeMux()
 
-	fp, err := GetFilePath()
-	if err != nil {
-		errx.FatalPerror(err)
-	}
-
-	contents, err := os.ReadFile(fp)
-	if err != nil {
-		errx.FatalPerror(err)
-	}
-
 	share := handlers.ShareData{
 		Version: lash.Version,
 	}
 
 	file := handlers.FileHandler{
 		Ctx: ctx,
-		FileData: handlers.FileData{
+	}
+
+	fps := flag.Args()
+	if len(fps) < 1 {
+		errx.FatalPerror(errx.ErrNotEnoughArgs)
+	}
+
+	for _, fp := range fps {
+		contents, err := os.ReadFile(fp)
+		if err != nil {
+			errx.FatalPerror(err)
+		}
+
+		file.FileData = append(file.FileData, handlers.FileData{
 			Contents: contents,
 			FileName: fp,
-		},
+		})
 	}
 
 	mux.HandleFunc("/", share.Handler)