Method ensureWorkspaceLoaded

engine/server/session_workspaces.go:60–94

ensureWorkspaceLoaded detects the workspace from the client's working directory and loads all configured modules onto the dagql server. Called from serveQuery (not initializeDaggerClient) because it requires the client's session attachables to access the client's filesystem for workspace detection.

(ctx context.Context, client *daggerClient)

// (not initializeDaggerClient) because it requires the client's session attachables
// to access the client's filesystem for workspace detection.
func (srv *Server) ensureWorkspaceLoaded(ctx context.Context, client *daggerClient) error {
	mode, workspaceRef := workspaceBindingMode(client)
	if mode == workspaceBindingInherit {
		return srv.inheritWorkspaceBinding(ctx, client)
	}

	client.workspaceMu.Lock()
	defer client.workspaceMu.Unlock()

	if client.workspaceLoaded {
		return client.workspaceErr
	}

	// Wait for the client's session attachables to be available.
	// Don't mark as loaded on failure — allow retry on next request.
	if _, err := client.getClientCaller(ctx, client.clientID); err != nil {
		return fmt.Errorf("waiting for client session attachables: %w", err)
	}

	var err error
	switch mode {
	case workspaceBindingDeclared:
		err = srv.loadWorkspaceFromDeclaredRef(ctx, client, workspaceRef)
	case workspaceBindingDetectHost:
		err = srv.loadWorkspaceFromHost(ctx, client)
	default:
		err = fmt.Errorf("unsupported workspace binding mode %q", mode)
	}
	if err != nil {
		client.workspaceErr = err
	}

	client.workspaceLoaded = true
	return client.workspaceErr
}

type workspaceBindingModeType string