|
|
/**
|
|
|
* @Author: jager
|
|
|
* @Email: lhj168os@gmail.com
|
|
|
* @File: main
|
|
|
* @Date: 2021/10/18 11:15 上午
|
|
|
* @package: wechat
|
|
|
* @Version: v1.0.0
|
|
|
*
|
|
|
* @Description:
|
|
|
*
|
|
|
*/
|
|
|
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
"github.com/jageros/hawox/contextx"
|
|
|
"github.com/jageros/hawox/wsc"
|
|
|
"log"
|
|
|
"net/http"
|
|
|
"unicode"
|
|
|
"wechat/types"
|
|
|
"wechat/view"
|
|
|
)
|
|
|
|
|
|
func main() {
|
|
|
ctx, cancel := contextx.Default()
|
|
|
defer cancel()
|
|
|
var err error
|
|
|
|
|
|
var sess *wsc.Session
|
|
|
m := wsc.New(ctx)
|
|
|
|
|
|
view.OnMessage("请发送本人手机号码加入聊天室！\n")
|
|
|
|
|
|
view.OnSendMsg(func(msg string) {
|
|
|
if sess == nil {
|
|
|
if len(msg) != 11 {
|
|
|
view.OnMessage("手机号码格式错误！\n")
|
|
|
return
|
|
|
}
|
|
|
|
|
|
for _, r := range msg {
|
|
|
if !unicode.Is(unicode.Number, r) {
|
|
|
view.OnMessage("手机号码格式错误！\n")
|
|
|
return
|
|
|
}
|
|
|
}
|
|
|
|
|
|
uid := msg
|
|
|
h := http.Header{}
|
|
|
h.Add("uid", uid)
|
|
|
sess, err = m.ConnectWithHeader("ws://wechat.hawtech.cn/ws/wechat/1", h, map[string]interface{}{"uid": uid})
|
|
|
if err != nil {
|
|
|
view.OnMessage(fmt.Sprintf("创建websocket链接错误，手机号码:%s 错误信息:%v\n", uid, err))
|
|
|
return
|
|
|
}
|
|
|
view.OnMessage("登录成功！\n")
|
|
|
return
|
|
|
}
|
|
|
data := &types.Msg{
|
|
|
MsgID: 1,
|
|
|
Msg: msg,
|
|
|
}
|
|
|
bytes, err := types.Marshal(data)
|
|
|
if err != nil {
|
|
|
view.OnMessage(fmt.Sprintf("编码错误: %v", err))
|
|
|
return
|
|
|
}
|
|
|
err = sess.WriteBinary(bytes)
|
|
|
if err != nil {
|
|
|
|
|
|
}
|
|
|
})
|
|
|
|
|
|
m.HandleMessageBinary(func(session *wsc.Session, bytes []byte) {
|
|
|
msg, err := types.Unmarshal(bytes)
|
|
|
if err != nil {
|
|
|
view.OnMessage(fmt.Sprintf("解码错误：%v", err))
|
|
|
return
|
|
|
}
|
|
|
switch msg.MsgID {
|
|
|
case 1:
|
|
|
view.OnMessage(msg.Msg)
|
|
|
|
|
|
case 2:
|
|
|
view.UpdateOnline(msg.Msg)
|
|
|
|
|
|
default:
|
|
|
log.Printf("MsgId=%d Msg=%s", msg.MsgID, msg.Msg)
|
|
|
}
|
|
|
})
|
|
|
|
|
|
err = view.Run()
|
|
|
if err != nil {
|
|
|
log.Fatal(err)
|
|
|
}
|
|
|
}
|