From dd84b9d64fb98746a230cd24233ff50a562c39c9 Mon Sep 17 00:00:00 2001
From: 简律纯 <hsiangnianian@outlook.com>
Date: Fri, 28 Apr 2023 01:36:44 +0800
Subject:

---
 cli/internal/ui/spinner.go | 80 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 cli/internal/ui/spinner.go

(limited to 'cli/internal/ui/spinner.go')

diff --git a/cli/internal/ui/spinner.go b/cli/internal/ui/spinner.go
new file mode 100644
index 0000000..6e47d2d
--- /dev/null
+++ b/cli/internal/ui/spinner.go
@@ -0,0 +1,80 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+package ui
+
+import (
+	"fmt"
+	"io"
+	"os"
+	"time"
+
+	"github.com/briandowns/spinner"
+)
+
+// startStopper is the interface to interact with the spinner.
+type startStopper interface {
+	Start()
+	Stop()
+}
+
+// Spinner represents an indicator that an asynchronous operation is taking place.
+//
+// For short operations, less than 4 seconds, display only the spinner with the Start and Stop methods.
+// For longer operations, display intermediate progress events using the Events method.
+type Spinner struct {
+	spin startStopper
+}
+
+// NewSpinner returns a spinner that outputs to w.
+func NewSpinner(w io.Writer) *Spinner {
+	interval := 125 * time.Millisecond
+	if os.Getenv("CI") == "true" {
+		interval = 30 * time.Second
+	}
+	s := spinner.New(charset, interval, spinner.WithHiddenCursor(true))
+	s.Writer = w
+	s.Color("faint")
+	return &Spinner{
+		spin: s,
+	}
+}
+
+// Start starts the spinner suffixed with a label.
+func (s *Spinner) Start(label string) {
+	s.suffix(fmt.Sprintf(" %s", label))
+	s.spin.Start()
+}
+
+// Stop stops the spinner and replaces it with a label.
+func (s *Spinner) Stop(label string) {
+	s.finalMSG(fmt.Sprint(label))
+	s.spin.Stop()
+}
+
+func (s *Spinner) lock() {
+	if spinner, ok := s.spin.(*spinner.Spinner); ok {
+		spinner.Lock()
+	}
+}
+
+func (s *Spinner) unlock() {
+	if spinner, ok := s.spin.(*spinner.Spinner); ok {
+		spinner.Unlock()
+	}
+}
+
+func (s *Spinner) suffix(label string) {
+	s.lock()
+	defer s.unlock()
+	if spinner, ok := s.spin.(*spinner.Spinner); ok {
+		spinner.Suffix = label
+	}
+}
+
+func (s *Spinner) finalMSG(label string) {
+	s.lock()
+	defer s.unlock()
+	if spinner, ok := s.spin.(*spinner.Spinner); ok {
+		spinner.FinalMSG = label
+	}
+}
-- 
cgit v1.2.3-70-g09d2