add otp service (#88)

This commit is contained in:
Asim Aslam
2021-04-28 10:09:17 +01:00
committed by GitHub
parent 2c8708ad65
commit b96bb863e2
14 changed files with 720 additions and 0 deletions

1
go.mod
View File

@@ -18,6 +18,7 @@ require (
github.com/miekg/dns v1.1.31 // indirect
github.com/nats-io/nats-streaming-server v0.21.1
github.com/paulmach/go.geo v0.0.0-20180829195134-22b514266d33
github.com/pquerna/otp v1.3.0
github.com/stoewer/go-strcase v1.2.0
github.com/stretchr/testify v1.6.1
github.com/ulikunitz/xz v0.5.8 // indirect

4
go.sum
View File

@@ -80,6 +80,8 @@ github.com/blang/semver v3.5.1+incompatible/go.mod h1:kRBLl5iJ+tD4TcOOxsy/0fnweb
github.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869 h1:DDGfHa7BWjL4YnC6+E63dPcxHo2sUxDIu8g3QgEJdRY=
github.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869/go.mod h1:Ekp36dRnpXw/yCqJaO+ZrUyxD+3VXMFFr56k5XYrpB4=
github.com/boltdb/bolt v1.3.1/go.mod h1:clJnj/oiGkjum5o1McbSZDSLxVThjynRyGBgiAx27Ps=
github.com/boombuler/barcode v1.0.1-0.20190219062509-6c824513bacc h1:biVzkmvwrH8WK8raXaxBx6fRVTlJILwEwQGL1I/ByEI=
github.com/boombuler/barcode v1.0.1-0.20190219062509-6c824513bacc/go.mod h1:paBWMcWSl3LHKBqUq+rly7CNSldXjb2rDl3JlRe0mD8=
github.com/caddyserver/certmagic v0.10.6 h1:sCya6FmfaN74oZE46kqfaFOVoROD/mF36rTQfjN7TZc=
github.com/caddyserver/certmagic v0.10.6/go.mod h1:Y8jcUBctgk/IhpAzlHKfimZNyXCkfGgRTC0orl8gROQ=
github.com/cenkalti/backoff/v4 v4.0.0 h1:6VeaLF9aI+MAUQ95106HwWzYZgJJpZ4stumjj6RFYAU=
@@ -471,6 +473,8 @@ github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/pquerna/otp v1.3.0 h1:oJV/SkzR33anKXwQU3Of42rL4wbrffP4uvUf1SvS5Xs=
github.com/pquerna/otp v1.3.0/go.mod h1:dkJfzwRKNiegxyNb54X/3fLwhCynbMspSyWKnvi1AEg=
github.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=
github.com/prometheus/client_golang v0.9.2/go.mod h1:OsXs2jCmiKlQ1lTBmv21f2mNfw4xf/QclQDMrYNZzcM=
github.com/prometheus/client_golang v0.9.3-0.20190127221311-3c4408c8b829/go.mod h1:p2iRAGwDERtqlqzRXnrOVns+ignqQo//hLXqYxZYVNs=

2
otp/.gitignore vendored Normal file
View File

@@ -0,0 +1,2 @@
otp

3
otp/Dockerfile Normal file
View File

@@ -0,0 +1,3 @@
FROM alpine
ADD otp /otp
ENTRYPOINT [ "/otp" ]

26
otp/Makefile Normal file
View File

@@ -0,0 +1,26 @@
GOPATH:=$(shell go env GOPATH)
.PHONY: init
init:
go get -u github.com/golang/protobuf/proto
go get -u github.com/golang/protobuf/protoc-gen-go
go get github.com/micro/micro/v3/cmd/protoc-gen-micro
.PHONY: proto
proto:
protoc --openapi_out=. --proto_path=. --micro_out=. --go_out=. proto/otp.proto
docs:
protoc --openapi_out=. --proto_path=. --micro_out=. --go_out=. proto/otp.proto
@redoc-cli bundle api-otp.json
.PHONY: build
build:
go build -o otp *.go
.PHONY: test
test:
go test -v ./... -cover
.PHONY: docker
docker:
docker build . -t otp:latest

23
otp/README.md Normal file
View File

@@ -0,0 +1,23 @@
# Otp Service
This is the Otp service
Generated with
```
micro new otp
```
## Usage
Generate the proto code
```
make proto
```
Run the service
```
micro run .
```

3
otp/generate.go Normal file
View File

@@ -0,0 +1,3 @@
package main
//go:generate make proto

92
otp/handler/otp.go Normal file
View File

@@ -0,0 +1,92 @@
package handler
import (
"context"
"time"
"github.com/micro/micro/v3/service/errors"
"github.com/micro/micro/v3/service/logger"
pb "github.com/micro/services/otp/proto"
"github.com/micro/services/pkg/cache"
"github.com/pquerna/otp"
"github.com/pquerna/otp/totp"
)
type Otp struct{}
func (e *Otp) Generate(ctx context.Context, req *pb.GenerateRequest, rsp *pb.GenerateResponse) error {
if len(req.Id) == 0 {
return errors.BadRequest("otp.generate", "missing id")
}
// check if a key exists for the user
var secret string
if err := cache.Get(req.Id, &secret); err != nil {
// generate a key
key, err := totp.Generate(totp.GenerateOpts{
Issuer: "Micro",
AccountName: req.Id,
})
if err != nil {
logger.Error("Failed to generate secret: %v", err)
return errors.InternalServerError("otp.generate", "failed to generate code")
}
secret = key.Secret()
if err := cache.Put(req.Id, secret, time.Now().Add(time.Minute*5)); err != nil {
logger.Error("Failed to store secret: %v", err)
return errors.InternalServerError("otp.generate", "failed to generate code")
}
}
// generate a new code
code, err := totp.GenerateCodeCustom(secret, time.Now(), totp.ValidateOpts{
Period: 60,
Skew: 1,
Digits: otp.DigitsSix,
Algorithm: otp.AlgorithmSHA512,
})
if err != nil {
return errors.InternalServerError("otp.generate", "failed to generate code: %v", err)
}
// return the code
rsp.Code = code
return nil
}
func (e *Otp) Validate(ctx context.Context, req *pb.ValidateRequest, rsp *pb.ValidateResponse) error {
if len(req.Id) == 0 {
return errors.BadRequest("otp.generate", "missing id")
}
if len(req.Code) == 0 {
return errors.BadRequest("otp.generate", "missing code")
}
var secret string
if err := cache.Get(req.Id, &secret); err != nil {
logger.Error("Failed to get secret from store: %v", err)
return errors.InternalServerError("otp.generate", "failed to validate code")
}
ok, err := totp.ValidateCustom(req.Code, secret, time.Now(), totp.ValidateOpts{
Period: 60,
Skew: 1,
Digits: otp.DigitsSix,
Algorithm: otp.AlgorithmSHA1,
})
if err != nil {
return errors.InternalServerError("otp.generate", "failed to validate code")
}
// set the response
rsp.Success = ok
return nil
}

24
otp/main.go Normal file
View File

@@ -0,0 +1,24 @@
package main
import (
"github.com/micro/services/otp/handler"
pb "github.com/micro/services/otp/proto"
"github.com/micro/micro/v3/service"
"github.com/micro/micro/v3/service/logger"
)
func main() {
// Create service
srv := service.New(
service.Name("otp"),
)
// Register handler
pb.RegisterOtpHandler(srv.Server(), new(handler.Otp))
// Run service
if err := srv.Run(); err != nil {
logger.Fatal(err)
}
}

1
otp/micro.mu Normal file
View File

@@ -0,0 +1 @@
service otp

357
otp/proto/otp.pb.go Normal file
View File

@@ -0,0 +1,357 @@
// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// protoc-gen-go v1.26.0
// protoc v3.15.6
// source: proto/otp.proto
package otp
import (
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
reflect "reflect"
sync "sync"
)
const (
// Verify that this generated code is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
// Verify that runtime/protoimpl is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)
// Generate an OTP (one time pass) code
type GenerateRequest struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
// unique id, email or user to generate an OTP for
Id string `protobuf:"bytes,1,opt,name=id,proto3" json:"id,omitempty"`
}
func (x *GenerateRequest) Reset() {
*x = GenerateRequest{}
if protoimpl.UnsafeEnabled {
mi := &file_proto_otp_proto_msgTypes[0]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *GenerateRequest) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*GenerateRequest) ProtoMessage() {}
func (x *GenerateRequest) ProtoReflect() protoreflect.Message {
mi := &file_proto_otp_proto_msgTypes[0]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use GenerateRequest.ProtoReflect.Descriptor instead.
func (*GenerateRequest) Descriptor() ([]byte, []int) {
return file_proto_otp_proto_rawDescGZIP(), []int{0}
}
func (x *GenerateRequest) GetId() string {
if x != nil {
return x.Id
}
return ""
}
type GenerateResponse struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
// 6 digit one time pass code
Code string `protobuf:"bytes,1,opt,name=code,proto3" json:"code,omitempty"`
}
func (x *GenerateResponse) Reset() {
*x = GenerateResponse{}
if protoimpl.UnsafeEnabled {
mi := &file_proto_otp_proto_msgTypes[1]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *GenerateResponse) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*GenerateResponse) ProtoMessage() {}
func (x *GenerateResponse) ProtoReflect() protoreflect.Message {
mi := &file_proto_otp_proto_msgTypes[1]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use GenerateResponse.ProtoReflect.Descriptor instead.
func (*GenerateResponse) Descriptor() ([]byte, []int) {
return file_proto_otp_proto_rawDescGZIP(), []int{1}
}
func (x *GenerateResponse) GetCode() string {
if x != nil {
return x.Code
}
return ""
}
// Validate the code
type ValidateRequest struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
// unique id, email or user for which the code was generated
Id string `protobuf:"bytes,1,opt,name=id,proto3" json:"id,omitempty"`
// 6 digit one time pass code to validate
Code string `protobuf:"bytes,2,opt,name=code,proto3" json:"code,omitempty"`
}
func (x *ValidateRequest) Reset() {
*x = ValidateRequest{}
if protoimpl.UnsafeEnabled {
mi := &file_proto_otp_proto_msgTypes[2]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *ValidateRequest) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*ValidateRequest) ProtoMessage() {}
func (x *ValidateRequest) ProtoReflect() protoreflect.Message {
mi := &file_proto_otp_proto_msgTypes[2]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use ValidateRequest.ProtoReflect.Descriptor instead.
func (*ValidateRequest) Descriptor() ([]byte, []int) {
return file_proto_otp_proto_rawDescGZIP(), []int{2}
}
func (x *ValidateRequest) GetId() string {
if x != nil {
return x.Id
}
return ""
}
func (x *ValidateRequest) GetCode() string {
if x != nil {
return x.Code
}
return ""
}
type ValidateResponse struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
// returns true if successful
Success bool `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"`
}
func (x *ValidateResponse) Reset() {
*x = ValidateResponse{}
if protoimpl.UnsafeEnabled {
mi := &file_proto_otp_proto_msgTypes[3]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *ValidateResponse) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*ValidateResponse) ProtoMessage() {}
func (x *ValidateResponse) ProtoReflect() protoreflect.Message {
mi := &file_proto_otp_proto_msgTypes[3]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use ValidateResponse.ProtoReflect.Descriptor instead.
func (*ValidateResponse) Descriptor() ([]byte, []int) {
return file_proto_otp_proto_rawDescGZIP(), []int{3}
}
func (x *ValidateResponse) GetSuccess() bool {
if x != nil {
return x.Success
}
return false
}
var File_proto_otp_proto protoreflect.FileDescriptor
var file_proto_otp_proto_rawDesc = []byte{
0x0a, 0x0f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f, 0x6f, 0x74, 0x70, 0x2e, 0x70, 0x72, 0x6f, 0x74,
0x6f, 0x12, 0x03, 0x6f, 0x74, 0x70, 0x22, 0x21, 0x0a, 0x0f, 0x47, 0x65, 0x6e, 0x65, 0x72, 0x61,
0x74, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18,
0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x02, 0x69, 0x64, 0x22, 0x26, 0x0a, 0x10, 0x47, 0x65, 0x6e,
0x65, 0x72, 0x61, 0x74, 0x65, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x12, 0x0a,
0x04, 0x63, 0x6f, 0x64, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x63, 0x6f, 0x64,
0x65, 0x22, 0x35, 0x0a, 0x0f, 0x56, 0x61, 0x6c, 0x69, 0x64, 0x61, 0x74, 0x65, 0x52, 0x65, 0x71,
0x75, 0x65, 0x73, 0x74, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09,
0x52, 0x02, 0x69, 0x64, 0x12, 0x12, 0x0a, 0x04, 0x63, 0x6f, 0x64, 0x65, 0x18, 0x02, 0x20, 0x01,
0x28, 0x09, 0x52, 0x04, 0x63, 0x6f, 0x64, 0x65, 0x22, 0x2c, 0x0a, 0x10, 0x56, 0x61, 0x6c, 0x69,
0x64, 0x61, 0x74, 0x65, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x18, 0x0a, 0x07,
0x73, 0x75, 0x63, 0x63, 0x65, 0x73, 0x73, 0x18, 0x01, 0x20, 0x01, 0x28, 0x08, 0x52, 0x07, 0x73,
0x75, 0x63, 0x63, 0x65, 0x73, 0x73, 0x32, 0x7b, 0x0a, 0x03, 0x4f, 0x74, 0x70, 0x12, 0x39, 0x0a,
0x08, 0x47, 0x65, 0x6e, 0x65, 0x72, 0x61, 0x74, 0x65, 0x12, 0x14, 0x2e, 0x6f, 0x74, 0x70, 0x2e,
0x47, 0x65, 0x6e, 0x65, 0x72, 0x61, 0x74, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a,
0x15, 0x2e, 0x6f, 0x74, 0x70, 0x2e, 0x47, 0x65, 0x6e, 0x65, 0x72, 0x61, 0x74, 0x65, 0x52, 0x65,
0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x12, 0x39, 0x0a, 0x08, 0x56, 0x61, 0x6c, 0x69,
0x64, 0x61, 0x74, 0x65, 0x12, 0x14, 0x2e, 0x6f, 0x74, 0x70, 0x2e, 0x56, 0x61, 0x6c, 0x69, 0x64,
0x61, 0x74, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x15, 0x2e, 0x6f, 0x74, 0x70,
0x2e, 0x56, 0x61, 0x6c, 0x69, 0x64, 0x61, 0x74, 0x65, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73,
0x65, 0x22, 0x00, 0x42, 0x0d, 0x5a, 0x0b, 0x2e, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x3b, 0x6f,
0x74, 0x70, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var (
file_proto_otp_proto_rawDescOnce sync.Once
file_proto_otp_proto_rawDescData = file_proto_otp_proto_rawDesc
)
func file_proto_otp_proto_rawDescGZIP() []byte {
file_proto_otp_proto_rawDescOnce.Do(func() {
file_proto_otp_proto_rawDescData = protoimpl.X.CompressGZIP(file_proto_otp_proto_rawDescData)
})
return file_proto_otp_proto_rawDescData
}
var file_proto_otp_proto_msgTypes = make([]protoimpl.MessageInfo, 4)
var file_proto_otp_proto_goTypes = []interface{}{
(*GenerateRequest)(nil), // 0: otp.GenerateRequest
(*GenerateResponse)(nil), // 1: otp.GenerateResponse
(*ValidateRequest)(nil), // 2: otp.ValidateRequest
(*ValidateResponse)(nil), // 3: otp.ValidateResponse
}
var file_proto_otp_proto_depIdxs = []int32{
0, // 0: otp.Otp.Generate:input_type -> otp.GenerateRequest
2, // 1: otp.Otp.Validate:input_type -> otp.ValidateRequest
1, // 2: otp.Otp.Generate:output_type -> otp.GenerateResponse
3, // 3: otp.Otp.Validate:output_type -> otp.ValidateResponse
2, // [2:4] is the sub-list for method output_type
0, // [0:2] is the sub-list for method input_type
0, // [0:0] is the sub-list for extension type_name
0, // [0:0] is the sub-list for extension extendee
0, // [0:0] is the sub-list for field type_name
}
func init() { file_proto_otp_proto_init() }
func file_proto_otp_proto_init() {
if File_proto_otp_proto != nil {
return
}
if !protoimpl.UnsafeEnabled {
file_proto_otp_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*GenerateRequest); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_proto_otp_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*GenerateResponse); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_proto_otp_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*ValidateRequest); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_proto_otp_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*ValidateResponse); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
}
type x struct{}
out := protoimpl.TypeBuilder{
File: protoimpl.DescBuilder{
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
RawDescriptor: file_proto_otp_proto_rawDesc,
NumEnums: 0,
NumMessages: 4,
NumExtensions: 0,
NumServices: 1,
},
GoTypes: file_proto_otp_proto_goTypes,
DependencyIndexes: file_proto_otp_proto_depIdxs,
MessageInfos: file_proto_otp_proto_msgTypes,
}.Build()
File_proto_otp_proto = out.File
file_proto_otp_proto_rawDesc = nil
file_proto_otp_proto_goTypes = nil
file_proto_otp_proto_depIdxs = nil
}

