Add initial Python plugin

Author: lbeckman314

buf.gen.yaml

@@ -1,11 +1,23 @@
 version: v1
 plugins:
+  # Go (Protobuf)
   - plugin: buf.build/protocolbuffers/go
     out: .
     opt:
       - paths=source_relative
+
+  # Go (gRPC)
   - plugin: buf.build/grpc/go:v1.3.0
     out: .
     opt:
       - paths=source_relative
       - require_unimplemented_servers=false
+
+  # Python (Protobuf)
+  - plugin: buf.build/protocolbuffers/python:v24.4
+    out: plugin-python
+
+  # Python (gRPC)
+  - plugin: buf.build/grpc/python:v1.58.1
+    out: plugin-python
+

plugin/auth_impl.go plugin-go/auth_impl.goplugin/auth_impl.go renamed to plugin-go/auth_impl.go

@@ -1,6 +1,3 @@
-// Copyright (c) HashiCorp, Inc.
-// SPDX-License-Identifier: MPL-2.0
-
 package main
 
 import (
@@ -16,23 +13,24 @@ import (
 type Authorize struct{}
 
 func (Authorize) Get(user string, host string) ([]byte, error) {
-	shared.Logger.Info("Get", "user", user, "host", host)
-
 	if user == "" {
 		return nil, fmt.Errorf("user is required (e.g. ./authorize <USER> <HOST>)")
 	}
 
+	shared.Logger.Info("Get", "user", user, "host", host)
 	resp, err := http.Get(host + user)
 	if err != nil {
 		return nil, fmt.Errorf("error making request: %w", err)
 	}
 	defer resp.Body.Close()
 
+	shared.Logger.Info("Response", "status", resp.Status)
 	body, err := io.ReadAll(resp.Body)
 	if err != nil {
 		return nil, fmt.Errorf("error reading response body: %w", err)
 	}
 
+	shared.Logger.Info("Response", "body", string(body))
 	return body, nil
 }
 

plugin/auth_impl_test.go plugin-go/auth_impl_test.goplugin/auth_impl_test.go renamed to plugin-go/auth_impl_test.go

@@ -0,0 +1,46 @@
+from concurrent import futures
+import sys
+import time
+
+import grpc
+
+from proto import auth_pb2
+from proto import auth_pb2_grpc
+
+from grpc_health.v1.health import HealthServicer
+from grpc_health.v1 import health_pb2, health_pb2_grpc
+
+class AuthorizeServicer(auth_pb2_grpc.AuthorizeServicer):
+    """Implementation of Authorize service."""
+
+    def Get(self, request, context):
+        filename = "auth_"+request.key
+        with open(filename, 'r+b') as f:
+            result = auth_pb2.GetResponse()
+            result.value = f.read()
+            return result
+
+def serve():
+    # We need to build a health service to work with go-plugin
+    health = HealthServicer()
+    health.set("plugin", health_pb2.HealthCheckResponse.ServingStatus.Value('SERVING'))
+
+    # Start the server.
+    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
+    auth_pb2_grpc.add_AuthorizeServicer_to_server(AuthorizeServicer(), server)
+    health_pb2_grpc.add_HealthServicer_to_server(health, server)
+    server.add_insecure_port('127.0.0.1:1234')
+    server.start()
+
+    # Output information
+    print("1|1|tcp|127.0.0.1:1234|grpc")
+    sys.stdout.flush()
+
+    try:
+        while True:
+            time.sleep(60 * 60 * 24)
+    except KeyboardInterrupt:
+        server.stop(0)
+
+if __name__ == '__main__':
+    serve()

plugin-python/README.md

@@ -0,0 +1,66 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from proto import auth_pb2 as proto_dot_auth__pb2
+
+
+class AuthorizeStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Get = channel.unary_unary(
+                '/proto.Authorize/Get',
+                request_serializer=proto_dot_auth__pb2.GetRequest.SerializeToString,
+                response_deserializer=proto_dot_auth__pb2.GetResponse.FromString,
+                )
+
+
+class AuthorizeServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def Get(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_AuthorizeServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Get': grpc.unary_unary_rpc_method_handler(
+                    servicer.Get,
+                    request_deserializer=proto_dot_auth__pb2.GetRequest.FromString,
+                    response_serializer=proto_dot_auth__pb2.GetResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'proto.Authorize', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class Authorize(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def Get(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/proto.Authorize/Get',
+            proto_dot_auth__pb2.GetRequest.SerializeToString,
+            proto_dot_auth__pb2.GetResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

plugin-python/main.py

@@ -0,0 +1,7 @@
+[project]
+name = "funnel-auth-plugin"
+version = "0.1.0"
+description = "Python version of the Funnel Authoritzation Plugin"
+readme = "README.md"
+requires-python = ">=3.13"
+dependencies = []

plugin-python/proto/auth_pb2.py

@@ -0,0 +1,3 @@
+grpcio
+grpcio-health-checking
+protobuf