mirror of
https://github.com/kubernetes-sigs/kind.git
synced 2025-12-01 07:26:05 +07:00
84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
/*
|
|
Copyright 2018 The Kubernetes Authors.
|
|
|
|
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 cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"sigs.k8s.io/kind/pkg/log"
|
|
)
|
|
|
|
// Status is used to track ongoing status in a CLI, with a nice loading spinner
|
|
// when attached to a terminal
|
|
type Status struct {
|
|
spinner *Spinner
|
|
status string
|
|
logger log.Logger
|
|
}
|
|
|
|
// StatusForLogger returns a new status object for the logger l,
|
|
// if l is the kind cli logger and the writer is a Spinner, that spinner
|
|
// will be used for the status
|
|
func StatusForLogger(l log.Logger) *Status {
|
|
s := &Status{
|
|
logger: l,
|
|
}
|
|
// if we're using the CLI logger, check for if it has a spinner setup
|
|
// and wire the status to that
|
|
if v, ok := l.(*Logger); ok {
|
|
if v2, ok := v.Writer.(*Spinner); ok {
|
|
s.spinner = v2
|
|
}
|
|
}
|
|
return s
|
|
}
|
|
|
|
// Start starts a new phase of the status, if attached to a terminal
|
|
// there will be a loading spinner with this status
|
|
func (s *Status) Start(status string) {
|
|
s.End(true)
|
|
// set new status
|
|
s.status = status
|
|
if s.spinner != nil {
|
|
s.spinner.SetSuffix(fmt.Sprintf(" %s ", s.status))
|
|
s.spinner.Start()
|
|
} else {
|
|
s.logger.V(0).Infof(" • %s ...\n", s.status)
|
|
}
|
|
}
|
|
|
|
// End completes the current status, ending any previous spinning and
|
|
// marking the status as success or failure
|
|
func (s *Status) End(success bool) {
|
|
if s.status == "" {
|
|
return
|
|
}
|
|
|
|
if s.spinner != nil {
|
|
s.spinner.Stop()
|
|
fmt.Fprint(s.spinner.writer, "\r")
|
|
}
|
|
|
|
if success {
|
|
s.logger.V(0).Infof(" ✓ %s\n", s.status)
|
|
} else {
|
|
s.logger.V(0).Infof("✗ %s\n", s.status)
|
|
}
|
|
|
|
s.status = ""
|
|
}
|