feat: add --port and --key-file flags using cobra

Co-authored-by: aider (openrouter/qwen/qwen3-coder) <[email protected]>

Author: jmelahman

README.md

@@ -7,6 +7,10 @@
 
 This serves the [NYT Connections TUI](https://github.com/jmelahman/connections) over SSH.
 
+```shell
+$ ssh connections.lahman.dev
+```
+
 ## Running
 
 In one terminal, start the server,
@@ -22,13 +26,13 @@ In a separate terminal,
 $ ssh -p 2222 localhost
 ```
 
-This assumes the server has an [SSH key](https://wiki.archlinux.org/title/SSH_keys) at [~/.ssh/id_rsa](https://github.com/jmelahman/connections-ssh/blob/12b9ba7d3ec6059a349d23ea85e7b948b16517a1/main.go#L32).
+By default, the server looks for an [SSH key](https://wiki.archlinux.org/title/SSH_keys) at [~/.ssh/id_rsa](https://github.com/jmelahman/connections-ssh/blob/12b9ba7d3ec6059a349d23ea85e7b948b16517a1/main.go#L32).
+This can be overridden with the `--key-file` flag.
 
-Moreover, if running on port `22` is desired, you'll like need elevated privileges (not recommended),
+Moreover, if running on port `22` is desired, you'll likely need elevated privileges (not recommended),
 
 ```shell
-$ go build .
-$ sudo ./connections-ssh
+$ sudo connections-ssh --port 22
 2025/08/07 22:55:16 Starting SSH server on :22
 ```
 

go.mod

@@ -6,15 +6,18 @@ require (
 	github.com/gdamore/tcell/v2 v2.8.1
 	github.com/gliderlabs/ssh v0.3.8
 	github.com/jmelahman/connections v0.1.17
+	github.com/spf13/cobra v1.9.1
 )
 
 require (
 	github.com/anmitsu/go-shlex v0.0.0-20200514113438-38f4b401e2be // indirect
 	github.com/gdamore/encoding v1.0.1 // indirect
+	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
 	github.com/mattn/go-runewidth v0.0.16 // indirect
 	github.com/rivo/tview v0.0.0-20250625164341-a4a78f1e05cb // indirect
 	github.com/rivo/uniseg v0.4.7 // indirect
+	github.com/spf13/pflag v1.0.6 // indirect
 	golang.org/x/crypto v0.41.0 // indirect
 	golang.org/x/sys v0.35.0 // indirect
 	golang.org/x/term v0.34.0 // indirect

go.sum

@@ -1,12 +1,15 @@
 github.com/anmitsu/go-shlex v0.0.0-20200514113438-38f4b401e2be h1:9AeTilPcZAjCFIImctFaOjnTIavg87rW78vTPkQqLI8=
 github.com/anmitsu/go-shlex v0.0.0-20200514113438-38f4b401e2be/go.mod h1:ySMOLuWl6zY27l47sB3qLNK6tF2fkHG55UZxx8oIVo4=
+github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
 github.com/gdamore/encoding v1.0.1 h1:YzKZckdBL6jVt2Gc+5p82qhrGiqMdG/eNs6Wy0u3Uhw=
 github.com/gdamore/encoding v1.0.1/go.mod h1:0Z0cMFinngz9kS1QfMjCP8TY7em3bZYeeklsSDPivEo=
 github.com/gdamore/tcell/v2 v2.8.1 h1:KPNxyqclpWpWQlPLx6Xui1pMk8S+7+R37h3g07997NU=
 github.com/gdamore/tcell/v2 v2.8.1/go.mod h1:bj8ori1BG3OYMjmb3IklZVWfZUJ1UBQt9JXrOCOhGWw=
 github.com/gliderlabs/ssh v0.3.8 h1:a4YXD1V7xMF9g5nTkdfnja3Sxy1PVDCj1Zg4Wb8vY6c=
 github.com/gliderlabs/ssh v0.3.8/go.mod h1:xYoytBv1sV0aL3CavoDuJIQNURXkkfPA/wxQ1pL1fAU=
 github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
+github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
+github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
 github.com/jmelahman/connections v0.1.17 h1:6CCN3e3NNnSSSnFl8DJdhjYIO0rfjPBWjF1LO+YBFvw=
 github.com/jmelahman/connections v0.1.17/go.mod h1:mTgdz687JXTa3HlSeIsejJrXjsv0WT1waXyOw4k1PXw=
 github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=
@@ -19,6 +22,11 @@ github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJ
 github.com/rivo/uniseg v0.4.3/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
 github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
 github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/spf13/cobra v1.9.1 h1:CXSaggrXdbHK9CF+8ywj8Amf7PBRmPCOJugH954Nnlo=
+github.com/spf13/cobra v1.9.1/go.mod h1:nDyEzZ8ogv936Cinf6g1RU9MRY64Ir93oCnqb9wxYW0=
+github.com/spf13/pflag v1.0.6 h1:jFzHGLGAlb3ruxLB8MhbI6A8+AQX/2eW4qeyNZXNp2o=
+github.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
@@ -89,3 +97,5 @@ golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
 golang.org/x/tools v0.13.0/go.mod h1:HvlwmtVNQAhOuCjW7xxvovg8wbNq7LwfXh/k7wXUl58=
 golang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d/go.mod h1:aiJjzUbINMkxbQROHiO6hDPo2LHcIPhhQsa9DLh0yGk=
 golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

main.go

@@ -11,9 +11,32 @@ import (
 	"github.com/gdamore/tcell/v2/terminfo"
 	"github.com/gliderlabs/ssh"
 	"github.com/jmelahman/connections/game"
+	"github.com/spf13/cobra"
+)
+
+var (
+	port    string
+	keyFile string
 )
 
 func main() {
+	var rootCmd = &cobra.Command{
+		Use:   "ssh-game",
+		Short: "SSH game server",
+		Run: func(cmd *cobra.Command, args []string) {
+			serve()
+		},
+	}
+
+	rootCmd.Flags().StringVar(&port, "port", "2222", "Port to listen on")
+	rootCmd.Flags().StringVar(&keyFile, "key-file", "", "Path to SSH host key file")
+
+	if err := rootCmd.Execute(); err != nil {
+		log.Fatal(err)
+	}
+}
+
+func serve() {
 	ssh.Handle(func(s ssh.Session) {
 		screen, err := NewSessionScreen(s)
 		if err != nil {
@@ -25,13 +48,20 @@ func main() {
 		}
 	})
 
-	log.Println("Starting SSH server on :2222")
-	home, err := os.UserHomeDir()
-	if err != nil {
-		panic(err)
+	log.Printf("Starting SSH server on :%s", port)
+
+	// Use provided key file or default to ~/.ssh/id_rsa
+	hostKeyFile := keyFile
+	if hostKeyFile == "" {
+		home, err := os.UserHomeDir()
+		if err != nil {
+			panic(err)
+		}
+		hostKeyFile = filepath.Join(home, ".ssh", "id_rsa")
 	}
-	log.Fatal(ssh.ListenAndServe(":2222", nil,
-		ssh.HostKeyFile(filepath.Join(home, ".ssh", "id_rsa")),
+
+	log.Fatal(ssh.ListenAndServe(":"+port, nil,
+		ssh.HostKeyFile(hostKeyFile),
 	))
 }