Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in
Toggle navigation
K
koko
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
ops
koko
Commits
cccf0ae5
Commit
cccf0ae5
authored
Apr 30, 2019
by
Eric
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[update] add perm service
parent
907750c9
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
44 additions
and
8 deletions
+44
-8
perms.go
pkg/service/perms.go
+38
-4
terminal.go
pkg/service/terminal.go
+2
-1
users.go
pkg/service/users.go
+4
-3
No files found.
pkg/service/perms.go
View file @
cccf0ae5
package
service
import
(
"fmt"
"cocogo/pkg/logger"
"cocogo/pkg/model"
)
func
GetUserAssets
(
userId
string
)
(
assets
model
.
AssetList
)
{
return
model
.
AssetList
{{
Id
:
"xxxxxxxxx"
,
Hostname
:
"test"
,
Ip
:
"192.168.244.185"
,
Port
:
22
}}
func
GetUserAssets
(
userId
,
cachePolicy
string
)
(
assets
model
.
AssetList
)
{
if
cachePolicy
==
""
{
cachePolicy
=
"0"
}
params
:=
map
[
string
]
string
{
"cache_policy"
:
cachePolicy
}
Url
:=
authClient
.
ParseUrlQuery
(
fmt
.
Sprintf
(
UserAssetsURL
,
userId
),
params
)
err
:=
authClient
.
Get
(
Url
,
&
assets
)
if
err
!=
nil
{
logger
.
Error
(
err
)
}
return
}
func
GetUserNodes
(
userId
string
)
(
nodes
model
.
NodeList
)
{
return
model
.
NodeList
{{
Id
:
"XXXXXXX"
,
Name
:
"test"
}}
func
GetUserNodes
(
userId
,
cachePolicy
string
)
(
nodes
model
.
NodeList
)
{
if
cachePolicy
==
""
{
cachePolicy
=
"0"
}
params
:=
map
[
string
]
string
{
"cache_policy"
:
cachePolicy
}
Url
:=
authClient
.
ParseUrlQuery
(
fmt
.
Sprintf
(
UserNodesAssetsURL
,
userId
),
params
)
err
:=
authClient
.
Get
(
Url
,
&
nodes
)
if
err
!=
nil
{
logger
.
Error
(
err
)
}
return
}
func
ValidateUserAssetPermission
(
userId
,
assetId
,
systemUserId
string
)
bool
{
params
:=
map
[
string
]
string
{
"user_id"
:
userId
,
"asset_id"
:
assetId
,
"system_user_id"
:
systemUserId
,
"cache_policy"
:
"1"
,
}
Url
:=
authClient
.
ParseUrlQuery
(
ValidateUserAssetPermissionURL
,
params
)
err
:=
authClient
.
Get
(
Url
,
nil
)
if
err
!=
nil
{
logger
.
Error
(
err
)
return
false
}
return
true
}
pkg/service/terminal.go
View file @
cccf0ae5
...
...
@@ -35,7 +35,8 @@ func TerminalHeartBeat(sIds []string) (res []model.TerminalTask) {
func
CreateSession
(
data
map
[
string
]
interface
{})
bool
{
var
res
map
[
string
]
interface
{}
err
:=
authClient
.
Post
(
SessionListURL
,
data
,
&
res
)
Url
:=
authClient
.
ParseUrlQuery
(
SessionListURL
,
nil
)
err
:=
authClient
.
Post
(
Url
,
data
,
&
res
)
if
err
==
nil
{
return
true
}
...
...
pkg/service/users.go
View file @
cccf0ae5
...
...
@@ -19,7 +19,7 @@ func Authenticate(username, password, publicKey, remoteAddr, loginType string) (
User
model
.
User
`json:"user"`
}
err
:=
client
.
Post
(
baseHost
+
UserAuthURL
,
data
,
&
resp
)
err
:=
client
.
Post
(
UserAuthURL
,
data
,
&
resp
)
if
err
!=
nil
{
logger
.
Error
(
err
)
}
...
...
@@ -27,7 +27,7 @@ func Authenticate(username, password, publicKey, remoteAddr, loginType string) (
}
func
GetUserProfile
(
userId
string
)
(
user
model
.
User
)
{
Url
:=
fmt
.
Sprintf
(
baseHost
+
UserUserURL
,
userId
)
Url
:=
fmt
.
Sprintf
(
UserUserURL
,
userId
)
err
:=
authClient
.
Get
(
Url
,
&
user
)
if
err
!=
nil
{
logger
.
Error
(
err
)
...
...
@@ -38,7 +38,8 @@ func GetUserProfile(userId string) (user model.User) {
func
CheckUserCookie
(
sessionId
,
csrfToken
string
)
(
user
model
.
User
)
{
client
.
SetCookie
(
"csrftoken"
,
csrfToken
)
client
.
SetCookie
(
"sessionid"
,
sessionId
)
err
:=
client
.
Get
(
baseHost
+
UserProfileURL
,
&
user
)
Url
:=
client
.
ParseUrlQuery
(
UserProfileURL
,
nil
)
err
:=
client
.
Get
(
Url
,
&
user
)
if
err
!=
nil
{
logger
.
Error
(
err
)
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment