140 lines
5.4 KiB
Go
140 lines
5.4 KiB
Go
package pluginsdk
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// Plugin is the interface a plugin author implements. It is delivered RPCs
|
|
// from kandev over the go-plugin gRPC transport (§3 of
|
|
// docs/plans/plugins/GRPC-CONTRACT.md): DeliverEvent -> OnEvent,
|
|
// HandleWebhook -> HandleWebhook.
|
|
type Plugin interface {
|
|
// OnEvent handles a single bus event delivery. A non-nil error causes
|
|
// kandev's delivery subsystem to retry per §5 (3 retries, 5s/15s/45s).
|
|
OnEvent(ctx context.Context, e *Event) error
|
|
|
|
// HandleWebhook handles an inbound webhook relayed by kandev's
|
|
// POST /api/plugins/{id}/webhooks/{key} endpoint.
|
|
HandleWebhook(ctx context.Context, req *WebhookRequest) (*WebhookResponse, error)
|
|
}
|
|
|
|
// ActionHandler is an optional browser-action extension. Keeping it separate
|
|
// from Plugin preserves source compatibility for plugins that only handle
|
|
// events and public webhooks.
|
|
type ActionHandler interface {
|
|
HandleAction(context.Context, *PluginActionRequest) (*PluginActionResponse, error)
|
|
}
|
|
|
|
// EntityReferenceSearcher is an optional manifest-owned composer source.
|
|
type EntityReferenceSearcher interface {
|
|
SearchEntityReferences(context.Context, *SearchEntityReferencesRequest) (*SearchEntityReferencesResponse, error)
|
|
}
|
|
|
|
// EntityReferenceAuthorizer is an optional live authorization extension for
|
|
// canonical composer references.
|
|
type EntityReferenceAuthorizer interface {
|
|
AuthorizeEntityReference(context.Context, *AuthorizeEntityReferenceRequest) (*AuthorizeEntityReferenceResponse, error)
|
|
}
|
|
|
|
// EntityReferenceHandler is the convenient optional extension for plugins
|
|
// that own a complete reference source. The narrower interfaces above remain
|
|
// available so the transport can return a precise unsupported error per RPC.
|
|
type EntityReferenceHandler interface {
|
|
EntityReferenceSearcher
|
|
EntityReferenceAuthorizer
|
|
}
|
|
|
|
// GitCredentialResolver is an optional provider-neutral credential resolver.
|
|
// Returned credentials are transient and must never be logged or persisted.
|
|
type GitCredentialResolver interface {
|
|
ResolveGitCredential(context.Context, *ResolveGitCredentialRequest) (*ResolveGitCredentialResponse, error)
|
|
}
|
|
|
|
// GitCredentialBinder is an optional companion to GitCredentialResolver. It
|
|
// provides a non-secret opaque revision for an exact provider lease scope.
|
|
// Generic provider leases require this extension so credential rotation and
|
|
// disconnect invalidate an already-issued helper lease without a secret read.
|
|
type GitCredentialBinder interface {
|
|
GetGitCredentialBinding(context.Context, *GitCredentialBindingRequest) (*GitCredentialBindingResponse, error)
|
|
}
|
|
|
|
// GitCredentialHandler is the complete provider-neutral Git credential
|
|
// extension. It keeps existing resolvers source-compatible while allowing
|
|
// hosts to fail closed when a generic provider lacks binding support.
|
|
type GitCredentialHandler interface {
|
|
GitCredentialResolver
|
|
GitCredentialBinder
|
|
}
|
|
|
|
// AgentToolPlugin is an optional extension implemented by plugins that
|
|
// declare agent_tools in their manifest. Keeping it separate from Plugin
|
|
// preserves source compatibility for existing plugins.
|
|
type AgentToolPlugin interface {
|
|
InvokeAgentTool(context.Context, *AgentToolRequest) (*AgentToolResult, error)
|
|
}
|
|
|
|
// HostSetter is implemented by Plugin values that want Serve to inject the
|
|
// Host once the broker connection back to kandev is established (see the
|
|
// "Host injection" section of the serve.go file header). UnimplementedPlugin
|
|
// implements this, so embedding it is the easiest way to opt in.
|
|
type HostSetter interface {
|
|
SetHost(Host)
|
|
}
|
|
|
|
// UnimplementedPlugin is an embeddable no-op base for Plugin. Authors that
|
|
// only care about one or two RPCs embed this and override the rest. It also
|
|
// implements HostSetter, storing the injected Host for retrieval via Host().
|
|
//
|
|
// SetHost is called from a background goroutine (see the GRPCPlugin doc
|
|
// comment in serve.go for why), so access to the stored Host is
|
|
// mutex-guarded — Host() may be called concurrently with SetHost from
|
|
// another goroutine.
|
|
type UnimplementedPlugin struct {
|
|
mu sync.RWMutex
|
|
host Host
|
|
}
|
|
|
|
var (
|
|
_ Plugin = (*UnimplementedPlugin)(nil)
|
|
_ HostSetter = (*UnimplementedPlugin)(nil)
|
|
)
|
|
|
|
// OnEvent is a no-op default: it accepts the event without doing anything.
|
|
//
|
|
// Pointer receiver (like every other UnimplementedPlugin method): the type
|
|
// embeds a sync.RWMutex, and a value receiver would trip go vet's copylocks
|
|
// check even though this method never touches it.
|
|
func (*UnimplementedPlugin) OnEvent(context.Context, *Event) error {
|
|
return nil
|
|
}
|
|
|
|
// HandleWebhook is a no-op default: it returns 404, since an unhandled
|
|
// webhook path has nothing sensible to serve.
|
|
func (*UnimplementedPlugin) HandleWebhook(context.Context, *WebhookRequest) (*WebhookResponse, error) {
|
|
return &WebhookResponse{Status: 404}, nil
|
|
}
|
|
|
|
func (*UnimplementedPlugin) InvokeAgentTool(context.Context, *AgentToolRequest) (*AgentToolResult, error) {
|
|
return nil, status.Error(codes.Unimplemented, "agent tool invocation is not implemented")
|
|
}
|
|
|
|
// SetHost stores the Host injected by Serve. Call Host() from your
|
|
// overridden methods to reach kandev.
|
|
func (p *UnimplementedPlugin) SetHost(h Host) {
|
|
p.mu.Lock()
|
|
defer p.mu.Unlock()
|
|
p.host = h
|
|
}
|
|
|
|
// Host returns the Host injected by Serve, or nil if the broker connection
|
|
// hasn't completed yet (or Serve hasn't been called, e.g. in unit tests).
|
|
func (p *UnimplementedPlugin) Host() Host {
|
|
p.mu.RLock()
|
|
defer p.mu.RUnlock()
|
|
return p.host
|
|
}
|