go.temporal.io/server/tools/tdbg/prompter.go

80 LOC · 34 covered · 46 uncovered · 13 ranges · 112 concepts · 12 introducers · 47 tests

File neighbourhood

The centred file is linked to every concept that introduces one of its ranges, every test that runs code from the file, and the gray connector concepts standing between those tests and the file's own introducer concepts. Undirected links join concepts to every file where they introduce source and concepts to the tests they introduce; arrows show specialization between the displayed concepts and bridge only concepts omitted from this view. Concept colors match the source ranges below; connector concepts have no source color and are shown in gray.

Focused file, its introducer and connector concepts, their introduced files, and tests that run code from the file

In the embedded map, ordinary wheel input scrolls the page; use the visible controls to zoom and drag to pan. Open the full-screen map for canvas navigation: wheel pans, Ctrl/Command plus wheel zooms, and arrow keys pan when this region is focused. On touch screens, open the full-screen map to pan or pinch. If JavaScript or WebGL is unavailable, use the related-file, concept, and source links on this page.

  1
  
  package tdbg
  
  2
  
  3
  
  import (
  
  4
  
  	"bufio"
  
  5
  
  	"fmt"
  
  6
  
  	"io"
  
  7
  
  	"os"
  
  8
  
  	"strings"
  
  9
  
  )
  
  10
  
  11
  
  type (
  
  12
  
  	// Prompter is a helper for prompting the user for confirmation.
  
  13
  
  	Prompter struct {
  
  14
  
  		writer     io.Writer
  
  15
  
  		reader     io.Reader
  
  16
  
  		exiter     func(code int)
  
  17
  
  		flagLookup BoolFlagLookup
  
  18
  
  	}
  
  19
  
  	// PrompterParams is used to configure a new Prompter.
  
  20
  
  	PrompterParams struct {
  
  21
  
  		// Reader defaults to os.Stdin.
  
  22
  
  		Reader io.Reader
  
  23
  
  		// Writer defaults to os.Stdout.
  
  24
  
  		Writer io.Writer
  
  25
  
  		// Exiter defaults to [os.Exit].
  
  26
  
  		Exiter func(code int)
  
  27
  
  	}
  
  28
  
  	// PrompterOption is used to override default PrompterParams.
  
  29
  
  	PrompterOption func(*PrompterParams)
  
  30
  
  	// BoolFlagLookup can be satisfied by [github.com/urfave/cli/v2.Context].
  
  31
  
  	BoolFlagLookup interface {
  
  32
  
  		Bool(name string) bool
  
  33
  
  	}
  
  34
  
  	PrompterFactory func(c BoolFlagLookup) *Prompter
  
  35
  
  )
  
  36
  
  37
  >
  func NewPrompterFactory(opts ...PrompterOption) PrompterFactory {
  
tdbg_commands.go ×12

  38
  >
  	return func(c BoolFlagLookup) *Prompter {
  
  39
  >
  		return NewPrompter(c, opts...)
  
dlq_service.go ×3

  40
  >
  	}
  
  41
  
  }
  
  42
  
  43
  
  // NewPrompter creates a new Prompter. In most cases, the first argument should be [github.com/urfave/cli/v2.Context].
  
  44
  >
  func NewPrompter(c BoolFlagLookup, opts ...PrompterOption) *Prompter {
  
prompter.go ×2

  45
  >
  	params := PrompterParams{
  
  46
  >
  		Writer: os.Stdout,
  
  47
  >
  		Reader: os.Stdin,
  
  48
  >
  		Exiter: os.Exit,
  
  49
  >
  	}
  
  50
  >
  	for _, opt := range opts {
  
  51
  >
  		opt(&params)
  
prompter.go ×1

  52
  >
  	}
  
  53
  >
  	return &Prompter{
  
prompter.go ×2

  54
  >
  		writer:     params.Writer,
  
  55
  >
  		reader:     params.Reader,
  
  56
  >
  		exiter:     params.Exiter,
  
  57
  >
  		flagLookup: c,
  
  58
  >
  	}
  
  59
  
  }
  
  60
  
  61
  
  // Prompt the user for confirmation. If the user does not respond with "y" or "yes" (case-insensitive and without
  
  62
  
  // leading or trailing space), the process will exit with code 1.
  
  63
  >
  func (p *Prompter) Prompt(msg string) {
  
prompter.go ×1

  64
  >
  	if p.flagLookup.Bool(FlagYes) {
  
  65
  >
  		return
  
prompter.go ×1

  66
  >
  	}
  
  67
  >
  	_, err := p.writer.Write([]byte(msg + " [y/N]: "))
  
prompter.go ×1

  68
  >
  	if err != nil {
  
  69
  >
  		panic(fmt.Errorf("failed to write prompt: %w", err))
  
prompter.go ×1

  70
  
  	}
  
  71
  >
  	reader := bufio.NewReader(p.reader)
  
prompter.go ×1

  72
  >
  	text, err := reader.ReadString('\n')
  
  73
  >
  	if err != nil {
  
  74
  >
  		panic(fmt.Errorf("failed to read prompt: %w", err))
  
prompter.go ×1

  75
  
  	}
  
  76
  >
  	textLower := strings.ToLower(strings.TrimSpace(text))
  
prompter.go ×1

  77
  >
  	if textLower != "y" && textLower != "yes" {
  
  78
  >
  		p.exiter(1)
  
prompter.go ×1

  79
  >
  	}
  
  80
  
  }