main.go

package main

import (
  "fmt"
  "os"
  "os/exec"

  tea "github.com/charmbracelet/bubbletea"
)

type menuOption string

const (
  optionLock      menuOption = "Lock"
  optionReboot    menuOption = "Reboot"
  optionPowerOff  menuOption = "Power Off"
)

var options = []menuOption{optionLock, optionReboot, optionPowerOff}

type model struct {
  cursor   int
  quitting bool
}

func (m model) Init() tea.Cmd {
  return nil
}

func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
  switch msg := msg.(type) {
  case tea.KeyMsg:
    switch msg.String() {
    case "q", "esc":
      m.quitting = true
      return m, tea.Quit
    case "up":
      if m.cursor > 0 {
        m.cursor--
      }
    case "down":
      if m.cursor < len(options) - 1 {
        m.cursor++
      }
    case "enter":
      return m, executeOption(options[m.cursor])
    }
  }
  return m, nil
}

func (m model) View() string {
  if m.quitting {
    return "Goodbye!\n"
  }

  s := "Power Menu\n\n"

  for i, option := range options {
    cursor := " "
    if m.cursor == i {
      cursor = ">"
    }
    s += fmt.Sprintf("%s %s\n", cursor, option)
  }

  s += "\n[↑/↓] Navigate [Enter] Select [q/Esc] Quit\n"
  return s
}

func executeOption(option menuOption) tea.Cmd {
  return func() tea.Msg {
    var cmd *exec.Cmd
    switch option {
    case optionLock:
			cmd = exec.Command("swaylock", "-f", "-i", "~/.dotfiles-swayzy/wallpapers/sakura-rose-blur.png")
    case optionReboot:
      cmd = exec.Command("systemctl", "reboot")
    case optionPowerOff:
      cmd = exec.Command("systemctl", "poweroff")
    default:
      return nil
    }

    // run command and check for errors
    err := cmd.Run()
    if err != nil {
      fmt.Fprintf(os.Stderr, "Error executing %s: %v\n", option, err)
    }
    
    return tea.Quit()
  }
} 

func main() {
  p := tea.NewProgram(model{})
  _, err := p.Run()
  if err != nil {
    fmt.Fprintf(os.Stderr, "Error: %v\n", err)
    os.Exit(1)
  }
}