110
otp/proto/otp.pb.micro.go Normal file
View File

@@ -0,0 +1,110 @@
// Code generated by protoc-gen-micro. DO NOT EDIT.
// source: proto/otp.proto
package otp
import (
fmt "fmt"
proto "github.com/golang/protobuf/proto"
math "math"
)
import (
context "context"
api "github.com/micro/micro/v3/service/api"
client "github.com/micro/micro/v3/service/client"
server "github.com/micro/micro/v3/service/server"
)
// Reference imports to suppress errors if they are not otherwise used.
var _ = proto.Marshal
var _ = fmt.Errorf
var _ = math.Inf
// This is a compile-time assertion to ensure that this generated file
// is compatible with the proto package it is being compiled against.
// A compilation error at this line likely means your copy of the
// proto package needs to be updated.
const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package
// Reference imports to suppress errors if they are not otherwise used.
var _ api.Endpoint
var _ context.Context
var _ client.Option
var _ server.Option
// Api Endpoints for Otp service
func NewOtpEndpoints() []*api.Endpoint {
return []*api.Endpoint{}
}
// Client API for Otp service
type OtpService interface {
Generate(ctx context.Context, in *GenerateRequest, opts ...client.CallOption) (*GenerateResponse, error)
Validate(ctx context.Context, in *ValidateRequest, opts ...client.CallOption) (*ValidateResponse, error)
}
type otpService struct {
c client.Client
name string
}
func NewOtpService(name string, c client.Client) OtpService {
return &otpService{
c: c,
name: name,
}
}
func (c *otpService) Generate(ctx context.Context, in *GenerateRequest, opts ...client.CallOption) (*GenerateResponse, error) {
req := c.c.NewRequest(c.name, "Otp.Generate", in)
out := new(GenerateResponse)
err := c.c.Call(ctx, req, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *otpService) Validate(ctx context.Context, in *ValidateRequest, opts ...client.CallOption) (*ValidateResponse, error) {
req := c.c.NewRequest(c.name, "Otp.Validate", in)
out := new(ValidateResponse)
err := c.c.Call(ctx, req, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
// Server API for Otp service
type OtpHandler interface {
Generate(context.Context, *GenerateRequest, *GenerateResponse) error
Validate(context.Context, *ValidateRequest, *ValidateResponse) error
}
func RegisterOtpHandler(s server.Server, hdlr OtpHandler, opts ...server.HandlerOption) error {
type otp interface {
Generate(ctx context.Context, in *GenerateRequest, out *GenerateResponse) error
Validate(ctx context.Context, in *ValidateRequest, out *ValidateResponse) error
}
type Otp struct {
otp
}
h := &otpHandler{hdlr}
return s.Handle(s.NewHandler(&Otp{h}, opts...))
}
type otpHandler struct {
OtpHandler
}
func (h *otpHandler) Generate(ctx context.Context, in *GenerateRequest, out *GenerateResponse) error {
return h.OtpHandler.Generate(ctx, in, out)
}
func (h *otpHandler) Validate(ctx context.Context, in *ValidateRequest, out *ValidateResponse) error {
return h.OtpHandler.Validate(ctx, in, out)
}

34
otp/proto/otp.proto Normal file
View File

@@ -0,0 +1,34 @@
syntax = "proto3";
package otp;
option go_package = "./proto;otp";
service Otp {
rpc Generate(GenerateRequest) returns (GenerateResponse) {}
rpc Validate(ValidateRequest) returns (ValidateResponse) {}
}
// Generate an OTP (one time pass) code
message GenerateRequest {
// unique id, email or user to generate an OTP for
string id = 1;
}
message GenerateResponse {
// 6 digit one time pass code
string code = 1;
}
// Validate the code
message ValidateRequest {
// unique id, email or user for which the code was generated
string id = 1;
// 6 digit one time pass code to validate
string code = 2;
}
message ValidateResponse {
// returns true if successful
bool success = 1;
}

40
pkg/cache/cache.go vendored Normal file
View File

@@ -0,0 +1,40 @@
// Cache provides a simple marshaling layer on top of the store
package cache
import (
"encoding/json"
"time"
"github.com/micro/micro/v3/service/store"
)
func Get(key string, val interface{}) error {
recs, err := store.Read(key, store.ReadLimit(1))
if err != nil {
return err
}
if len(recs) == 0 {
return store.ErrNotFound
}
if err := json.Unmarshal(recs[0].Value, val); err != nil {
return err
}
return nil
}
func Put(key string, val interface{}, expires time.Time) error {
b, err := json.Marshal(val)
if err != nil {
return err
}
expiry := expires.Sub(time.Now())
return store.Write(&store.Record{
Key: key,
Value: b,
Expiry: expiry,
})
}
func Delete(key string) error {
return store.Delete(key)
}