forked from mconintet/KiwiChat-Server
/
addBuddy.go
140 lines (115 loc) · 2.96 KB
/
addBuddy.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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
package main
import (
"database/sql"
"github.com/bitly/go-simplejson"
_ "github.com/go-sql-driver/mysql"
"github.com/mconintet/kiwi"
)
func handAddBuddy(j *simplejson.Json, s kiwi.MessageSender, m *kiwi.Message) error {
uid, err := j.Get("uid").Int()
if err != nil {
return ErrDeformedRequest
}
token, err := j.Get("token").String()
if err != nil {
return ErrDeformedRequest
}
buddyID, err := j.Get("buddyID").Int()
if err != nil {
return ErrDeformedRequest
}
db, err := sql.Open("mysql", dbConnInfo)
if err != nil {
return err
}
defer db.Close()
// validate request info
stmtOut, err := db.Prepare("SELECT COUNT(*) FROM user WHERE id=? AND token=?")
if err != nil {
return err
}
var count int
err = stmtOut.QueryRow(uid, token).Scan(&count)
if err != nil {
return err
}
stmtOut.Close()
if count == 0 {
return ErrIllegalRequest
}
// check if buddy is exists or not
stmtOut, err = db.Prepare("SELECT COUNT(*) FROM user WHERE id=?")
if err != nil {
return err
}
defer stmtOut.Close()
err = stmtOut.QueryRow(buddyID).Scan(&count)
if err != nil {
return err
}
if count == 0 {
retMsg := simplejson.New()
retMsg.Set("action", "add-buddy-return")
retMsg.Set("errCode", ErrBuddyNotFound)
retMsg.Set("errMsg", ErrCodeMsg[ErrBuddyNotFound])
byts, _ := retMsg.MarshalJSON()
s.SendWholeBytes(byts, false)
return nil
}
// check if relation is exists or not
stmtOut, err = db.Prepare("SELECT COUNT(*) FROM user_buddy WHERE user=? AND buddy=?")
if err != nil {
return err
}
defer stmtOut.Close()
err = stmtOut.QueryRow(uid, buddyID).Scan(&count)
if err != nil {
return err
}
if count == 1 {
retMsg := simplejson.New()
retMsg.Set("action", "add-buddy-return")
retMsg.Set("errCode", ErrBuddyAlreadyExists)
retMsg.Set("errMsg", ErrCodeMsg[ErrBuddyAlreadyExists])
byts, _ := retMsg.MarshalJSON()
s.SendWholeBytes(byts, false)
return nil
}
stmtIns, err := db.Prepare("INSERT INTO user_buddy (user, buddy) VALUES(?, ?)")
if err != nil {
return err
}
defer stmtIns.Close()
_, err = stmtIns.Exec(uid, buddyID)
if err != nil {
return err
}
// become friends each other
_, err = stmtIns.Exec(buddyID, uid)
if err != nil {
return err
}
// return buddy info
stmtOut, err = db.Prepare("SELECT nickname,network,avatar FROM user WHERE id=?")
if err != nil {
return err
}
var nickname sql.NullString
var network sql.NullInt64
var avatar []byte
err = stmtOut.QueryRow(buddyID).Scan(&nickname, &network, &avatar)
if err != nil {
return err
}
retMsg := simplejson.New()
retMsg.Set("action", "add-buddy-return")
retMsg.SetPath([]string{"buddy", "uid"}, buddyID)
retMsg.SetPath([]string{"buddy", "nickname"}, nickname.String)
retMsg.SetPath([]string{"buddy", "network"}, network.Int64)
retMsg.SetPath([]string{"buddy", "avatar"}, encodeAvatar(avatar))
retMsg.Set("errCode", ErrNone)
retMsg.Set("errMsg", ErrCodeMsg[ErrNone])
byts, _ := retMsg.MarshalJSON()
s.SendWholeBytes(byts, false)
return nil
}