3
0
mirror of https://github.com/ergochat/ergo.git synced 2024-11-10 22:19:31 +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] [server]
name = "irc.example.com" ; required, usually a hostname name = "localhost" ; required, usually a hostname
database = "ergonomadic.db" ; path relative to this file database = "ergonomadic.db" ; path relative to this file
listen = "localhost:6667" ; see `net.Listen` for examples listen = "localhost:6667" ; see `net.Listen` for examples
listen = "[::1]:6667" ; multiple `listen`s are allowed. listen = "[::1]:6667" ; multiple `listen`s are allowed.
wslisten = ":8080" ; websocket listen
log = "debug" ; error, warn, info, debug log = "debug" ; error, warn, info, debug
motd = "motd.txt" ; path relative to this file motd = "motd.txt" ; path relative to this file
password = "JDJhJDA0JHJzVFFlNXdOUXNhLmtkSGRUQVVEVHVYWXRKUmdNQ3FKVTRrczRSMTlSWGRPZHRSMVRzQmtt" ; 'test' password = "JDJhJDA0JHJzVFFlNXdOUXNhLmtkSGRUQVVEVHVYWXRKUmdNQ3FKVTRrczRSMTlSWGRPZHRSMVRzQmtt" ; 'test'

View File

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

View File

@ -6,6 +6,7 @@ import (
"fmt" "fmt"
"log" "log"
"net" "net"
"net/http"
"os" "os"
"os/signal" "os/signal"
"strings" "strings"
@ -72,6 +73,10 @@ func NewServer(config *Config) *Server {
server.listen(addr) server.listen(addr)
} }
if config.Server.Wslisten != "" {
server.wslisten(config.Server.Wslisten)
}
signal.Notify(server.signals, SERVER_SIGNALS...) signal.Notify(server.signals, SERVER_SIGNALS...)
return server 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 // 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)
}