Method Handle

cmd/dagger/shell.go:348–435

(ctx context.Context, line string)

var _ idtui.ShellHandler = (*shellCallHandler)(nil)

func (h *shellCallHandler) Handle(ctx context.Context, line string) (rerr error) {
	// Quick sanitization
	line = strings.TrimSpace(line)

	// If in exit command
	if line == "exit" || line == "/exit" {
		h.cancel()
		return nil
	}

	// Empty input
	if line == "" {
		// add an immediately-canceled blank span, to emulate submitting blank shell
		// commands to space things apart
		_, span := Tracer().Start(ctx, "",
			telemetry.Reveal(),
			trace.WithAttributes(attribute.Bool(telemetry.CanceledAttr, true)))
		span.End()
		return nil
	}

	// Handle based on mode
	if h.mode == modePrompt {
		// NB: no span in this case, just let the LLM APIs create the user/assistant
		// message spans

		llm, err := h.llm(ctx)
		if err != nil {
			return err
		}
		newLLM, err := llm.WithPrompt(ctx, line)
		if err != nil {
			return err
		}
		h.llmSession = newLLM
		h.llmModel = newLLM.model
		return nil
	}

	// Ensure we always see new telemetry for shell commands, rather than
	// "resurrecting" the same telemetry from previous commands
	if bag, err := baggage.Parse("repeat-telemetry=true"); err == nil {
		ctx = baggage.ContextWithBaggage(ctx, bag)
	}

	// Create a new span for this command
	var span trace.Span
	ctx, span = Tracer().Start(ctx, line,
		telemetry.Reveal(),
		trace.WithAttributes(
			attribute.String(telemetry.ContentTypeAttr, h.mode.ContentType()),
		))
	var telemetryErr error
	defer telemetry.EndWithCause(span, &telemetryErr)
	defer func() {
		if errors.Is(rerr, context.Canceled) {
			span.SetAttributes(attribute.Bool(telemetry.CanceledAttr, true))
		} else {