133 lines
2.8 KiB
Go
133 lines
2.8 KiB
Go
package steps
|
|
|
|
import (
|
|
"miasma-installer/tui/styles"
|
|
|
|
"github.com/charmbracelet/bubbles/textinput"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
)
|
|
|
|
type EncryptionModel struct {
|
|
EnableEncryption bool
|
|
Password string
|
|
ConfirmPassword string
|
|
passwordInput textinput.Model
|
|
confirmInput textinput.Model
|
|
focusIndex int
|
|
Finished bool
|
|
showInputs bool
|
|
}
|
|
|
|
func NewEncryptionModel() EncryptionModel {
|
|
pi := textinput.New()
|
|
pi.Placeholder = "Enter encryption password"
|
|
pi.EchoMode = textinput.EchoPassword
|
|
pi.Focus()
|
|
|
|
ci := textinput.New()
|
|
ci.Placeholder = "Confirm password"
|
|
ci.EchoMode = textinput.EchoPassword
|
|
|
|
return EncryptionModel{
|
|
EnableEncryption: true,
|
|
passwordInput: pi,
|
|
confirmInput: ci,
|
|
showInputs: true,
|
|
}
|
|
}
|
|
|
|
func (m EncryptionModel) Init() tea.Cmd {
|
|
return textinput.Blink
|
|
}
|
|
|
|
func (m EncryptionModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
if m.Finished {
|
|
return m, nil
|
|
}
|
|
|
|
switch msg := msg.(type) {
|
|
case tea.KeyMsg:
|
|
switch msg.String() {
|
|
case "enter":
|
|
if !m.showInputs {
|
|
m.Finished = true
|
|
return m, nil
|
|
}
|
|
|
|
if m.focusIndex == 0 {
|
|
m.focusIndex = 1
|
|
m.passwordInput.Blur()
|
|
return m, m.confirmInput.Focus()
|
|
} else if m.focusIndex == 1 {
|
|
if m.passwordInput.Value() == m.confirmInput.Value() && m.passwordInput.Value() != "" {
|
|
m.Password = m.passwordInput.Value()
|
|
m.Finished = true
|
|
return m, nil
|
|
}
|
|
}
|
|
|
|
case "y":
|
|
if !m.showInputs {
|
|
m.EnableEncryption = true
|
|
m.showInputs = true
|
|
return m, m.passwordInput.Focus()
|
|
}
|
|
|
|
case "n":
|
|
if !m.showInputs {
|
|
m.EnableEncryption = false
|
|
m.Finished = true
|
|
return m, nil
|
|
}
|
|
|
|
case "tab", "shift+tab":
|
|
if m.showInputs {
|
|
if m.focusIndex == 0 {
|
|
m.focusIndex = 1
|
|
m.passwordInput.Blur()
|
|
return m, m.confirmInput.Focus()
|
|
} else {
|
|
m.focusIndex = 0
|
|
m.confirmInput.Blur()
|
|
return m, m.passwordInput.Focus()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var cmd tea.Cmd
|
|
if m.showInputs {
|
|
if m.focusIndex == 0 {
|
|
m.passwordInput, cmd = m.passwordInput.Update(msg)
|
|
} else {
|
|
m.confirmInput, cmd = m.confirmInput.Update(msg)
|
|
}
|
|
}
|
|
|
|
return m, cmd
|
|
}
|
|
|
|
func (m EncryptionModel) View() string {
|
|
s := styles.TitleStyle.Render("Disk Encryption")
|
|
s += "\n\n"
|
|
|
|
if !m.showInputs {
|
|
s += "Enable LUKS2 disk encryption? (recommended)\n\n"
|
|
s += styles.HelpStyle.Render("[y] Yes [n] No")
|
|
} else {
|
|
s += "Enter encryption password:\n\n"
|
|
s += m.passwordInput.View() + "\n\n"
|
|
s += m.confirmInput.View() + "\n\n"
|
|
|
|
if m.passwordInput.Value() != "" && m.confirmInput.Value() != "" {
|
|
if m.passwordInput.Value() != m.confirmInput.Value() {
|
|
s += styles.HelpStyle.Render("Passwords do not match\n")
|
|
}
|
|
}
|
|
|
|
s += styles.HelpStyle.Render("[Tab] Switch fields [Enter] Continue")
|
|
}
|
|
|
|
return styles.MainStyle.Render(s)
|
|
}
|