First commit, forked from wormhole

main.go

@@ -0,0 +1,73 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+
+	log "github.com/sirupsen/logrus"
+)
+
+const BUFFERSIZE = 1024
+const numberConnections = 8
+
+// Build flags
+var server, file string
+
+// Global varaibles
+var serverAddress, fileName string
+
+func main() {
+	flag.StringVar(&serverAddress, "server", "", "(run as client) server address to connect to")
+	flag.StringVar(&fileName, "file", "", "(run as server) file to serve")
+	flag.Parse()
+	// Check build flags too, which take precedent
+	if server != "" {
+		serverAddress = server
+	}
+	if file != "" {
+		fileName = file
+	}
+	fmt.Println(`
+             *     ,MMM8&&&.            *
+                  MMMM88&&&&&    .
+                 MMMM88&&&&&&&
+     *           MMM88&&&&&&&&
+                 MMM88&&&&&&&&
+                 'MMM88&&&&&&'
+                   'MMM8&&&'      *
+          |\___/|
+          )     (             .              '
+         =\     /=
+           )===(       *
+          /     \
+          |     |
+         /       \
+         \       /
+  _/\_/\_/\__  _/_/\_/\_/\_/\_/\_/\_/\_/\_/\_
+  |  |  |  |( (  |  |  |  |  |  |  |  |  |  |
+  |  |  |  | ) ) |  |  |  |  |  |  |  |  |  |
+  |  |  |  |(_(  |  |  |  |  |  |  |  |  |  |
+  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |
+  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  
+`)
+	if len(fileName) != 0 {
+		runServer()
+	} else if len(serverAddress) != 0 {
+		runClient()
+	} else {
+		fmt.Println("You must specify either -file (for running as a server) or -server (for running as a client)")
+	}
+}
+
+func init() {
+	// Log as JSON instead of the default ASCII formatter.
+	// log.SetFormatter(&log.JSONFormatter{})
+	log.SetFormatter(&log.TextFormatter{})
+
+	// Output to stdout instead of the default stderr
+	// Can be any io.Writer, see below for File example
+	// log.SetOutput(os.Stdout)
+
+	// Only log the warning severity or above.
+	log.SetLevel(log.DebugLevel)
+}