mirror of
https://github.com/ergochat/ergo.git
synced 2024-12-22 18:52:41 +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:
parent
89bdb8a321
commit
6a69a65860
@ -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'
|
||||||
|
@ -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
|
||||||
|
@ -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
56
irc/websocket.go
Normal 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)
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user