mirror of
https://github.com/dstotijn/hetty.git
synced 2025-07-01 18:47:29 -04:00
Replace GraphQL server with Connect RPC
This commit is contained in:
David Stotijn
340
pkg/proj/proj.connect.go
Normal file
340
pkg/proj/proj.connect.go
Normal file
@ -0,0 +1,340 @@
|
||||
// Code generated by protoc-gen-connect-go. DO NOT EDIT.
|
||||
//
|
||||
// Source: proj/proj.proto
|
||||
|
||||
package proj
|
||||
|
||||
import (
|
||||
connect "connectrpc.com/connect"
|
||||
context "context"
|
||||
errors "errors"
|
||||
http "net/http"
|
||||
strings "strings"
|
||||
)
|
||||
|
||||
// This is a compile-time assertion to ensure that this generated file and the connect package are
|
||||
// compatible. If you get a compiler error that this constant is not defined, this code was
|
||||
// generated with a version of connect newer than the one compiled into your binary. You can fix the
|
||||
// problem by either regenerating this code with an older version of connect or updating the connect
|
||||
// version compiled into your binary.
|
||||
const _ = connect.IsAtLeastVersion1_13_0
|
||||
|
||||
const (
|
||||
// ProjectServiceName is the fully-qualified name of the ProjectService service.
|
||||
ProjectServiceName = "hetty.proj.v1.ProjectService"
|
||||
)
|
||||
|
||||
// These constants are the fully-qualified names of the RPCs defined in this package. They're
|
||||
// exposed at runtime as Spec.Procedure and as the final two segments of the HTTP route.
|
||||
//
|
||||
// Note that these are different from the fully-qualified method names used by
|
||||
// google.golang.org/protobuf/reflect/protoreflect. To convert from these constants to
|
||||
// reflection-formatted method names, remove the leading slash and convert the remaining slash to a
|
||||
// period.
|
||||
const (
|
||||
// ProjectServiceCreateProjectProcedure is the fully-qualified name of the ProjectService's
|
||||
// CreateProject RPC.
|
||||
ProjectServiceCreateProjectProcedure = "/hetty.proj.v1.ProjectService/CreateProject"
|
||||
// ProjectServiceOpenProjectProcedure is the fully-qualified name of the ProjectService's
|
||||
// OpenProject RPC.
|
||||
ProjectServiceOpenProjectProcedure = "/hetty.proj.v1.ProjectService/OpenProject"
|
||||
// ProjectServiceCloseProjectProcedure is the fully-qualified name of the ProjectService's
|
||||
// CloseProject RPC.
|
||||
ProjectServiceCloseProjectProcedure = "/hetty.proj.v1.ProjectService/CloseProject"
|
||||
// ProjectServiceDeleteProjectProcedure is the fully-qualified name of the ProjectService's
|
||||
// DeleteProject RPC.
|
||||
ProjectServiceDeleteProjectProcedure = "/hetty.proj.v1.ProjectService/DeleteProject"
|
||||
// ProjectServiceGetActiveProjectProcedure is the fully-qualified name of the ProjectService's
|
||||
// GetActiveProject RPC.
|
||||
ProjectServiceGetActiveProjectProcedure = "/hetty.proj.v1.ProjectService/GetActiveProject"
|
||||
// ProjectServiceListProjectsProcedure is the fully-qualified name of the ProjectService's
|
||||
// ListProjects RPC.
|
||||
ProjectServiceListProjectsProcedure = "/hetty.proj.v1.ProjectService/ListProjects"
|
||||
// ProjectServiceUpdateInterceptSettingsProcedure is the fully-qualified name of the
|
||||
// ProjectService's UpdateInterceptSettings RPC.
|
||||
ProjectServiceUpdateInterceptSettingsProcedure = "/hetty.proj.v1.ProjectService/UpdateInterceptSettings"
|
||||
// ProjectServiceSetScopeRulesProcedure is the fully-qualified name of the ProjectService's
|
||||
// SetScopeRules RPC.
|
||||
ProjectServiceSetScopeRulesProcedure = "/hetty.proj.v1.ProjectService/SetScopeRules"
|
||||
// ProjectServiceSetRequestLogsFilterProcedure is the fully-qualified name of the ProjectService's
|
||||
// SetRequestLogsFilter RPC.
|
||||
ProjectServiceSetRequestLogsFilterProcedure = "/hetty.proj.v1.ProjectService/SetRequestLogsFilter"
|
||||
)
|
||||
|
||||
// ProjectServiceClient is a client for the hetty.proj.v1.ProjectService service.
|
||||
type ProjectServiceClient interface {
|
||||
CreateProject(context.Context, *connect.Request[CreateProjectRequest]) (*connect.Response[CreateProjectResponse], error)
|
||||
OpenProject(context.Context, *connect.Request[OpenProjectRequest]) (*connect.Response[OpenProjectResponse], error)
|
||||
CloseProject(context.Context, *connect.Request[CloseProjectRequest]) (*connect.Response[CloseProjectResponse], error)
|
||||
DeleteProject(context.Context, *connect.Request[DeleteProjectRequest]) (*connect.Response[DeleteProjectResponse], error)
|
||||
GetActiveProject(context.Context, *connect.Request[GetActiveProjectRequest]) (*connect.Response[GetActiveProjectResponse], error)
|
||||
ListProjects(context.Context, *connect.Request[ListProjectsRequest]) (*connect.Response[ListProjectsResponse], error)
|
||||
UpdateInterceptSettings(context.Context, *connect.Request[UpdateInterceptSettingsRequest]) (*connect.Response[UpdateInterceptSettingsResponse], error)
|
||||
SetScopeRules(context.Context, *connect.Request[SetScopeRulesRequest]) (*connect.Response[SetScopeRulesResponse], error)
|
||||
SetRequestLogsFilter(context.Context, *connect.Request[SetRequestLogsFilterRequest]) (*connect.Response[SetRequestLogsFilterResponse], error)
|
||||
}
|
||||
|
||||
// NewProjectServiceClient constructs a client for the hetty.proj.v1.ProjectService service. By
|
||||
// default, it uses the Connect protocol with the binary Protobuf Codec, asks for gzipped responses,
|
||||
// and sends uncompressed requests. To use the gRPC or gRPC-Web protocols, supply the
|
||||
// connect.WithGRPC() or connect.WithGRPCWeb() options.
|
||||
//
|
||||
// The URL supplied here should be the base URL for the Connect or gRPC server (for example,
|
||||
// http://api.acme.com or https://acme.com/grpc).
|
||||
func NewProjectServiceClient(httpClient connect.HTTPClient, baseURL string, opts ...connect.ClientOption) ProjectServiceClient {
|
||||
baseURL = strings.TrimRight(baseURL, "/")
|
||||
projectServiceMethods := File_proj_proj_proto.Services().ByName("ProjectService").Methods()
|
||||
return &projectServiceClient{
|
||||
createProject: connect.NewClient[CreateProjectRequest, CreateProjectResponse](
|
||||
httpClient,
|
||||
baseURL+ProjectServiceCreateProjectProcedure,
|
||||
connect.WithSchema(projectServiceMethods.ByName("CreateProject")),
|
||||
connect.WithClientOptions(opts...),
|
||||
),
|
||||
openProject: connect.NewClient[OpenProjectRequest, OpenProjectResponse](
|
||||
httpClient,
|
||||
baseURL+ProjectServiceOpenProjectProcedure,
|
||||
connect.WithSchema(projectServiceMethods.ByName("OpenProject")),
|
||||
connect.WithClientOptions(opts...),
|
||||
),
|
||||
closeProject: connect.NewClient[CloseProjectRequest, CloseProjectResponse](
|
||||
httpClient,
|
||||
baseURL+ProjectServiceCloseProjectProcedure,
|
||||
connect.WithSchema(projectServiceMethods.ByName("CloseProject")),
|
||||
connect.WithClientOptions(opts...),
|
||||
),
|
||||
deleteProject: connect.NewClient[DeleteProjectRequest, DeleteProjectResponse](
|
||||
httpClient,
|
||||
baseURL+ProjectServiceDeleteProjectProcedure,
|
||||
connect.WithSchema(projectServiceMethods.ByName("DeleteProject")),
|
||||
connect.WithClientOptions(opts...),
|
||||
),
|
||||
getActiveProject: connect.NewClient[GetActiveProjectRequest, GetActiveProjectResponse](
|
||||
httpClient,
|
||||
baseURL+ProjectServiceGetActiveProjectProcedure,
|
||||
connect.WithSchema(projectServiceMethods.ByName("GetActiveProject")),
|
||||
connect.WithClientOptions(opts...),
|
||||
),
|
||||
listProjects: connect.NewClient[ListProjectsRequest, ListProjectsResponse](
|
||||
httpClient,
|
||||
baseURL+ProjectServiceListProjectsProcedure,
|
||||
connect.WithSchema(projectServiceMethods.ByName("ListProjects")),
|
||||
connect.WithClientOptions(opts...),
|
||||
),
|
||||
updateInterceptSettings: connect.NewClient[UpdateInterceptSettingsRequest, UpdateInterceptSettingsResponse](
|
||||
httpClient,
|
||||
baseURL+ProjectServiceUpdateInterceptSettingsProcedure,
|
||||
connect.WithSchema(projectServiceMethods.ByName("UpdateInterceptSettings")),
|
||||
connect.WithClientOptions(opts...),
|
||||
),
|
||||
setScopeRules: connect.NewClient[SetScopeRulesRequest, SetScopeRulesResponse](
|
||||
httpClient,
|
||||
baseURL+ProjectServiceSetScopeRulesProcedure,
|
||||
connect.WithSchema(projectServiceMethods.ByName("SetScopeRules")),
|
||||
connect.WithClientOptions(opts...),
|
||||
),
|
||||
setRequestLogsFilter: connect.NewClient[SetRequestLogsFilterRequest, SetRequestLogsFilterResponse](
|
||||
httpClient,
|
||||
baseURL+ProjectServiceSetRequestLogsFilterProcedure,
|
||||
connect.WithSchema(projectServiceMethods.ByName("SetRequestLogsFilter")),
|
||||
connect.WithClientOptions(opts...),
|
||||
),
|
||||
}
|
||||
}
|
||||
|
||||
// projectServiceClient implements ProjectServiceClient.
|
||||
type projectServiceClient struct {
|
||||
createProject *connect.Client[CreateProjectRequest, CreateProjectResponse]
|
||||
openProject *connect.Client[OpenProjectRequest, OpenProjectResponse]
|
||||
closeProject *connect.Client[CloseProjectRequest, CloseProjectResponse]
|
||||
deleteProject *connect.Client[DeleteProjectRequest, DeleteProjectResponse]
|
||||
getActiveProject *connect.Client[GetActiveProjectRequest, GetActiveProjectResponse]
|
||||
listProjects *connect.Client[ListProjectsRequest, ListProjectsResponse]
|
||||
updateInterceptSettings *connect.Client[UpdateInterceptSettingsRequest, UpdateInterceptSettingsResponse]
|
||||
setScopeRules *connect.Client[SetScopeRulesRequest, SetScopeRulesResponse]
|
||||
setRequestLogsFilter *connect.Client[SetRequestLogsFilterRequest, SetRequestLogsFilterResponse]
|
||||
}
|
||||
|
||||
// CreateProject calls hetty.proj.v1.ProjectService.CreateProject.
|
||||
func (c *projectServiceClient) CreateProject(ctx context.Context, req *connect.Request[CreateProjectRequest]) (*connect.Response[CreateProjectResponse], error) {
|
||||
return c.createProject.CallUnary(ctx, req)
|
||||
}
|
||||
|
||||
// OpenProject calls hetty.proj.v1.ProjectService.OpenProject.
|
||||
func (c *projectServiceClient) OpenProject(ctx context.Context, req *connect.Request[OpenProjectRequest]) (*connect.Response[OpenProjectResponse], error) {
|
||||
return c.openProject.CallUnary(ctx, req)
|
||||
}
|
||||
|
||||
// CloseProject calls hetty.proj.v1.ProjectService.CloseProject.
|
||||
func (c *projectServiceClient) CloseProject(ctx context.Context, req *connect.Request[CloseProjectRequest]) (*connect.Response[CloseProjectResponse], error) {
|
||||
return c.closeProject.CallUnary(ctx, req)
|
||||
}
|
||||
|
||||
// DeleteProject calls hetty.proj.v1.ProjectService.DeleteProject.
|
||||
func (c *projectServiceClient) DeleteProject(ctx context.Context, req *connect.Request[DeleteProjectRequest]) (*connect.Response[DeleteProjectResponse], error) {
|
||||
return c.deleteProject.CallUnary(ctx, req)
|
||||
}
|
||||
|
||||
// GetActiveProject calls hetty.proj.v1.ProjectService.GetActiveProject.
|
||||
func (c *projectServiceClient) GetActiveProject(ctx context.Context, req *connect.Request[GetActiveProjectRequest]) (*connect.Response[GetActiveProjectResponse], error) {
|
||||
return c.getActiveProject.CallUnary(ctx, req)
|
||||
}
|
||||
|
||||
// ListProjects calls hetty.proj.v1.ProjectService.ListProjects.
|
||||
func (c *projectServiceClient) ListProjects(ctx context.Context, req *connect.Request[ListProjectsRequest]) (*connect.Response[ListProjectsResponse], error) {
|
||||
return c.listProjects.CallUnary(ctx, req)
|
||||
}
|
||||
|
||||
// UpdateInterceptSettings calls hetty.proj.v1.ProjectService.UpdateInterceptSettings.
|
||||
func (c *projectServiceClient) UpdateInterceptSettings(ctx context.Context, req *connect.Request[UpdateInterceptSettingsRequest]) (*connect.Response[UpdateInterceptSettingsResponse], error) {
|
||||
return c.updateInterceptSettings.CallUnary(ctx, req)
|
||||
}
|
||||
|
||||
// SetScopeRules calls hetty.proj.v1.ProjectService.SetScopeRules.
|
||||
func (c *projectServiceClient) SetScopeRules(ctx context.Context, req *connect.Request[SetScopeRulesRequest]) (*connect.Response[SetScopeRulesResponse], error) {
|
||||
return c.setScopeRules.CallUnary(ctx, req)
|
||||
}
|
||||
|
||||
// SetRequestLogsFilter calls hetty.proj.v1.ProjectService.SetRequestLogsFilter.
|
||||
func (c *projectServiceClient) SetRequestLogsFilter(ctx context.Context, req *connect.Request[SetRequestLogsFilterRequest]) (*connect.Response[SetRequestLogsFilterResponse], error) {
|
||||
return c.setRequestLogsFilter.CallUnary(ctx, req)
|
||||
}
|
||||
|
||||
// ProjectServiceHandler is an implementation of the hetty.proj.v1.ProjectService service.
|
||||
type ProjectServiceHandler interface {
|
||||
CreateProject(context.Context, *connect.Request[CreateProjectRequest]) (*connect.Response[CreateProjectResponse], error)
|
||||
OpenProject(context.Context, *connect.Request[OpenProjectRequest]) (*connect.Response[OpenProjectResponse], error)
|
||||
CloseProject(context.Context, *connect.Request[CloseProjectRequest]) (*connect.Response[CloseProjectResponse], error)
|
||||
DeleteProject(context.Context, *connect.Request[DeleteProjectRequest]) (*connect.Response[DeleteProjectResponse], error)
|
||||
GetActiveProject(context.Context, *connect.Request[GetActiveProjectRequest]) (*connect.Response[GetActiveProjectResponse], error)
|
||||
ListProjects(context.Context, *connect.Request[ListProjectsRequest]) (*connect.Response[ListProjectsResponse], error)
|
||||
UpdateInterceptSettings(context.Context, *connect.Request[UpdateInterceptSettingsRequest]) (*connect.Response[UpdateInterceptSettingsResponse], error)
|
||||
SetScopeRules(context.Context, *connect.Request[SetScopeRulesRequest]) (*connect.Response[SetScopeRulesResponse], error)
|
||||
SetRequestLogsFilter(context.Context, *connect.Request[SetRequestLogsFilterRequest]) (*connect.Response[SetRequestLogsFilterResponse], error)
|
||||
}
|
||||
|
||||
// NewProjectServiceHandler builds an HTTP handler from the service implementation. It returns the
|
||||
// path on which to mount the handler and the handler itself.
|
||||
//
|
||||
// By default, handlers support the Connect, gRPC, and gRPC-Web protocols with the binary Protobuf
|
||||
// and JSON codecs. They also support gzip compression.
|
||||
func NewProjectServiceHandler(svc ProjectServiceHandler, opts ...connect.HandlerOption) (string, http.Handler) {
|
||||
projectServiceMethods := File_proj_proj_proto.Services().ByName("ProjectService").Methods()
|
||||
projectServiceCreateProjectHandler := connect.NewUnaryHandler(
|
||||
ProjectServiceCreateProjectProcedure,
|
||||
svc.CreateProject,
|
||||
connect.WithSchema(projectServiceMethods.ByName("CreateProject")),
|
||||
connect.WithHandlerOptions(opts...),
|
||||
)
|
||||
projectServiceOpenProjectHandler := connect.NewUnaryHandler(
|
||||
ProjectServiceOpenProjectProcedure,
|
||||
svc.OpenProject,
|
||||
connect.WithSchema(projectServiceMethods.ByName("OpenProject")),
|
||||
connect.WithHandlerOptions(opts...),
|
||||
)
|
||||
projectServiceCloseProjectHandler := connect.NewUnaryHandler(
|
||||
ProjectServiceCloseProjectProcedure,
|
||||
svc.CloseProject,
|
||||
connect.WithSchema(projectServiceMethods.ByName("CloseProject")),
|
||||
connect.WithHandlerOptions(opts...),
|
||||
)
|
||||
projectServiceDeleteProjectHandler := connect.NewUnaryHandler(
|
||||
ProjectServiceDeleteProjectProcedure,
|
||||
svc.DeleteProject,
|
||||
connect.WithSchema(projectServiceMethods.ByName("DeleteProject")),
|
||||
connect.WithHandlerOptions(opts...),
|
||||
)
|
||||
projectServiceGetActiveProjectHandler := connect.NewUnaryHandler(
|
||||
ProjectServiceGetActiveProjectProcedure,
|
||||
svc.GetActiveProject,
|
||||
connect.WithSchema(projectServiceMethods.ByName("GetActiveProject")),
|
||||
connect.WithHandlerOptions(opts...),
|
||||
)
|
||||
projectServiceListProjectsHandler := connect.NewUnaryHandler(
|
||||
ProjectServiceListProjectsProcedure,
|
||||
svc.ListProjects,
|
||||
connect.WithSchema(projectServiceMethods.ByName("ListProjects")),
|
||||
connect.WithHandlerOptions(opts...),
|
||||
)
|
||||
projectServiceUpdateInterceptSettingsHandler := connect.NewUnaryHandler(
|
||||
ProjectServiceUpdateInterceptSettingsProcedure,
|
||||
svc.UpdateInterceptSettings,
|
||||
connect.WithSchema(projectServiceMethods.ByName("UpdateInterceptSettings")),
|
||||
connect.WithHandlerOptions(opts...),
|
||||
)
|
||||
projectServiceSetScopeRulesHandler := connect.NewUnaryHandler(
|
||||
ProjectServiceSetScopeRulesProcedure,
|
||||
svc.SetScopeRules,
|
||||
connect.WithSchema(projectServiceMethods.ByName("SetScopeRules")),
|
||||
connect.WithHandlerOptions(opts...),
|
||||
)
|
||||
projectServiceSetRequestLogsFilterHandler := connect.NewUnaryHandler(
|
||||
ProjectServiceSetRequestLogsFilterProcedure,
|
||||
svc.SetRequestLogsFilter,
|
||||
connect.WithSchema(projectServiceMethods.ByName("SetRequestLogsFilter")),
|
||||
connect.WithHandlerOptions(opts...),
|
||||
)
|
||||
return "/hetty.proj.v1.ProjectService/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
switch r.URL.Path {
|
||||
case ProjectServiceCreateProjectProcedure:
|
||||
projectServiceCreateProjectHandler.ServeHTTP(w, r)
|
||||
case ProjectServiceOpenProjectProcedure:
|
||||
projectServiceOpenProjectHandler.ServeHTTP(w, r)
|
||||
case ProjectServiceCloseProjectProcedure:
|
||||
projectServiceCloseProjectHandler.ServeHTTP(w, r)
|
||||
case ProjectServiceDeleteProjectProcedure:
|
||||
projectServiceDeleteProjectHandler.ServeHTTP(w, r)
|
||||
case ProjectServiceGetActiveProjectProcedure:
|
||||
projectServiceGetActiveProjectHandler.ServeHTTP(w, r)
|
||||
case ProjectServiceListProjectsProcedure:
|
||||
projectServiceListProjectsHandler.ServeHTTP(w, r)
|
||||
case ProjectServiceUpdateInterceptSettingsProcedure:
|
||||
projectServiceUpdateInterceptSettingsHandler.ServeHTTP(w, r)
|
||||
case ProjectServiceSetScopeRulesProcedure:
|
||||
projectServiceSetScopeRulesHandler.ServeHTTP(w, r)
|
||||
case ProjectServiceSetRequestLogsFilterProcedure:
|
||||
projectServiceSetRequestLogsFilterHandler.ServeHTTP(w, r)
|
||||
default:
|
||||
http.NotFound(w, r)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// UnimplementedProjectServiceHandler returns CodeUnimplemented from all methods.
|
||||
type UnimplementedProjectServiceHandler struct{}
|
||||
|
||||
func (UnimplementedProjectServiceHandler) CreateProject(context.Context, *connect.Request[CreateProjectRequest]) (*connect.Response[CreateProjectResponse], error) {
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hetty.proj.v1.ProjectService.CreateProject is not implemented"))
|
||||
}
|
||||
|
||||
func (UnimplementedProjectServiceHandler) OpenProject(context.Context, *connect.Request[OpenProjectRequest]) (*connect.Response[OpenProjectResponse], error) {
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hetty.proj.v1.ProjectService.OpenProject is not implemented"))
|
||||
}
|
||||
|
||||
func (UnimplementedProjectServiceHandler) CloseProject(context.Context, *connect.Request[CloseProjectRequest]) (*connect.Response[CloseProjectResponse], error) {
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hetty.proj.v1.ProjectService.CloseProject is not implemented"))
|
||||
}
|
||||
|
||||
func (UnimplementedProjectServiceHandler) DeleteProject(context.Context, *connect.Request[DeleteProjectRequest]) (*connect.Response[DeleteProjectResponse], error) {
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hetty.proj.v1.ProjectService.DeleteProject is not implemented"))
|
||||
}
|
||||
|
||||
func (UnimplementedProjectServiceHandler) GetActiveProject(context.Context, *connect.Request[GetActiveProjectRequest]) (*connect.Response[GetActiveProjectResponse], error) {
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hetty.proj.v1.ProjectService.GetActiveProject is not implemented"))
|
||||
}
|
||||
|
||||
func (UnimplementedProjectServiceHandler) ListProjects(context.Context, *connect.Request[ListProjectsRequest]) (*connect.Response[ListProjectsResponse], error) {
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hetty.proj.v1.ProjectService.ListProjects is not implemented"))
|
||||
}
|
||||
|
||||
func (UnimplementedProjectServiceHandler) UpdateInterceptSettings(context.Context, *connect.Request[UpdateInterceptSettingsRequest]) (*connect.Response[UpdateInterceptSettingsResponse], error) {
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hetty.proj.v1.ProjectService.UpdateInterceptSettings is not implemented"))
|
||||
}
|
||||
|
||||
func (UnimplementedProjectServiceHandler) SetScopeRules(context.Context, *connect.Request[SetScopeRulesRequest]) (*connect.Response[SetScopeRulesResponse], error) {
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hetty.proj.v1.ProjectService.SetScopeRules is not implemented"))
|
||||
}
|
||||
|
||||
func (UnimplementedProjectServiceHandler) SetRequestLogsFilter(context.Context, *connect.Request[SetRequestLogsFilterRequest]) (*connect.Response[SetRequestLogsFilterResponse], error) {
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hetty.proj.v1.ProjectService.SetRequestLogsFilter is not implemented"))
|
||||
}
|
||||
356
pkg/proj/proj.go
356
pkg/proj/proj.go
@ -4,12 +4,11 @@ import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"math/rand"
|
||||
"regexp"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/oklog/ulid"
|
||||
connect "connectrpc.com/connect"
|
||||
"github.com/oklog/ulid/v2"
|
||||
|
||||
"github.com/dstotijn/hetty/pkg/filter"
|
||||
"github.com/dstotijn/hetty/pkg/proxy/intercept"
|
||||
@ -18,27 +17,16 @@ import (
|
||||
"github.com/dstotijn/hetty/pkg/sender"
|
||||
)
|
||||
|
||||
//nolint:gosec
|
||||
var ulidEntropy = rand.New(rand.NewSource(time.Now().UnixNano()))
|
||||
|
||||
type Service struct {
|
||||
repo Repository
|
||||
interceptSvc *intercept.Service
|
||||
reqLogSvc *reqlog.Service
|
||||
senderSvc *sender.Service
|
||||
scope *scope.Scope
|
||||
activeProjectID ulid.ULID
|
||||
activeProjectID string
|
||||
mu sync.RWMutex
|
||||
}
|
||||
|
||||
type Project struct {
|
||||
ID ulid.ULID
|
||||
Name string
|
||||
Settings Settings
|
||||
|
||||
isActive bool
|
||||
}
|
||||
|
||||
type Settings struct {
|
||||
// Request log settings
|
||||
ReqLogBypassOutOfScope bool
|
||||
@ -87,216 +75,324 @@ func NewService(cfg Config) (*Service, error) {
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (svc *Service) CreateProject(ctx context.Context, name string) (Project, error) {
|
||||
if !nameRegexp.MatchString(name) {
|
||||
return Project{}, ErrInvalidName
|
||||
func (svc *Service) CreateProject(ctx context.Context, req *connect.Request[CreateProjectRequest]) (*connect.Response[CreateProjectResponse], error) {
|
||||
if !nameRegexp.MatchString(req.Msg.Name) {
|
||||
return nil, ErrInvalidName
|
||||
}
|
||||
|
||||
project := Project{
|
||||
ID: ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy),
|
||||
Name: name,
|
||||
project := &Project{
|
||||
Id: ulid.Make().String(),
|
||||
Name: req.Msg.Name,
|
||||
}
|
||||
|
||||
err := svc.repo.UpsertProject(ctx, project)
|
||||
if err != nil {
|
||||
return Project{}, fmt.Errorf("proj: could not create project: %w", err)
|
||||
return nil, fmt.Errorf("proj: could not create project: %w", err)
|
||||
}
|
||||
|
||||
return project, nil
|
||||
return &connect.Response[CreateProjectResponse]{
|
||||
Msg: &CreateProjectResponse{
|
||||
Project: project,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
// CloseProject closes the currently open project (if there is one).
|
||||
func (svc *Service) CloseProject() error {
|
||||
func (svc *Service) CloseProject(ctx context.Context, _ *connect.Request[CloseProjectRequest]) (*connect.Response[CloseProjectResponse], error) {
|
||||
svc.mu.Lock()
|
||||
defer svc.mu.Unlock()
|
||||
|
||||
if svc.activeProjectID.Compare(ulid.ULID{}) == 0 {
|
||||
return nil
|
||||
if svc.activeProjectID == "" {
|
||||
return nil, connect.NewError(connect.CodeFailedPrecondition, ErrNoProject)
|
||||
}
|
||||
|
||||
svc.activeProjectID = ulid.ULID{}
|
||||
svc.reqLogSvc.SetActiveProjectID(ulid.ULID{})
|
||||
svc.activeProjectID = ""
|
||||
svc.reqLogSvc.SetActiveProjectID("")
|
||||
svc.reqLogSvc.SetBypassOutOfScopeRequests(false)
|
||||
svc.reqLogSvc.SetFindReqsFilter(reqlog.FindRequestsFilter{})
|
||||
svc.reqLogSvc.SetRequestLogsFilter(nil)
|
||||
svc.interceptSvc.UpdateSettings(intercept.Settings{
|
||||
RequestsEnabled: false,
|
||||
ResponsesEnabled: false,
|
||||
RequestFilter: nil,
|
||||
ResponseFilter: nil,
|
||||
})
|
||||
svc.senderSvc.SetActiveProjectID(ulid.ULID{})
|
||||
svc.senderSvc.SetFindReqsFilter(sender.FindRequestsFilter{})
|
||||
svc.senderSvc.SetActiveProjectID("")
|
||||
svc.scope.SetRules(nil)
|
||||
|
||||
return nil
|
||||
return &connect.Response[CloseProjectResponse]{}, nil
|
||||
}
|
||||
|
||||
// DeleteProject removes a project from the repository.
|
||||
func (svc *Service) DeleteProject(ctx context.Context, projectID ulid.ULID) error {
|
||||
if svc.activeProjectID.Compare(projectID) == 0 {
|
||||
return fmt.Errorf("proj: project (%v) is active", projectID.String())
|
||||
func (svc *Service) DeleteProject(ctx context.Context, req *connect.Request[DeleteProjectRequest]) (*connect.Response[DeleteProjectResponse], error) {
|
||||
if svc.activeProjectID == "" {
|
||||
return nil, connect.NewError(connect.CodeFailedPrecondition, ErrNoProject)
|
||||
}
|
||||
|
||||
if err := svc.repo.DeleteProject(ctx, projectID); err != nil {
|
||||
return fmt.Errorf("proj: could not delete project: %w", err)
|
||||
if err := svc.repo.DeleteProject(ctx, req.Msg.ProjectId); err != nil {
|
||||
return nil, fmt.Errorf("proj: could not delete project: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
return &connect.Response[DeleteProjectResponse]{}, nil
|
||||
}
|
||||
|
||||
// OpenProject sets a project as the currently active project.
|
||||
func (svc *Service) OpenProject(ctx context.Context, projectID ulid.ULID) (Project, error) {
|
||||
func (svc *Service) OpenProject(ctx context.Context, req *connect.Request[OpenProjectRequest]) (*connect.Response[OpenProjectResponse], error) {
|
||||
svc.mu.Lock()
|
||||
defer svc.mu.Unlock()
|
||||
|
||||
project, err := svc.repo.FindProjectByID(ctx, projectID)
|
||||
p, err := svc.repo.FindProjectByID(ctx, req.Msg.ProjectId)
|
||||
if errors.Is(err, ErrProjectNotFound) {
|
||||
return nil, connect.NewError(connect.CodeNotFound, ErrProjectNotFound)
|
||||
}
|
||||
if err != nil {
|
||||
return Project{}, fmt.Errorf("proj: failed to get project: %w", err)
|
||||
return nil, connect.NewError(connect.CodeInternal, fmt.Errorf("proj: failed to find project: %w", err))
|
||||
}
|
||||
|
||||
svc.activeProjectID = project.ID
|
||||
svc.activeProjectID = p.Id
|
||||
|
||||
interceptSettings := intercept.Settings{
|
||||
RequestsEnabled: p.InterceptRequests,
|
||||
ResponsesEnabled: p.InterceptResponses,
|
||||
}
|
||||
|
||||
if p.InterceptRequestFilterExpr != "" {
|
||||
expr, err := filter.ParseQuery(p.InterceptRequestFilterExpr)
|
||||
if err != nil {
|
||||
return nil, connect.NewError(connect.CodeInternal, fmt.Errorf("proj: failed to parse intercept request filter: %w", err))
|
||||
}
|
||||
interceptSettings.RequestFilter = expr
|
||||
}
|
||||
|
||||
if p.InterceptResponseFilterExpr != "" {
|
||||
expr, err := filter.ParseQuery(p.InterceptResponseFilterExpr)
|
||||
if err != nil {
|
||||
return nil, connect.NewError(connect.CodeInternal, fmt.Errorf("proj: failed to parse intercept response filter: %w", err))
|
||||
}
|
||||
interceptSettings.ResponseFilter = expr
|
||||
}
|
||||
|
||||
// Request log settings.
|
||||
svc.reqLogSvc.SetFindReqsFilter(reqlog.FindRequestsFilter{
|
||||
ProjectID: project.ID,
|
||||
OnlyInScope: project.Settings.ReqLogOnlyFindInScope,
|
||||
SearchExpr: project.Settings.ReqLogSearchExpr,
|
||||
})
|
||||
svc.reqLogSvc.SetBypassOutOfScopeRequests(project.Settings.ReqLogBypassOutOfScope)
|
||||
svc.reqLogSvc.SetActiveProjectID(project.ID)
|
||||
svc.reqLogSvc.SetActiveProjectID(p.Id)
|
||||
svc.reqLogSvc.SetBypassOutOfScopeRequests(p.ReqLogBypassOutOfScope)
|
||||
svc.reqLogSvc.SetRequestLogsFilter(p.ReqLogFilter)
|
||||
|
||||
// Intercept settings.
|
||||
svc.interceptSvc.UpdateSettings(intercept.Settings{
|
||||
RequestsEnabled: project.Settings.InterceptRequests,
|
||||
ResponsesEnabled: project.Settings.InterceptResponses,
|
||||
RequestFilter: project.Settings.InterceptRequestFilter,
|
||||
ResponseFilter: project.Settings.InterceptResponseFilter,
|
||||
})
|
||||
svc.interceptSvc.UpdateSettings(interceptSettings)
|
||||
|
||||
// Sender settings.
|
||||
svc.senderSvc.SetActiveProjectID(project.ID)
|
||||
svc.senderSvc.SetFindReqsFilter(sender.FindRequestsFilter{
|
||||
ProjectID: project.ID,
|
||||
OnlyInScope: project.Settings.SenderOnlyFindInScope,
|
||||
SearchExpr: project.Settings.SenderSearchExpr,
|
||||
svc.senderSvc.SetActiveProjectID(p.Id)
|
||||
svc.senderSvc.SetRequestsFilter(&sender.RequestsFilter{
|
||||
OnlyInScope: p.SenderOnlyFindInScope,
|
||||
SearchExpr: p.SenderSearchExpr,
|
||||
})
|
||||
|
||||
// Scope settings.
|
||||
svc.scope.SetRules(project.Settings.ScopeRules)
|
||||
|
||||
return project, nil
|
||||
}
|
||||
|
||||
func (svc *Service) ActiveProject(ctx context.Context) (Project, error) {
|
||||
activeProjectID := svc.activeProjectID
|
||||
if activeProjectID.Compare(ulid.ULID{}) == 0 {
|
||||
return Project{}, ErrNoProject
|
||||
}
|
||||
|
||||
project, err := svc.repo.FindProjectByID(ctx, activeProjectID)
|
||||
scopeRules, err := p.ParseScopeRules()
|
||||
if err != nil {
|
||||
return Project{}, fmt.Errorf("proj: failed to get active project: %w", err)
|
||||
return nil, connect.NewError(connect.CodeInvalidArgument, fmt.Errorf("proj: failed to parse scope rules: %w", err))
|
||||
}
|
||||
svc.scope.SetRules(scopeRules)
|
||||
|
||||
p.IsActive = true
|
||||
|
||||
return &connect.Response[OpenProjectResponse]{
|
||||
Msg: &OpenProjectResponse{
|
||||
Project: p,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (svc *Service) GetActiveProject(ctx context.Context, _ *connect.Request[GetActiveProjectRequest]) (*connect.Response[GetActiveProjectResponse], error) {
|
||||
project, err := svc.activeProject(ctx)
|
||||
if errors.Is(err, ErrNoProject) {
|
||||
return nil, connect.NewError(connect.CodeNotFound, err)
|
||||
}
|
||||
if err != nil {
|
||||
return nil, connect.NewError(connect.CodeInternal, err)
|
||||
}
|
||||
|
||||
project.isActive = true
|
||||
project.IsActive = true
|
||||
|
||||
return &connect.Response[GetActiveProjectResponse]{
|
||||
Msg: &GetActiveProjectResponse{
|
||||
Project: project,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (svc *Service) activeProject(ctx context.Context) (*Project, error) {
|
||||
if svc.activeProjectID == "" {
|
||||
return nil, ErrNoProject
|
||||
}
|
||||
|
||||
project, err := svc.repo.FindProjectByID(ctx, svc.activeProjectID)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("proj: failed to get active project: %w", err)
|
||||
}
|
||||
|
||||
return project, nil
|
||||
}
|
||||
|
||||
func (svc *Service) Projects(ctx context.Context) ([]Project, error) {
|
||||
func (svc *Service) ListProjects(ctx context.Context, _ *connect.Request[ListProjectsRequest]) (*connect.Response[ListProjectsResponse], error) {
|
||||
projects, err := svc.repo.Projects(ctx)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("proj: could not get projects: %w", err)
|
||||
}
|
||||
|
||||
return projects, nil
|
||||
for _, project := range projects {
|
||||
if svc.IsProjectActive(project.Id) {
|
||||
project.IsActive = true
|
||||
}
|
||||
}
|
||||
|
||||
return &connect.Response[ListProjectsResponse]{
|
||||
Msg: &ListProjectsResponse{
|
||||
Projects: projects,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (svc *Service) Scope() *scope.Scope {
|
||||
return svc.scope
|
||||
}
|
||||
|
||||
func (svc *Service) SetScopeRules(ctx context.Context, rules []scope.Rule) error {
|
||||
project, err := svc.ActiveProject(ctx)
|
||||
func (svc *Service) SetScopeRules(ctx context.Context, req *connect.Request[SetScopeRulesRequest]) (*connect.Response[SetScopeRulesResponse], error) {
|
||||
p, err := svc.activeProject(ctx)
|
||||
if errors.Is(err, ErrNoProject) {
|
||||
return nil, connect.NewError(connect.CodeFailedPrecondition, err)
|
||||
}
|
||||
if err != nil {
|
||||
return nil, connect.NewError(connect.CodeInternal, err)
|
||||
}
|
||||
|
||||
p.ScopeRules = req.Msg.Rules
|
||||
|
||||
err = svc.repo.UpsertProject(ctx, p)
|
||||
if err != nil {
|
||||
return nil, connect.NewError(connect.CodeInternal, fmt.Errorf("proj: failed to update project: %w", err))
|
||||
}
|
||||
|
||||
scopeRules, err := p.ParseScopeRules()
|
||||
if err != nil {
|
||||
return nil, connect.NewError(connect.CodeInvalidArgument, fmt.Errorf("proj: failed to parse scope rules: %w", err))
|
||||
}
|
||||
|
||||
svc.scope.SetRules(scopeRules)
|
||||
|
||||
return &connect.Response[SetScopeRulesResponse]{}, nil
|
||||
}
|
||||
|
||||
func (p *Project) ParseScopeRules() ([]scope.Rule, error) {
|
||||
var err error
|
||||
scopeRules := make([]scope.Rule, len(p.ScopeRules))
|
||||
|
||||
for i, rule := range p.ScopeRules {
|
||||
scopeRules[i] = scope.Rule{}
|
||||
if rule.UrlRegexp != "" {
|
||||
scopeRules[i].URL, err = regexp.Compile(rule.UrlRegexp)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to parse scope rule's URL field: %w", err)
|
||||
}
|
||||
}
|
||||
if rule.HeaderKeyRegexp != "" {
|
||||
scopeRules[i].Header.Key, err = regexp.Compile(rule.HeaderKeyRegexp)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to parse scope rule's header key field: %w", err)
|
||||
}
|
||||
}
|
||||
if rule.HeaderValueRegexp != "" {
|
||||
scopeRules[i].Header.Value, err = regexp.Compile(rule.HeaderValueRegexp)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to parse scope rule's header value field: %w", err)
|
||||
}
|
||||
}
|
||||
if rule.BodyRegexp != "" {
|
||||
scopeRules[i].Body, err = regexp.Compile(rule.BodyRegexp)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to parse scope rule's body field: %w", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return scopeRules, nil
|
||||
}
|
||||
|
||||
func (svc *Service) SetRequestLogsFilter(ctx context.Context, req *connect.Request[SetRequestLogsFilterRequest]) (*connect.Response[SetRequestLogsFilterResponse], error) {
|
||||
project, err := svc.activeProject(ctx)
|
||||
if errors.Is(err, ErrNoProject) {
|
||||
return nil, connect.NewError(connect.CodeFailedPrecondition, err)
|
||||
}
|
||||
if err != nil {
|
||||
return nil, connect.NewError(connect.CodeInternal, err)
|
||||
}
|
||||
|
||||
project.ReqLogFilter = req.Msg.Filter
|
||||
|
||||
err = svc.repo.UpsertProject(ctx, project)
|
||||
if err != nil {
|
||||
return nil, connect.NewError(connect.CodeInternal, fmt.Errorf("proj: failed to update project: %w", err))
|
||||
}
|
||||
|
||||
svc.reqLogSvc.SetRequestLogsFilter(req.Msg.Filter)
|
||||
|
||||
return &connect.Response[SetRequestLogsFilterResponse]{}, nil
|
||||
}
|
||||
|
||||
func (svc *Service) SetSenderRequestFindFilter(ctx context.Context, filter *sender.RequestsFilter) error {
|
||||
project, err := svc.activeProject(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
project.Settings.ScopeRules = rules
|
||||
project.SenderOnlyFindInScope = filter.OnlyInScope
|
||||
project.SenderSearchExpr = filter.SearchExpr
|
||||
|
||||
err = svc.repo.UpsertProject(ctx, project)
|
||||
if err != nil {
|
||||
return fmt.Errorf("proj: failed to update project: %w", err)
|
||||
}
|
||||
|
||||
svc.scope.SetRules(rules)
|
||||
svc.senderSvc.SetRequestsFilter(filter)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (svc *Service) SetRequestLogFindFilter(ctx context.Context, filter reqlog.FindRequestsFilter) error {
|
||||
project, err := svc.ActiveProject(ctx)
|
||||
func (svc *Service) IsProjectActive(projectID string) bool {
|
||||
return projectID == svc.activeProjectID
|
||||
}
|
||||
|
||||
func (svc *Service) UpdateInterceptSettings(ctx context.Context, req *connect.Request[UpdateInterceptSettingsRequest]) (*connect.Response[UpdateInterceptSettingsResponse], error) {
|
||||
project, err := svc.activeProject(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
return nil, err
|
||||
}
|
||||
|
||||
filter.ProjectID = project.ID
|
||||
|
||||
project.Settings.ReqLogOnlyFindInScope = filter.OnlyInScope
|
||||
project.Settings.ReqLogSearchExpr = filter.SearchExpr
|
||||
project.InterceptRequests = req.Msg.RequestsEnabled
|
||||
project.InterceptResponses = req.Msg.ResponsesEnabled
|
||||
project.InterceptRequestFilterExpr = req.Msg.RequestFilterExpr
|
||||
project.InterceptResponseFilterExpr = req.Msg.ResponseFilterExpr
|
||||
|
||||
err = svc.repo.UpsertProject(ctx, project)
|
||||
if err != nil {
|
||||
return fmt.Errorf("proj: failed to update project: %w", err)
|
||||
return nil, connect.NewError(connect.CodeInternal, fmt.Errorf("proj: failed to update project: %w", err))
|
||||
}
|
||||
|
||||
svc.reqLogSvc.SetFindReqsFilter(filter)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (svc *Service) SetSenderRequestFindFilter(ctx context.Context, filter sender.FindRequestsFilter) error {
|
||||
project, err := svc.ActiveProject(ctx)
|
||||
reqFilterExpr, err := filter.ParseQuery(req.Msg.RequestFilterExpr)
|
||||
if err != nil {
|
||||
return err
|
||||
return nil, connect.NewError(connect.CodeInvalidArgument, fmt.Errorf("proj: failed to parse intercept request filter: %w", err))
|
||||
}
|
||||
|
||||
filter.ProjectID = project.ID
|
||||
|
||||
project.Settings.SenderOnlyFindInScope = filter.OnlyInScope
|
||||
project.Settings.SenderSearchExpr = filter.SearchExpr
|
||||
|
||||
err = svc.repo.UpsertProject(ctx, project)
|
||||
respFilterExpr, err := filter.ParseQuery(req.Msg.ResponseFilterExpr)
|
||||
if err != nil {
|
||||
return fmt.Errorf("proj: failed to update project: %w", err)
|
||||
return nil, connect.NewError(connect.CodeInvalidArgument, fmt.Errorf("proj: failed to parse intercept response filter: %w", err))
|
||||
}
|
||||
|
||||
svc.senderSvc.SetFindReqsFilter(filter)
|
||||
svc.interceptSvc.UpdateSettings(intercept.Settings{
|
||||
RequestsEnabled: req.Msg.RequestsEnabled,
|
||||
ResponsesEnabled: req.Msg.ResponsesEnabled,
|
||||
RequestFilter: reqFilterExpr,
|
||||
ResponseFilter: respFilterExpr,
|
||||
})
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (svc *Service) IsProjectActive(projectID ulid.ULID) bool {
|
||||
return projectID.Compare(svc.activeProjectID) == 0
|
||||
}
|
||||
|
||||
func (svc *Service) UpdateInterceptSettings(ctx context.Context, settings intercept.Settings) error {
|
||||
project, err := svc.ActiveProject(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
project.Settings.InterceptRequests = settings.RequestsEnabled
|
||||
project.Settings.InterceptResponses = settings.ResponsesEnabled
|
||||
project.Settings.InterceptRequestFilter = settings.RequestFilter
|
||||
project.Settings.InterceptResponseFilter = settings.ResponseFilter
|
||||
|
||||
err = svc.repo.UpsertProject(ctx, project)
|
||||
if err != nil {
|
||||
return fmt.Errorf("proj: failed to update project: %w", err)
|
||||
}
|
||||
|
||||
svc.interceptSvc.UpdateSettings(settings)
|
||||
|
||||
return nil
|
||||
return &connect.Response[UpdateInterceptSettingsResponse]{}, nil
|
||||
}
|
||||
|
||||
1181
pkg/proj/proj.pb.go
Normal file
1181
pkg/proj/proj.pb.go
Normal file
File diff suppressed because it is too large
Load Diff
@ -2,14 +2,12 @@ package proj
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/oklog/ulid"
|
||||
)
|
||||
|
||||
type Repository interface {
|
||||
FindProjectByID(ctx context.Context, id ulid.ULID) (Project, error)
|
||||
UpsertProject(ctx context.Context, project Project) error
|
||||
DeleteProject(ctx context.Context, id ulid.ULID) error
|
||||
Projects(ctx context.Context) ([]Project, error)
|
||||
FindProjectByID(ctx context.Context, id string) (*Project, error)
|
||||
UpsertProject(ctx context.Context, project *Project) error
|
||||
DeleteProject(ctx context.Context, id string) error
|
||||
Projects(ctx context.Context) ([]*Project, error)
|
||||
Close() error
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user