publish as package

.gitignore

@@ -1 +1,2 @@
 *.sum
+portmux

README.md

@@ -1,11 +1,15 @@
 # portmux
 
+[![go.dev reference](https://img.shields.io/badge/go.dev-reference-007d9c?logo=go&logoColor=white)](https://pkg.go.dev/github.com/btwiuse/portmux?tab=doc)
+[![Go 1.18+](https://img.shields.io/github/go-mod/go-version/btwiuse/portmux)](https://golang.org/dl/)
+[![License](https://img.shields.io/github/license/btwiuse/portmux?color=%23000&style=flat-round)](https://github.com/btwiuse/portmux/blob/main/LICENSE)
+
 Port multiplexer: a reverse proxy that exposes HTTP / WS endpoints with UI on a single PORT
 
 ## Install
 
 ```
-$ go install github.com/btwiuse/portmux@latest
+$ go install github.com/btwiuse/portmux/cmd/portmux@latest
 ```
 
 ## Examples

main.go

@@ -1,4 +1,4 @@
-package main
+package portmux
 
 import (
 	"fmt"
@@ -140,18 +140,17 @@ func (p *PortMux) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	}
 }
 
-func main() {
+func Run(args []string) error {
 	port := envPORT(":8000")
 	mux := NewPortMux(&Options{
-		argv: portmuxArgv(),
+		argv: args,
 		ui:   portmuxUI(),
 		ws:   portmuxWS(),
 		http: portmuxHTTP(),
 	})
 	if u := portmuxUFO(); u != nil {
-		log.Fatalln(wtf.Serve(*u, mux))
-		return
+		return wtf.Serve(*u, mux)
 	}
-	log.Println(fmt.Sprintf("listening on http://127.0.0.1%s", port))
-	log.Fatalln(http.ListenAndServe(port, mux))
+	log.Println(fmt.Sprintf("No PORTMUX_UFO envvar found, listening on http://127.0.0.1%s", port))
+	return http.ListenAndServe(port, mux)
 }