scottk/databunker
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/databunker.git synced 2025-11-02 02:48:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

use custom rootaccess code

Browse Source
This commit is contained in:
Yuli
2020-07-23 15:13:02 +03:00
parent 0e18a3f0ff
commit 5c1197f32a
2 changed files with 22 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/xtokens_db.go
View File

@@ -20,7 +20,7 @@ func (dbobj dbcon) getRootXtoken() (string, error) {
return record["xtoken"].(string), nil
}
func (dbobj dbcon) createRootXtoken(demo bool) (string, error) {
func (dbobj dbcon) createRootXtoken(customRootXtoken string) (string, error) {
rootToken, err := dbobj.getRootXtoken()
if err != nil {
return "", err
@@ -32,8 +32,11 @@ func (dbobj dbcon) createRootXtoken(demo bool) (string, error) {
if err != nil {
return "", err
}
if demo {
rootToken = "DEMO"
if len(customRootXtoken) > 0 {
if customRootXtoken != "DEMO" && !isValidUUID(customRootXtoken) {
return "bad-custom-root-token-format", nil
}
rootToken = customRootXtoken
}
bdoc := bson.M{}
bdoc["xtoken"] = hashString(dbobj.hash, rootToken)