mirror of
https://github.com/mitchell/selfpass.git
synced 2025-12-14 21:27:22 +00:00
Move service related filed to services folder
This commit is contained in:
mitchell
parent
347fbe7268
commit
7d770ef150
41 changed files with 50 additions and 50 deletions
254
services/credentials/transport/encoding.go
Normal file
254
services/credentials/transport/encoding.go
Normal file
|
|
@ -0,0 +1,254 @@
|
|||
package transport
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/golang/protobuf/ptypes"
|
||||
|
||||
"github.com/mitchell/selfpass/services/credentials/endpoints"
|
||||
"github.com/mitchell/selfpass/services/credentials/protobuf"
|
||||
"github.com/mitchell/selfpass/services/credentials/types"
|
||||
)
|
||||
|
||||
func decodeGetAllMetadataRequest(ctx context.Context, request interface{}) (interface{}, error) {
|
||||
r := request.(protobuf.GetAllMetadataRequest)
|
||||
return endpoints.GetAllMetadataRequest{
|
||||
SourceHost: r.SourceHost,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func EncodeGetAllMetadataRequest(ctx context.Context, request interface{}) (interface{}, error) {
|
||||
r := request.(endpoints.GetAllMetadataRequest)
|
||||
return protobuf.GetAllMetadataRequest{
|
||||
SourceHost: r.SourceHost,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func encodeDumpResponse(ctx context.Context, response interface{}) (interface{}, error) {
|
||||
r := response.(endpoints.DumpResponse)
|
||||
return protobuf.DumpResponse{
|
||||
Contents: r.Contents,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func encodeMetadataStreamResponse(ctx context.Context, response interface{}) (interface{}, error) {
|
||||
r := response.(endpoints.MetadataStream)
|
||||
pbmdch := make(chan protobuf.Metadata, 1)
|
||||
|
||||
go func() {
|
||||
defer close(pbmdch)
|
||||
|
||||
for md := range r.Metadata {
|
||||
createdAt, err := ptypes.TimestampProto(md.CreatedAt)
|
||||
if err != nil {
|
||||
r.Errors <- err
|
||||
return
|
||||
}
|
||||
|
||||
updatedAt, err := ptypes.TimestampProto(md.UpdatedAt)
|
||||
if err != nil {
|
||||
r.Errors <- err
|
||||
return
|
||||
}
|
||||
|
||||
pbmdch <- protobuf.Metadata{
|
||||
Id: md.ID,
|
||||
CreatedAt: createdAt,
|
||||
UpdatedAt: updatedAt,
|
||||
SourceHost: md.SourceHost,
|
||||
Primary: md.Primary,
|
||||
LoginUrl: md.LoginURL,
|
||||
Tag: md.Tag,
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
return ProtobufMetadataStream{
|
||||
Metadata: pbmdch,
|
||||
Errors: r.Errors,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func DecodeMetdataStreamResponse(ctx context.Context, r ProtobufMetadataStream) (endpoints.MetadataStream, error) {
|
||||
mdch := make(chan types.Metadata, 1)
|
||||
errch := make(chan error, 1)
|
||||
|
||||
go func() {
|
||||
defer close(mdch)
|
||||
|
||||
for pbmd := range r.Metadata {
|
||||
createdAt, err := ptypes.Timestamp(pbmd.CreatedAt)
|
||||
if err != nil {
|
||||
errch <- err
|
||||
return
|
||||
}
|
||||
|
||||
updatedAt, err := ptypes.Timestamp(pbmd.UpdatedAt)
|
||||
if err != nil {
|
||||
errch <- err
|
||||
return
|
||||
}
|
||||
|
||||
mdch <- types.Metadata{
|
||||
ID: pbmd.Id,
|
||||
SourceHost: pbmd.SourceHost,
|
||||
CreatedAt: createdAt,
|
||||
UpdatedAt: updatedAt,
|
||||
Primary: pbmd.Primary,
|
||||
LoginURL: pbmd.LoginUrl,
|
||||
Tag: pbmd.Tag,
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
return endpoints.MetadataStream{
|
||||
Metadata: mdch,
|
||||
Errors: errch,
|
||||
}, nil
|
||||
}
|
||||
|
||||
type ProtobufMetadataStream struct {
|
||||
Metadata <-chan protobuf.Metadata
|
||||
Errors chan error
|
||||
}
|
||||
|
||||
func decodeCredentialRequest(ctx context.Context, request interface{}) (interface{}, error) {
|
||||
r := request.(protobuf.CredentialRequest)
|
||||
|
||||
return types.CredentialInput{
|
||||
MetadataInput: types.MetadataInput{
|
||||
Primary: r.Primary,
|
||||
LoginURL: r.LoginUrl,
|
||||
SourceHost: r.SourceHost,
|
||||
Tag: r.Tag,
|
||||
},
|
||||
Username: r.Username,
|
||||
Email: r.Email,
|
||||
Password: r.Password,
|
||||
OTPSecret: r.OtpSecret,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func EncodeCredentialRequest(r types.CredentialInput) protobuf.CredentialRequest {
|
||||
return protobuf.CredentialRequest{
|
||||
Primary: r.Primary,
|
||||
Username: r.Username,
|
||||
Email: r.Email,
|
||||
Password: r.Password,
|
||||
OtpSecret: r.OTPSecret,
|
||||
SourceHost: r.SourceHost,
|
||||
LoginUrl: r.LoginURL,
|
||||
Tag: r.Tag,
|
||||
}
|
||||
}
|
||||
|
||||
func encodeCredentialResponse(ctx context.Context, response interface{}) (interface{}, error) {
|
||||
r := response.(types.Credential)
|
||||
|
||||
createdAt, err := ptypes.TimestampProto(r.CreatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
updatedAt, err := ptypes.TimestampProto(r.UpdatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return protobuf.Credential{
|
||||
Id: r.ID,
|
||||
CreatedAt: createdAt,
|
||||
UpdatedAt: updatedAt,
|
||||
Primary: r.Primary,
|
||||
SourceHost: r.SourceHost,
|
||||
LoginUrl: r.LoginURL,
|
||||
Tag: r.Tag,
|
||||
Username: r.Username,
|
||||
Email: r.Email,
|
||||
Password: r.Password,
|
||||
OtpSecret: r.OTPSecret,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func DecodeCredential(r protobuf.Credential) (c types.Credential, err error) {
|
||||
|
||||
createdAt, err := ptypes.Timestamp(r.CreatedAt)
|
||||
if err != nil {
|
||||
return c, err
|
||||
}
|
||||
|
||||
updatedAt, err := ptypes.Timestamp(r.UpdatedAt)
|
||||
if err != nil {
|
||||
return c, err
|
||||
}
|
||||
|
||||
return types.Credential{
|
||||
Metadata: types.Metadata{
|
||||
ID: r.Id,
|
||||
SourceHost: r.SourceHost,
|
||||
CreatedAt: createdAt,
|
||||
UpdatedAt: updatedAt,
|
||||
Primary: r.Primary,
|
||||
LoginURL: r.LoginUrl,
|
||||
Tag: r.Tag,
|
||||
},
|
||||
Username: r.Username,
|
||||
Email: r.Email,
|
||||
Password: r.Password,
|
||||
OTPSecret: r.OtpSecret,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func decodeUpdateRequest(ctx context.Context, request interface{}) (interface{}, error) {
|
||||
r := request.(protobuf.UpdateRequest)
|
||||
|
||||
return endpoints.UpdateRequest{
|
||||
ID: r.Id,
|
||||
Credential: types.CredentialInput{
|
||||
MetadataInput: types.MetadataInput{
|
||||
Primary: r.Credential.Primary,
|
||||
SourceHost: r.Credential.SourceHost,
|
||||
LoginURL: r.Credential.LoginUrl,
|
||||
Tag: r.Credential.Tag,
|
||||
},
|
||||
Username: r.Credential.Username,
|
||||
Email: r.Credential.Email,
|
||||
Password: r.Credential.Password,
|
||||
OTPSecret: r.Credential.OtpSecret,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func EncodeUpdateRequest(r endpoints.UpdateRequest) protobuf.UpdateRequest {
|
||||
c := r.Credential
|
||||
return protobuf.UpdateRequest{
|
||||
Id: r.ID,
|
||||
Credential: &protobuf.CredentialRequest{
|
||||
Primary: c.Primary,
|
||||
Username: c.Username,
|
||||
Email: c.Email,
|
||||
Password: c.Password,
|
||||
OtpSecret: c.OTPSecret,
|
||||
SourceHost: c.SourceHost,
|
||||
LoginUrl: c.LoginURL,
|
||||
Tag: c.Tag,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func decodeIdRequest(ctx context.Context, request interface{}) (interface{}, error) {
|
||||
r := request.(protobuf.IdRequest)
|
||||
return endpoints.IDRequest{
|
||||
ID: r.Id,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func EncodeIdRequest(r endpoints.IDRequest) protobuf.IdRequest {
|
||||
return protobuf.IdRequest{
|
||||
Id: r.ID,
|
||||
}
|
||||
}
|
||||
|
||||
func noOp(context.Context, interface{}) (interface{}, error) {
|
||||
return nil, nil
|
||||
}
|
||||
170
services/credentials/transport/grpc_server.go
Normal file
170
services/credentials/transport/grpc_server.go
Normal file
|
|
@ -0,0 +1,170 @@
|
|||
package transport
|
||||
|
||||
import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"github.com/go-kit/kit/log"
|
||||
"github.com/go-kit/kit/transport/grpc"
|
||||
"google.golang.org/grpc/codes"
|
||||
"google.golang.org/grpc/status"
|
||||
|
||||
"github.com/mitchell/selfpass/services/credentials/endpoints"
|
||||
"github.com/mitchell/selfpass/services/credentials/protobuf"
|
||||
"github.com/mitchell/selfpass/services/credentials/types"
|
||||
)
|
||||
|
||||
func NewGRPCServer(svc types.Service, logger log.Logger) GRPCServer {
|
||||
return GRPCServer{
|
||||
getAllMetadata: grpc.NewServer(
|
||||
endpoints.MakeGetAllMetadataEndpoint(svc),
|
||||
decodeGetAllMetadataRequest,
|
||||
encodeMetadataStreamResponse,
|
||||
grpc.ServerErrorLogger(logger),
|
||||
),
|
||||
get: grpc.NewServer(
|
||||
endpoints.MakeGetEndpoint(svc),
|
||||
decodeIdRequest,
|
||||
encodeCredentialResponse,
|
||||
grpc.ServerErrorLogger(logger),
|
||||
),
|
||||
create: grpc.NewServer(
|
||||
endpoints.MakeCreateEndpoint(svc),
|
||||
decodeCredentialRequest,
|
||||
encodeCredentialResponse,
|
||||
grpc.ServerErrorLogger(logger),
|
||||
),
|
||||
update: grpc.NewServer(
|
||||
endpoints.MakeUpdateEndpoint(svc),
|
||||
decodeUpdateRequest,
|
||||
encodeCredentialResponse,
|
||||
grpc.ServerErrorLogger(logger),
|
||||
),
|
||||
delete: grpc.NewServer(
|
||||
endpoints.MakeDeleteEndpoint(svc),
|
||||
decodeIdRequest,
|
||||
noOp,
|
||||
grpc.ServerErrorLogger(logger),
|
||||
),
|
||||
dump: grpc.NewServer(
|
||||
endpoints.MakeDumpEndpoint(svc),
|
||||
noOp,
|
||||
encodeDumpResponse,
|
||||
grpc.ServerErrorLogger(logger),
|
||||
),
|
||||
}
|
||||
}
|
||||
|
||||
type GRPCServer struct {
|
||||
getAllMetadata *grpc.Server
|
||||
get *grpc.Server
|
||||
create *grpc.Server
|
||||
update *grpc.Server
|
||||
delete *grpc.Server
|
||||
dump *grpc.Server
|
||||
}
|
||||
|
||||
func (s GRPCServer) GetAllMetadata(r *protobuf.GetAllMetadataRequest, srv protobuf.CredentialService_GetAllMetadataServer) (err error) {
|
||||
defer func() { err = handlerGRPCError(err) }()
|
||||
|
||||
var i interface{}
|
||||
ctx := srv.Context()
|
||||
|
||||
ctx, i, err = s.getAllMetadata.ServeGRPC(ctx, *r)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
mds := i.(ProtobufMetadataStream)
|
||||
|
||||
receiveLoop:
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
break receiveLoop
|
||||
case err = <-mds.Errors:
|
||||
break receiveLoop
|
||||
case md, ok := <-mds.Metadata:
|
||||
if !ok {
|
||||
break receiveLoop
|
||||
}
|
||||
if err = srv.Send(&md); err != nil {
|
||||
break receiveLoop
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (s GRPCServer) Get(ctx context.Context, r *protobuf.IdRequest) (*protobuf.Credential, error) {
|
||||
ctx, i, err := s.get.ServeGRPC(ctx, *r)
|
||||
if err != nil {
|
||||
err = handlerGRPCError(err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
c := &protobuf.Credential{}
|
||||
*c = i.(protobuf.Credential)
|
||||
return c, nil
|
||||
}
|
||||
|
||||
func (s GRPCServer) Create(ctx context.Context, r *protobuf.CredentialRequest) (*protobuf.Credential, error) {
|
||||
ctx, i, err := s.create.ServeGRPC(ctx, *r)
|
||||
if err != nil {
|
||||
err = handlerGRPCError(err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
c := &protobuf.Credential{}
|
||||
*c = i.(protobuf.Credential)
|
||||
return c, nil
|
||||
}
|
||||
|
||||
func (s GRPCServer) Update(ctx context.Context, r *protobuf.UpdateRequest) (*protobuf.Credential, error) {
|
||||
ctx, i, err := s.update.ServeGRPC(ctx, *r)
|
||||
if err != nil {
|
||||
err = handlerGRPCError(err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
c := &protobuf.Credential{}
|
||||
*c = i.(protobuf.Credential)
|
||||
return c, nil
|
||||
}
|
||||
|
||||
func (s GRPCServer) Delete(ctx context.Context, r *protobuf.IdRequest) (*protobuf.DeleteResponse, error) {
|
||||
ctx, _, err := s.delete.ServeGRPC(ctx, *r)
|
||||
if err != nil {
|
||||
err = handlerGRPCError(err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &protobuf.DeleteResponse{Success: true}, nil
|
||||
}
|
||||
|
||||
func (s GRPCServer) Dump(ctx context.Context, r *protobuf.EmptyRequest) (*protobuf.DumpResponse, error) {
|
||||
ctx, i, err := s.dump.ServeGRPC(ctx, *r)
|
||||
if err != nil {
|
||||
err = handlerGRPCError(err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
res := &protobuf.DumpResponse{}
|
||||
*res = i.(protobuf.DumpResponse)
|
||||
return res, nil
|
||||
}
|
||||
|
||||
func handlerGRPCError(err error) error {
|
||||
if err != nil {
|
||||
switch {
|
||||
case strings.HasPrefix(err.Error(), types.InvalidArgument):
|
||||
err = status.Error(codes.InvalidArgument, err.Error())
|
||||
case strings.HasPrefix(err.Error(), types.NotFound):
|
||||
err = status.Error(codes.NotFound, err.Error())
|
||||
default:
|
||||
err = status.Error(codes.Internal, "an internal error has occurred")
|
||||
}
|
||||
}
|
||||
return err
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue