maint: update todo

Makefile

@@ -5,7 +5,7 @@ GO?= go
 all: build
 
 build:
-	$(GO) build -o lash ./cmd/lash/
+	$(GO) build -o lash ./bin/main.go
 
 lint:
 	$(GO) fmt ./...

TODO

@@ -1,9 +1 @@
-maint: clean up source code
-maint: comment source code
-maint: document more
-feat: improve flags
-  -p for port
-feat: replace uuid dep with custom id generator
-feat: multiple files
-feat: share link
-maint: make the colors more cross platform
+feat: sumbit code with press of enter key

bin/main.go

@@ -0,0 +1,44 @@
+package main
+
+import (
+	_ "embed"
+	"flag"
+	"fmt"
+	"lash"
+	"lash/internal/app"
+	"lash/internal/errx"
+	share "lash/internal/shareLink"
+	"net/http"
+)
+
+var versionFlag = flag.Bool("version", false, "Print out version and exit.")
+
+var port = flag.Int("p", 1337, "Set the port for LASH exchanges.")
+var numberOfShares = flag.Int("n", -1, "Number of shares to allow before killing the server.")
+
+func main() {
+	flag.Parse()
+
+	if *versionFlag {
+		fmt.Print("lash/vxc v", lash.Version)
+		return
+	}
+
+	ctx := lash.LashContext{
+		N: *numberOfShares,
+	}
+
+	srv := app.New(&ctx)
+
+	server := http.Server{
+		Addr:    fmt.Sprintf(":%d", *port),
+		Handler: srv,
+	}
+
+	errx.Log("Your share link is %s", share.GenerateShareLink(*port))
+	errx.Log("Your token is \033[1;92m%s\033[0m", lash.Token)
+	errx.Log("starting server at http://0.0.0.0:%d", *port)
+	if err := server.ListenAndServe(); err != nil {
+		errx.FatalPerror(err)
+	}
+}

cmd/lash/main.go

@@ -1,31 +0,0 @@
-package main
-
-import (
-	_ "embed"
-	"flag"
-	"fmt"
-	"lash"
-	"lash/internal/app"
-	"lash/internal/errx"
-	"net/http"
-)
-
-var versionFlag = flag.Bool("version", false, "Print out version and exit.")
-
-func main() {
-	flag.Parse()
-
-	if *versionFlag {
-		fmt.Print("lash/vxc v", lash.Version)
-		return
-	}
-
-	srv := app.New()
-
-	errx.Log("Your token is \033[1;92m%s\033[0m", lash.Token)
-
-	errx.Log("starting server at http://0.0.0.0:1337")
-	if err := http.ListenAndServe(":1337", srv); err != nil {
-		errx.FatalPerror(err)
-	}
-}

go.mod

@@ -2,4 +2,4 @@ module lash
 
 go 1.26.3
 
-require github.com/google/uuid v1.6.0 // indirect
+require github.com/google/uuid v1.6.0

internal/app/file.go

