forked from jumpserver/koko
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathwebfolder.go
More file actions
55 lines (43 loc) · 1.02 KB
/
webfolder.go
File metadata and controls
55 lines (43 loc) · 1.02 KB
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
package httpd
import (
"github.com/jumpserver/koko/pkg/jms-sdk-go/service"
"strings"
)
var _ Handler = (*webFolder)(nil)
type webFolder struct {
ws *UserWebsocket
done chan struct{}
targetId string
volume *UserVolume
jmsService *service.JMService
}
func (h *webFolder) Name() string {
return WebFolderName
}
func (h *webFolder) CheckValidation() bool {
jmsServiceCopy := h.jmsService.Copy()
if langCode, err := h.ws.ctx.Cookie("django_language"); err == nil {
jmsServiceCopy.SetCookie("django_language", langCode)
}
switch strings.TrimSpace(h.targetId) {
case "_":
h.volume = NewUserVolume(jmsServiceCopy, h.ws.CurrentUser(), h.ws.ClientIP(), "")
default:
h.volume = NewUserVolume(jmsServiceCopy, h.ws.CurrentUser(), h.ws.ClientIP(), strings.TrimSpace(h.targetId))
}
return true
}
func (h *webFolder) HandleMessage(*Message) {
}
func (h *webFolder) CleanUp() {
close(h.done)
h.volume.Close()
}
func (h *webFolder) GetVolume() *UserVolume {
select {
case <-h.done:
return nil
default:
return h.volume
}
}