-
Notifications
You must be signed in to change notification settings - Fork 3.6k
/
main.go
120 lines (104 loc) · 2.07 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package main
import (
"bufio"
"fmt"
"io"
"net"
"github.com/ttacon/chalk"
)
// User holds a user of the chat room
type User struct {
Name string
Output chan Message
}
// Message holds a message for the chat room
type Message struct {
Username string
Text string
}
// ChatServer holds all the data about the current state of the chat room
type ChatServer struct {
Users map[string]User
Join chan User
Leave chan User
Input chan Message
}
// Run runs the server
func (cs *ChatServer) Run() {
for {
select {
case user := <-cs.Join:
cs.Users[user.Name] = user
go func() {
cs.Input <- Message{
Username: "SYSTEM",
Text: fmt.Sprintf("%s joined", user.Name),
}
}()
case user := <-cs.Leave:
delete(cs.Users, user.Name)
go func() {
cs.Input <- Message{
Username: "SYSTEM",
Text: fmt.Sprintf("%s left", user.Name),
}
}()
case msg := <-cs.Input:
for _, user := range cs.Users {
select {
case user.Output <- msg:
default:
}
}
}
}
}
func handleConn(chatServer *ChatServer, conn net.Conn) {
defer conn.Close()
io.WriteString(conn, "Enter your username: ")
scanner := bufio.NewScanner(conn)
scanner.Scan()
user := User{
Name: scanner.Text(),
Output: make(chan Message, 10),
}
chatServer.Join <- user
defer func() {
chatServer.Leave <- user
}()
go func() {
scanner := bufio.NewScanner(conn)
for scanner.Scan() {
ln := scanner.Text()
chatServer.Input <- Message{user.Name, ln}
}
close(user.Output)
}()
for msg := range user.Output {
_, err := fmt.Fprintf(conn, "%s%s: %s%s%s\n", chalk.Yellow, msg.Username, chalk.White, msg.Text, chalk.ResetColor)
if err != nil {
break
}
}
}
func main() {
lnr, err := net.Listen("tcp", ":9000")
if err != nil {
panic(err)
}
defer lnr.Close()
chatServer := &ChatServer{
Users: make(map[string]User),
Join: make(chan User),
Leave: make(chan User),
Input: make(chan Message),
}
go chatServer.Run()
for {
conn, err := lnr.Accept()
if err != nil {
panic(err)
}
go handleConn(chatServer, conn)
}
}