mirror of
https://github.com/IllusionMan1212/gorc.git
synced 2026-10-24 02:12:33 +00:00
add bubbletea, bubbles, lipgloss and term as deps. initial work on ui
This commit is contained in:
@@ -0,0 +1,296 @@
|
||||
// gorc project
|
||||
// Copyright (C) 2021 IllusionMan1212 and contributors
|
||||
//
|
||||
// This program is free software; you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// the Free Software Foundation; either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License along
|
||||
// with this program; if not, see https://www.gnu.org/licenses.
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"github.com/charmbracelet/bubbles/textinput"
|
||||
tea "github.com/charmbracelet/bubbletea"
|
||||
"github.com/charmbracelet/lipgloss"
|
||||
"golang.org/x/term"
|
||||
)
|
||||
|
||||
var (
|
||||
border = lipgloss.Border{
|
||||
Top: "─",
|
||||
Bottom: "─",
|
||||
Left: "│",
|
||||
Right: "│",
|
||||
TopLeft: "╭",
|
||||
TopRight: "╮",
|
||||
BottomLeft: "╰",
|
||||
BottomRight: "╯",
|
||||
}
|
||||
|
||||
mainStyle = lipgloss.NewStyle().
|
||||
Width(172).
|
||||
Height(40)
|
||||
|
||||
dialogStyle = lipgloss.NewStyle().
|
||||
Border(border, true).
|
||||
Align(lipgloss.Center).
|
||||
Width(50).
|
||||
Padding(3, 6).
|
||||
BorderForeground(lipgloss.Color("105"))
|
||||
|
||||
welcomeMsgStyle = lipgloss.NewStyle().
|
||||
Padding(0, 5).
|
||||
MarginBottom(5)
|
||||
|
||||
cursorStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("105"))
|
||||
focusedStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("105"))
|
||||
noStyle = lipgloss.NewStyle()
|
||||
|
||||
focusedButton = lipgloss.NewStyle().
|
||||
Background(lipgloss.Color("105")).
|
||||
Foreground(lipgloss.Color("255")).
|
||||
MarginTop(1).
|
||||
Padding(0, 2).
|
||||
Align(lipgloss.Center).
|
||||
Render("Connect")
|
||||
blurredButton = lipgloss.NewStyle().
|
||||
Background(lipgloss.Color("#809070")).
|
||||
Foreground(lipgloss.Color("#EEEEEE")).
|
||||
MarginTop(1).
|
||||
Padding(0, 2).
|
||||
Align(lipgloss.Center).
|
||||
Render("Connect")
|
||||
|
||||
focusedCheckbox = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("105")).
|
||||
MarginTop(1).
|
||||
Render("[ ] Enable TLS")
|
||||
|
||||
blurredCheckbox = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#EEEEEE")).
|
||||
MarginTop(1).
|
||||
Render("[ ] Enable TLS")
|
||||
|
||||
focusedCheckedCheckbox = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("105")).
|
||||
MarginTop(1).
|
||||
Render("[x] Enable TLS")
|
||||
|
||||
blurredCheckedCheckbox = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#EEEEEE")).
|
||||
MarginTop(1).
|
||||
Render("[x] Enable TLS")
|
||||
)
|
||||
|
||||
type loginState struct {
|
||||
focusIndex int
|
||||
inputs []textinput.Model
|
||||
|
||||
tls bool
|
||||
}
|
||||
|
||||
// TODO: have a global state that holds a separate client state and separate tui state
|
||||
type state struct {
|
||||
host string
|
||||
port string
|
||||
|
||||
nick string
|
||||
password string
|
||||
|
||||
windowWidth int
|
||||
windowHeight int
|
||||
|
||||
login loginState
|
||||
}
|
||||
|
||||
func initialLoginState() loginState {
|
||||
ls := loginState{
|
||||
inputs: make([]textinput.Model, 5),
|
||||
}
|
||||
|
||||
var t textinput.Model
|
||||
for i := range ls.inputs {
|
||||
t = textinput.NewModel()
|
||||
t.CursorStyle = cursorStyle
|
||||
|
||||
switch i {
|
||||
case 0:
|
||||
t.Placeholder = "Host"
|
||||
t.Focus()
|
||||
t.CharLimit = 40
|
||||
t.TextStyle = focusedStyle
|
||||
case 1:
|
||||
t.Placeholder = "Port"
|
||||
t.CharLimit = 5
|
||||
case 2:
|
||||
t.Placeholder = "Channel"
|
||||
t.CharLimit = 32
|
||||
case 3:
|
||||
t.Placeholder = "Nickname"
|
||||
t.CharLimit = 32
|
||||
case 4:
|
||||
t.Placeholder = "Password"
|
||||
t.CharLimit = 32
|
||||
t.EchoMode = textinput.EchoPassword
|
||||
t.EchoCharacter = '•'
|
||||
}
|
||||
|
||||
ls.inputs[i] = t
|
||||
}
|
||||
|
||||
return ls
|
||||
}
|
||||
|
||||
func initialState() state {
|
||||
width, height, err := term.GetSize(int(os.Stdout.Fd()))
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
s := state{
|
||||
login: initialLoginState(),
|
||||
windowWidth: width,
|
||||
windowHeight: height,
|
||||
}
|
||||
|
||||
return s
|
||||
}
|
||||
|
||||
func (s state) Init() tea.Cmd {
|
||||
return textinput.Blink
|
||||
}
|
||||
|
||||
func (s state) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
case tea.KeyMsg:
|
||||
switch msg.String() {
|
||||
case "ctrl+c":
|
||||
return s, tea.Quit
|
||||
case "tab", "shift+tab", "enter", "up", "down", "space":
|
||||
str := msg.String()
|
||||
|
||||
// run the newClient cmd when pressing "enter" while focused on the connect button
|
||||
if str == "enter" && s.login.focusIndex == len(s.login.inputs)+1 {
|
||||
return s, tea.Quit // TODO: change this to create a newClient and connect using a tea.Cmd
|
||||
}
|
||||
|
||||
// toggle tls state when pressing "enter" while focused on the checkbox
|
||||
// TODO: figure out why space isn't working here. all other keys works just fine
|
||||
if (str == "enter" || str == "space") && s.login.focusIndex == len(s.login.inputs) {
|
||||
s.login.tls = !s.login.tls
|
||||
return s, nil
|
||||
}
|
||||
|
||||
if str == "up" || str == "shift+tab" {
|
||||
s.login.focusIndex--
|
||||
} else {
|
||||
s.login.focusIndex++
|
||||
}
|
||||
|
||||
if s.login.focusIndex > len(s.login.inputs)+1 {
|
||||
s.login.focusIndex = 0
|
||||
} else if s.login.focusIndex < 0 {
|
||||
s.login.focusIndex = len(s.login.inputs) + 1
|
||||
}
|
||||
|
||||
cmds := make([]tea.Cmd, len(s.login.inputs))
|
||||
for i := 0; i <= len(s.login.inputs)-1; i++ {
|
||||
if i == s.login.focusIndex {
|
||||
cmds[i] = s.login.inputs[i].Focus()
|
||||
s.login.inputs[i].PromptStyle = focusedStyle
|
||||
s.login.inputs[i].TextStyle = focusedStyle
|
||||
continue
|
||||
}
|
||||
|
||||
s.login.inputs[i].Blur()
|
||||
s.login.inputs[i].PromptStyle = noStyle
|
||||
s.login.inputs[i].TextStyle = noStyle
|
||||
}
|
||||
|
||||
return s, tea.Batch(cmds...)
|
||||
}
|
||||
case tea.WindowSizeMsg:
|
||||
// NOTE: this kinda works. still need to fix it tho
|
||||
width, height, err := term.GetSize(int(os.Stdout.Fd()))
|
||||
if err != nil {
|
||||
log.Print(err)
|
||||
return s, tea.Quit
|
||||
}
|
||||
|
||||
s.windowWidth = width
|
||||
s.windowHeight = height
|
||||
|
||||
mainStyle = mainStyle.
|
||||
Width(width).
|
||||
Height(height)
|
||||
|
||||
return s, nil
|
||||
}
|
||||
|
||||
cmd := s.updateInputs(msg)
|
||||
|
||||
return s, cmd
|
||||
}
|
||||
|
||||
func (s *state) updateInputs(msg tea.Msg) tea.Cmd {
|
||||
var cmds = make([]tea.Cmd, len(s.login.inputs))
|
||||
|
||||
for i := range s.login.inputs {
|
||||
s.login.inputs[i], cmds[i] = s.login.inputs[i].Update(msg)
|
||||
}
|
||||
|
||||
return tea.Batch(cmds...)
|
||||
}
|
||||
|
||||
func (s state) View() string {
|
||||
var sb strings.Builder
|
||||
|
||||
for i := range s.login.inputs {
|
||||
sb.WriteString(s.login.inputs[i].View())
|
||||
if i < len(s.login.inputs)-1 {
|
||||
sb.WriteRune('\n')
|
||||
}
|
||||
}
|
||||
|
||||
checkbox := &blurredCheckbox
|
||||
// if the checkbox is focused
|
||||
if s.login.focusIndex == len(s.login.inputs) {
|
||||
checkbox = &focusedCheckbox
|
||||
// if tls is enabled
|
||||
if s.login.tls {
|
||||
checkbox = &focusedCheckedCheckbox
|
||||
}
|
||||
}
|
||||
|
||||
// if the checkbox is not focused and tls is enabled
|
||||
if s.login.focusIndex != len(s.login.inputs) && s.login.tls {
|
||||
checkbox = &blurredCheckedCheckbox
|
||||
}
|
||||
|
||||
fmt.Fprintf(&sb, "\n%s\n", *checkbox)
|
||||
|
||||
button := &blurredButton
|
||||
if s.login.focusIndex == len(s.login.inputs)+1 {
|
||||
button = &focusedButton
|
||||
}
|
||||
fmt.Fprintf(&sb, "\n%s\n", *button)
|
||||
|
||||
ui := lipgloss.JoinVertical(0, welcomeMsgStyle.Render(welcomeMsg), dialogStyle.Render(sb.String()))
|
||||
|
||||
final := lipgloss.Place(s.windowWidth, s.windowHeight, lipgloss.Center, lipgloss.Top, ui)
|
||||
|
||||
return mainStyle.Render(final)
|
||||
}
|
||||
Reference in New Issue
Block a user