refactor: create an interface for status console

Signed-off-by: Terry Howe <[email protected]>

cmd/oras/internal/display/status/console/console.go

@@ -16,10 +16,9 @@ limitations under the License.
 package console
 
 import (
-	"os"
-
-	"github.com/containerd/console"
+	containerd "github.com/containerd/console"
 	"github.com/morikuni/aec"
+	"os"
 )
 
 const (
@@ -33,60 +32,75 @@ const (
 	Restore = "\0338"
 )
 
-// Console is a wrapper around containerd's console.Console and ANSI escape
-// codes.
-type Console struct {
-	console.Console
+// Console is a wrapper around containerd's Console and ANSI escape codes.
+type Console interface {
+	containerd.Console
+	GetHeightWidth() (height, width int)
+	Save()
+	NewRow()
+	OutputTo(upCnt uint, str string)
+	Restore()
+}
+
+type console struct {
+	containerd.Console
+}
+
+// NewConsole generates a console from a file.
+func NewConsole(f *os.File) (Console, error) {
+	c, err := containerd.ConsoleFromFile(f)
+	if err != nil {
+		return nil, err
+	}
+	return &console{c}, nil
 }
 
 // Size returns the width and height of the console.
 // If the console size cannot be determined, returns a default value of 80x10.
-func (c *Console) Size() (width, height int) {
-	width = MinWidth
-	height = MinHeight
-	size, err := c.Console.Size()
-	if err == nil {
-		if size.Height > MinHeight {
-			height = int(size.Height)
+func (c *console) Size() (size containerd.WinSize, err error) {
+	size, err = c.Console.Size()
+	if err != nil {
+		size.Height = MinHeight
+		size.Width = MinWidth
+	} else {
+		if size.Height < MinHeight {
+			size.Height = MinHeight
 		}
-		if size.Width > MinWidth {
-			width = int(size.Width)
+		if size.Width < MinWidth {
+			size.Width = MinWidth
 		}
 	}
 	return
 }
 
-// New generates a Console from a file.
-func New(f *os.File) (*Console, error) {
-	c, err := console.ConsoleFromFile(f)
-	if err != nil {
-		return nil, err
-	}
-	return &Console{c}, nil
+// GetHeightWidth returns the width and height of the console.
+func (c *console) GetHeightWidth() (height, width int) {
+	windowSize, _ := c.Size()
+	return int(windowSize.Height), int(windowSize.Width)
 }
 
 // Save saves the current cursor position.
-func (c *Console) Save() {
+func (c *console) Save() {
 	_, _ = c.Write([]byte(aec.Hide.Apply(Save)))
 }
 
 // NewRow allocates a horizontal space to the output area with scroll if needed.
-func (c *Console) NewRow() {
+func (c *console) NewRow() {
 	_, _ = c.Write([]byte(Restore))
 	_, _ = c.Write([]byte("\n"))
 	_, _ = c.Write([]byte(Save))
 }
 
 // OutputTo outputs a string to a specific line.
-func (c *Console) OutputTo(upCnt uint, str string) {
+func (c *console) OutputTo(upCnt uint, str string) {
 	_, _ = c.Write([]byte(Restore))
 	_, _ = c.Write([]byte(aec.PreviousLine(upCnt).Apply(str)))
 	_, _ = c.Write([]byte("\n"))
 	_, _ = c.Write([]byte(aec.EraseLine(aec.EraseModes.Tail).String()))
 }
 
 // Restore restores the saved cursor position.
-func (c *Console) Restore() {
+func (c *console) Restore() {
 	// cannot use aec.Restore since DEC has better compatibility than SCO
 	_, _ = c.Write([]byte(Restore))
 	_, _ = c.Write([]byte(aec.Column(0).

cmd/oras/internal/display/status/console/console_test.go

@@ -1,5 +1,3 @@
-//go:build freebsd || linux || netbsd || openbsd || solaris
-
 /*
 Copyright The ORAS Authors.
 Licensed under the Apache License, Version 2.0 (the "License");
@@ -18,11 +16,22 @@ limitations under the License.
 package console
 
 import (
+	containerd "github.com/containerd/console"
 	"testing"
 
-	"github.com/containerd/console"
+	"oras.land/oras/internal/testutils"
 )
 
+func givenConsole(t *testing.T) (Console, containerd.Console) {
+	pty, _, err := containerd.NewPty()
+	if err != nil {
+		t.Fatal(err)
+	}
+	return &console{
+		Console: pty,
+	}, pty
+}
+
 func validateSize(t *testing.T, gotWidth, gotHeight, wantWidth, wantHeight int) {
 	t.Helper()
 	if gotWidth != wantWidth {
@@ -33,31 +42,65 @@ func validateSize(t *testing.T, gotWidth, gotHeight, wantWidth, wantHeight int)
 	}
 }
 
-func TestConsole_Size(t *testing.T) {
-	pty, _, err := console.NewPty()
+func TestNewConsole(t *testing.T) {
+	_, device, err := testutils.NewPty()
 	if err != nil {
 		t.Fatal(err)
 	}
-	c := &Console{
-		Console: pty,
+	defer device.Close()
+
+	c, err := NewConsole(device)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	s, err := c.Size()
+	if err != nil {
+		t.Errorf("Unexpect error %v", err)
+	}
+	if s.Height != 10 {
+		t.Errorf("Expected height 10 got %d", s.Height)
 	}
+	if s.Width != 80 {
+		t.Errorf("Expected height 80 got %d", s.Width)
+	}
+}
+
+func TestConsole_Size(t *testing.T) {
+	c, pty := givenConsole(t)
 
 	// minimal width and height
-	gotWidth, gotHeight := c.Size()
+	gotHeight, gotWidth := c.GetHeightWidth()
 	validateSize(t, gotWidth, gotHeight, MinWidth, MinHeight)
 
 	// zero width
-	_ = pty.Resize(console.WinSize{Width: 0, Height: MinHeight})
-	gotWidth, gotHeight = c.Size()
+	_ = pty.Resize(containerd.WinSize{Width: 0, Height: MinHeight})
+	gotHeight, gotWidth = c.GetHeightWidth()
 	validateSize(t, gotWidth, gotHeight, MinWidth, MinHeight)
 
 	// zero height
-	_ = pty.Resize(console.WinSize{Width: MinWidth, Height: 0})
-	gotWidth, gotHeight = c.Size()
+	_ = pty.Resize(containerd.WinSize{Width: MinWidth, Height: 0})
+	gotHeight, gotWidth = c.GetHeightWidth()
 	validateSize(t, gotWidth, gotHeight, MinWidth, MinHeight)
 
 	// valid zero and height
-	_ = pty.Resize(console.WinSize{Width: 200, Height: 100})
-	gotWidth, gotHeight = c.Size()
+	_ = pty.Resize(containerd.WinSize{Width: 200, Height: 100})
+	gotHeight, gotWidth = c.GetHeightWidth()
 	validateSize(t, gotWidth, gotHeight, 200, 100)
 }
+
+func TestConsole_GetHeightWidth(t *testing.T) {
+
+}
+
+func TestConsole_NewRow(t *testing.T) {
+
+}
+
+func TestConsole_OutputTo(t *testing.T) {
+
+}
+
+func TestConsole_Restore(t *testing.T) {
+
+}

cmd/oras/internal/display/status/progress/manager.go

@@ -44,15 +44,15 @@ type Manager interface {
 type manager struct {
 	status       []*status
 	statusLock   sync.RWMutex
-	console      *console.Console
+	console      console.Console
 	updating     sync.WaitGroup
 	renderDone   chan struct{}
 	renderClosed chan struct{}
 }
 
 // NewManager initialized a new progress manager.
-func NewManager(f *os.File) (Manager, error) {
-	c, err := console.New(f)
+func NewManager(tty *os.File) (Manager, error) {
+	c, err := console.NewConsole(tty)
 	if err != nil {
 		return nil, err
 	}
@@ -88,7 +88,7 @@ func (m *manager) render() {
 	m.statusLock.RLock()
 	defer m.statusLock.RUnlock()
 	// todo: update size in another routine
-	width, height := m.console.Size()
+	height, width := m.console.GetHeightWidth()
 	lineCount := len(m.status) * 2
 	offset := 0
 	if lineCount > height {

cmd/oras/internal/display/status/progress/manager_test.go

@@ -31,10 +31,15 @@ func Test_manager_render(t *testing.T) {
 		t.Fatal(err)
 	}
 	defer device.Close()
+	sole, err := console.NewConsole(device)
+	if err != nil {
+		t.Fatal(err)
+	}
+
 	m := &manager{
-		console: &console.Console{Console: pty},
+		console: sole,
 	}
-	_, height := m.console.Size()
+	height, _ := m.console.GetHeightWidth()
 	for i := 0; i < height; i++ {
 		if _, err := m.Add(); err != nil {
 			t.Fatal(err)

internal/testutils/console.go

@@ -23,13 +23,13 @@ import (
 	"strings"
 	"sync"
 
-	"github.com/containerd/console"
+	containerd "github.com/containerd/console"
 )
 
 // NewPty creates a new pty pair for testing, caller is responsible for closing
 // the returned device file if err is not nil.
-func NewPty() (console.Console, *os.File, error) {
-	pty, devicePath, err := console.NewPty()
+func NewPty() (containerd.Console, *os.File, error) {
+	pty, devicePath, err := containerd.NewPty()
 	if err != nil {
 		return nil, nil, err
 	}
@@ -42,7 +42,7 @@ func NewPty() (console.Console, *os.File, error) {
 
 // MatchPty checks that the output matches the expected strings in specified
 // order.
-func MatchPty(pty console.Console, device *os.File, expected ...string) error {
+func MatchPty(pty containerd.Console, device *os.File, expected ...string) error {
 	var wg sync.WaitGroup
 	wg.Add(1)
 	var buffer bytes.Buffer