mirror of
https://github.com/supleed2/go-chat.git
synced 2024-09-19 13:26:17 +00:00
Initial Commit
This commit is contained in:
commit
6a94af7daf
SSH key fingerprint:
SHA256:CkbNRs0yVzXEiUp2zd0PSxsfRUMFF9bLlKXtE1xEbKM
24
.gitignore
vendored
Normal file
24
.gitignore
vendored
Normal file
|
|
@ -0,0 +1,24 @@
|
|||
# If you prefer the allow list template instead of the deny list, see community template:
|
||||
# https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
|
||||
#
|
||||
# Binaries for programs and plugins
|
||||
*.exe
|
||||
*.exe~
|
||||
*.dll
|
||||
*.so
|
||||
*.dylib
|
||||
|
||||
# Test binary, built with `go test -c`
|
||||
*.test
|
||||
|
||||
# Output of the go coverage tool, specifically when used with LiteIDE
|
||||
*.out
|
||||
|
||||
# Dependency directories (remove the comment below to include it)
|
||||
vendor/
|
||||
|
||||
# Go workspace file
|
||||
go.work
|
||||
|
||||
# Project specific files
|
||||
nicks.json
|
||||
334
client/main.go
Normal file
334
client/main.go
Normal file
|
|
@ -0,0 +1,334 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
c "go-chat/common"
|
||||
|
||||
"github.com/alexflint/go-arg"
|
||||
"github.com/charmbracelet/bubbles/help"
|
||||
"github.com/charmbracelet/bubbles/key"
|
||||
"github.com/charmbracelet/bubbles/textinput"
|
||||
"github.com/charmbracelet/bubbles/viewport"
|
||||
tea "github.com/charmbracelet/bubbletea"
|
||||
"github.com/charmbracelet/lipgloss"
|
||||
ws "nhooyr.io/websocket"
|
||||
"nhooyr.io/websocket/wsjson"
|
||||
)
|
||||
|
||||
const manText string = `The current available commands are:
|
||||
man
|
||||
prints this message
|
||||
mv <string>
|
||||
set your nick
|
||||
ls
|
||||
get the available rooms
|
||||
cd <string>
|
||||
connect to a room
|
||||
who
|
||||
list users in the current room
|
||||
moo
|
||||
:)`
|
||||
|
||||
const mooText string = `
|
||||
(__)
|
||||
(oo)
|
||||
/------\/
|
||||
/ | ||
|
||||
* /\---/\
|
||||
~~ ~~
|
||||
..."Have you mooed today?"...`
|
||||
|
||||
type showTim int
|
||||
|
||||
const (
|
||||
off showTim = iota
|
||||
short
|
||||
full
|
||||
)
|
||||
|
||||
type model struct {
|
||||
history viewport.Model
|
||||
msgs []c.SMsg
|
||||
showTim showTim
|
||||
tz time.Location
|
||||
input textinput.Model
|
||||
idStyle lipgloss.Style
|
||||
pStyle lipgloss.Style
|
||||
help help.Model
|
||||
recvCh chan c.SMsg
|
||||
sendCh chan c.CMsg
|
||||
conn *ws.Conn
|
||||
exitCh chan exit
|
||||
}
|
||||
|
||||
type args struct {
|
||||
Address string `arg:"positional" default:"localhost:8000" help:"address to connect to, without ws://" placeholder:"HOST[:PORT]"`
|
||||
Timestamps showTim `arg:"-t" default:"off" help:"display timestamps of messages, ctrl+t to cycle after startup [off, short, full]" placeholder:"CHOICE"`
|
||||
Nick *string `arg:"-n" help:"attempt to automatically set nick after connecting"`
|
||||
Password *string `arg:"-p" help:"password, if required"`
|
||||
}
|
||||
|
||||
func main() {
|
||||
ctx := context.Background()
|
||||
|
||||
var a args
|
||||
arg.MustParse(&a)
|
||||
|
||||
conn, _, err := ws.Dial(ctx, "ws://"+a.Address, nil)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
defer conn.Close(ws.StatusNormalClosure, "")
|
||||
|
||||
local, err := time.LoadLocation("Local")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
p := tea.NewProgram(initModel(ctx, conn, a, *local), tea.WithAltScreen(), tea.WithMouseCellMotion())
|
||||
if _, err := p.Run(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func initModel(ctx context.Context, conn *ws.Conn, a args, tz time.Location) model {
|
||||
discCtx, disc := context.WithCancel(ctx)
|
||||
exitCh := make(chan exit)
|
||||
|
||||
recvCh := make(chan c.SMsg)
|
||||
go func() {
|
||||
smsg := c.SMsg{}
|
||||
for {
|
||||
err := wsjson.Read(ctx, conn, &smsg)
|
||||
if err != nil {
|
||||
if ws.CloseStatus(err) != ws.StatusNormalClosure {
|
||||
log.Println(err)
|
||||
}
|
||||
disc()
|
||||
return
|
||||
}
|
||||
recvCh <- smsg
|
||||
}
|
||||
}()
|
||||
|
||||
sendCh := make(chan c.CMsg)
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case cmsg := <-sendCh:
|
||||
err := wsjson.Write(ctx, conn, cmsg)
|
||||
if err != nil {
|
||||
recvCh <- c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("wsjson error when sending message: %v", err)}
|
||||
}
|
||||
case <-discCtx.Done():
|
||||
exitCh <- exit{}
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
ta := textinput.New()
|
||||
ta.Placeholder = "Send a message (or a command with /)"
|
||||
ta.Focus()
|
||||
ta.CharLimit = 128
|
||||
ta.Width = 60
|
||||
|
||||
vp := viewport.New(60, 5)
|
||||
vp.KeyMap = viewport.KeyMap{
|
||||
PageDown: key.NewBinding(
|
||||
key.WithKeys("pgdown"),
|
||||
key.WithHelp("pgdn", "page down"),
|
||||
),
|
||||
PageUp: key.NewBinding(
|
||||
key.WithKeys("pgup"),
|
||||
key.WithHelp("pgup", "page up"),
|
||||
),
|
||||
Up: key.NewBinding(
|
||||
key.WithKeys("up"),
|
||||
key.WithHelp("↑", "up"),
|
||||
),
|
||||
Down: key.NewBinding(
|
||||
key.WithKeys("down"),
|
||||
key.WithHelp("↓", "down"),
|
||||
),
|
||||
}
|
||||
|
||||
if a.Nick != nil {
|
||||
login := *a.Nick
|
||||
if a.Password != nil {
|
||||
login += ":" + *a.Password
|
||||
}
|
||||
sendCh <- c.CMsg{Typ: c.Mv, Msg: login}
|
||||
}
|
||||
|
||||
messages := []c.SMsg{{Tim: time.Now(), Id: "system", Msg: "Welcome to the chat room! Press Enter to send, /man for more info :)"}}
|
||||
|
||||
return model{
|
||||
input: ta,
|
||||
msgs: messages,
|
||||
showTim: a.Timestamps,
|
||||
tz: tz,
|
||||
history: vp,
|
||||
idStyle: lipgloss.NewStyle().Width(30),
|
||||
pStyle: lipgloss.NewStyle().Bold(true),
|
||||
help: help.New(),
|
||||
recvCh: recvCh,
|
||||
sendCh: sendCh,
|
||||
conn: conn,
|
||||
exitCh: exitCh,
|
||||
}
|
||||
}
|
||||
|
||||
func (m model) Init() tea.Cmd {
|
||||
return tea.Batch(
|
||||
tea.SetWindowTitle("go-chat by 8bit"),
|
||||
textinput.Blink,
|
||||
getNextSMsg(m.recvCh),
|
||||
getExitMsg(m.exitCh),
|
||||
)
|
||||
}
|
||||
|
||||
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
var tiCmd, vpCmd, clCmd, smCmd tea.Cmd
|
||||
m.input, tiCmd = m.input.Update(msg)
|
||||
m.history, vpCmd = m.history.Update(msg)
|
||||
|
||||
switch msg := msg.(type) {
|
||||
case exit:
|
||||
return m, tea.Quit
|
||||
case c.SMsg:
|
||||
m.msgs = append(m.msgs, msg)
|
||||
m.history.SetContent(m.viewMessages())
|
||||
m.history.GotoBottom()
|
||||
smCmd = getNextSMsg(m.recvCh)
|
||||
case tea.KeyMsg:
|
||||
switch msg.Type {
|
||||
case tea.KeyCtrlC, tea.KeyEsc:
|
||||
return m, tea.Quit
|
||||
case tea.KeyCtrlT:
|
||||
m.showTim = (m.showTim + 1) % 3
|
||||
m.history.SetContent(m.viewMessages())
|
||||
case tea.KeyEnter:
|
||||
text := strings.TrimSpace(m.input.Value())
|
||||
if text, ok := strings.CutPrefix(text, "/"); ok {
|
||||
if text == "man" {
|
||||
m.recvCh <- c.SMsg{Tim: time.Now(), Id: "system", Msg: manText}
|
||||
} else if text, ok := strings.CutPrefix(text, "mv "); ok {
|
||||
m.sendCh <- c.CMsg{Typ: c.Mv, Msg: text}
|
||||
} else if text == "ls" {
|
||||
m.sendCh <- c.CMsg{Typ: c.Ls, Msg: ""}
|
||||
} else if text, ok := strings.CutPrefix(text, "cd "); ok {
|
||||
m.sendCh <- c.CMsg{Typ: c.Cd, Msg: text}
|
||||
} else if text == "who" {
|
||||
m.sendCh <- c.CMsg{Typ: c.Who, Msg: ""}
|
||||
} else if text, ok := strings.CutPrefix(text, "sudo "); ok {
|
||||
m.sendCh <- c.CMsg{Typ: c.Sudo, Msg: text}
|
||||
} else if text == "moo" {
|
||||
m.recvCh <- c.SMsg{Tim: time.Now(), Id: "cow", Msg: mooText}
|
||||
} else {
|
||||
m.recvCh <- c.SMsg{Tim: time.Now(), Id: "system", Msg: "Unrecognised command, use /man for more info"}
|
||||
}
|
||||
} else if text != "" {
|
||||
m.sendCh <- c.CMsg{Typ: c.Echo, Msg: text}
|
||||
}
|
||||
m.input.Reset()
|
||||
}
|
||||
case tea.WindowSizeMsg:
|
||||
clCmd = tea.Sequence(tea.ExitAltScreen, tea.EnterAltScreen)
|
||||
m.history.Height = msg.Height - 2
|
||||
m.history.Width = msg.Width
|
||||
m.history.GotoBottom()
|
||||
m.input.Width = msg.Width - 3
|
||||
m.idStyle.Width(msg.Width)
|
||||
m.help.Width = msg.Width - 1
|
||||
m.history.SetContent(m.viewMessages())
|
||||
}
|
||||
|
||||
return m, tea.Batch(tiCmd, vpCmd, clCmd, smCmd)
|
||||
}
|
||||
|
||||
func (m model) View() string {
|
||||
return fmt.Sprintf(
|
||||
"%s\n%s\n%s",
|
||||
m.history.View(),
|
||||
m.input.View(),
|
||||
m.help.View(m),
|
||||
)
|
||||
}
|
||||
|
||||
func (m model) ShortHelp() []key.Binding {
|
||||
return []key.Binding{
|
||||
m.history.KeyMap.PageDown,
|
||||
m.history.KeyMap.PageUp,
|
||||
m.history.KeyMap.Down,
|
||||
m.history.KeyMap.Up,
|
||||
key.NewBinding(
|
||||
key.WithKeys("ctrl+t"),
|
||||
key.WithHelp("ctrl+t", "toggle timestamps"),
|
||||
),
|
||||
}
|
||||
}
|
||||
|
||||
func (m model) FullHelp() [][]key.Binding {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m model) viewMessages() string {
|
||||
s := ""
|
||||
for i := range m.msgs {
|
||||
prefix := ""
|
||||
if m.showTim == short {
|
||||
prefix += m.msgs[i].Tim.In(&m.tz).Format(time.TimeOnly) + " "
|
||||
} else if m.showTim == full {
|
||||
prefix += m.msgs[i].Tim.In(&m.tz).Format(time.DateTime) + " "
|
||||
}
|
||||
if m.msgs[i].Id == "system" {
|
||||
prefix += m.pStyle.Foreground(lipgloss.Color("201")).Render("system:")
|
||||
} else {
|
||||
prefix += m.pStyle.Foreground(lipgloss.Color(prefixColor(m.msgs[i].Id))).Render(m.msgs[i].Id + ":")
|
||||
}
|
||||
s += m.idStyle.SetString(prefix).Render(m.msgs[i].Msg) + "\n"
|
||||
}
|
||||
return s[:len(s)-1]
|
||||
}
|
||||
|
||||
func prefixColor(s string) string {
|
||||
if len(s) == 0 {
|
||||
s = "missing"
|
||||
}
|
||||
return fmt.Sprint(uint(s[0]+s[len(s)-1]) % 8)
|
||||
}
|
||||
|
||||
func getNextSMsg(c <-chan c.SMsg) tea.Cmd {
|
||||
return func() tea.Msg {
|
||||
return <-c
|
||||
}
|
||||
}
|
||||
|
||||
type exit struct{}
|
||||
|
||||
func getExitMsg(c <-chan exit) tea.Cmd {
|
||||
return func() tea.Msg {
|
||||
return <-c
|
||||
}
|
||||
}
|
||||
|
||||
func (st *showTim) UnmarshalText(b []byte) error {
|
||||
s := string(b)
|
||||
switch s {
|
||||
case "off":
|
||||
*st = off
|
||||
case "short":
|
||||
*st = short
|
||||
case "full":
|
||||
*st = full
|
||||
default:
|
||||
return fmt.Errorf("invalid choice: %s [off, short, full]", s)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
25
common/types.go
Normal file
25
common/types.go
Normal file
|
|
@ -0,0 +1,25 @@
|
|||
package common
|
||||
|
||||
import "time"
|
||||
|
||||
type SMsg struct {
|
||||
Tim time.Time
|
||||
Id string
|
||||
Msg string
|
||||
}
|
||||
|
||||
type CMsgT int
|
||||
|
||||
const (
|
||||
Sudo CMsgT = iota
|
||||
Echo
|
||||
Mv
|
||||
Ls
|
||||
Cd
|
||||
Who
|
||||
)
|
||||
|
||||
type CMsg struct {
|
||||
Typ CMsgT
|
||||
Msg string
|
||||
}
|
||||
32
go.mod
Normal file
32
go.mod
Normal file
|
|
@ -0,0 +1,32 @@
|
|||
module go-chat
|
||||
|
||||
go 1.21.5
|
||||
|
||||
require (
|
||||
github.com/alexflint/go-arg v1.4.3
|
||||
github.com/charmbracelet/bubbles v0.17.1
|
||||
github.com/charmbracelet/bubbletea v0.25.0
|
||||
github.com/charmbracelet/lipgloss v0.9.1
|
||||
nhooyr.io/websocket v1.8.10
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/alexflint/go-scalar v1.2.0 // indirect
|
||||
github.com/atotto/clipboard v0.1.4 // indirect
|
||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
|
||||
github.com/containerd/console v1.0.4-0.20230313162750-1ae8d489ac81 // indirect
|
||||
github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/mattn/go-localereader v0.0.1 // indirect
|
||||
github.com/mattn/go-runewidth v0.0.15 // indirect
|
||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
|
||||
github.com/muesli/cancelreader v0.2.2 // indirect
|
||||
github.com/muesli/reflow v0.3.0 // indirect
|
||||
github.com/muesli/termenv v0.15.2 // indirect
|
||||
github.com/rivo/uniseg v0.4.4 // indirect
|
||||
github.com/stretchr/testify v1.8.4 // indirect
|
||||
golang.org/x/sync v0.6.0 // indirect
|
||||
golang.org/x/sys v0.16.0 // indirect
|
||||
golang.org/x/term v0.16.0 // indirect
|
||||
golang.org/x/text v0.14.0 // indirect
|
||||
)
|
||||
64
go.sum
Normal file
64
go.sum
Normal file
|
|
@ -0,0 +1,64 @@
|
|||
github.com/alexflint/go-arg v1.4.3 h1:9rwwEBpMXfKQKceuZfYcwuc/7YY7tWJbFsgG5cAU/uo=
|
||||
github.com/alexflint/go-arg v1.4.3/go.mod h1:3PZ/wp/8HuqRZMUUgu7I+e1qcpUbvmS258mRXkFH4IA=
|
||||
github.com/alexflint/go-scalar v1.1.0/go.mod h1:LoFvNMqS1CPrMVltza4LvnGKhaSpc3oyLEBUZVhhS2o=
|
||||
github.com/alexflint/go-scalar v1.2.0 h1:WR7JPKkeNpnYIOfHRa7ivM21aWAdHD0gEWHCx+WQBRw=
|
||||
github.com/alexflint/go-scalar v1.2.0/go.mod h1:LoFvNMqS1CPrMVltza4LvnGKhaSpc3oyLEBUZVhhS2o=
|
||||
github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4=
|
||||
github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI=
|
||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
|
||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
|
||||
github.com/charmbracelet/bubbles v0.17.1 h1:0SIyjOnkrsfDo88YvPgAWvZMwXe26TP6drRvmkjyUu4=
|
||||
github.com/charmbracelet/bubbles v0.17.1/go.mod h1:9HxZWlkCqz2PRwsCbYl7a3KXvGzFaDHpYbSYMJ+nE3o=
|
||||
github.com/charmbracelet/bubbletea v0.25.0 h1:bAfwk7jRz7FKFl9RzlIULPkStffg5k6pNt5dywy4TcM=
|
||||
github.com/charmbracelet/bubbletea v0.25.0/go.mod h1:EN3QDR1T5ZdWmdfDzYcqOCAps45+QIJbLOBxmVNWNNg=
|
||||
github.com/charmbracelet/lipgloss v0.9.1 h1:PNyd3jvaJbg4jRHKWXnCj1akQm4rh8dbEzN1p/u1KWg=
|
||||
github.com/charmbracelet/lipgloss v0.9.1/go.mod h1:1mPmG4cxScwUQALAAnacHaigiiHB9Pmr+v1VEawJl6I=
|
||||
github.com/containerd/console v1.0.4-0.20230313162750-1ae8d489ac81 h1:q2hJAaP1k2wIvVRd/hEHD7lacgqrCPS+k8g1MndzfWY=
|
||||
github.com/containerd/console v1.0.4-0.20230313162750-1ae8d489ac81/go.mod h1:YynlIjWYF8myEu6sdkwKIvGQq+cOckRm6So2avqoYAk=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=
|
||||
github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2JC/oIi4=
|
||||
github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88=
|
||||
github.com/mattn/go-runewidth v0.0.12/go.mod h1:RAqKPSqVFrSLVXbA8x7dzmKdmGzieGRCM46jaSJTDAk=
|
||||
github.com/mattn/go-runewidth v0.0.15 h1:UNAjwbU9l54TA3KzvqLGxwWjHmMgBUVhBiTjelZgg3U=
|
||||
github.com/mattn/go-runewidth v0.0.15/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
|
||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=
|
||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo=
|
||||
github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
|
||||
github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo=
|
||||
github.com/muesli/reflow v0.3.0 h1:IFsN6K9NfGtjeggFP+68I4chLZV2yIKsXJFNZ+eWh6s=
|
||||
github.com/muesli/reflow v0.3.0/go.mod h1:pbwTDkVPibjO2kyvBQRBxTWEEGDGq0FlB1BIKtnHY/8=
|
||||
github.com/muesli/termenv v0.15.2 h1:GohcuySI0QmI3wN8Ok9PtKGkgkFIk7y6Vpb5PvrY+Wo=
|
||||
github.com/muesli/termenv v0.15.2/go.mod h1:Epx+iuz8sNs7mNKhxzH4fWXGNpZwUaJKRS1noLXviQ8=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/rivo/uniseg v0.1.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
|
||||
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
|
||||
github.com/rivo/uniseg v0.4.4 h1:8TfxU8dW6PdqD27gjM8MVNuicgxIjxpm4K7x4jp8sis=
|
||||
github.com/rivo/uniseg v0.4.4/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
|
||||
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||
golang.org/x/sync v0.6.0 h1:5BMeUDZ7vkXGfEr1x9B4bRcTH4lpkTkpdh0T/J+qjbQ=
|
||||
golang.org/x/sync v0.6.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
|
||||
golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.16.0 h1:xWw16ngr6ZMtmxDyKyIgsE93KNKz5HKmMa3b8ALHidU=
|
||||
golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/term v0.16.0 h1:m+B6fahuftsE9qjo0VWp2FW0mB3MTJvR0BaMQrq0pmE=
|
||||
golang.org/x/term v0.16.0/go.mod h1:yn7UURbUtPyrVJPGPq404EukNFxcm/foM+bV/bfcDsY=
|
||||
golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
|
||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
nhooyr.io/websocket v1.8.10 h1:mv4p+MnGrLDcPlBoWsvPP7XCzTYMXP9F9eIGoKbgx7Q=
|
||||
nhooyr.io/websocket v1.8.10/go.mod h1:rN9OFWIUwuxg4fR5tELlYC04bXYowCP9GX47ivo2l+c=
|
||||
359
server/main.go
Normal file
359
server/main.go
Normal file
|
|
@ -0,0 +1,359 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
c "go-chat/common"
|
||||
|
||||
"github.com/alexflint/go-arg"
|
||||
ws "nhooyr.io/websocket"
|
||||
"nhooyr.io/websocket/wsjson"
|
||||
)
|
||||
|
||||
type user struct {
|
||||
room string
|
||||
nick string
|
||||
}
|
||||
|
||||
type conns struct {
|
||||
sm sync.Mutex
|
||||
cm map[*ws.Conn]user
|
||||
}
|
||||
|
||||
type server struct {
|
||||
admin string
|
||||
logFn func(string, ...interface{})
|
||||
conns *conns
|
||||
rooms map[string]struct{}
|
||||
rhist map[string][]c.SMsg
|
||||
rhlen int
|
||||
nickm map[string]string
|
||||
}
|
||||
|
||||
func main() {
|
||||
log := log.New(os.Stderr, "ws server 🚀 ", log.LstdFlags|log.Lshortfile|log.Lmsgprefix)
|
||||
|
||||
var args struct {
|
||||
Admin string `arg:"-a" default:"8bit" help:"admin user nick, allows access to /sudo" placeholder:"NICK"`
|
||||
HistLen uint `arg:"-h" default:"10" help:"set message history size" placeholder:"N"`
|
||||
Port uint `arg:"positional" default:"0" help:"port to listen on, random available port if not set"`
|
||||
NickMap *string `arg:"-n" help:"path to nick:pass JSON file" placeholder:"FILE"`
|
||||
}
|
||||
arg.MustParse(&args)
|
||||
|
||||
nickMap, err := loadNickMap(args.NickMap)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
err = run("localhost:"+fmt.Sprint(args.Port), nickMap, args.Admin, int(args.HistLen), log)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func run(addr string, nickMap map[string]string, admin string, rhlen int, log *log.Logger) error {
|
||||
listener, err := net.Listen("tcp4", addr)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
log.Printf("listening on ws://%v", listener.Addr())
|
||||
|
||||
server := &http.Server{
|
||||
Handler: server{
|
||||
admin: admin,
|
||||
logFn: log.Printf,
|
||||
conns: &conns{cm: make(map[*ws.Conn]user)},
|
||||
rooms: map[string]struct{}{"general": {}, "test1": {}, "test2": {}},
|
||||
rhist: make(map[string][]c.SMsg),
|
||||
rhlen: rhlen,
|
||||
nickm: nickMap,
|
||||
},
|
||||
ReadTimeout: 10 * time.Second,
|
||||
WriteTimeout: 10 * time.Second,
|
||||
}
|
||||
|
||||
errch := make(chan error, 1)
|
||||
go func() {
|
||||
errch <- server.Serve(listener)
|
||||
}()
|
||||
|
||||
signals := make(chan os.Signal, 1)
|
||||
signal.Notify(signals, os.Interrupt)
|
||||
select {
|
||||
case err := <-errch:
|
||||
log.Printf("failed to serve: %v", err)
|
||||
case signal := <-signals:
|
||||
log.Printf("quitting: %v", signal)
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
defer cancel()
|
||||
|
||||
return server.Shutdown(ctx)
|
||||
}
|
||||
|
||||
func (s server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
conn, err := ws.Accept(w, r, nil)
|
||||
if err != nil {
|
||||
s.logFn("%v", err)
|
||||
return
|
||||
}
|
||||
defer conn.CloseNow()
|
||||
|
||||
if conn.Subprotocol() != "" {
|
||||
return
|
||||
}
|
||||
|
||||
port := strings.Split(r.RemoteAddr, ":")[1]
|
||||
s.conns.sm.Lock()
|
||||
s.conns.cm[conn] = user{room: "general", nick: port}
|
||||
s.conns.sm.Unlock()
|
||||
defer func() {
|
||||
s.conns.sm.Lock()
|
||||
delete(s.conns.cm, conn)
|
||||
s.logFn("Remaining connections: %v", len(s.conns.cm))
|
||||
s.conns.sm.Unlock()
|
||||
}()
|
||||
|
||||
s.logFn("connected: %v", r.RemoteAddr)
|
||||
for i := range s.rhist["general"] {
|
||||
wsjson.Write(ctx, conn, s.rhist["general"][i])
|
||||
}
|
||||
cmsg := c.CMsg{}
|
||||
smsg := c.SMsg{Id: port}
|
||||
for {
|
||||
err := func(ctx context.Context, conn *ws.Conn) error {
|
||||
err := wsjson.Read(ctx, conn, &cmsg)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
switch cmsg.Typ {
|
||||
case c.Sudo:
|
||||
s.logFn("(%v) sudo: %v", smsg.Id, cmsg.Msg)
|
||||
if smsg.Id == s.admin {
|
||||
cmd := strings.Split(cmsg.Msg, " ")
|
||||
if len(cmd) == 2 {
|
||||
if cmd[0] == "mk" {
|
||||
if _, ok := s.rooms[cmd[1]]; ok {
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("Room exists: %v", cmd)})
|
||||
} else {
|
||||
s.rooms[cmd[1]] = struct{}{}
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("Created room: %v", cmd)})
|
||||
}
|
||||
} else if cmd[0] == "rm" {
|
||||
if _, ok := s.rooms[cmd[1]]; ok && cmd[1] != "general" {
|
||||
delete(s.rooms, cmd[1])
|
||||
s.rhist[cmd[1]] = []c.SMsg{}
|
||||
tim := time.Now()
|
||||
s.conns.sm.Lock()
|
||||
for cn, r := range s.conns.cm {
|
||||
if r.room == cmd[1] {
|
||||
r.room = "general"
|
||||
s.conns.cm[cn] = r
|
||||
wsjson.Write(ctx, cn, c.SMsg{Tim: tim, Id: "system", Msg: "room deleted, reconnected to general"})
|
||||
}
|
||||
}
|
||||
s.conns.sm.Unlock()
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("Deleted room: %v", cmd)})
|
||||
} else {
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("Room does not exist: %v", cmd)})
|
||||
}
|
||||
} else if cmd[0] == "yeet" {
|
||||
found := false
|
||||
s.conns.sm.Lock()
|
||||
for c, r := range s.conns.cm {
|
||||
if r.nick == cmd[1] {
|
||||
c.Close(ws.StatusNormalClosure, "Kicked")
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
s.conns.sm.Unlock()
|
||||
if found {
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("Yeet: %v", cmd[1])})
|
||||
} else {
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("Not found: %v", cmd[1])})
|
||||
}
|
||||
} else {
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("Invalid command: %v", cmd)})
|
||||
}
|
||||
} else if cmd[0] == "wc" {
|
||||
s.conns.sm.Lock()
|
||||
wc := len(s.conns.cm)
|
||||
s.conns.sm.Unlock()
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("Online: %v", wc)})
|
||||
} else if cmd[0] == "man" {
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: "Available commands: man, mk, rm, wc, yeet"})
|
||||
} else {
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("Invalid command: %v", cmd)})
|
||||
}
|
||||
} else {
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: "Unrecognised command, use /man for more info"})
|
||||
}
|
||||
case c.Echo:
|
||||
s.logFn("(%v) echo: %v", smsg.Id, cmsg.Msg)
|
||||
s.conns.sm.Lock()
|
||||
room := s.conns.cm[conn].room
|
||||
s.conns.sm.Unlock()
|
||||
smsg.Tim = time.Now()
|
||||
smsg.Msg = cmsg.Msg
|
||||
if len(s.rhist[room]) < s.rhlen {
|
||||
s.rhist[room] = append(s.rhist[room], smsg)
|
||||
} else {
|
||||
s.rhist[room] = append(s.rhist[room][1:], smsg)
|
||||
}
|
||||
s.conns.sm.Lock()
|
||||
for c, r := range s.conns.cm {
|
||||
if r.room == room {
|
||||
wsjson.Write(ctx, c, &smsg)
|
||||
}
|
||||
}
|
||||
s.conns.sm.Unlock()
|
||||
case c.Mv:
|
||||
switch nick, valid := verifyNick(&s, cmsg.Msg); valid {
|
||||
case nickOk:
|
||||
s.logFn("(%v) mv: %v", smsg.Id, cmsg.Msg)
|
||||
smsg.Id = nick
|
||||
s.conns.sm.Lock()
|
||||
u := s.conns.cm[conn]
|
||||
u.nick = smsg.Id
|
||||
s.conns.cm[conn] = u
|
||||
s.conns.sm.Unlock()
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("nick set: %v", nick)})
|
||||
case nickUsed:
|
||||
s.logFn("(%v) mv used: %v", smsg.Id, cmsg.Msg)
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("nick in use: %v", cmsg.Msg)})
|
||||
case nickInvalid:
|
||||
s.logFn("(%v) mv invalid: %v", smsg.Id, cmsg.Msg)
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("invalid nick: %v", cmsg.Msg)})
|
||||
}
|
||||
case c.Ls:
|
||||
s.logFn("(%v) ls", smsg.Id)
|
||||
s.conns.sm.Lock()
|
||||
room := s.conns.cm[conn].room
|
||||
s.conns.sm.Unlock()
|
||||
avRooms := ""
|
||||
for r := range s.rooms {
|
||||
avRooms += r + ", "
|
||||
}
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("connected to: %v, available: %v", room, avRooms[:len(avRooms)-2])})
|
||||
case c.Cd:
|
||||
if _, ok := s.rooms[cmsg.Msg]; ok {
|
||||
s.logFn("(%v) cd: %v", smsg.Id, cmsg.Msg)
|
||||
s.conns.sm.Lock()
|
||||
u := s.conns.cm[conn]
|
||||
u.room = cmsg.Msg
|
||||
s.conns.cm[conn] = u
|
||||
s.conns.sm.Unlock()
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("connected to: %v", u.room)})
|
||||
rhistSize := max(len(s.rhist[u.room])-10, 0)
|
||||
recentHistory := s.rhist[u.room][rhistSize:]
|
||||
for i := range recentHistory {
|
||||
wsjson.Write(ctx, conn, recentHistory[i])
|
||||
}
|
||||
} else {
|
||||
s.logFn("(%v) cd invalid: %v", smsg.Id, cmsg.Msg)
|
||||
wsjson.Write(ctx, conn, c.SMsg{Tim: time.Now(), Id: "system", Msg: fmt.Sprintf("unchanged, invalid room: %v", cmsg.Msg)})
|
||||
}
|
||||
case c.Who:
|
||||
s.conns.sm.Lock()
|
||||
room := s.conns.cm[conn].room
|
||||
s.logFn("(%v) who: %v", smsg.Id, room)
|
||||
users := fmt.Sprintf("users in %v: ", room)
|
||||
for _, r := range s.conns.cm {
|
||||
if r.room == room {
|
||||
users += fmt.Sprintf("%v, ", r.nick)
|
||||
}
|
||||
}
|
||||
s.conns.sm.Unlock()
|
||||
wsjson.Write(ctx, conn, &c.SMsg{Tim: time.Now(), Id: "system", Msg: users[:len(users)-2]})
|
||||
}
|
||||
return nil
|
||||
}(ctx, conn)
|
||||
|
||||
if ws.CloseStatus(err) == ws.StatusNormalClosure {
|
||||
s.logFn("disconnected: %v", r.RemoteAddr)
|
||||
return
|
||||
}
|
||||
if err != nil {
|
||||
s.logFn("failed, addr %v: %v", r.RemoteAddr, err)
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
type nickErr int
|
||||
|
||||
const (
|
||||
nickOk nickErr = iota
|
||||
nickUsed
|
||||
nickInvalid
|
||||
)
|
||||
|
||||
func verifyNick(s *server, n string) (string, nickErr) {
|
||||
nick, pass, _ := strings.Cut(n, ":")
|
||||
|
||||
s.conns.sm.Lock()
|
||||
defer s.conns.sm.Unlock()
|
||||
for _, u := range s.conns.cm {
|
||||
if u.nick == nick {
|
||||
return "", nickUsed
|
||||
}
|
||||
}
|
||||
|
||||
expPass, needAuth := s.nickm[nick]
|
||||
|
||||
if (!needAuth || pass == expPass) && alphanumeric(nick) {
|
||||
return nick, nickOk
|
||||
} else {
|
||||
return "", nickInvalid
|
||||
}
|
||||
}
|
||||
|
||||
func loadNickMap(m *string) (map[string]string, error) {
|
||||
nm := make(map[string]string)
|
||||
if m == nil {
|
||||
return nm, nil
|
||||
}
|
||||
|
||||
path, err := filepath.Abs(*m)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
file, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = json.Unmarshal(file, &nm)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return nm, nil
|
||||
}
|
||||
|
||||
func alphanumeric(s string) bool {
|
||||
for _, r := range s {
|
||||
if (r < 'a' || r > 'z') && (r < 'A' || r > 'Z') && (r < '0' || r > '9') {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
Loading…
Reference in a new issue