增加edge-api token --role=[admin|user|api]命令用来快速查询节点Token

This commit is contained in:
GoEdgeLab
2023-03-19 21:24:24 +08:00
parent 016148c65c
commit f5c0b4600f
2 changed files with 68 additions and 0 deletions

View File

@@ -755,6 +755,41 @@ func (this *APINode) listenSock() error {
"code": teaconst.InstanceCode,
},
})
case "lookupToken":
var role = maps.NewMap(cmd.Params).GetString("role")
switch role {
case "admin", "user", "api":
tokens, err := models.SharedApiTokenDAO.FindAllEnabledAPITokens(nil, role)
if err != nil {
_ = cmd.Reply(&gosock.Command{
Params: map[string]any{
"isOk": false,
"err": err.Error(),
},
})
} else {
var tokenMaps = []maps.Map{}
for _, token := range tokens {
tokenMaps = append(tokenMaps, maps.Map{
"nodeId": token.NodeId,
"secret": token.Secret,
})
}
_ = cmd.Reply(&gosock.Command{
Params: map[string]any{
"isOk": true,
"tokens": tokenMaps,
},
})
}
default:
_ = cmd.Reply(&gosock.Command{
Params: map[string]any{
"isOk": false,
"err": "unsupported role '" + role + "'",
},
})
}
}
})