/
controller.go
84 lines (71 loc) · 2.09 KB
/
controller.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
package system
import (
"encoding/json"
"github.com/bernishen/lion-go/oauth2/controller/system/domain"
"github.com/bernishen/lion-go/oauth2/route"
"github.com/bernishen/lion-go/oauth2/service/sessionservice"
"github.com/bernishen/lion-go/utils/router"
"github.com/bernishen/lion-go/utils/websocket"
ws "github.com/gorilla/websocket"
)
var wsService *websocket.WSService
func init() {
wsService = websocket.InitWebSocket(20, callRoute)
s := router.InitController("/system/{wsID}").
Get(wsService.Handler, "wsID")
route.Router.Register(s)
}
func callRoute(wsID string, msg *websocket.WSMessage) *websocket.WSReturn {
switch msg.MessageType {
case ws.TextMessage:
f := initFunc(msg)
if f == nil {
return websocket.Return(returnFailed(301, "The parameter can not be parsed."))
}
switch f.name {
case "Verify":
return verify(wsID, f.data)
case "SignOut":
return signOut(f.data)
default:
return websocket.Return(returnFailed(301, "Found not function name of "+f.name+"."))
}
default:
return websocket.Return(returnOK(nil))
}
}
func verify(wsID string, param string) *websocket.WSReturn {
pByte := []byte(param)
var c domain.Client
err := json.Unmarshal(pByte, c)
if err != nil {
return websocket.Return(returnFailed(301, err.Error()))
}
s, ex := sessionservice.VerifyGlobal(c.AccessToken)
if ex != nil {
return websocket.Return(returnFailed(301, err.Error()))
}
r, ok := s.Roles[c.SystemID]
if !ok {
return websocket.Return(returnFailed(301, "This user does not have this system permission."))
}
_, ex = sessionservice.VerifySystem(c.SystemID, c.AccessToken)
if ex != nil {
}
sToken, ex := sessionservice.NewSystem(c.SystemID, c.AccessToken, &r)
if ex != nil {
return websocket.Return(returnFailed(301, ex.Message))
}
ret := []byte(sToken)
return websocket.Return(returnOK(&ret))
}
func signOut(param string) *websocket.WSReturn {
pByte := []byte(param)
var c domain.Client
err := json.Unmarshal(pByte, c)
if err != nil {
return websocket.Return(returnFailed(301, err.Error()))
}
//sessionservice.SignOutAll(c.AccessToken)
return websocket.NoReturn()
}