zirconium-go/cmd/zr/main.go

37 lines
828 B
Go

package main
import (
"log"
"net/http"
"github.com/ChronosX88/zirconium/internal"
"github.com/gorilla/mux"
"github.com/gorilla/websocket"
)
var connectionHandler = internal.NewConnectionHandler()
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
}
func main() {
router := mux.NewRouter()
router.HandleFunc("/", func(response http.ResponseWriter, request *http.Request) {
response.Write([]byte("Zirconium server is up and running!"))
}).Methods("GET")
router.HandleFunc("/ws", wsHandler)
log.Println("Zirconium successfully started!")
log.Fatal(http.ListenAndServe(":8844", router))
}
func wsHandler(w http.ResponseWriter, r *http.Request) {
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Fatal(err)
}
connectionHandler.HandleNewConnection(ws)
}