feat: simple downloader

Author: traviolus

.gitignore

@@ -18,3 +18,5 @@ Thumbs.db
 
 # GoLand specific files
 *.iml
+
+*.weave

go.mod

@@ -3,6 +3,7 @@ module github.com/initia-labs/weave
 go 1.22.6
 
 require (
+	github.com/charmbracelet/bubbles v0.20.0
 	github.com/charmbracelet/bubbletea v1.1.0
 	github.com/charmbracelet/lipgloss v0.13.0
 	github.com/spf13/cobra v1.8.1
@@ -12,6 +13,7 @@ require (
 
 require (
 	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
+	github.com/charmbracelet/harmonica v0.2.0 // indirect
 	github.com/charmbracelet/x/ansi v0.2.3 // indirect
 	github.com/charmbracelet/x/term v0.2.0 // indirect
 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect

go.sum

@@ -1,7 +1,11 @@
 github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
 github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
+github.com/charmbracelet/bubbles v0.20.0 h1:jSZu6qD8cRQ6k9OMfR1WlM+ruM8fkPWkHvQWD9LIutE=
+github.com/charmbracelet/bubbles v0.20.0/go.mod h1:39slydyswPy+uVOHZ5x/GjwVAFkCsV8IIVy+4MhzwwU=
 github.com/charmbracelet/bubbletea v1.1.0 h1:FjAl9eAL3HBCHenhz/ZPjkKdScmaS5SK69JAK2YJK9c=
 github.com/charmbracelet/bubbletea v1.1.0/go.mod h1:9Ogk0HrdbHolIKHdjfFpyXJmiCzGwy+FesYkZr7hYU4=
+github.com/charmbracelet/harmonica v0.2.0 h1:8NxJWRWg/bzKqqEaaeFNipOu77YR5t8aSwG4pgaUBiQ=
+github.com/charmbracelet/harmonica v0.2.0/go.mod h1:KSri/1RMQOZLbw7AHqgcBycp8pgJnQMYYT8QZRqZ1Ao=
 github.com/charmbracelet/lipgloss v0.13.0 h1:4X3PPeoWEDCMvzDvGmTajSyYPcZM4+y8sCA/SsA3cjw=
 github.com/charmbracelet/lipgloss v0.13.0/go.mod h1:nw4zy0SBX/F/eAO1cWdcvy6qnkDUxr8Lw7dvFrAIbbY=
 github.com/charmbracelet/x/ansi v0.2.3 h1:VfFN0NUpcjBRd4DnKfRaIRo53KRgey/nhOoEqosGDEY=
@@ -13,6 +17,8 @@ github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSs
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
+github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
 github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f h1:Y/CXytFA4m6baUTXGLOoWe4PQhGxaX0KpnayAqC48p4=
 github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f/go.mod h1:vw97MGsxSvLiUE2X8qFplwetxpGLQrlU1Q9AUEIzCaM=
 github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=

models/weaveinit/run_l1_node.go

@@ -869,34 +869,34 @@ func (m *StateSyncEndpointInput) View() string {
 }
 
 type SnapshotDownloadLoading struct {
-	utils.Loading
+	utils.Downloader
 	state *RunL1NodeState
 }
 
 func NewSnapshotDownloadLoading(state *RunL1NodeState) *SnapshotDownloadLoading {
 	return &SnapshotDownloadLoading{
-		Loading: utils.NewLoading("Downloading snapshot from the provided URL...", utils.DefaultWait()),
-		state:   state,
+		Downloader: *utils.NewDownloader(
+			"Downloading snapshot from the provided URL",
+			state.snapshotEndpoint,
+			"snapshot.weave",
+		),
+		state: state,
 	}
 }
 
 func (m *SnapshotDownloadLoading) Init() tea.Cmd {
-	return m.Loading.Init()
+	return m.Downloader.Init()
 }
 
 func (m *SnapshotDownloadLoading) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
-	loader, cmd := m.Loading.Update(msg)
-	m.Loading = loader
-	if m.Loading.Completing {
-		m.state.weave.PreviousResponse += styles.RenderPreviousResponse(styles.NoSeparator, "Snapshot downloaded successfully.", []string{}, "")
+	if m.GetCompletion() {
 		return m, tea.Quit
 	}
+	downloader, cmd := m.Downloader.Update(msg)
+	m.Downloader = *downloader
 	return m, cmd
 }
 
 func (m *SnapshotDownloadLoading) View() string {
-	if m.Completing {
-		return m.state.weave.PreviousResponse
-	}
-	return m.state.weave.PreviousResponse + m.Loading.View()
+	return m.state.weave.PreviousResponse + m.Downloader.View()
 }

styles/text.go

@@ -64,12 +64,12 @@ var (
 	QuestionMark    string = Text("? ", Cyan)
 	CorrectMark     string = Text("✓ ", Green)
 	InformationMark string = Text("i ", Cyan)
-	SelectorCursor  string = Text("> ", Yellow)
+	SelectorCursor  string = Text("> ", Cyan)
 )
 
 // RenderPrompt highlights phrases in the text if they match any phrase in the highlights list
 func RenderPrompt(text string, highlights []string, status PromptStatus) string {
-	prompt := "\n"
+	prompt := ""
 	switch status {
 	case Question:
 		prompt += QuestionMark

utils/downloader.go

@@ -0,0 +1,139 @@
+package utils
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"time"
+
+	"github.com/charmbracelet/bubbles/progress"
+	tea "github.com/charmbracelet/bubbletea"
+
+	"github.com/initia-labs/weave/styles"
+)
+
+type Downloader struct {
+	progress progress.Model
+	total    int64
+	current  int64
+	text     string
+	url      string
+	dest     string
+	done     bool
+}
+
+func NewDownloader(text, url, dest string) *Downloader {
+	return &Downloader{
+		progress: progress.New(progress.WithGradient(string(styles.Cyan), "#008B8B")),
+		total:    0,
+		text:     text,
+		url:      url,
+		dest:     dest,
+	}
+}
+
+func (m *Downloader) startDownload() tea.Cmd {
+	return func() tea.Msg {
+		// Use a high buffer size for faster I/O operations
+		const bufferSize = 8192 // 8 KB buffer
+
+		resp, err := http.Get(m.url)
+		if err != nil {
+			m.done = true
+			return nil
+		}
+		defer resp.Body.Close()
+
+		m.total = resp.ContentLength
+		if m.total <= 0 {
+			m.total = 1
+		}
+
+		file, err := os.Create(m.dest)
+		if err != nil {
+			m.done = true
+			return nil
+		}
+		defer file.Close()
+
+		buffer := make([]byte, bufferSize)
+		var totalDownloaded int64
+		for {
+			n, err := resp.Body.Read(buffer)
+			if err != nil && err != io.EOF {
+				m.done = true
+				return nil
+			}
+			if n == 0 {
+				break
+			}
+
+			// Write to file and update progress in a single step
+			if _, err := file.Write(buffer[:n]); err != nil {
+				m.done = true
+				return nil
+			}
+
+			totalDownloaded += int64(n)
+			m.current = totalDownloaded
+
+			// Simulate network delay for smoother updates
+			time.Sleep(10 * time.Millisecond)
+		}
+
+		m.done = true
+		return nil
+	}
+}
+
+func (m *Downloader) Init() tea.Cmd {
+	return tea.Batch(m.tick(), m.startDownload())
+}
+
+func (m *Downloader) Update(msg tea.Msg) (*Downloader, tea.Cmd) {
+	switch msg := msg.(type) {
+	case TickMsg:
+		return m, m.tick()
+
+	case tea.KeyMsg:
+		if msg.String() == "q" || msg.String() == "ctrl+c" {
+			return m, tea.Quit
+		}
+	}
+
+	model, cmd := m.progress.Update(msg)
+	m.progress = model.(progress.Model)
+	return m, cmd
+}
+
+func (m *Downloader) View() string {
+	if m.done {
+		return fmt.Sprintf("%sDownload Complete!\nTotal Size: %d bytes\n", styles.CorrectMark, m.total)
+	}
+	percentage := float64(m.current) / float64(m.total)
+	return fmt.Sprintf("%s: %s / %s \n%s", m.text, ByteCountSI(m.current), ByteCountSI(m.total), m.progress.ViewAs(percentage))
+}
+
+func (m *Downloader) GetCompletion() bool {
+	return m.done
+}
+
+func (m *Downloader) tick() tea.Cmd {
+	return tea.Tick(100*time.Millisecond, func(t time.Time) tea.Msg {
+		return TickMsg(t)
+	})
+}
+
+func ByteCountSI(b int64) string {
+	const unit = 1000
+	if b < unit {
+		return fmt.Sprintf("%d B", b)
+	}
+	div, exp := int64(unit), 0
+	for n := b / unit; n >= unit; n /= unit {
+		div *= unit
+		exp++
+	}
+	return fmt.Sprintf("%.3f %cB", float64(b)/float64(div), "kMGTPE"[exp])
+}