Compare commits
No commits in common. "3563d53365dc5d49e21d2cf36e35a58e215b8f2b" and "214e2348aa9339b92054ab9f9f1be77abe089305" have entirely different histories.
3563d53365
...
214e2348aa
23
handler.go
23
handler.go
|
|
@ -1,9 +1,7 @@
|
||||||
package aicra
|
package aicra
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
|
||||||
|
|
||||||
"git.xdrm.io/go/aicra/api"
|
"git.xdrm.io/go/aicra/api"
|
||||||
"git.xdrm.io/go/aicra/internal/config"
|
"git.xdrm.io/go/aicra/internal/config"
|
||||||
|
|
@ -52,27 +50,8 @@ func (s Handler) resolve(w http.ResponseWriter, r *http.Request) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// replace format '[a]' in scope where 'a' is an existing input's name
|
|
||||||
scope := make([][]string, len(service.Scope))
|
|
||||||
for a, list := range service.Scope {
|
|
||||||
scope[a] = make([]string, len(list))
|
|
||||||
for b, perm := range list {
|
|
||||||
scope[a][b] = perm
|
|
||||||
for name, value := range input.Data {
|
|
||||||
var (
|
|
||||||
token = fmt.Sprintf("[%s]", name)
|
|
||||||
replacement = ""
|
|
||||||
)
|
|
||||||
if value != nil {
|
|
||||||
replacement = fmt.Sprintf("[%v]", value)
|
|
||||||
}
|
|
||||||
scope[a][b] = strings.ReplaceAll(scope[a][b], token, replacement)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var auth = api.Auth{
|
var auth = api.Auth{
|
||||||
Required: scope,
|
Required: service.Scope,
|
||||||
Active: []string{},
|
Active: []string{},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
276
handler_test.go
276
handler_test.go
|
|
@ -1,4 +1,4 @@
|
||||||
package aicra_test
|
package aicra
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
|
@ -9,35 +9,11 @@ import (
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"git.xdrm.io/go/aicra"
|
|
||||||
"git.xdrm.io/go/aicra/api"
|
"git.xdrm.io/go/aicra/api"
|
||||||
"git.xdrm.io/go/aicra/datatype/builtin"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func addBuiltinTypes(b *aicra.Builder) error {
|
|
||||||
if err := b.AddType(builtin.AnyDataType{}); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := b.AddType(builtin.BoolDataType{}); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := b.AddType(builtin.FloatDataType{}); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := b.AddType(builtin.IntDataType{}); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := b.AddType(builtin.StringDataType{}); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := b.AddType(builtin.UintDataType{}); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWith(t *testing.T) {
|
func TestWith(t *testing.T) {
|
||||||
builder := &aicra.Builder{}
|
builder := &Builder{}
|
||||||
if err := addBuiltinTypes(builder); err != nil {
|
if err := addBuiltinTypes(builder); err != nil {
|
||||||
t.Fatalf("unexpected error <%v>", err)
|
t.Fatalf("unexpected error <%v>", err)
|
||||||
}
|
}
|
||||||
|
|
@ -206,7 +182,7 @@ func TestWithAuth(t *testing.T) {
|
||||||
|
|
||||||
for _, tc := range tt {
|
for _, tc := range tt {
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
builder := &aicra.Builder{}
|
builder := &Builder{}
|
||||||
if err := addBuiltinTypes(builder); err != nil {
|
if err := addBuiltinTypes(builder); err != nil {
|
||||||
t.Fatalf("unexpected error <%v>", err)
|
t.Fatalf("unexpected error <%v>", err)
|
||||||
}
|
}
|
||||||
|
|
@ -263,249 +239,3 @@ func TestWithAuth(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDynamicScope(t *testing.T) {
|
|
||||||
tt := []struct {
|
|
||||||
name string
|
|
||||||
manifest string
|
|
||||||
path string
|
|
||||||
handler interface{}
|
|
||||||
url string
|
|
||||||
body string
|
|
||||||
permissions []string
|
|
||||||
granted bool
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
name: "replace one granted",
|
|
||||||
manifest: `[
|
|
||||||
{
|
|
||||||
"method": "POST",
|
|
||||||
"path": "/path/{id}",
|
|
||||||
"info": "info",
|
|
||||||
"scope": [["user[Input1]"]],
|
|
||||||
"in": {
|
|
||||||
"{id}": { "info": "info", "name": "Input1", "type": "uint" }
|
|
||||||
},
|
|
||||||
"out": {}
|
|
||||||
}
|
|
||||||
]`,
|
|
||||||
path: "/path/{id}",
|
|
||||||
handler: func(struct{ Input1 uint }) (*struct{}, api.Err) { return nil, api.ErrSuccess },
|
|
||||||
url: "/path/123",
|
|
||||||
body: ``,
|
|
||||||
permissions: []string{"user[123]"},
|
|
||||||
granted: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "replace one mismatch",
|
|
||||||
manifest: `[
|
|
||||||
{
|
|
||||||
"method": "POST",
|
|
||||||
"path": "/path/{id}",
|
|
||||||
"info": "info",
|
|
||||||
"scope": [["user[Input1]"]],
|
|
||||||
"in": {
|
|
||||||
"{id}": { "info": "info", "name": "Input1", "type": "uint" }
|
|
||||||
},
|
|
||||||
"out": {}
|
|
||||||
}
|
|
||||||
]`,
|
|
||||||
path: "/path/{id}",
|
|
||||||
handler: func(struct{ Input1 uint }) (*struct{}, api.Err) { return nil, api.ErrSuccess },
|
|
||||||
url: "/path/666",
|
|
||||||
body: ``,
|
|
||||||
permissions: []string{"user[123]"},
|
|
||||||
granted: false,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "replace one valid dot separated",
|
|
||||||
manifest: `[
|
|
||||||
{
|
|
||||||
"method": "POST",
|
|
||||||
"path": "/path/{id}",
|
|
||||||
"info": "info",
|
|
||||||
"scope": [["prefix.user[User].suffix"]],
|
|
||||||
"in": {
|
|
||||||
"{id}": { "info": "info", "name": "User", "type": "uint" }
|
|
||||||
},
|
|
||||||
"out": {}
|
|
||||||
}
|
|
||||||
]`,
|
|
||||||
path: "/path/{id}",
|
|
||||||
handler: func(struct{ User uint }) (*struct{}, api.Err) { return nil, api.ErrSuccess },
|
|
||||||
url: "/path/123",
|
|
||||||
body: ``,
|
|
||||||
permissions: []string{"prefix.user[123].suffix"},
|
|
||||||
granted: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "replace two valid dot separated",
|
|
||||||
manifest: `[
|
|
||||||
{
|
|
||||||
"method": "POST",
|
|
||||||
"path": "/prefix/{pid}/user/{uid}",
|
|
||||||
"info": "info",
|
|
||||||
"scope": [["prefix[Prefix].user[User].suffix"]],
|
|
||||||
"in": {
|
|
||||||
"{pid}": { "info": "info", "name": "Prefix", "type": "uint" },
|
|
||||||
"{uid}": { "info": "info", "name": "User", "type": "uint" }
|
|
||||||
},
|
|
||||||
"out": {}
|
|
||||||
}
|
|
||||||
]`,
|
|
||||||
path: "/prefix/{pid}/user/{uid}",
|
|
||||||
handler: func(struct {
|
|
||||||
Prefix uint
|
|
||||||
User uint
|
|
||||||
}) (*struct{}, api.Err) {
|
|
||||||
return nil, api.ErrSuccess
|
|
||||||
},
|
|
||||||
url: "/prefix/123/user/456",
|
|
||||||
body: ``,
|
|
||||||
permissions: []string{"prefix[123].user[456].suffix"},
|
|
||||||
granted: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "replace two invalid dot separated",
|
|
||||||
manifest: `[
|
|
||||||
{
|
|
||||||
"method": "POST",
|
|
||||||
"path": "/prefix/{pid}/user/{uid}",
|
|
||||||
"info": "info",
|
|
||||||
"scope": [["prefix[Prefix].user[User].suffix"]],
|
|
||||||
"in": {
|
|
||||||
"{pid}": { "info": "info", "name": "Prefix", "type": "uint" },
|
|
||||||
"{uid}": { "info": "info", "name": "User", "type": "uint" }
|
|
||||||
},
|
|
||||||
"out": {}
|
|
||||||
}
|
|
||||||
]`,
|
|
||||||
path: "/prefix/{pid}/user/{uid}",
|
|
||||||
handler: func(struct {
|
|
||||||
Prefix uint
|
|
||||||
User uint
|
|
||||||
}) (*struct{}, api.Err) {
|
|
||||||
return nil, api.ErrSuccess
|
|
||||||
},
|
|
||||||
url: "/prefix/123/user/666",
|
|
||||||
body: ``,
|
|
||||||
permissions: []string{"prefix[123].user[456].suffix"},
|
|
||||||
granted: false,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "replace three valid dot separated",
|
|
||||||
manifest: `[
|
|
||||||
{
|
|
||||||
"method": "POST",
|
|
||||||
"path": "/prefix/{pid}/user/{uid}/suffix/{sid}",
|
|
||||||
"info": "info",
|
|
||||||
"scope": [["prefix[Prefix].user[User].suffix[Suffix]"]],
|
|
||||||
"in": {
|
|
||||||
"{pid}": { "info": "info", "name": "Prefix", "type": "uint" },
|
|
||||||
"{uid}": { "info": "info", "name": "User", "type": "uint" },
|
|
||||||
"{sid}": { "info": "info", "name": "Suffix", "type": "uint" }
|
|
||||||
},
|
|
||||||
"out": {}
|
|
||||||
}
|
|
||||||
]`,
|
|
||||||
path: "/prefix/{pid}/user/{uid}/suffix/{sid}",
|
|
||||||
handler: func(struct {
|
|
||||||
Prefix uint
|
|
||||||
User uint
|
|
||||||
Suffix uint
|
|
||||||
}) (*struct{}, api.Err) {
|
|
||||||
return nil, api.ErrSuccess
|
|
||||||
},
|
|
||||||
url: "/prefix/123/user/456/suffix/789",
|
|
||||||
body: ``,
|
|
||||||
permissions: []string{"prefix[123].user[456].suffix[789]"},
|
|
||||||
granted: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "replace three invalid dot separated",
|
|
||||||
manifest: `[
|
|
||||||
{
|
|
||||||
"method": "POST",
|
|
||||||
"path": "/prefix/{pid}/user/{uid}/suffix/{sid}",
|
|
||||||
"info": "info",
|
|
||||||
"scope": [["prefix[Prefix].user[User].suffix[Suffix]"]],
|
|
||||||
"in": {
|
|
||||||
"{pid}": { "info": "info", "name": "Prefix", "type": "uint" },
|
|
||||||
"{uid}": { "info": "info", "name": "User", "type": "uint" },
|
|
||||||
"{sid}": { "info": "info", "name": "Suffix", "type": "uint" }
|
|
||||||
},
|
|
||||||
"out": {}
|
|
||||||
}
|
|
||||||
]`,
|
|
||||||
path: "/prefix/{pid}/user/{uid}/suffix/{sid}",
|
|
||||||
handler: func(struct {
|
|
||||||
Prefix uint
|
|
||||||
User uint
|
|
||||||
Suffix uint
|
|
||||||
}) (*struct{}, api.Err) {
|
|
||||||
return nil, api.ErrSuccess
|
|
||||||
},
|
|
||||||
url: "/prefix/123/user/666/suffix/789",
|
|
||||||
body: ``,
|
|
||||||
permissions: []string{"prefix[123].user[456].suffix[789]"},
|
|
||||||
granted: false,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tc := range tt {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
builder := &aicra.Builder{}
|
|
||||||
if err := addBuiltinTypes(builder); err != nil {
|
|
||||||
t.Fatalf("unexpected error <%v>", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// tester middleware (last executed)
|
|
||||||
builder.WithAuth(func(next api.AuthHandlerFunc) api.AuthHandlerFunc {
|
|
||||||
return func(a api.Auth, w http.ResponseWriter, r *http.Request) {
|
|
||||||
if a.Granted() == tc.granted {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if a.Granted() {
|
|
||||||
t.Fatalf("unexpected granted auth")
|
|
||||||
} else {
|
|
||||||
t.Fatalf("expected granted auth")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
// update permissions
|
|
||||||
builder.WithAuth(func(next api.AuthHandlerFunc) api.AuthHandlerFunc {
|
|
||||||
return func(a api.Auth, w http.ResponseWriter, r *http.Request) {
|
|
||||||
a.Active = tc.permissions
|
|
||||||
next(a, w, r)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
err := builder.Setup(strings.NewReader(tc.manifest))
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("setup: unexpected error <%v>", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := builder.Bind(http.MethodPost, tc.path, tc.handler); err != nil {
|
|
||||||
t.Fatalf("bind: unexpected error <%v>", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
handler, err := builder.Build()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("build: unexpected error <%v>", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
response := httptest.NewRecorder()
|
|
||||||
body := strings.NewReader(tc.body)
|
|
||||||
request := httptest.NewRequest(http.MethodPost, tc.url, body)
|
|
||||||
|
|
||||||
// test request
|
|
||||||
handler.ServeHTTP(response, request)
|
|
||||||
if response.Body == nil {
|
|
||||||
t.Fatalf("response has no body")
|
|
||||||
}
|
|
||||||
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue