3
0
mirror of https://github.com/ergochat/ergo.git synced 2024-12-22 10:42:52 +01:00

WebSocket layer to be able to connect "web" client

Currently working with a mini irc js implem, the flow:

* PASS
* NICK
* USER
* JOIN
* PRIVMSG

works and the ping/pong timeout keep the communication open.
This commit is contained in:
Niels Freier 2015-05-04 09:47:26 +04:00
parent 89bdb8a321
commit 6a69a65860
4 changed files with 95 additions and 1 deletions

View File

@ -1,8 +1,9 @@
[server]
name = "irc.example.com" ; required, usually a hostname
name = "localhost" ; required, usually a hostname
database = "ergonomadic.db" ; path relative to this file
listen = "localhost:6667" ; see `net.Listen` for examples
listen = "[::1]:6667" ; multiple `listen`s are allowed.
wslisten = ":8080" ; websocket listen
log = "debug" ; error, warn, info, debug
motd = "motd.txt" ; path relative to this file
password = "JDJhJDA0JHJzVFFlNXdOUXNhLmtkSGRUQVVEVHVYWXRKUmdNQ3FKVTRrczRSMTlSWGRPZHRSMVRzQmtt" ; 'test'

View File

@ -23,6 +23,7 @@ type Config struct {
PassConfig
Database string
Listen []string
Wslisten string
Log string
MOTD string
Name string

View File

@ -6,6 +6,7 @@ import (
"fmt"
"log"
"net"
"net/http"
"os"
"os/signal"
"strings"
@ -72,6 +73,10 @@ func NewServer(config *Config) *Server {
server.listen(addr)
}
if config.Server.Wslisten != "" {
server.wslisten(config.Server.Wslisten)
}
signal.Notify(server.signals, SERVER_SIGNALS...)
return server
@ -203,6 +208,37 @@ func (s *Server) listen(addr string) {
}()
}
//
// websocket listen goroutine
//
func (s *Server) wslisten(addr string) {
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
Log.error.Printf("%s method not allowed", s)
return
}
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
Log.error.Printf("%s websocket upgrade error: %s", s, err)
return
}
wsc := WSContainer{
conn: ws,
}
s.newConns <- wsc
})
go func() {
Log.info.Printf("%s listening on %s", s, addr)
err := http.ListenAndServe(addr, nil)
if err != nil {
Log.error.Printf("%s listenAndServe error: %s", s, err)
}
}()
}
//
// server functionality
//

56
irc/websocket.go Normal file
View File

@ -0,0 +1,56 @@
package irc
import (
"github.com/gorilla/websocket"
"net"
"net/http"
"time"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
CheckOrigin: func(r *http.Request) bool { return true },
}
type WSContainer struct {
conn *websocket.Conn
}
func (this WSContainer) Close() error {
return this.conn.Close()
}
func (this WSContainer) LocalAddr() net.Addr {
return this.conn.LocalAddr()
}
func (this WSContainer) RemoteAddr() net.Addr {
return this.conn.RemoteAddr()
}
func (this WSContainer) Read(msg []byte) (int, error) {
_, tmp, err := this.conn.ReadMessage()
str := (string)(tmp)
n := copy(msg, ([]byte)(str+CRLF+CRLF))
return n, err
}
func (this WSContainer) Write(msg []byte) (int, error) {
err := this.conn.WriteMessage(1, msg)
return len(msg), err
}
func (this WSContainer) SetDeadline(t time.Time) error {
err := this.conn.SetWriteDeadline(t)
err = this.conn.SetReadDeadline(t)
return err
}
func (this WSContainer) SetReadDeadline(t time.Time) error {
return this.conn.SetReadDeadline(t)
}
func (this WSContainer) SetWriteDeadline(t time.Time) error {
return this.conn.SetWriteDeadline(t)
}