@@ -1,19 +0,0 @@
-package app
-
-import (
-	"errors"
-	"os"
-	"strings"
-)
-
-func GetFilePath() (string, error) {
-	fp := ""
-
-	for _, v := range os.Args[1:] {
-		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"
@@ -8,30 +9,37 @@ import (
 	"os"
 )
 
-func New() http.Handler {
+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,
+		Version: lash.Version,
 	}
 
-	file := handlers.FileData{
-		Contents: contents,
-		FileName: fp,
+	file := handlers.FileHandler{
+		Ctx: ctx,
+	}
+
+	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)
 	mux.HandleFunc("/api/receive-token", file.APIHandler)
+	mux.HandleFunc("/"+lash.ShareLinkToken, file.FileHandler)
 
 	return mux
 }

internal/errx/define.go

@@ -0,0 +1,5 @@
+package errx
+
+import "errors"
+
+var ErrNotEnoughArgs error = errors.New("not enough arguments")

internal/generator/generator.go

@@ -0,0 +1,55 @@
+package generator
+
+import (
+	"math/rand/v2"
+)
+
+var keys []string = []string{
+	"octo",
+	"cool",
+	"whip",
+	"keys",
+	"cats",
+	"part",
+	"rate",
+	"face",
+	"lard",
+	"larp",
+	"dogs",
+	"cash",
+	"city",
+	"cold",
+	"desk",
+	"down",
+	"dirt",
+	"long",
+	"mean",
+	"news",
+	"only",
+	"open",
+	"year",
+	"wood",
+	"wing",
+	"work",
+	"wash",
+	"vote",
+	"onyx",
+	"snow",
+	"john",
+	"json",
+	"fang",
+}
+
+func Generate(n int) (code string) {
+	codeLen := n
+
+	start := rand.IntN(len(keys))
+
+	for i := range codeLen {
+		code += keys[(start+i)%len(keys)]
+		if i != codeLen-1 {
+			code += "-"
+		}
+	}
+	return
+}

internal/handlers/file.go

@@ -1,13 +1,20 @@
 package handlers
 
 import (
+	"archive/zip"
 	"encoding/json"
 	"fmt"
 	"lash"
+	"lash/internal/errx"
 	"net/http"
 	"strconv"
 )
 
+type FileHandler struct {
+	Ctx      *lash.LashContext
+	FileData []FileData
+}
+
 type FileData struct {
 	Contents []byte
 	FileName string
@@ -17,7 +24,9 @@ type ValidateRequest struct {
 	Token string
 }
 
-func (h FileData) APIHandler(w http.ResponseWriter, r *http.Request) {
+var sent int = 0
+
+func (h FileHandler) APIHandler(w http.ResponseWriter, r *http.Request) {
 	decoder := json.NewDecoder(r.Body)
 
 	var t ValidateRequest
@@ -32,10 +41,43 @@ func (h FileData) APIHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s", h.FileName))
-	w.Header().Set("Content-Type", "application/octet-stream")
-	w.Header().Set("Content-Length", strconv.Itoa(len(h.Contents)))
-
-	w.WriteHeader(http.StatusOK)
-	w.Write(h.Contents)
+	http.Redirect(w, r, fmt.Sprintf("/%s", lash.ShareLinkToken), http.StatusTemporaryRedirect)
+}
+
+func (h FileHandler) FileHandler(w http.ResponseWriter, r *http.Request) {
+	if sent >= h.Ctx.N && h.Ctx.N != -1 {
+		http.Error(w, "Too many requests", http.StatusTooManyRequests)
+		return
+	}
+
+	filename := h.FileData[0].FileName
+	if len(h.FileData) > 1 {
+		filename = "lash.zip"
+	}
+
+	w.Header().Set("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, filename))
+	w.Header().Set("Content-Type", "application/octet-stream")
+
+	if len(h.FileData) == 1 {
+		w.Header().Set("Content-Length", strconv.Itoa(len(h.FileData[0].Contents)))
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write(h.FileData[0].Contents)
+	} else {
+		w.WriteHeader(http.StatusOK)
+		zw := zip.NewWriter(w)
+		defer zw.Close()
+
+		for _, f := range h.FileData {
+			fw, err := zw.Create(f.FileName)
+			if err != nil {
+				errx.FatalPerror(err)
+			}
+			_, err = fw.Write(f.Contents)
+			if err != nil {
+				errx.FatalPerror(err)
+			}
+		}
+	}
+
+	sent++
 }

internal/shareLink/shareLink.go

@@ -0,0 +1,29 @@
+package sharelink
+
+import (
+	"fmt"
+	"lash"
+	"lash/internal/errx"
+	"net"
+)
+
+func GenerateShareLink(port int) string {
+	link := "http://"
+	link += GetLocalIP()
+	link += fmt.Sprintf(":%d", port)
+	link += "/"
+	link += lash.ShareLinkToken
+	return link
+}
+
+func GetLocalIP() string {
+	conn, err := net.Dial("udp", "8.8.8.8:80")
+	if err != nil {
+		errx.FatalPerror(err)
+	}
+	defer conn.Close()
+
+	localAddress := conn.LocalAddr().(*net.UDPAddr)
+
+	return localAddress.IP.String()
+}

lash.go

@@ -2,14 +2,21 @@ package lash
 
 import (
 	"embed"
-
-	"github.com/google/uuid"
+	"lash/internal/generator"
+	"net/http"
 )
 
+type LashContext struct {
+	N      int
+	Server *http.Server
+}
+
 //go:embed templates/*.html
 var Templates embed.FS
 
 //go:embed version
 var Version string
 
-var Token string = uuid.New().String()
+var Token string = generator.Generate(1)
+
+var ShareLinkToken string = generator.Generate(1)

version

@@ -1 +1 @@
-1.0.0
+6