some fix

Author: shiyuhang0

Makefile

@@ -22,9 +22,9 @@ setup: deps devtools ## Set up dev env
 generate-mocks: ## Generate mock objects
 	@echo "==> Generating mock objects"
 	go install github.com/vektra/mockery/v2@v2.53.5
-	mockery --name TiDBCloudClient --recursive --output=internal/mock --outpkg mock --filename api_client.go
-	mockery --name EventsSender --recursive --output=internal/mock --outpkg mock --filename sender.go
-	mockery --name Uploader --recursive --output=internal/mock --outpkg mock --filename uploader.go
+	mockery --name TiDBCloudClient --recursive --dir=internal/service/cloud --output=internal/mock --outpkg mock --filename api_client.go
+	mockery --name EventsSender --recursive --dir=internal/telemetry --output=internal/mock --outpkg mock --filename sender.go
+	mockery --name Uploader --recursive --dir=internal/service/aws/s3 --output=internal/mock --outpkg mock --filename uploader.go
 
 .PHONY: addcopy
 addcopy: ## Add copyright to all files

internal/flag/flag.go

@@ -15,6 +15,7 @@
 package flag
 
 const (
+	ExampleID            string = "example-id"
 	ClusterID            string = "cluster-id"
 	ClusterIDShort       string = "c"
 	LocalConcurrency     string = "local.concurrency"

internal/ui/test_one_input_model.go

@@ -0,0 +1,91 @@
+// Copyright 2026 PingCAP, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package ui
+
+import (
+	"fmt"
+
+	"github.com/charmbracelet/bubbles/textinput"
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/juju/errors"
+	"github.com/tidbcloud/tidbcloud-cli/internal/config"
+	"github.com/tidbcloud/tidbcloud-cli/internal/util"
+)
+
+type TextOneInputModel struct {
+	Prompt      string
+	Input       textinput.Model
+	Err         error
+	Interrupted bool
+}
+
+func (m TextOneInputModel) Init() tea.Cmd {
+	return textinput.Blink
+}
+
+func (m TextOneInputModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	var cmd tea.Cmd
+
+	switch msg := msg.(type) {
+	case tea.KeyMsg:
+		switch msg.Type {
+		case tea.KeyEnter:
+			return m, tea.Quit
+		case tea.KeyCtrlC, tea.KeyEsc:
+			m.Interrupted = true
+			return m, tea.Quit
+		}
+
+	case errMsg:
+		m.Err = msg
+		return m, nil
+	}
+
+	m.Input, cmd = m.Input.Update(msg)
+	return m, cmd
+}
+
+func (m TextOneInputModel) View() string {
+	return fmt.Sprintf(
+		"%s\n\n%s\n\n%s\n\n",
+		m.Prompt,
+		m.Input.View(),
+		helpMessageStyle("Press Enter to submit (esc to quit)"),
+	)
+}
+
+// InitialOneInputModel runs an interactive single-line input and returns the model and any error.
+// view is the prompt line shown above the input. placeholder is used as the input placeholder text.
+func InitialOneInputModel(prompt, placeholder string) (TextOneInputModel, error) {
+	ti := textinput.New()
+	ti.Placeholder = placeholder
+	ti.Focus()
+	ti.PromptStyle = config.FocusedStyle
+	ti.TextStyle = config.FocusedStyle
+
+	p := tea.NewProgram(TextOneInputModel{Prompt: prompt, Input: ti})
+	model, err := p.Run()
+	finalModel := model.(TextOneInputModel)
+	if err != nil {
+		return finalModel, errors.Trace(err)
+	}
+	if finalModel.Interrupted {
+		return finalModel, util.InterruptError
+	}
+	if finalModel.Err != nil {
+		return finalModel, finalModel.Err
+	}
+	return finalModel, nil
+}