WIP PROTI

Author: keyurboss

.vscode/launch.json

@@ -4,12 +4,20 @@
     // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
     "version": "0.2.0",
     "configurations": [
+
         {
             "name": "Go Run",
             "type": "go",
             "request": "launch",
             "mode": "auto",
             "program": "main.go"
+        },
+        {
+            "name": "Limit Server Go",
+            "type": "go",
+            "request": "launch",
+            "mode": "auto",
+            "program": "${workspaceFolder}/servers/limit-server/main.go"
         }
     ]
 }

demo.go

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package proto;
 
-option go_package = "src/limit-server";
+option go_package = "servers/limit-server/src";
 
 enum BuySell {
     BUY = 0;

proto/limit-server.proto

@@ -0,0 +1,7 @@
+package main
+
+import limitserver "github.com/rpsoftech/bullion-server/servers/limit-server/src"
+
+func main() {
+	limitserver.Start()
+}

servers/limit-server/main.go

@@ -0,0 +1,52 @@
+package interceptor
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/rpsoftech/bullion-server/src/env"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/metadata"
+	"google.golang.org/grpc/status"
+)
+
+var (
+	errMissingMetadata = status.Errorf(codes.InvalidArgument, "missing metadata")
+	errInvalidToken    = status.Errorf(codes.Unauthenticated, "invalid token")
+)
+
+// func(ctx context.Context, req any, info *UnaryServerInfo, handler UnaryHandler) (resp any, err error)
+
+func OAuth2Interceptor(ctx context.Context, req any, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp any, err error) {
+
+	// return nil, nil
+	md, ok := metadata.FromIncomingContext(ctx)
+	if !ok {
+		return nil, errMissingMetadata
+	}
+	if !valid(md[env.RequestTokenHeaderKey]) {
+		return nil, errInvalidToken
+	}
+	m, err := handler(ctx, req)
+	if err != nil {
+		logger("RPC failed with error: %v", err)
+	}
+	return m, err
+}
+func logger(format string, a ...any) {
+	fmt.Printf("LOG:\t"+format+"\n", a...)
+}
+
+func valid(authorization []string) bool {
+	if len(authorization) < 1 {
+		return false
+	}
+	token := strings.TrimPrefix(authorization[0], "Bearer ")
+	println(authorization)
+	// Perform the token validation here. For the sake of this example, the code
+	// here forgoes any of the usual OAuth2 token validation and instead checks
+	// for a token matching an arbitrary string.
+	return token == "some-secret-token"
+}

servers/limit-server/src/Interceptor/oAuth2.go

@@ -5,6 +5,7 @@ import (
 	"io"
 	"net"
 
+	interceptor "github.com/rpsoftech/bullion-server/servers/limit-server/src/Interceptor"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/reflection"
 )
@@ -21,14 +22,17 @@ func Start() {
 		panic(err)
 	}
 	srv := grpc.NewServer(
+		grpc.UnaryInterceptor(interceptor.OAuth2Interceptor),
+	// grpc.MaxConcurrentStreams(3),
+	// grpc.
 	// grpc.UnaryInterceptor()
 	)
 	// re
 	RegisterLimitServerServer(srv, &LimitServerService{})
 	reflection.Register(srv)
 
 	if e := srv.Serve(lis); e != nil {
-		panic(err)
+		panic(e)
 	}
 }