jb
138 lines
2.5 KiB
Go
138 lines
2.5 KiB
Go
package tui
|
|
|
|
import (
|
|
"math/rand"
|
|
|
|
tea "charm.land/bubbletea/v2"
|
|
)
|
|
|
|
func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
if key, ok := msg.(tea.KeyPressMsg); ok {
|
|
switch key.String() {
|
|
case "ctrl+c":
|
|
return m, tea.Quit
|
|
}
|
|
}
|
|
|
|
switch m.currentStep {
|
|
case StepCheckDocker:
|
|
return m.updateCheckDocker(msg)
|
|
case StepDockerInstall:
|
|
return m.updateDockerInstall(msg)
|
|
case StepDownloadImage:
|
|
return m.updateDownloadImage(msg)
|
|
case StepIPQuestion:
|
|
return m.updateIPQuestion(msg)
|
|
//case StepInstallWireguard
|
|
// return m.updateInstallWireguard(msg)
|
|
case StepServerConfig:
|
|
done, cmd := m.serverForm.Update(msg)
|
|
|
|
if done {
|
|
m.currentStep = StepDatabaseConfig
|
|
}
|
|
|
|
return m, cmd
|
|
case StepDatabaseConfig:
|
|
done, cmd := m.dbForm.Update(msg)
|
|
|
|
if done {
|
|
m.currentStep = StepCertConfig
|
|
}
|
|
|
|
return m, cmd
|
|
case StepCertConfig:
|
|
done, cmd := m.certForm.Update(msg)
|
|
|
|
if done {
|
|
m.currentStep = StepDone
|
|
}
|
|
|
|
return m, cmd
|
|
case StepDone:
|
|
return m, tea.Quit
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
func (m Model) updateCheckDocker(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case DockerCheckedMsg:
|
|
m.dockerInstalled = msg.Installed
|
|
m.checkDockerDone = true
|
|
|
|
case TickMsg:
|
|
if m.checkProgress < 1.0 {
|
|
m.checkProgress += 0.15 + (rand.Float64()*0.15 - 0.15)
|
|
if m.checkProgress > 1.0 {
|
|
m.checkProgress = 1.0
|
|
}
|
|
return m, tickCmd()
|
|
}
|
|
case tea.KeyPressMsg:
|
|
if m.checkDockerDone {
|
|
if m.dockerInstalled {
|
|
m.currentStep = StepIPQuestion
|
|
} else {
|
|
m.currentStep = StepDockerInstall
|
|
}
|
|
}
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
func (m Model) updateDockerInstall(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case tea.KeyPressMsg:
|
|
return m, tea.Quit
|
|
case DockerInstalledMsg:
|
|
if msg.Err != nil {
|
|
m.err = msg.Err
|
|
} else {
|
|
m.currentStep = StepIPQuestion
|
|
}
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
func (m Model) updateIPQuestion(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
numOptions := 2
|
|
|
|
switch msg := msg.(type) {
|
|
case tea.KeyPressMsg:
|
|
switch msg.String() {
|
|
case "up", "k":
|
|
if m.cursor > 0 {
|
|
m.cursor--
|
|
}
|
|
case "down", "j":
|
|
if m.cursor < numOptions-1 {
|
|
m.cursor++
|
|
}
|
|
case "enter":
|
|
// Yes
|
|
if m.cursor == 0 {
|
|
m.currentStep = StepServerConfig
|
|
return m, nil
|
|
}
|
|
|
|
m.currentStep = StepInstallWireguard
|
|
return m, nil
|
|
}
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
func (m Model) updateWaitForProgramQuit(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg.(type) {
|
|
case tea.KeyPressMsg:
|
|
return m, tea.Quit
|
|
}
|
|
|
|
return m, nil
|
|
}
|