Skip to content

Commit eea53ab

Browse files
committed
refactor code
1 parent d3c5d0c commit eea53ab

File tree

3 files changed

+41
-31
lines changed

3 files changed

+41
-31
lines changed

docker-build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/sh
22

33
VERSION=$(cat ./version.txt)
4-
docker build --progress=plain -t securitybunker/databunker:$VERSION .
4+
BUILDKIT_PROGRESS=plain docker build -t securitybunker/databunker:$VERSION .
55
docker tag securitybunker/databunker:$VERSION securitybunker/databunker:latest

src/userlist_api.go

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
package main
2+
3+
import (
4+
"fmt"
5+
"log"
6+
"net/http"
7+
8+
"github.com/julienschmidt/httprouter"
9+
"github.com/securitybunker/databunker/src/utils"
10+
)
11+
12+
func (e mainEnv) userList(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
13+
if e.EnforceAdmin(w, r, nil) == "" {
14+
return
15+
}
16+
if e.conf.Generic.ListUsers == false {
17+
utils.ReturnError(w, r, "access denied", 403, nil, nil)
18+
return
19+
}
20+
var offset int32 = 0
21+
var limit int32 = 10
22+
args := r.URL.Query()
23+
if value, ok := args["offset"]; ok {
24+
offset = utils.Atoi(value[0])
25+
}
26+
if value, ok := args["limit"]; ok {
27+
limit = utils.Atoi(value[0])
28+
}
29+
resultJSON, counter, _ := e.db.getUsers(offset, limit)
30+
log.Printf("Total count of events: %d\n", counter)
31+
w.Header().Set("Content-Type", "application/json; charset=utf-8")
32+
w.WriteHeader(200)
33+
if counter == 0 {
34+
str := fmt.Sprintf(`{"status":"ok","total":%d,"rows":[]}`, counter)
35+
w.Write([]byte(str))
36+
} else {
37+
str := fmt.Sprintf(`{"status":"ok","total":%d,"rows":%s}`, counter, resultJSON)
38+
w.Write([]byte(str))
39+
}
40+
}

src/users_api.go

Lines changed: 0 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -167,36 +167,6 @@ func (e mainEnv) userGet(w http.ResponseWriter, r *http.Request, ps httprouter.P
167167
w.Write([]byte(finalJSON))
168168
}
169169

170-
func (e mainEnv) userList(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
171-
if e.EnforceAdmin(w, r, nil) == "" {
172-
return
173-
}
174-
if e.conf.Generic.ListUsers == false {
175-
utils.ReturnError(w, r, "access denied", 403, nil, nil)
176-
return
177-
}
178-
var offset int32 = 0
179-
var limit int32 = 10
180-
args := r.URL.Query()
181-
if value, ok := args["offset"]; ok {
182-
offset = utils.Atoi(value[0])
183-
}
184-
if value, ok := args["limit"]; ok {
185-
limit = utils.Atoi(value[0])
186-
}
187-
resultJSON, counter, _ := e.db.getUsers(offset, limit)
188-
log.Printf("Total count of events: %d\n", counter)
189-
w.Header().Set("Content-Type", "application/json; charset=utf-8")
190-
w.WriteHeader(200)
191-
if counter == 0 {
192-
str := fmt.Sprintf(`{"status":"ok","total":%d,"rows":[]}`, counter)
193-
w.Write([]byte(str))
194-
} else {
195-
str := fmt.Sprintf(`{"status":"ok","total":%d,"rows":%s}`, counter, resultJSON)
196-
w.Write([]byte(str))
197-
}
198-
}
199-
200170
func (e mainEnv) userChange(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
201171
identity := ps.ByName("identity")
202172
mode := ps.ByName("mode")

0 commit comments

Comments
 (0)