73 lines
2.1 KiB
Go
73 lines
2.1 KiB
Go
|
//Package middleware provides middlewares for rpc server
|
||
|
|
//
|
||
|
|
//Copyright (C) 2022 Alexander Kiryukhin <i@neonxp.dev>
|
||
|
|
//
|
||
|
|
//This file is part of go.neonxp.dev/jsonrpc2 project.
|
||
|
|
//
|
||
|
|
//This program is free software: you can redistribute it and/or modify
|
||
|
|
//it under the terms of the GNU General Public License as published by
|
||
|
|
//the Free Software Foundation, either version 3 of the License, or
|
||
|
|
//(at your option) any later version.
|
||
|
|
//
|
||
|
|
//This program is distributed in the hope that it will be useful,
|
||
|
|
//but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
//MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
//GNU General Public License for more details.
|
||
|
|
//
|
||
|
|
//You should have received a copy of the GNU General Public License
|
||
|
|
//along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/qri-io/jsonschema"
|
||
|
|
|
||
|
|
"go.neonxp.dev/jsonrpc2/rpc"
|
||
|
|
)
|
||
|
|
|
||
|
|
type MethodSchema struct {
|
||
|
|
Request jsonschema.Schema
|
||
|
|
Response jsonschema.Schema
|
||
|
|
}
|
||
|
|
|
||
|
|
func Validation(serviceSchema map[string]MethodSchema) (rpc.Middleware, error) {
|
||
|
|
return func(handler rpc.RpcHandler) rpc.RpcHandler {
|
||
|
|
return func(ctx context.Context, req *rpc.RpcRequest) *rpc.RpcResponse {
|
||
|
|
if rs, ok := serviceSchema[strings.ToLower(req.Method)]; ok {
|
||
|
|
if errResp := formatError(ctx, req.Id, rs.Request, req.Params); errResp != nil {
|
||
|
|
return errResp
|
||
|
|
}
|
||
|
|
resp := handler(ctx, req)
|
||
|
|
if errResp := formatError(ctx, req.Id, rs.Response, resp.Result); errResp != nil {
|
||
|
|
return errResp
|
||
|
|
}
|
||
|
|
return resp
|
||
|
|
}
|
||
|
|
return handler(ctx, req)
|
||
|
|
}
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func formatError(ctx context.Context, requestId any, schema jsonschema.Schema, data json.RawMessage) *rpc.RpcResponse {
|
||
|
|
errs, err := schema.ValidateBytes(ctx, data)
|
||
|
|
if err != nil {
|
||
|
|
return rpc.ErrorResponse(requestId, err)
|
||
|
|
}
|
||
|
|
if errs != nil && len(errs) > 0 {
|
||
|
|
messages := []string{}
|
||
|
|
for _, msg := range errs {
|
||
|
|
messages = append(messages, fmt.Sprintf("%s: %s", msg.PropertyPath, msg.Message))
|
||
|
|
}
|
||
|
|
return rpc.ErrorResponse(requestId, rpc.Error{
|
||
|
|
Code: rpc.ErrCodeInvalidParams,
|
||
|
|
Message: strings.Join(messages, "\n"),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|