vail/main.go

60 lines
1.0 KiB
Go
Raw Normal View History

2020-04-09 23:09:33 -06:00
package main
import (
2020-05-03 09:37:50 -06:00
"os"
2020-04-09 23:09:33 -06:00
"log"
"net/http"
"golang.org/x/net/websocket"
)
var book Book
2020-04-12 20:40:52 -06:00
2020-04-09 23:09:33 -06:00
type Client struct {
repeaterName string
2020-04-09 23:09:33 -06:00
}
func (c Client) Handle(ws *websocket.Conn) {
ws.MaxPayloadBytes = 500
book.Join(c.repeaterName, ws)
defer book.Part(c.repeaterName, ws)
2020-04-09 23:09:33 -06:00
for {
buf := make([]byte, ws.MaxPayloadBytes)
if n, err := ws.Read(buf); err != nil {
break
} else {
buf = buf[:n]
}
book.Send(c.repeaterName, buf)
2020-04-09 23:09:33 -06:00
}
}
func ChatHandler(w http.ResponseWriter, r *http.Request) {
c := Client {
repeaterName: r.FormValue("repeater"),
2020-04-09 23:09:33 -06:00
}
// This API is confusing as hell.
// I suspect there's a better way to do this.
websocket.Handler(c.Handle).ServeHTTP(w, r)
2020-04-09 23:09:33 -06:00
}
func main() {
book = NewBook()
http.Handle("/chat", http.HandlerFunc(ChatHandler))
2020-04-09 23:09:33 -06:00
http.Handle("/", http.FileServer(http.Dir("static")))
go book.Run()
2020-05-17 12:52:25 -06:00
2020-05-03 09:37:50 -06:00
port := os.Getenv("PORT")
if port == "" {
port = "8080"
}
log.Println("Listening on port", port)
err := http.ListenAndServe(":" + port, nil)
2020-04-09 23:09:33 -06:00
if err != nil {
log.Fatal(err.Error())
}
}