/
status.go
127 lines (111 loc) · 2.92 KB
/
status.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
// Copyright (c) 2021 Terminus, Inc.
//
// This program is free software: you can use, redistribute, and/or modify
// it under the terms of the GNU Affero General Public License, version 3
// or later ("AGPL"), as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
package status
import (
"encoding/json"
"os"
"path/filepath"
"time"
"github.com/erda-project/erda/tools/cli/dicedir"
)
const (
sessionFile = "sessions"
)
// {
// "terminus-org.app.terminus.io": {
// "sessionid": "",
// "orgID": 1,
// "id": "0001",
// "nickName": "username",
// },
// "terminus-org.test.terminus.io": {
// "sessionid": "",
// "orgID": 2,
// "id": "0002",
// "nickName": "username",
// }
// }
var SessionInfos = map[string]StatusInfo{}
type StatusInfo struct {
SessionID string `json:"sessionid"`
ExpiredAt *time.Time `json:"expiredAt"`
OrgID uint64 `json:"orgID"`
UserInfo
}
type UserInfo struct {
ID string `json:"id"`
Email string `json:"email"`
NickName string `json:"nickName"`
Enabled bool `json:"enabled"`
CreatedAt string `json:"createdAt"`
UpdatedAt string `json:"updatedAt"`
LastLoginAt string `json:"lastLoginAt"`
}
// GetSessionInfos fetch sessions
func GetSessionInfos() (map[string]StatusInfo, error) {
// check directory ~/.dice.d if exist
diceDir, err := dicedir.FindGlobalDiceDir()
if err != nil {
return nil, err
}
// load file ~/.dice.d/sessions
if _, err := os.Stat(filepath.Join(diceDir, sessionFile)); err != nil {
if os.IsNotExist(err) {
return nil, dicedir.NotExist
}
return nil, err
}
f, err := os.Open(filepath.Join(diceDir, sessionFile))
if err != nil {
return nil, err
}
defer f.Close()
if err := json.NewDecoder(f).Decode(&SessionInfos); err != nil {
return nil, err
}
return SessionInfos, nil
}
// StoreSessionInfo write session info to file ~/.dice.d/sessions
func StoreSessionInfo(host string, stat StatusInfo) error {
diceDir, err := dicedir.FindGlobalDiceDir()
if err != nil {
if err != dicedir.NotExist {
return err
}
diceDir, err = dicedir.CreateGlobalDiceDir()
if err != nil {
return err
}
}
sessions := make(map[string]StatusInfo)
sessionPath := filepath.Join(diceDir, sessionFile)
if _, err := os.Stat(sessionPath); err == nil {
sessions, err = GetSessionInfos()
if err != nil {
return err
}
}
f, err := os.OpenFile(sessionPath, os.O_CREATE|os.O_TRUNC|os.O_RDWR, 0666)
if err != nil {
return err
}
defer f.Close()
// add new session
sessions[host] = stat
content, err := json.MarshalIndent(sessions, "", " ")
if err != nil {
return err
}
f.Write(content)
return nil
}