parseFlags

1 files changed, 15 insertions, 0 deletions

diff --git a/src/flags.go b/src/flags.go
new file mode 100644
index 0000000..1fd28df
--- /dev/null
+++ b/src/flags.go
@@ -0,0 +1,15 @@
+package main
+
+import "flag"
+
+func parseFlags() config {
+	sshPort := flag.Int("sshPort", 2222, "the port the ssh server is listening on")
+	httpPort := flag.Int("httpPort", 8084, "the port the http server is listening on")
+
+	flag.Parse()
+
+	return config{
+		sshPort:  *sshPort,
+		httpPort: *httpPort,
+	}
+}