feat: handle error for downloading and extrating

Author: Benzbeeb

models/weaveinit/run_l1_node.go

@@ -932,7 +932,7 @@ func NewExistingDataReplaceSelect(state *RunL1NodeState) *ExistingDataReplaceSel
 			},
 		},
 		state:    state,
-		question: fmt.Sprintf("Existing %s detected. Would you like to use the current one or replace it", utils.InitiaDataDirectory),
+		question: fmt.Sprintf("Existing %s detected. Would you like to use the current one or replace it?", utils.InitiaDataDirectory),
 	}
 }
 
@@ -998,7 +998,8 @@ func (m *SnapshotEndpointInput) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	input, cmd, done := m.TextInput.Update(msg)
 	if done {
 		m.state.snapshotEndpoint = input.Text
-		// m.state.weave.PreviousResponse += styles.RenderPreviousResponse(styles.DotsSeparator, m.GetQuestion(), []string{"snapshot url"}, input.Text)
+		m.state.weave.PushPreviousResponse(styles.RenderPreviousResponse(styles.DotsSeparator, m.GetQuestion(), []string{"snapshot url"}, input.Text))
+
 		if snapshotDownload, err := NewSnapshotDownloadLoading(m.state); err == nil {
 			return snapshotDownload, snapshotDownload.Init()
 		} else {
@@ -1065,8 +1066,7 @@ type SnapshotDownloadLoading struct {
 func NewSnapshotDownloadLoading(state *RunL1NodeState) (*SnapshotDownloadLoading, error) {
 	userHome, err := os.UserHomeDir()
 	if err != nil {
-		fmt.Printf("[error] Failed to get user home: %v\n", err)
-		return nil, err
+		return nil, fmt.Errorf("[error] Failed to get user home: %v", err)
 	}
 
 	return &SnapshotDownloadLoading{
@@ -1087,7 +1087,7 @@ func (m *SnapshotDownloadLoading) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	if err := m.GetError(); err != nil {
 		model := NewSnapshotEndpointInput(m.state)
 		model.err = err
-		return model, nil
+		return model, model.Init()
 	}
 
 	if m.GetCompletion() {
@@ -1104,8 +1104,7 @@ func (m *SnapshotDownloadLoading) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 }
 
 func (m *SnapshotDownloadLoading) View() string {
-	view := m.state.weave.Render() + m.Downloader.View()
-	return view
+	return m.state.weave.Render() + m.Downloader.View()
 }
 
 type SnapshotExtractLoading struct {
@@ -1127,6 +1126,15 @@ func (m *SnapshotExtractLoading) Init() tea.Cmd {
 func (m *SnapshotExtractLoading) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	loader, cmd := m.Loading.Update(msg)
 	m.Loading = loader
+	switch msg := msg.(type) {
+	case utils.ErrorLoading:
+		m.state.weave.PopPreviousResponse()
+		m.state.weave.PopPreviousResponse()
+		model := NewSnapshotEndpointInput(m.state)
+		model.err = msg.Err
+		return model, cmd
+	}
+
 	if m.Loading.Completing {
 		m.state.weave.PushPreviousResponse(styles.RenderPreviousResponse(styles.NoSeparator, fmt.Sprintf("Snapshot extracted to %s successfully.", utils.InitiaDataDirectory), []string{}, ""))
 		return m, tea.Quit
@@ -1145,8 +1153,7 @@ func snapshotExtractor() tea.Cmd {
 	return func() tea.Msg {
 		userHome, err := os.UserHomeDir()
 		if err != nil {
-			fmt.Printf("[error] Failed to get user home: %v\n", err)
-			// TODO: Return error
+			return utils.ErrorLoading{Err: fmt.Errorf("[error] Failed to get user home: %v", err)}
 		}
 
 		targetDir := filepath.Join(userHome, utils.InitiaDirectory)
@@ -1156,8 +1163,7 @@ func snapshotExtractor() tea.Cmd {
 
 		err = cmd.Run()
 		if err != nil {
-			fmt.Printf("[error] Failed to extract snapshot: %v\n", err)
-			// TODO: Return error
+			return utils.ErrorLoading{Err: fmt.Errorf("[error] Failed to extract snapshot: %v", err)}
 		}
 		return utils.EndLoading{}
 	}

utils/downloader.go

@@ -1,6 +1,7 @@
 package utils
 
 import (
+	"bytes"
 	"fmt"
 	"io"
 	"net/http"
@@ -101,12 +102,49 @@ func (m *Downloader) startDownload() tea.Cmd {
 			m.current = totalDownloaded
 		}
 
+		// Now, check if the file is a valid .tar.lz4 file
+		if err := m.validateTarLz4Header(); err != nil {
+			m.done = true
+			m.err = fmt.Errorf("invalid file format: %w", err)
+			return nil
+		}
+
 		m.done = true
 		m.err = nil
 		return nil
 	}
 }
 
+// LZ4 magic number for LZ4 frame format
+var lz4MagicNumber = []byte{0x04, 0x22, 0x4D, 0x18}
+
+// validateTarLz4Header checks if the downloaded file is a valid .tar.lz4 file based on the file header.
+func (m *Downloader) validateTarLz4Header() error {
+	// Open the .lz4 file
+	file, err := os.Open(m.dest)
+	if err != nil {
+		return fmt.Errorf("failed to open file: %w", err)
+	}
+	defer file.Close()
+
+	// Read the first few bytes of the file (header)
+	header := make([]byte, 4)
+	_, err = file.Read(header)
+	if err != nil {
+		return fmt.Errorf("failed to read file header: %w", err)
+	}
+
+	// Check if the header matches the LZ4 magic number
+	if !bytes.Equal(header, lz4MagicNumber) {
+		return fmt.Errorf("invalid file format: the file is not a valid .lz4 file")
+	}
+
+	// If the header matches, we assume it's a valid .lz4 file.
+	// You could continue checking the contents further if needed, but this verifies the file is compressed with LZ4.
+
+	return nil
+}
+
 func (m *Downloader) Init() tea.Cmd {
 	return tea.Batch(m.tick(), m.startDownload())
 }

utils/loading.go

@@ -30,6 +30,7 @@ type Loading struct {
 	quitting   bool
 	frame      int
 	executeFn  tea.Cmd
+	Err        error
 }
 
 func NewLoading(text string, executeFn tea.Cmd) Loading {
@@ -64,6 +65,9 @@ func (m Loading) Update(msg tea.Msg) (Loading, tea.Cmd) {
 	case EndLoading:
 		m.Completing = true
 		return m, nil
+	case ErrorLoading:
+		m.Err = msg.Err
+		return m, nil
 	default:
 		return m, nil
 	}