Merge pull request #51 from initia-labs/feat/clipboard

Click to copy mnemonic!

Author: WasinWatt

go.mod

@@ -8,13 +8,15 @@ require (
 	github.com/BurntSushi/toml v1.4.0
 	github.com/PuerkitoBio/goquery v1.10.0
 	github.com/amplitude/analytics-go v1.0.2
+	github.com/atotto/clipboard v0.1.4
 	github.com/btcsuite/btcd/btcec/v2 v2.3.4
 	github.com/btcsuite/btcutil v1.0.2
 	github.com/charmbracelet/bubbles v0.20.0
 	github.com/charmbracelet/bubbletea v1.1.0
 	github.com/charmbracelet/lipgloss v0.13.0
 	github.com/charmbracelet/x/term v0.2.0
 	github.com/fynelabs/selfupdate v0.2.0
+	github.com/google/uuid v1.6.0
 	github.com/muesli/reflow v0.3.0
 	github.com/pelletier/go-toml/v2 v2.2.2
 	github.com/spf13/cobra v1.8.1
@@ -39,7 +41,6 @@ require (
 	github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
 	github.com/fsnotify/fsnotify v1.7.0 // indirect
 	github.com/golang/protobuf v1.5.3 // indirect
-	github.com/google/uuid v1.6.0 // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect

go.sum

@@ -11,6 +11,8 @@ github.com/amplitude/analytics-go v1.0.2 h1:GiKYxFwuuEcEgNpw8p4lPiIXHL6z9iEXa8Ql
 github.com/amplitude/analytics-go v1.0.2/go.mod h1:kAQG8OQ6aPOxZrEZ3+/NFCfxdYSyjqXZhgkjWFD3/vo=
 github.com/andybalholm/cascadia v1.3.2 h1:3Xi6Dw5lHF15JtdcmAHD3i1+T8plmv7BQ/nsViSLyss=
 github.com/andybalholm/cascadia v1.3.2/go.mod h1:7gtRlve5FxPPgIgX36uWBX58OdBsSS6lUvCFb+h7KvU=
+github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4=
+github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI=
 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/btcsuite/btcd v0.20.1-beta/go.mod h1:wVuoA8VJLEcwgqHBwHmzLRazpKxTv13Px/pDuV7OomQ=

io/filesystem.go

@@ -10,6 +10,8 @@ import (
 	"path/filepath"
 	"runtime"
 
+	"github.com/atotto/clipboard"
+
 	"github.com/initia-labs/weave/client"
 )
 
@@ -149,3 +151,12 @@ func CopyDirectory(src, des string) error {
 	}
 	return nil
 }
+
+// CopyToClipboard copies the given string to the clipboard.
+func CopyToClipboard(text string) error {
+	// Copy the text to clipboard
+	if err := clipboard.WriteAll(text); err != nil {
+		return fmt.Errorf("failed to copy to clipboard: %v", err)
+	}
+	return nil
+}

models/initialize.go

@@ -3,6 +3,7 @@ package models
 import (
 	"context"
 	"fmt"
+	"strings"
 	"time"
 
 	tea "github.com/charmbracelet/bubbletea"
@@ -11,6 +12,7 @@ import (
 	"github.com/initia-labs/weave/config"
 	weavecontext "github.com/initia-labs/weave/context"
 	"github.com/initia-labs/weave/crypto"
+	"github.com/initia-labs/weave/io"
 	"github.com/initia-labs/weave/styles"
 	"github.com/initia-labs/weave/types"
 	"github.com/initia-labs/weave/ui"
@@ -167,7 +169,8 @@ func (m *GenerateGasStationLoading) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		m.Ctx = m.Loading.EndContext
 		state := weavecontext.PushPageAndGetState[ExistingCheckerState](m)
 		state.weave.PushPreviousResponse(styles.RenderPreviousResponse(styles.NoSeparator, "Gas Station account has been successfully generated.", []string{}, ""))
-		return NewSystemKeysMnemonicDisplayInput(weavecontext.SetCurrentState(m.Ctx, state)), nil
+		model := NewSystemKeysMnemonicDisplayInput(weavecontext.SetCurrentState(m.Ctx, state))
+		return model, model.Init()
 	}
 	return m, cmd
 }
@@ -179,15 +182,28 @@ func (m *GenerateGasStationLoading) View() string {
 
 type GasStationMnemonicDisplayInput struct {
 	ui.TextInput
+	ui.Clickable
 	weavecontext.BaseModel
-	question string
+	question          string
+	generatedMnemonic string
 }
 
 func NewSystemKeysMnemonicDisplayInput(ctx context.Context) *GasStationMnemonicDisplayInput {
+	state := weavecontext.GetCurrentState[ExistingCheckerState](ctx)
 	model := &GasStationMnemonicDisplayInput{
 		TextInput: ui.NewTextInput(true),
-		BaseModel: weavecontext.BaseModel{Ctx: ctx, CannotBack: true},
-		question:  "Please type `continue` to proceed after you have securely stored the mnemonic.",
+		Clickable: *ui.NewClickable(
+			ui.NewClickableItem(
+				map[bool]string{
+					true:  "Copied! Click to copy again",
+					false: "Click here to copy",
+				}, func() error {
+					return io.CopyToClipboard(state.generatedMnemonic)
+				},
+			)),
+		BaseModel:         weavecontext.BaseModel{Ctx: ctx, CannotBack: true},
+		question:          "Please type `continue` to proceed after you have securely stored the mnemonic.",
+		generatedMnemonic: state.generatedMnemonic,
 	}
 	model.WithPlaceholder("Type `continue` to continue, Ctrl+C to quit.")
 	model.WithValidatorFn(common.ValidateExactString("continue"))
@@ -199,19 +215,24 @@ func (m *GasStationMnemonicDisplayInput) GetQuestion() string {
 }
 
 func (m *GasStationMnemonicDisplayInput) Init() tea.Cmd {
-	return nil
+	return m.Clickable.Init()
 }
 
 func (m *GasStationMnemonicDisplayInput) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	if model, cmd, handled := weavecontext.HandleCommonCommands[ExistingCheckerState](m, msg); handled {
 		return model, cmd
 	}
 
+	err := m.Clickable.ClickableUpdate(msg)
+	if err != nil {
+		return m, m.HandlePanic(err)
+	}
+
 	input, cmd, done := m.TextInput.Update(msg)
 	if done {
 		state := weavecontext.PushPageAndGetState[ExistingCheckerState](m)
 		model := NewWeaveAppInitialization(m.Ctx, state.generatedMnemonic)
-		return model, model.Init()
+		return model, tea.Batch(model.Init(), m.Clickable.PostUpdate())
 	}
 	m.TextInput = input
 	return m, cmd
@@ -225,12 +246,17 @@ func (m *GasStationMnemonicDisplayInput) View() string {
 	}
 
 	var mnemonicText string
-	mnemonicText += styles.RenderMnemonic("Gas Station", gasStationAddress, state.generatedMnemonic)
+	mnemonicText += styles.RenderMnemonic("Gas Station", gasStationAddress, m.generatedMnemonic, m.Clickable.ClickableView(0))
 
-	return m.WrapView(InitHeader(state.isFirstTime) + "\n" + state.weave.Render() + "\n" +
+	viewText := m.WrapView(InitHeader(state.isFirstTime) + "\n" + state.weave.Render() + "\n" +
 		styles.BoldUnderlineText("Important", styles.Yellow) + "\n" +
 		styles.Text("Write down these mnemonic phrases and store them in a safe place. \nIt is the only way to recover your system keys.", styles.Yellow) + "\n\n" +
-		mnemonicText + styles.RenderPrompt(m.GetQuestion(), []string{"`continue`"}, styles.Question) + m.TextInput.View())
+		mnemonicText + "\n" + styles.RenderPrompt(m.GetQuestion(), []string{"`continue`"}, styles.Question) + m.TextInput.View())
+	err = m.Clickable.ClickableUpdatePositions(viewText)
+	if err != nil {
+		m.HandlePanic(err)
+	}
+	return viewText
 }
 
 type GasStationMnemonicInput struct {
@@ -269,7 +295,7 @@ func (m *GasStationMnemonicInput) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		state.weave.PushPreviousResponse(
 			styles.RenderPreviousResponse(styles.DotsSeparator, "Please set up a Gas Station account", []string{"Gas Station account"}, styles.HiddenMnemonicText),
 		)
-		model := NewWeaveAppInitialization(weavecontext.SetCurrentState(m.Ctx, state), input.Text)
+		model := NewWeaveAppInitialization(weavecontext.SetCurrentState(m.Ctx, state), strings.Trim(input.Text, "\n"))
 		return model, model.Init()
 	}
 	m.TextInput = input

models/minitia/launch.go

@@ -2520,7 +2520,8 @@ func (m *GenerateOrRecoverSystemKeysLoading) Update(msg tea.Msg) (tea.Model, tea
 
 		if state.generateKeys {
 			state.weave.PushPreviousResponse(styles.RenderPreviousResponse(styles.NoSeparator, "System keys have been successfully generated.", []string{}, ""))
-			return NewSystemKeysMnemonicDisplayInput(weavecontext.SetCurrentState(m.Ctx, state)), nil
+			model := NewSystemKeysMnemonicDisplayInput(weavecontext.SetCurrentState(m.Ctx, state))
+			return model, model.Init()
 		} else {
 			state.weave.PushPreviousResponse(styles.RenderPreviousResponse(styles.NoSeparator, "System keys have been successfully recovered.", []string{}, ""))
 			model, err := NewFundGasStationConfirmationInput(weavecontext.SetCurrentState(m.Ctx, state))
@@ -2540,13 +2541,49 @@ func (m *GenerateOrRecoverSystemKeysLoading) View() string {
 
 type SystemKeysMnemonicDisplayInput struct {
 	ui.TextInput
+	//ui.Clickable
 	weavecontext.BaseModel
 	question string
 }
 
+// NewSystemKeysMnemonicDisplayInput
+// TODO: Switch to the comment version if we manage to fix the height content overflow issue
 func NewSystemKeysMnemonicDisplayInput(ctx context.Context) *SystemKeysMnemonicDisplayInput {
+	//state := weavecontext.GetCurrentState[LaunchState](ctx)
 	model := &SystemKeysMnemonicDisplayInput{
 		TextInput: ui.NewTextInput(true),
+		//Clickable: *ui.NewClickable([]*ui.ClickableItem{
+		//	ui.NewClickableItem(map[bool]string{
+		//		true:  "Copied! Click to copy again",
+		//		false: "Click here to copy",
+		//	}, func() error {
+		//		return io.CopyToClipboard(state.systemKeyOperatorMnemonic)
+		//	}),
+		//	ui.NewClickableItem(map[bool]string{
+		//		true:  "Copied! Click to copy again",
+		//		false: "Click here to copy",
+		//	}, func() error {
+		//		return io.CopyToClipboard(state.systemKeyBridgeExecutorMnemonic)
+		//	}),
+		//	ui.NewClickableItem(map[bool]string{
+		//		true:  "Copied! Click to copy again",
+		//		false: "Click here to copy",
+		//	}, func() error {
+		//		return io.CopyToClipboard(state.systemKeyOutputSubmitterMnemonic)
+		//	}),
+		//	ui.NewClickableItem(map[bool]string{
+		//		true:  "Copied! Click to copy again",
+		//		false: "Click here to copy",
+		//	}, func() error {
+		//		return io.CopyToClipboard(state.systemKeyBatchSubmitterMnemonic)
+		//	}),
+		//	ui.NewClickableItem(map[bool]string{
+		//		true:  "Copied! Click to copy again",
+		//		false: "Click here to copy",
+		//	}, func() error {
+		//		return io.CopyToClipboard(state.systemKeyChallengerMnemonic)
+		//	}),
+		//}...),
 		BaseModel: weavecontext.BaseModel{Ctx: ctx, CannotBack: true},
 		question:  "Type `continue` to proceed after you have securely stored the mnemonic.",
 	}
@@ -2560,6 +2597,7 @@ func (m *SystemKeysMnemonicDisplayInput) GetQuestion() string {
 }
 
 func (m *SystemKeysMnemonicDisplayInput) Init() tea.Cmd {
+	//return m.Clickable.Init()
 	return nil
 }
 
@@ -2568,13 +2606,19 @@ func (m *SystemKeysMnemonicDisplayInput) Update(msg tea.Msg) (tea.Model, tea.Cmd
 		return model, cmd
 	}
 
+	//err := m.Clickable.ClickableUpdate(msg)
+	//if err != nil {
+	//	return m, m.HandlePanic(err)
+	//}
+
 	input, cmd, done := m.TextInput.Update(msg)
 	if done {
 		_ = weavecontext.PushPageAndGetState[LaunchState](m)
 		model, err := NewFundGasStationConfirmationInput(m.Ctx)
 		if err != nil {
 			return m, m.HandlePanic(err)
 		}
+		//return model, m.Clickable.PostUpdate()
 		return model, nil
 	}
 	m.TextInput = input
@@ -2585,16 +2629,31 @@ func (m *SystemKeysMnemonicDisplayInput) View() string {
 	state := weavecontext.GetCurrentState[LaunchState](m.Ctx)
 
 	var mnemonicText string
-	mnemonicText += styles.RenderMnemonic("Operator", state.systemKeyOperatorAddress, state.systemKeyOperatorMnemonic)
-	mnemonicText += styles.RenderMnemonic("Bridge Executor", state.systemKeyBridgeExecutorAddress, state.systemKeyBridgeExecutorMnemonic)
-	mnemonicText += styles.RenderMnemonic("Output Submitter", state.systemKeyOutputSubmitterAddress, state.systemKeyOutputSubmitterMnemonic)
-	mnemonicText += styles.RenderMnemonic("Batch Submitter", state.systemKeyBatchSubmitterAddress, state.systemKeyBatchSubmitterMnemonic)
-	mnemonicText += styles.RenderMnemonic("Challenger", state.systemKeyChallengerAddress, state.systemKeyChallengerMnemonic)
-
-	return m.WrapView(state.weave.Render() + "\n" +
+	//mnemonicText += styles.RenderMnemonic("Operator", state.systemKeyOperatorAddress, state.systemKeyOperatorMnemonic, m.Clickable.ClickableView(0))
+	//mnemonicText += styles.RenderMnemonic("Bridge Executor", state.systemKeyBridgeExecutorAddress, state.systemKeyBridgeExecutorMnemonic, m.Clickable.ClickableView(1))
+	//mnemonicText += styles.RenderMnemonic("Output Submitter", state.systemKeyOutputSubmitterAddress, state.systemKeyOutputSubmitterMnemonic, m.Clickable.ClickableView(2))
+	//mnemonicText += styles.RenderMnemonic("Batch Submitter", state.systemKeyBatchSubmitterAddress, state.systemKeyBatchSubmitterMnemonic, m.Clickable.ClickableView(3))
+	//mnemonicText += styles.RenderMnemonic("Challenger", state.systemKeyChallengerAddress, state.systemKeyChallengerMnemonic, m.Clickable.ClickableView(4))
+	mnemonicText += styles.RenderMnemonic("Operator", state.systemKeyOperatorAddress, state.systemKeyOperatorMnemonic, "")
+	mnemonicText += styles.RenderMnemonic("Bridge Executor", state.systemKeyBridgeExecutorAddress, state.systemKeyBridgeExecutorMnemonic, "")
+	mnemonicText += styles.RenderMnemonic("Output Submitter", state.systemKeyOutputSubmitterAddress, state.systemKeyOutputSubmitterMnemonic, "")
+	mnemonicText += styles.RenderMnemonic("Batch Submitter", state.systemKeyBatchSubmitterAddress, state.systemKeyBatchSubmitterMnemonic, "")
+	mnemonicText += styles.RenderMnemonic("Challenger", state.systemKeyChallengerAddress, state.systemKeyChallengerMnemonic, "")
+
+	userHome, err := os.UserHomeDir()
+	if err != nil {
+		m.HandlePanic(err)
+	}
+	configFilePath := filepath.Join(userHome, common.WeaveDataDirectory, LaunchConfigFilename)
+	viewText := m.WrapView(state.weave.Render() + "\n" +
 		styles.BoldUnderlineText("Important", styles.Yellow) + "\n" +
-		styles.Text("Write down these mnemonic phrases and store them in a safe place. \nIt is the only way to recover your system keys.", styles.Yellow) + "\n\n" +
+		styles.Text(fmt.Sprintf("Write down these mnemonic phrases and store them in a safe place. \nIt is the only way to recover your system keys.\n\nNote that before launching, these mnemonic phrases along with other configuration details will be stored\nin %s. You can revisit them anytime.", configFilePath), styles.Yellow) + "\n\n" +
 		mnemonicText + styles.RenderPrompt(m.GetQuestion(), []string{"`continue`"}, styles.Question) + m.TextInput.View())
+	//err = m.Clickable.ClickableUpdatePositions(viewText)
+	if err != nil {
+		m.HandlePanic(err)
+	}
+	return viewText
 }
 
 type FundGasStationConfirmationInput struct {

models/minitia/tx.go

@@ -84,15 +84,15 @@ func (lsk *L1SystemKeys) FundAccountsWithGasStation(state *LaunchState) (*FundAc
 			return nil, fmt.Errorf("failed to get active rpc for celestia: %v", err)
 		}
 
-		celestiaMinGasPrice, err := celestiaRegistry.GetMinGasPriceByDenom(DefaultCelestiaGasDenom)
-		if err != nil {
-			return nil, fmt.Errorf("failed to get celestia minimum gas price: %v", err)
-		}
+		//celestiaMinGasPrice, err := celestiaRegistry.GetMinGasPriceByDenom(DefaultCelestiaGasDenom)
+		//if err != nil {
+		//	return nil, fmt.Errorf("failed to get celestia minimum gas price: %v", err)
+		//}
 
 		celestiaChainId := celestiaRegistry.GetChainId()
 		sendCmd := exec.Command(state.celestiaBinaryPath, "tx", "bank", "send", common.WeaveGasStationKeyName,
 			lsk.BatchSubmitter.Address, fmt.Sprintf("%sutia", lsk.BatchSubmitter.Coins), "--node", celestiaRpc,
-			"--chain-id", celestiaChainId, "--gas", "200000", "--gas-prices", celestiaMinGasPrice, "--output", "json", "-y",
+			"--chain-id", celestiaChainId, "--gas", "200000", "--gas-prices", "0.1utia", "--output", "json", "-y",
 		)
 		broadcastRes, err := sendCmd.CombinedOutput()
 		if err != nil {