Generate endpoints + transports

examples/go-kit/templates/{{.File.Package}}/gen/client/client.go.tmpl

@@ -1 +0,0 @@
-package {{.File.Package}}_client

examples/go-kit/templates/{{.File.Package}}/gen/cmd/{{.File.Package}}/main.go.tmpl

@@ -1 +0,0 @@
-package main

examples/go-kit/templates/{{.File.Package}}/gen/endpoints/endpoints.go.tmpl

@@ -1 +1,49 @@
-package {{.File.Package}}_endpoints
+package {{.File.Package}}_endpoints
+
+{{$file := .File}}
+
+import (
+	"fmt"
+	context "golang.org/x/net/context"
+	pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/{{.File.Package}}"
+	"github.com/go-kit/kit/endpoint"
+)
+
+var _ = fmt.Errorf
+
+type Endpoints struct {
+	{{range .Service.Method}}
+	{{.Name}}Endpoint endpoint.Endpoint
+	{{end}}
+}
+
+{{range .Service.Method}}
+func (e *Endpoints){{.Name}}(ctx context.Context, in *pb.{{.Name}}Request) (*pb.{{.Name}}Response, error) {
+	out, err := e.{{.Name}}Endpoint(ctx, in)
+	if err != nil {
+		return &pb.{{.Name}}Response{ErrMsg: err.Error()}, err
+	}
+	return out.(*pb.{{.Name}}Reply), err
+}
+{{end}}
+
+{{range .Service.Method}}
+func Make{{.Name}}Endpoint(svc pb.{{$file.Package | title}}ServiceServer) endpoint.Endpoint {
+	return func(ctx context.Context, request interface{}) (interface{}, error) {
+		req := request.(*pb.{{.Name}}Request)
+		rep, err := svc.{{.Name}}(ctx, req)
+		if err != nil {
+			return &pb.{{.Name}}Reply{ErrMsg: err.Error()}, err
+		}
+		return rep, nil
+	}
+}
+{{end}}
+
+func MakeEndpoints(svc pb.{{.File.Package | title}}ServiceServer) Endpoints {
+	return Endpoints{
+		{{range .Service.Method}}
+		{{.Name}}Endpoint: Make{{.Name}}Endpoint(svc),
+		{{end}}
+	}
+}

examples/go-kit/templates/{{.File.Package}}/gen/init/init.go.tmpl

@@ -1 +0,0 @@
-package {{.File.Package}}_init

examples/go-kit/templates/{{.File.Package}}/gen/transports/grpc/grpc.go.tmpl

@@ -1 +1,59 @@
-package {{.File.Package}}_transportgrpc
+package {{.File.Package}}_transportgrpc
+
+{{$file := .File}}
+
+import (
+        "fmt"
+
+	context "golang.org/x/net/context"
+	pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/{{.File.Package}}"
+	grpctransport "github.com/go-kit/kit/transport/grpc"
+	endpoint "github.com/moul/protoc-gen-gotemplate/examples/go-kit/{{.File.Package}}/gen/endpoints"
+)
+
+// avoid import errors
+var _ = fmt.Errorf
+
+func MakeGRPCServer(ctx context.Context, endpoints endpoint.Endpoints) pb.{{.File.Package | title}}ServiceServer {
+	options := []grpctransport.ServerOption{}
+	return &grpcServer{
+		{{range .Service.Method}}
+                {{if not .ServerStreaming}}
+                {{if not .ClientStreaming}}
+		{{.Name | lower}}: grpctransport.NewServer(
+			ctx,
+			endpoints.{{.Name}}Endpoint,
+			decode{{.Name}}Request,
+			encode{{.Name}}Response,
+			options,
+		),
+                {{end}}
+		{{end}}
+                {{end}}
+	}
+}
+
+type grpcServer struct {
+	{{range .Service.Method}}
+	{{.Name | lower}} grpctransport.Handler
+	{{end}}
+}
+
+{{range .Service.Method}}
+func (s *grpcServer) {{.Name}}(ctx context.Context, req *pb.{{.Name}}Request) (*pb.{{.Name}}Reply, error) {
+	_, rep, err := s.{{.Name | lower}}.ServeGRPC(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+	return rep.(*pb.{{.Name}}Reply), nil
+}
+
+func decode{{.Name}}Request(ctx context.Context, grpcReq interface{}) (interface{}, error) {
+	return grpcReq, nil
+}
+
+func encode{{.Name}}Response(ctx context.Context, response interface{}) (interface{}, error) {
+	resp := response.(*pb.{{.Name}}Reply)
+	return resp, nil
+}
+{{end}}

examples/go-kit/templates/{{.File.Package}}/gen/transports/http/http.go.tmpl

@@ -1 +1,47 @@
-package {{.File.Package}}_transporthttp
+package {{.File.Package}}_transporthttp
+
+{{$file := .File}}
+
+import (
+       "log"
+	"net/http"
+	"encoding/json"
+	context "golang.org/x/net/context"
+
+	pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/{{.File.Package}}"
+        gokit_endpoint "github.com/go-kit/kit/endpoint"
+	httptransport "github.com/go-kit/kit/transport/http"
+	endpoints "github.com/moul/protoc-gen-gotemplate/examples/go-kit/{{.File.Package}}/gen/endpoints"
+)
+
+{{range .Service.Method}}
+func Make{{.Name}}Handler(ctx context.Context, svc pb.{{$file.Package | title}}ServiceServer, endpoint gokit_endpoint.Endpoint) *httptransport.Server {
+	return httptransport.NewServer(
+		ctx,
+		endpoint,
+		decode{{.Name}}Request,
+		encode{{.Name}}Response,
+		append([]httptransport.ServerOption{}, httptransport.ServerBefore(jwt.ToHTTPContext()))...,
+	)
+}
+
+func decode{{.Name}}Request(ctx context.Context, r *http.Request) (interface{}, error) {
+	var req pb.{{.Name}}Request
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		return nil, err
+	}
+	return &req, nil
+}
+
+func encode{{.Name}}Response(ctx context.Context, w http.ResponseWriter, response interface{}) error {
+	return json.NewEncoder(w).Encode(response)
+}
+{{end}}
+
+func RegisterHandlers(ctx context.Context, svc pb.{{$file.Package | title}}ServiceServer, mux *http.ServeMux, endpoints endpoints.Endpoints) error {
+	{{range .Service.Method}}
+        log.Println("new HTTP endpoint: \"/{{.Name}}\" (service={{$file.Package | title}})")
+	mux.Handle("/{{.Name}}", Make{{.Name}}Handler(ctx, svc, endpoints.{{.Name}}Endpoint))
+	{{end}}
+	return nil
+}