diff --git a/TODO b/TODO index dfd81de..49e8f3c 100644 --- a/TODO +++ b/TODO @@ -2,7 +2,6 @@ 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 diff --git a/cmd/lash/main.go b/cmd/lash/main.go index 1f994c6..b2ccab9 100644 --- a/cmd/lash/main.go +++ b/cmd/lash/main.go @@ -11,6 +11,7 @@ import ( ) var versionFlag = flag.Bool("version", false, "Print out version and exit.") +var port = flag.Int("p", 1337, "Set the port for LASH exchanges.") func main() { flag.Parse() @@ -24,8 +25,8 @@ func main() { 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.Log("starting server at http://0.0.0.0:%d", *port) + if err := http.ListenAndServe(fmt.Sprintf(":%d", *port), srv); err != nil { errx.FatalPerror(err) } } diff --git a/internal/app/file.go b/internal/app/file.go index 3b42355..126ba59 100644 --- a/internal/app/file.go +++ b/internal/app/file.go @@ -2,14 +2,14 @@ package app import ( "errors" - "os" + "flag" "strings" ) func GetFilePath() (string, error) { fp := "" - for _, v := range os.Args[1:] { + for _, v := range flag.Args() { if !strings.HasPrefix(v, "-") { fp = v return fp, nil