wisp/cmd/impl/convert_model.go at ff21085c498534eeac5536cb48a371168f6d9e8f · jumpserver/wisp · GitHub

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
package impl

import (
	"github.com/jumpserver-dev/sdk-go/common"
	"github.com/jumpserver-dev/sdk-go/model"
	pb "github.com/jumpserver/wisp/protobuf-go/protobuf"
)

var modelLoginFrom = map[pb.Session_LoginFrom]model.LabelField{
	pb.Session_WT: model.LoginFromWeb,
	pb.Session_ST: model.LoginFromSSH,
	pb.Session_RT: model.LoginFromRT,
	pb.Session_DT: model.LoginFromDT,
}

func ConvertModelLoginFrom(lf pb.Session_LoginFrom) model.LabelField {
	return modelLoginFrom[lf]
}

func ConvertToSession(sees *pb.Session) model.Session {
	return model.Session{
		ID:         sees.Id,
		User:       sees.User,
		Asset:      sees.Asset,
		Account:    sees.Account,
		LoginFrom:  ConvertModelLoginFrom(sees.LoginFrom),
		RemoteAddr: sees.RemoteAddr,
		Protocol:   sees.Protocol,
		DateStart:  common.ParseUnixTime(sees.DateStart),
		OrgID:      sees.OrgId,
		UserID:     sees.UserId,
		AssetID:    sees.AssetId,
		AccountID:  sees.AccountId,
		Type:       model.NORMALType,
		TokenId:    sees.TokenId,
	}
}

func ConvertToCommand(cmd *pb.CommandRequest) model.Command {
	utc := ConvertUTCTime(cmd.Timestamp)
	return model.Command{
		SessionID:      cmd.Sid,
		OrgID:          cmd.OrgId,
		Input:          cmd.Input,
		Output:         cmd.Output,
		User:           cmd.User,
		Server:         cmd.Asset,
		Account:        cmd.Account,
		Timestamp:      cmd.Timestamp,
		CmdFilterAclId: cmd.CmdAclId,
		CmdGroupId:     cmd.CmdGroupId,
		RiskLevel:      convertRiskLevel(cmd.RiskLevel),
		DateCreated:    utc.UTC(),
	}
}

func ConvertUTCTime(t int64) common.UTCTime {
	return common.ParseUnixTime(t)
}

var riskLevelMap = map[pb.RiskLevel]int64{
	pb.RiskLevel_Normal:       model.NormalLevel,
	pb.RiskLevel_Warning:      model.WarningLevel,
	pb.RiskLevel_Reject:       model.RejectLevel,
	pb.RiskLevel_ReviewReject: model.ReviewReject,
	pb.RiskLevel_ReviewAccept: model.ReviewAccept,
	pb.RiskLevel_ReviewCancel: model.ReviewCancel,
}

func convertRiskLevel(lvl pb.RiskLevel) int64 {
	if v, ok := riskLevelMap[lvl]; ok {
		return v
	}
	return model.NormalLevel
}

func ConvertToReqInfo(req *pb.ReqInfo) model.ReqInfo {
	return model.ReqInfo{
		Method: req.GetMethod(),
		URL:    req.GetUrl(),
	}
}

var LifecycleEventMap = map[pb.SessionLifecycleLogRequest_EventType]model.LifecycleEvent{
	pb.SessionLifecycleLogRequest_AssetConnectSuccess:  model.AssetConnectSuccess,
	pb.SessionLifecycleLogRequest_AssetConnectFinished: model.AssetConnectFinished,
	pb.SessionLifecycleLogRequest_CreateShareLink:      model.CreateShareLink,
	pb.SessionLifecycleLogRequest_UserJoinSession:      model.UserJoinSession,
	pb.SessionLifecycleLogRequest_UserLeaveSession:     model.UserLeaveSession,
	pb.SessionLifecycleLogRequest_AdminJoinMonitor:     model.AdminJoinMonitor,
	pb.SessionLifecycleLogRequest_AdminExitMonitor:     model.AdminExitMonitor,
	pb.SessionLifecycleLogRequest_ReplayConvertStart:   model.ReplayConvertStart,
	pb.SessionLifecycleLogRequest_ReplayConvertSuccess: model.ReplayUploadStart,
	pb.SessionLifecycleLogRequest_ReplayConvertFailure: model.ReplayConvertFailure,
	pb.SessionLifecycleLogRequest_ReplayUploadStart:    model.ReplayConvertStart,
	pb.SessionLifecycleLogRequest_ReplayUploadSuccess:  model.ReplayUploadSuccess,
	pb.SessionLifecycleLogRequest_ReplayUploadFailure:  model.ReplayUploadFailure,
}