Remove -port flag

Author: ajeetdsouza

config.go

@@ -20,8 +20,7 @@ var (
 	pathDb      = filepath.Join(pathClidle, "db.json")
 	pathHostKey = filepath.Join(pathClidle, "hostkey")
 
-	flagServe = flag.Bool("serve", false, "Spawns an SSH server")
-	flagPort  = flag.Int("port", 22, "Changes the server port")
+	flagServe = flag.String("serve", "", "Spawns an SSH server on the given address (format: 0.0.0.0:1337)")
 
 	teaOptions = []tea.ProgramOption{tea.WithAltScreen(), tea.WithOutput(os.Stderr)}
 )

main.go

@@ -19,8 +19,8 @@ import (
 
 func main() {
 	flag.Parse()
-	if *flagServe {
-		runServer(*flagPort)
+	if addr := *flagServe; addr != "" {
+		runServer(addr)
 	} else {
 		runCli()
 	}
@@ -42,14 +42,14 @@ func runCli() {
 	os.Exit(exitCode)
 }
 
-func runServer(port int) {
+func runServer(addr string) {
 	withHostKey := wish.WithHostKeyPath(pathHostKey)
 	if pem, ok := os.LookupEnv(envHostKey); ok {
 		withHostKey = wish.WithHostKeyPEM([]byte(pem))
 	}
 
 	server, err := wish.NewServer(
-		wish.WithAddress(fmt.Sprintf("0.0.0.0:%d", port)),
+		wish.WithAddress(addr),
 		wish.WithMaxTimeout(30*time.Minute),
 		wish.WithMiddleware(
 			bm.Middleware(func(s ssh.Session) (tea.Model, []tea.ProgramOption) {