a61e8dcb62
This change helps the transition between `dagger input` and `#Plan.context`. In summary, the codebase now relies on a *context* for execution with mapping to *IDs*. In the future, *context* will come from a `#Plan.context`. In the meantime, a bridge converts `dagger input` to a plan context. This allows both *old* and *new* style configurations to co-exist with the same underlying engine. - Implement `plancontext`. Context holds the execution context for a plan. Currently this includes the platform, local directories, secrets and services (e.g. unix/npipe). - Contextual data can be registered at any point. In the future, this will be done by `#Plan.context` - Migrated the `dagger input` codebase to register inputs in a `plancontext` - Migrated low-level types/operations to the *Context ID* pattern. - `dagger.#Stream` now only includes an `id` (instead of `unix` path) - `dagger.#Secret` still includes only an ID, but now it's based off `plancontext` - `op.#Local` now only includes an `id` (instead of `path`, `include`, `exclude`. Signed-off-by: Andrea Luzzardi <aluzzardi@gmail.com>
52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package solver
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/moby/buildkit/session"
|
|
"github.com/moby/buildkit/session/sshforward"
|
|
"go.dagger.io/dagger/plancontext"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
type SocketProvider struct {
|
|
pctx *plancontext.Context
|
|
}
|
|
|
|
func NewDockerSocketProvider(pctx *plancontext.Context) session.Attachable {
|
|
return &SocketProvider{pctx}
|
|
}
|
|
|
|
func (sp *SocketProvider) Register(server *grpc.Server) {
|
|
sshforward.RegisterSSHServer(server, sp)
|
|
}
|
|
|
|
func (sp *SocketProvider) CheckAgent(ctx context.Context, req *sshforward.CheckAgentRequest) (*sshforward.CheckAgentResponse, error) {
|
|
return &sshforward.CheckAgentResponse{}, nil
|
|
}
|
|
|
|
func (sp *SocketProvider) ForwardAgent(stream sshforward.SSH_ForwardAgentServer) error {
|
|
id := sshforward.DefaultID
|
|
|
|
opts, _ := metadata.FromIncomingContext(stream.Context()) // if no metadata continue with empty object
|
|
|
|
if v, ok := opts[sshforward.KeySSHID]; ok && len(v) > 0 && v[0] != "" {
|
|
id = v[0]
|
|
}
|
|
|
|
service := sp.pctx.Services.Get(plancontext.ContextKey(id))
|
|
if service == nil {
|
|
return fmt.Errorf("invalid socket id %q", id)
|
|
}
|
|
|
|
conn, err := dialService(service)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to connect to %s: %w", id, err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
return sshforward.Copy(context.TODO(), conn, stream, nil)
|
|
}
|