scottk/databunker
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/databunker.git synced 2026-01-09 00:51:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

user can now view his own requests

Browse Source
This commit is contained in:
Yuli
2020-05-12 23:29:07 +03:00
parent 8fe7a4f165
commit 40e98aaa34
8 changed files with 311 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/bunker.go
View File

@@ -200,6 +200,7 @@ func (e mainEnv) setupRouter() *httprouter.Router {
router.GET("/v1/request/:request", e.getUserRequest)
router.POST("/v1/request/:request", e.approveUserRequest)
router.DELETE("/v1/request/:request", e.cancelUserRequest)
router.GET("/v1/requests/:mode/:address", e.getCustomUserRequests)
router.GET("/v1/requests", e.getUserRequests)
router.GET("/v1/consent/:mode/:address", e.consentAllUserRecords)

56
src/requests_api.go
View File

@@ -6,6 +6,7 @@ import (
"net/http"
"github.com/julienschmidt/httprouter"
"go.mongodb.org/mongo-driver/bson"
)
func (e mainEnv) getUserRequests(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
@@ -39,6 +40,61 @@ func (e mainEnv) getUserRequests(w http.ResponseWriter, r *http.Request, ps http
w.Write([]byte(str))
}
func (e mainEnv) getCustomUserRequests(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
address := ps.ByName("address")
mode := ps.ByName("mode")
event := audit("get user privacy requests", address, mode, address)
defer func() { event.submit(e.db) }()
if validateMode(mode) == false {
returnError(w, r, "bad mode", 405, nil, event)
return
}
userTOKEN := address
var userBson bson.M
if mode == "token" {
if enforceUUID(w, address, event) == false {
return
}
userBson, _ = e.db.lookupUserRecord(address)
} else {
userBson, _ = e.db.lookupUserRecordByIndex(mode, address, e.conf)
if userBson != nil {
userTOKEN = userBson["token"].(string)
event.Record = userTOKEN
}
}
if userBson == nil {
returnError(w, r, "internal error", 405, nil, event)
return
}
if e.enforceAuth(w, r, event) == "" {
return
}
var offset int32
var limit int32 = 10
args := r.URL.Query()
if value, ok := args["offset"]; ok {
offset = atoi(value[0])
}
if value, ok := args["limit"]; ok {
limit = atoi(value[0])
}
resultJSON, counter, err := e.db.getUserRequests(userTOKEN, offset, limit)
if err != nil {
returnError(w, r, "internal error", 405, err, nil)
return
}
fmt.Printf("Total count of custom user requests: %d\n", counter)
//fmt.Fprintf(w, "<html><head><title>title</title></head>")
w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.WriteHeader(200)
str := fmt.Sprintf(`{"status":"ok","total":%d,"rows":%s}`, counter, resultJSON)
w.Write([]byte(str))
}
func (e mainEnv) getUserRequest(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
request := ps.ByName("request")
event := audit("get user request by request token", request, "request", request)

28
src/requests_db.go
View File

@@ -77,6 +77,34 @@ func (dbobj dbcon) getRequests(status string, offset int32, limit int32) ([]byte
return resultJSON, count, nil
}
func (dbobj dbcon) getUserRequests(userTOKEN string, offset int32, limit int32) ([]byte, int64, error) {
//var results []*auditEvent
count, err := dbobj.store.CountRecords(storage.TblName.Requests, "token", userTOKEN)
if err != nil {
return nil, 0, err
}
var results []bson.M
records, err := dbobj.store.GetList(storage.TblName.Requests, "token", userTOKEN, offset, limit)
if err != nil {
return nil, 0, err
}
for _, element := range records {
element["more"] = false
if _, ok := element["change"]; ok {
element["more"] = true
delete(element, "change")
}
results = append(results, element)
}
resultJSON, err := json.Marshal(records)
if err != nil {
return nil, 0, err
}
//fmt.Printf("Found multiple documents (array of pointers): %+v\n", results)
return resultJSON, count, nil
}
func (dbobj dbcon) getRequest(rtoken string) (bson.M, error) {
record, err := dbobj.store.GetRecord(storage.TblName.Requests, "rtoken", rtoken)
if err != nil {