bec3a43f55
Every CoreServices interface (core/plugin/deps.go) now has a guest-side stub
that marshals to the WO-WZ-001 capability messages and dispatches through the
generic host_call transport, so plugin service code compiles and runs
unchanged against content.Content, settings.Settings, plugin.PluginBridge, etc.
- core/plugin/wasmguest/caps/: one file per family (17 families, 38 methods),
a var _ <iface> = (*stub)(nil) compile proof each, and NewCoreServices(call)
assembling them. The transport is injected (CallFunc) so marshaling is
natively testable; the wasm shim binds it to CallHost, DESCRIBE probes pass
nil (capability calls fail cleanly instead of nil-panicking).
- Error mapping wraps AbiError with <family>.<method> context and maps
DEADLINE_EXCEEDED onto context.DeadlineExceeded.
- RAGService.RegisterContentFetcher stays guest-side (RAGStub) for
HOOK_RAG_FETCH dispatch; Query/OnContentChanged marshal out. dispatch.go and
describe.go now source fetchers from the caps RAG stub.
- caps_roundtrip_test.go: fake transport + 76 deterministic golden payloads
(family_method_{req,resp}.pb) covering 100% of families, plus error-mapping,
deadline, nil-transport, and guest-side-fetcher tests. WO-WZ-006 replays the
same goldens to prevent host/guest drift.
- Acceptance: testdata/fixture Load hook calls deps.Content/Settings/Bridge
unchanged (compiles for wasip1); caps_wasmhost_test.go drives it end-to-end
through a real wazero module + fake host_call table.
- Disposition table in docs/wasm-abi.md: every member stub | host-side
(Pool, Interceptors, AppURL/MediaPath, CoreServiceBindings host-side).
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
79 lines
2.6 KiB
Go
79 lines
2.6 KiB
Go
package caps
|
|
|
|
import (
|
|
"context"
|
|
"sort"
|
|
|
|
abiv1 "git.dev.alexdunmow.com/block/core/abi/v1"
|
|
"git.dev.alexdunmow.com/block/core/plugin"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
// RAGStub implements plugin.RAGService. Query and OnContentChanged marshal out
|
|
// to rag.* capability calls, while RegisterContentFetcher records the fetcher
|
|
// guest-side: the host re-indexes by inverting the call as a HOOK_RAG_FETCH
|
|
// callback, so the wasm shim reads the recorded fetchers to dispatch it. The
|
|
// stub is exported so the shim can reach FetcherTypes/Fetcher.
|
|
type RAGStub struct {
|
|
base
|
|
fetchers map[string]plugin.ContentFetcher
|
|
}
|
|
|
|
var _ plugin.RAGService = (*RAGStub)(nil)
|
|
|
|
// NewRAGStub builds a RAG stub with the given transport. A nil transport
|
|
// (DESCRIBE probe, native build) still records fetchers so their content
|
|
// types can be captured into the manifest.
|
|
func NewRAGStub(call CallFunc) *RAGStub {
|
|
return &RAGStub{
|
|
base: base{family: "rag", call: call},
|
|
fetchers: make(map[string]plugin.ContentFetcher),
|
|
}
|
|
}
|
|
|
|
func (s *RAGStub) RegisterContentFetcher(contentType string, fetcher plugin.ContentFetcher) {
|
|
s.fetchers[contentType] = fetcher
|
|
}
|
|
|
|
func (s *RAGStub) Query(ctx context.Context, query string, limit int) ([]plugin.RAGResult, error) {
|
|
req := &abiv1.RagQueryRequest{Query: query, Limit: int32(limit)}
|
|
resp := &abiv1.RagQueryResponse{}
|
|
if err := s.invoke(ctx, "query", req, resp); err != nil {
|
|
return nil, err
|
|
}
|
|
results := make([]plugin.RAGResult, 0, len(resp.GetResults()))
|
|
for _, r := range resp.GetResults() {
|
|
results = append(results, plugin.RAGResult{
|
|
Content: r.GetContent(),
|
|
Score: r.GetScore(),
|
|
Metadata: r.GetMetadata(),
|
|
})
|
|
}
|
|
return results, nil
|
|
}
|
|
|
|
// OnContentChanged has no error channel; a transport failure is dropped (the
|
|
// host will re-index on its own schedule regardless).
|
|
func (s *RAGStub) OnContentChanged(ctx context.Context, contentType string, contentID uuid.UUID) {
|
|
req := &abiv1.RagOnContentChangedRequest{ContentType: contentType, ContentId: contentID.String()}
|
|
_ = s.invoke(ctx, "on_content_changed", req, &abiv1.RagOnContentChangedResponse{})
|
|
}
|
|
|
|
// Fetcher returns the content fetcher registered for a content type, for
|
|
// HOOK_RAG_FETCH dispatch by the wasm shim.
|
|
func (s *RAGStub) Fetcher(contentType string) (plugin.ContentFetcher, bool) {
|
|
f, ok := s.fetchers[contentType]
|
|
return f, ok
|
|
}
|
|
|
|
// FetcherTypes lists the registered content-fetcher types in sorted order
|
|
// (deterministic manifest capture).
|
|
func (s *RAGStub) FetcherTypes() []string {
|
|
types := make([]string, 0, len(s.fetchers))
|
|
for k := range s.fetchers {
|
|
types = append(types, k)
|
|
}
|
|
sort.Strings(types)
|
|
return types
|
|
}
|