Add cobra, move the server code into its own package

server/server.go

@@ -0,0 +1,124 @@
+package server
+
+import (
+	"log"
+	"net/http"
+	"strconv"
+	"strings"
+	"sync"
+
+	"github.com/khlieng/name_pending/Godeps/_workspace/src/github.com/julienschmidt/httprouter"
+	"github.com/khlieng/name_pending/Godeps/_workspace/src/golang.org/x/net/websocket"
+
+	"github.com/khlieng/name_pending/storage"
+)
+
+var (
+	channelStore *storage.ChannelStore
+	sessions     map[string]*Session
+	sessionLock  sync.Mutex
+	fs           http.Handler
+	files        []File
+)
+
+type File struct {
+	Path        string
+	ContentType string
+}
+
+func Run(port int, development bool) {
+	defer storage.Cleanup()
+
+	channelStore = storage.NewChannelStore()
+	sessions = make(map[string]*Session)
+	fs = http.FileServer(BindataFileSystem{})
+
+	files = []File{
+		File{"/bundle.js", "text/javascript"},
+		File{"/css/style.css", "text/css"},
+		File{"/css/fontello.css", "text/css"},
+		File{"/font/fontello.eot", "application/vnd.ms-fontobject"},
+		File{"/font/fontello.svg", "image/svg+xml"},
+		File{"/font/fontello.ttf", "application/x-font-ttf"},
+		File{"/font/fontello.woff", "application/font-woff"},
+	}
+
+	if !development {
+		reconnect()
+	}
+
+	router := httprouter.New()
+
+	router.Handler("GET", "/ws", websocket.Handler(handleWS))
+	router.NotFound = serveFiles
+
+	log.Println("Listening on port", port)
+	log.Fatal(http.ListenAndServe(":"+strconv.Itoa(port), router))
+}
+
+func reconnect() {
+	for _, user := range storage.LoadUsers() {
+		session := NewSession()
+		session.user = user
+		sessions[user.UUID] = session
+		go session.write()
+
+		channels := user.GetChannels()
+
+		for _, server := range user.GetServers() {
+			irc := NewIRC(server.Nick, server.Username)
+			irc.TLS = server.TLS
+			irc.Password = server.Password
+			irc.Realname = server.Realname
+
+			go func() {
+				err := irc.Connect(server.Address)
+				if err != nil {
+					log.Println(err)
+				} else {
+					session.setIRC(irc.Host, irc)
+
+					go handleMessages(irc, session)
+
+					var joining []string
+					for _, channel := range channels {
+						if channel.Server == server.Address {
+							joining = append(joining, channel.Name)
+						}
+					}
+					irc.Join(joining...)
+				}
+			}()
+		}
+	}
+}
+
+func serveFiles(w http.ResponseWriter, r *http.Request) {
+	var ext string
+
+	if strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
+		w.Header().Set("Content-Encoding", "gzip")
+		ext = ".gz"
+	}
+
+	if r.URL.Path == "/" {
+		w.Header().Set("Content-Type", "text/html")
+		r.URL.Path = "/index.html" + ext
+		fs.ServeHTTP(w, r)
+		return
+	}
+
+	for _, file := range files {
+		if strings.HasSuffix(r.URL.Path, file.Path) {
+			w.Header().Set("Content-Type", file.ContentType)
+			r.URL.Path = file.Path + ext
+			fs.ServeHTTP(w, r)
+			return
+		}
+	}
+
+	w.Header().Set("Content-Type", "text/html")
+	r.URL.Path = "/index.html" + ext
+
+	fs.ServeHTTP(w, r)
+}