(WIP) console-frontend: initial implementation of TUI dashboard, API client, WebSocket handling, and protocol buffer support

Signed-off-by: YoungSoo Shin <shinys000114@gmail.com>
This commit is contained in:
2025-12-12 18:31:20 +09:00
parent dde9611058
commit edf6cf40cb
5 changed files with 1474 additions and 0 deletions

View File

@@ -0,0 +1,35 @@
module console-frontend
go 1.24.0
toolchain go1.24.11
require (
github.com/charmbracelet/bubbles v0.21.0
github.com/charmbracelet/bubbletea v1.3.10
github.com/charmbracelet/lipgloss v1.1.0
github.com/gorilla/websocket v1.5.3
google.golang.org/protobuf v1.36.10
)
require (
github.com/atotto/clipboard v0.1.4 // indirect
github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
github.com/charmbracelet/colorprofile v0.2.3-0.20250311203215-f60798e515dc // indirect
github.com/charmbracelet/x/ansi v0.10.1 // indirect
github.com/charmbracelet/x/cellbuf v0.0.13-0.20250311204145-2c3ea96c31dd // indirect
github.com/charmbracelet/x/term v0.2.1 // indirect
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
github.com/hinshun/vt10x v0.0.0-20220301184237-5011da428d02 // indirect
github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
github.com/mattn/go-isatty v0.0.20 // indirect
github.com/mattn/go-localereader v0.0.1 // indirect
github.com/mattn/go-runewidth v0.0.16 // indirect
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
github.com/muesli/cancelreader v0.2.2 // indirect
github.com/muesli/termenv v0.16.0 // indirect
github.com/rivo/uniseg v0.4.7 // indirect
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect
golang.org/x/sys v0.36.0 // indirect
golang.org/x/text v0.3.8 // indirect
)

View File

@@ -0,0 +1,55 @@
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/charmbracelet/bubbles v0.21.0 h1:9TdC97SdRVg/1aaXNVWfFH3nnLAwOXr8Fn6u6mfQdFs=
github.com/charmbracelet/bubbles v0.21.0/go.mod h1:HF+v6QUR4HkEpz62dx7ym2xc71/KBHg+zKwJtMw+qtg=
github.com/charmbracelet/bubbletea v1.3.10 h1:otUDHWMMzQSB0Pkc87rm691KZ3SWa4KUlvF9nRvCICw=
github.com/charmbracelet/bubbletea v1.3.10/go.mod h1:ORQfo0fk8U+po9VaNvnV95UPWA1BitP1E0N6xJPlHr4=
github.com/charmbracelet/colorprofile v0.2.3-0.20250311203215-f60798e515dc h1:4pZI35227imm7yK2bGPcfpFEmuY1gc2YSTShr4iJBfs=
github.com/charmbracelet/colorprofile v0.2.3-0.20250311203215-f60798e515dc/go.mod h1:X4/0JoqgTIPSFcRA/P6INZzIuyqdFY5rm8tb41s9okk=
github.com/charmbracelet/lipgloss v1.1.0 h1:vYXsiLHVkK7fp74RkV7b2kq9+zDLoEU4MZoFqR/noCY=
github.com/charmbracelet/lipgloss v1.1.0/go.mod h1:/6Q8FR2o+kj8rz4Dq0zQc3vYf7X+B0binUUBwA0aL30=
github.com/charmbracelet/x/ansi v0.10.1 h1:rL3Koar5XvX0pHGfovN03f5cxLbCF2YvLeyz7D2jVDQ=
github.com/charmbracelet/x/ansi v0.10.1/go.mod h1:3RQDQ6lDnROptfpWuUVIUG64bD2g2BgntdxH0Ya5TeE=
github.com/charmbracelet/x/cellbuf v0.0.13-0.20250311204145-2c3ea96c31dd h1:vy0GVL4jeHEwG5YOXDmi86oYw2yuYUGqz6a8sLwg0X8=
github.com/charmbracelet/x/cellbuf v0.0.13-0.20250311204145-2c3ea96c31dd/go.mod h1:xe0nKWGd3eJgtqZRaN9RjMtK7xUYchjzPr7q6kcvCCs=
github.com/charmbracelet/x/term v0.2.1 h1:AQeHeLZ1OqSXhrAWpYUtZyX1T3zVxfpZuEQMIQaGIAQ=
github.com/charmbracelet/x/term v0.2.1/go.mod h1:oQ4enTYFV7QN4m0i9mzHrViD7TQKvNEEkHUMCmsxdUg=
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f h1:Y/CXytFA4m6baUTXGLOoWe4PQhGxaX0KpnayAqC48p4=
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f/go.mod h1:vw97MGsxSvLiUE2X8qFplwetxpGLQrlU1Q9AUEIzCaM=
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
github.com/hinshun/vt10x v0.0.0-20220301184237-5011da428d02 h1:AgcIVYPa6XJnU3phs104wLj8l5GEththEw6+F79YsIY=
github.com/hinshun/vt10x v0.0.0-20220301184237-5011da428d02/go.mod h1:Q48J4R4DvxnHolD5P8pOtXigYlRuPLGl6moFx3ulM68=
github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=
github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2JC/oIi4=
github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88=
github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo=
github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo=
github.com/muesli/termenv v0.16.0 h1:S5AlUN9dENB57rsbnkPyfdGuWIlkmzJjbFf0Tf5FWUc=
github.com/muesli/termenv v0.16.0/go.mod h1:ZRfOIKPFDYQoDFF4Olj7/QJbW60Ol/kL1pU3VfY/Cnk=
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no=
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM=
golang.org/x/exp v0.0.0-20220909182711-5c715a9e8561 h1:MDc5xs78ZrZr3HMQugiXOAkSZtfTpbJLDr/lwfgO53E=
golang.org/x/exp v0.0.0-20220909182711-5c715a9e8561/go.mod h1:cyybsKvd6eL0RnXn6p/Grxp8F5bW7iYuBgsNCOHpMYE=
golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k=
golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
golang.org/x/text v0.3.8 h1:nAL+RVCQ9uMn3vJZbV+MRnydTJFPf8qqY42YiA6MrqY=
golang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=
google.golang.org/protobuf v1.36.10 h1:AYd7cD/uASjIL6Q9LiTjz8JLcrh/88q5UObnmY3aOOE=
google.golang.org/protobuf v1.36.10/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=

View File

@@ -0,0 +1,688 @@
package main
import (
"bytes"
"encoding/json"
"fmt"
"io"
"net/http"
"net/url"
"os"
"strings"
"time"
"console-frontend/pb"
"github.com/charmbracelet/bubbles/textinput"
"github.com/charmbracelet/bubbles/viewport"
tea "github.com/charmbracelet/bubbletea"
"github.com/charmbracelet/lipgloss"
"github.com/gorilla/websocket"
"github.com/hinshun/vt10x"
"google.golang.org/protobuf/proto"
)
// --- Configuration ---
const DefaultBaseURL = "http://192.168.4.1"
// Global HTTP Client for reuse
var httpClient = &http.Client{
Timeout: 5 * time.Second,
}
// --- Styles ---
var (
titleStyle = lipgloss.NewStyle().
Foreground(lipgloss.Color("#FFFDF5")).
Background(lipgloss.Color("#25A065")).
Padding(0, 1)
infoStyle = lipgloss.NewStyle().
Foreground(lipgloss.Color("#FFFDF5")).
Padding(0, 1)
headerStyle = lipgloss.NewStyle().
Foreground(lipgloss.Color("205")).
Bold(true)
statusBarStyle = lipgloss.NewStyle().
Foreground(lipgloss.Color("#FFFDF5")).
Background(lipgloss.Color("#3C3C3C")).
Padding(0, 1)
commandModeStyle = lipgloss.NewStyle().
Foreground(lipgloss.Color("#000000")).
Background(lipgloss.Color("#D75F00")).
Padding(0, 1).
Bold(true)
errorStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("196"))
)
// --- Messages ---
type sessionMsg struct {
token string
}
type initStatusMsg struct {
mainOn bool
usbOn bool
}
type errMsg error
type wsMsg *pb.StatusMessage
// --- API Client ---
type LoginRequest struct {
Username string `json:"username"`
Password string `json:"password"`
}
type LoginResponse struct {
Token string `json:"token"`
}
type ControlRequest struct {
Load12vOn *bool `json:"load_12v_on,omitempty"`
Load5vOn *bool `json:"load_5v_on,omitempty"`
PowerTrigger *bool `json:"power_trigger,omitempty"`
ResetTrigger *bool `json:"reset_trigger,omitempty"`
}
type ControlStatusResponse struct {
Load12vOn bool `json:"load_12v_on"`
Load5vOn bool `json:"load_5v_on"`
}
func login(baseURL, username, password string) tea.Cmd {
return func() tea.Msg {
baseURL = strings.TrimRight(baseURL, "/")
reqBody, _ := json.Marshal(LoginRequest{Username: username, Password: password})
resp, err := httpClient.Post(baseURL+"/login", "application/json", bytes.NewBuffer(reqBody))
if err != nil {
return errMsg(err)
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
body, _ := io.ReadAll(resp.Body)
return errMsg(fmt.Errorf("login failed: %s", string(body)))
}
var loginResp LoginResponse
if err := json.NewDecoder(resp.Body).Decode(&loginResp); err != nil {
return errMsg(err)
}
return sessionMsg{token: loginResp.Token}
}
}
func fetchControlStatus(baseURL, token string) tea.Cmd {
return func() tea.Msg {
baseURL = strings.TrimRight(baseURL, "/")
req, _ := http.NewRequest("GET", baseURL+"/api/control", nil)
req.Header.Set("Authorization", "Bearer "+token)
resp, err := httpClient.Do(req)
if err != nil {
return errMsg(err)
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
return errMsg(fmt.Errorf("failed to fetch status: %d", resp.StatusCode))
}
var statusResp ControlStatusResponse
if err := json.NewDecoder(resp.Body).Decode(&statusResp); err != nil {
return errMsg(err)
}
return initStatusMsg{
mainOn: statusResp.Load12vOn,
usbOn: statusResp.Load5vOn,
}
}
}
func sendControl(baseURL, token string, payload ControlRequest) tea.Cmd {
return func() tea.Msg {
baseURL = strings.TrimRight(baseURL, "/")
reqBody, _ := json.Marshal(payload)
req, _ := http.NewRequest("POST", baseURL+"/api/control", bytes.NewBuffer(reqBody))
req.Header.Set("Authorization", "Bearer "+token)
req.Header.Set("Content-Type", "application/json")
resp, err := httpClient.Do(req)
if err != nil {
return errMsg(err)
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
body, _ := io.ReadAll(resp.Body)
return errMsg(fmt.Errorf("control failed (%d): %s", resp.StatusCode, string(body)))
}
return nil
}
}
// --- WebSocket Handling ---
func connectWebSocket(baseURL, token string, sendChan <-chan []byte) tea.Cmd {
return func() tea.Msg {
u, err := url.Parse(baseURL)
if err != nil {
return errMsg(err)
}
scheme := "ws"
if u.Scheme == "https" {
scheme = "wss"
}
wsURL := fmt.Sprintf("%s://%s/ws?token=%s", scheme, u.Host, token)
c, _, err := websocket.DefaultDialer.Dial(wsURL, nil)
if err != nil {
return errMsg(err)
}
// Writer
go func() {
for data := range sendChan {
err := c.WriteMessage(websocket.BinaryMessage, data)
if err != nil {
return
}
}
c.Close()
}()
// Reader
go func() {
defer c.Close()
for {
_, message, err := c.ReadMessage()
if err != nil {
return
}
var statusMsg pb.StatusMessage
if err := proto.Unmarshal(message, &statusMsg); err == nil {
program.Send(wsMsg(&statusMsg))
}
}
}()
return nil
}
}
var program *tea.Program
// --- Model ---
type state int
const (
stateLogin state = iota
stateDashboard
)
type model struct {
state state
baseURL string
token string
width int
height int
err error
wsSend chan []byte
serverInput textinput.Model
usernameInput textinput.Model
passwordInput textinput.Model
focusIndex int
sensorData *pb.SensorData
wifiStatus *pb.WifiStatus
swStatus *pb.LoadSwStatus
// Terminal Emulator (vt10x)
term vt10x.Terminal
logViewport viewport.Model
awaitingCommand bool
lastStatusMsg string
}
func initialModel() model {
s := textinput.New()
s.Placeholder = "http://192.168.4.1"
s.SetValue(DefaultBaseURL)
s.Focus()
u := textinput.New()
u.Placeholder = "Username"
p := textinput.New()
p.Placeholder = "Password"
p.EchoMode = textinput.EchoPassword
// 기본 80x24 초기화
vp := viewport.New(80, 24)
vp.SetContent("Waiting for data...")
term := vt10x.New(vt10x.WithWriter(io.Discard))
term.Resize(80, 24)
return model{
state: stateLogin,
serverInput: s,
usernameInput: u,
passwordInput: p,
logViewport: vp,
term: term,
swStatus: &pb.LoadSwStatus{Main: false, Usb: false},
wsSend: make(chan []byte, 100),
}
}
func (m model) Init() tea.Cmd {
return textinput.Blink
}
// renderVt10x: vt10x 상태를 ANSI 문자열로 직접 변환 (최적화됨)
func (m model) renderVt10x() string {
var sb strings.Builder
// 버퍼를 미리 할당 (가로 * 세로 * 문자당 평균 바이트 수 추정)
rows := m.logViewport.Height
cols := m.logViewport.Width
sb.Grow(rows * cols * 4)
cursor := m.term.Cursor()
// 현재 상태 추적용 변수 (중복 ANSI 코드 제거)
curFG := vt10x.DefaultFG
curBG := vt10x.DefaultBG
// 시작 시 스타일 초기화
sb.WriteString("\x1b[0m")
for y := 0; y < rows; y++ {
for x := 0; x < cols; x++ {
cell := m.term.Cell(x, y)
fg := cell.FG
bg := cell.BG
char := cell.Char
// 커서 위치 판별
isCursor := (x == cursor.X && y == cursor.Y)
// 상태가 변경되었을 때만 ANSI 코드 출력 (커서 위치 포함)
if fg != curFG || bg != curBG || isCursor {
sb.WriteString("\x1b[0m") // Reset
// 커서 위치면 반전(Reverse) 적용
if isCursor {
sb.WriteString("\x1b[7m")
}
if fg != vt10x.DefaultFG {
fmt.Fprintf(&sb, "\x1b[38;5;%dm", fg)
}
if bg != vt10x.DefaultBG {
fmt.Fprintf(&sb, "\x1b[48;5;%dm", bg)
}
curFG = fg
curBG = bg
// 커서가 지나간 후 다음 글자에서 스타일이 리셋되어야 하므로 상태 강제 변경
if isCursor {
curFG = vt10x.Color(65535)
curBG = vt10x.Color(65535)
}
}
sb.WriteRune(char)
}
// 줄바꿈 전 스타일 초기화
if y < rows-1 {
sb.WriteString("\x1b[0m\n")
curFG = vt10x.DefaultFG
curBG = vt10x.DefaultBG
}
}
return sb.String()
}
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
var cmd tea.Cmd
var cmds []tea.Cmd
switch msg := msg.(type) {
// --- Mouse Scroll Handling ---
case tea.MouseMsg:
if m.state == stateDashboard {
var data []byte
switch msg.Type {
case tea.MouseWheelUp:
data = []byte("\x1b[A") // Arrow Up
case tea.MouseWheelDown:
data = []byte("\x1b[B") // Arrow Down
}
if len(data) > 0 {
m.wsSend <- data
}
}
case tea.KeyMsg:
// LOGIN STATE
if m.state == stateLogin {
if msg.Type == tea.KeyCtrlC {
return m, tea.Quit
}
switch msg.String() {
case "tab":
m.focusIndex = (m.focusIndex + 1) % 3
m.serverInput.Blur()
m.usernameInput.Blur()
m.passwordInput.Blur()
switch m.focusIndex {
case 0:
m.serverInput.Focus()
case 1:
m.usernameInput.Focus()
case 2:
m.passwordInput.Focus()
}
return m, textinput.Blink
case "enter":
m.baseURL = m.serverInput.Value()
if m.baseURL == "" {
m.baseURL = DefaultBaseURL
}
return m, login(m.baseURL, m.usernameInput.Value(), m.passwordInput.Value())
case "esc":
return m, tea.Quit
default:
if m.focusIndex == 0 {
m.serverInput, cmd = m.serverInput.Update(msg)
cmds = append(cmds, cmd)
} else if m.focusIndex == 1 {
m.usernameInput, cmd = m.usernameInput.Update(msg)
cmds = append(cmds, cmd)
} else {
m.passwordInput, cmd = m.passwordInput.Update(msg)
cmds = append(cmds, cmd)
}
return m, tea.Batch(cmds...)
}
}
// DASHBOARD STATE
if m.state == stateDashboard {
keyStr := msg.String()
// 1. Ctrl+A Command Mode Trigger
if keyStr == "ctrl+a" {
m.awaitingCommand = !m.awaitingCommand
if m.awaitingCommand {
m.lastStatusMsg = "Command Mode: Press key (m:Main, u:USB, p:Power, r:Reset, q:Quit, a:Ctrl+A)"
} else {
m.lastStatusMsg = ""
}
return m, nil
}
// 2. Handle Command Mode Local Keys
if m.awaitingCommand {
m.awaitingCommand = false
m.lastStatusMsg = ""
switch keyStr {
case "m":
m.lastStatusMsg = "Toggling Main Power..."
newState := !m.swStatus.Main
payload := ControlRequest{Load12vOn: &newState}
return m, sendControl(m.baseURL, m.token, payload)
case "u":
m.lastStatusMsg = "Toggling USB Power..."
newState := !m.swStatus.Usb
payload := ControlRequest{Load5vOn: &newState}
return m, sendControl(m.baseURL, m.token, payload)
case "r":
m.lastStatusMsg = "Sending Reset Signal..."
t := true
payload := ControlRequest{ResetTrigger: &t}
return m, sendControl(m.baseURL, m.token, payload)
case "p":
m.lastStatusMsg = "Sending Power Action..."
t := true
payload := ControlRequest{PowerTrigger: &t}
return m, sendControl(m.baseURL, m.token, payload)
case "a": // 1. Send literal Ctrl+A
m.wsSend <- []byte{1} // 0x01
return m, nil
case "q":
return m, tea.Quit
default:
m.lastStatusMsg = "Command cancelled."
return m, nil
}
}
// 3. Send ALL other inputs to Device
var data []byte
switch msg.Type {
case tea.KeyRunes:
data = []byte(string(msg.Runes))
case tea.KeyEnter:
data = []byte{'\r'}
case tea.KeySpace:
data = []byte{' '}
case tea.KeyBackspace:
data = []byte{127}
case tea.KeyTab:
data = []byte{'\t'}
case tea.KeyEsc:
data = []byte{27}
case tea.KeyUp:
data = []byte("\x1b[A")
case tea.KeyDown:
data = []byte("\x1b[B")
case tea.KeyRight:
data = []byte("\x1b[C")
case tea.KeyLeft:
data = []byte("\x1b[D")
case tea.KeyHome:
data = []byte("\x1b[H")
case tea.KeyEnd:
data = []byte("\x1b[F")
case tea.KeyPgUp:
data = []byte("\x1b[5~")
case tea.KeyPgDown:
data = []byte("\x1b[6~")
case tea.KeyDelete:
data = []byte("\x1b[3~")
case tea.KeyInsert:
data = []byte("\x1b[2~")
default:
if len(keyStr) == 6 && strings.HasPrefix(keyStr, "ctrl+") {
c := keyStr[5]
if c >= 'a' && c <= 'z' {
data = []byte{byte(c - 'a' + 1)}
} else if c == '@' {
data = []byte{0}
} else if c == '[' {
data = []byte{27}
} else if c == '\\' {
data = []byte{28}
} else if c == ']' {
data = []byte{29}
} else if c == '^' {
data = []byte{30}
} else if c == '_' {
data = []byte{31}
}
} else if len(msg.Runes) > 0 {
data = []byte(string(msg.Runes))
}
}
if len(data) > 0 {
m.wsSend <- data
}
return m, nil
}
case tea.WindowSizeMsg:
m.width = msg.Width
m.height = msg.Height
headerHeight := 10
termW := msg.Width - 2
termH := msg.Height - headerHeight
if termH < 5 { termH = 5 }
m.logViewport.Width = termW
m.logViewport.Height = termH
// 터미널 에뮬레이터 리사이즈
m.term.Resize(termW, termH)
case sessionMsg:
m.token = msg.token
m.state = stateDashboard
return m, tea.Batch(
connectWebSocket(m.baseURL, m.token, m.wsSend),
fetchControlStatus(m.baseURL, m.token),
)
case initStatusMsg:
m.swStatus.Main = msg.mainOn
m.swStatus.Usb = msg.usbOn
return m, nil
case errMsg:
m.err = msg
return m, nil
case wsMsg:
switch payload := msg.Payload.(type) {
case *pb.StatusMessage_SensorData:
m.sensorData = payload.SensorData
case *pb.StatusMessage_WifiStatus:
m.wifiStatus = payload.WifiStatus
case *pb.StatusMessage_SwStatus:
m.swStatus = payload.SwStatus
case *pb.StatusMessage_UartData:
incoming := payload.UartData.Data
m.term.Write(incoming)
m.logViewport.SetContent(m.renderVt10x())
}
}
return m, tea.Batch(cmds...)
}
func (m model) View() string {
if m.err != nil {
return fmt.Sprintf("\nError: %v\n\nPress q to quit", errorStyle.Render(m.err.Error()))
}
if m.state == stateLogin {
return m.loginView()
}
return m.dashboardView()
}
func (m model) loginView() string {
var b strings.Builder
b.WriteString("\n\n")
b.WriteString(titleStyle.Render(" ODROID Power Mate Login "))
b.WriteString("\n\n")
b.WriteString("Server Address:\n")
b.WriteString(m.serverInput.View())
b.WriteString("\n\n")
b.WriteString("Username:\n")
b.WriteString(m.usernameInput.View())
b.WriteString("\n\nPassword:\n")
b.WriteString(m.passwordInput.View())
b.WriteString("\n\n(Press Tab to switch, Enter to login, Ctrl+C to quit)")
return lipgloss.Place(m.width, m.height, lipgloss.Center, lipgloss.Center, b.String())
}
func (m model) dashboardView() string {
header := titleStyle.Render(" ODROID Power Mate ")
if m.wifiStatus != nil {
ssid := m.wifiStatus.Ssid
if ssid == "" {
ssid = "Disconnected"
}
header += infoStyle.Render(fmt.Sprintf(" WiFi: %s (%s)", ssid, m.wifiStatus.IpAddress))
} else {
header += infoStyle.Render(" WiFi: --")
}
var statusText string
var statusStyle lipgloss.Style
if m.awaitingCommand {
statusText = " COMMAND MODE (Ctrl-A) >> Press: M(Main) U(USB) P(Power) R(Reset) Q(Quit) "
statusStyle = commandModeStyle
} else {
statusText = " TERMINAL MODE >> Press Ctrl-A for Commands "
if m.lastStatusMsg != "" {
statusText += "| " + m.lastStatusMsg
}
statusStyle = statusBarStyle
}
bar := statusStyle.Width(m.width).Render(statusText)
metrics := ""
if m.sensorData != nil {
metrics = fmt.Sprintf(
"USB: %.2f V / %.2f A / %.2f W\nMAIN: %.2f V / %.2f A / %.2f W\nVIN: %.2f V / %.2f A / %.2f W\nUptime: %ds",
m.sensorData.Usb.Voltage, m.sensorData.Usb.Current, m.sensorData.Usb.Power,
m.sensorData.Main.Voltage, m.sensorData.Main.Current, m.sensorData.Main.Power,
m.sensorData.Vin.Voltage, m.sensorData.Vin.Current, m.sensorData.Vin.Power,
m.sensorData.UptimeMs/1000,
)
} else {
metrics = "Waiting for sensor data..."
}
metricsBox := lipgloss.NewStyle().Border(lipgloss.RoundedBorder()).Padding(0, 1).Width(m.width/2 - 2).Render(metrics)
switches := "States:\n"
mainState := "[OFF]"
if m.swStatus.Main {
mainState = "[ON ]"
}
usbState := "[OFF]"
if m.swStatus.Usb {
usbState = "[ON ]"
}
switches += fmt.Sprintf("Main: %s\n", mainState)
switches += fmt.Sprintf("USB : %s\n", usbState)
switchBox := lipgloss.NewStyle().Border(lipgloss.RoundedBorder()).Padding(0, 1).Width(m.width/2 - 2).Render(switches)
topSection := lipgloss.JoinHorizontal(lipgloss.Top, metricsBox, switchBox)
termView := lipgloss.NewStyle().
Border(lipgloss.NormalBorder()).
BorderForeground(lipgloss.Color("240")).
Width(m.width - 2).
Height(m.logViewport.Height).
Render(m.logViewport.View())
return lipgloss.JoinVertical(lipgloss.Left, header, topSection, bar, termView)
}
func main() {
program = tea.NewProgram(initialModel(), tea.WithAltScreen(), tea.WithMouseCellMotion())
if _, err := program.Run(); err != nil {
fmt.Printf("Error running program: %v\n", err)
os.Exit(1)
}
}

View File

@@ -0,0 +1,647 @@
// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// protoc-gen-go v1.32.0
// protoc v3.21.12
// source: status.proto
package pb
import (
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
reflect "reflect"
sync "sync"
)
const (
// Verify that this generated code is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
// Verify that runtime/protoimpl is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)
// Represents data for a single sensor channel
type SensorChannelData struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Voltage float32 `protobuf:"fixed32,1,opt,name=voltage,proto3" json:"voltage,omitempty"`
Current float32 `protobuf:"fixed32,2,opt,name=current,proto3" json:"current,omitempty"`
Power float32 `protobuf:"fixed32,3,opt,name=power,proto3" json:"power,omitempty"`
}
func (x *SensorChannelData) Reset() {
*x = SensorChannelData{}
if protoimpl.UnsafeEnabled {
mi := &file_status_proto_msgTypes[0]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *SensorChannelData) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*SensorChannelData) ProtoMessage() {}
func (x *SensorChannelData) ProtoReflect() protoreflect.Message {
mi := &file_status_proto_msgTypes[0]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use SensorChannelData.ProtoReflect.Descriptor instead.
func (*SensorChannelData) Descriptor() ([]byte, []int) {
return file_status_proto_rawDescGZIP(), []int{0}
}
func (x *SensorChannelData) GetVoltage() float32 {
if x != nil {
return x.Voltage
}
return 0
}
func (x *SensorChannelData) GetCurrent() float32 {
if x != nil {
return x.Current
}
return 0
}
func (x *SensorChannelData) GetPower() float32 {
if x != nil {
return x.Power
}
return 0
}
// Contains data for all sensor channels and system info
type SensorData struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Usb *SensorChannelData `protobuf:"bytes,1,opt,name=usb,proto3" json:"usb,omitempty"`
Main *SensorChannelData `protobuf:"bytes,2,opt,name=main,proto3" json:"main,omitempty"`
Vin *SensorChannelData `protobuf:"bytes,3,opt,name=vin,proto3" json:"vin,omitempty"`
TimestampMs uint64 `protobuf:"varint,4,opt,name=timestamp_ms,json=timestampMs,proto3" json:"timestamp_ms,omitempty"`
UptimeMs uint64 `protobuf:"varint,5,opt,name=uptime_ms,json=uptimeMs,proto3" json:"uptime_ms,omitempty"`
}
func (x *SensorData) Reset() {
*x = SensorData{}
if protoimpl.UnsafeEnabled {
mi := &file_status_proto_msgTypes[1]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *SensorData) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*SensorData) ProtoMessage() {}
func (x *SensorData) ProtoReflect() protoreflect.Message {
mi := &file_status_proto_msgTypes[1]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use SensorData.ProtoReflect.Descriptor instead.
func (*SensorData) Descriptor() ([]byte, []int) {
return file_status_proto_rawDescGZIP(), []int{1}
}
func (x *SensorData) GetUsb() *SensorChannelData {
if x != nil {
return x.Usb
}
return nil
}
func (x *SensorData) GetMain() *SensorChannelData {
if x != nil {
return x.Main
}
return nil
}
func (x *SensorData) GetVin() *SensorChannelData {
if x != nil {
return x.Vin
}
return nil
}
func (x *SensorData) GetTimestampMs() uint64 {
if x != nil {
return x.TimestampMs
}
return 0
}
func (x *SensorData) GetUptimeMs() uint64 {
if x != nil {
return x.UptimeMs
}
return 0
}
// Contains WiFi connection status
type WifiStatus struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Connected bool `protobuf:"varint,1,opt,name=connected,proto3" json:"connected,omitempty"`
Ssid string `protobuf:"bytes,2,opt,name=ssid,proto3" json:"ssid,omitempty"`
Rssi int32 `protobuf:"varint,3,opt,name=rssi,proto3" json:"rssi,omitempty"`
IpAddress string `protobuf:"bytes,4,opt,name=ip_address,json=ipAddress,proto3" json:"ip_address,omitempty"`
}
func (x *WifiStatus) Reset() {
*x = WifiStatus{}
if protoimpl.UnsafeEnabled {
mi := &file_status_proto_msgTypes[2]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *WifiStatus) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*WifiStatus) ProtoMessage() {}
func (x *WifiStatus) ProtoReflect() protoreflect.Message {
mi := &file_status_proto_msgTypes[2]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use WifiStatus.ProtoReflect.Descriptor instead.
func (*WifiStatus) Descriptor() ([]byte, []int) {
return file_status_proto_rawDescGZIP(), []int{2}
}
func (x *WifiStatus) GetConnected() bool {
if x != nil {
return x.Connected
}
return false
}
func (x *WifiStatus) GetSsid() string {
if x != nil {
return x.Ssid
}
return ""
}
func (x *WifiStatus) GetRssi() int32 {
if x != nil {
return x.Rssi
}
return 0
}
func (x *WifiStatus) GetIpAddress() string {
if x != nil {
return x.IpAddress
}
return ""
}
// Contains raw UART data
type UartData struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Data []byte `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"`
}
func (x *UartData) Reset() {
*x = UartData{}
if protoimpl.UnsafeEnabled {
mi := &file_status_proto_msgTypes[3]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *UartData) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*UartData) ProtoMessage() {}
func (x *UartData) ProtoReflect() protoreflect.Message {
mi := &file_status_proto_msgTypes[3]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use UartData.ProtoReflect.Descriptor instead.
func (*UartData) Descriptor() ([]byte, []int) {
return file_status_proto_rawDescGZIP(), []int{3}
}
func (x *UartData) GetData() []byte {
if x != nil {
return x.Data
}
return nil
}
// Contains load sw status
type LoadSwStatus struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Main bool `protobuf:"varint,1,opt,name=main,proto3" json:"main,omitempty"`
Usb bool `protobuf:"varint,2,opt,name=usb,proto3" json:"usb,omitempty"`
}
func (x *LoadSwStatus) Reset() {
*x = LoadSwStatus{}
if protoimpl.UnsafeEnabled {
mi := &file_status_proto_msgTypes[4]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *LoadSwStatus) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*LoadSwStatus) ProtoMessage() {}
func (x *LoadSwStatus) ProtoReflect() protoreflect.Message {
mi := &file_status_proto_msgTypes[4]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use LoadSwStatus.ProtoReflect.Descriptor instead.
func (*LoadSwStatus) Descriptor() ([]byte, []int) {
return file_status_proto_rawDescGZIP(), []int{4}
}
func (x *LoadSwStatus) GetMain() bool {
if x != nil {
return x.Main
}
return false
}
func (x *LoadSwStatus) GetUsb() bool {
if x != nil {
return x.Usb
}
return false
}
// Top-level message for all websocket communication
type StatusMessage struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
// Types that are assignable to Payload:
//
// *StatusMessage_SensorData
// *StatusMessage_WifiStatus
// *StatusMessage_SwStatus
// *StatusMessage_UartData
Payload isStatusMessage_Payload `protobuf_oneof:"payload"`
}
func (x *StatusMessage) Reset() {
*x = StatusMessage{}
if protoimpl.UnsafeEnabled {
mi := &file_status_proto_msgTypes[5]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *StatusMessage) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*StatusMessage) ProtoMessage() {}
func (x *StatusMessage) ProtoReflect() protoreflect.Message {
mi := &file_status_proto_msgTypes[5]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use StatusMessage.ProtoReflect.Descriptor instead.
func (*StatusMessage) Descriptor() ([]byte, []int) {
return file_status_proto_rawDescGZIP(), []int{5}
}
func (m *StatusMessage) GetPayload() isStatusMessage_Payload {
if m != nil {
return m.Payload
}
return nil
}
func (x *StatusMessage) GetSensorData() *SensorData {
if x, ok := x.GetPayload().(*StatusMessage_SensorData); ok {
return x.SensorData
}
return nil
}
func (x *StatusMessage) GetWifiStatus() *WifiStatus {
if x, ok := x.GetPayload().(*StatusMessage_WifiStatus); ok {
return x.WifiStatus
}
return nil
}
func (x *StatusMessage) GetSwStatus() *LoadSwStatus {
if x, ok := x.GetPayload().(*StatusMessage_SwStatus); ok {
return x.SwStatus
}
return nil
}
func (x *StatusMessage) GetUartData() *UartData {
if x, ok := x.GetPayload().(*StatusMessage_UartData); ok {
return x.UartData
}
return nil
}
type isStatusMessage_Payload interface {
isStatusMessage_Payload()
}
type StatusMessage_SensorData struct {
SensorData *SensorData `protobuf:"bytes,1,opt,name=sensor_data,json=sensorData,proto3,oneof"`
}
type StatusMessage_WifiStatus struct {
WifiStatus *WifiStatus `protobuf:"bytes,2,opt,name=wifi_status,json=wifiStatus,proto3,oneof"`
}
type StatusMessage_SwStatus struct {
SwStatus *LoadSwStatus `protobuf:"bytes,3,opt,name=sw_status,json=swStatus,proto3,oneof"`
}
type StatusMessage_UartData struct {
UartData *UartData `protobuf:"bytes,4,opt,name=uart_data,json=uartData,proto3,oneof"`
}
func (*StatusMessage_SensorData) isStatusMessage_Payload() {}
func (*StatusMessage_WifiStatus) isStatusMessage_Payload() {}
func (*StatusMessage_SwStatus) isStatusMessage_Payload() {}
func (*StatusMessage_UartData) isStatusMessage_Payload() {}
var File_status_proto protoreflect.FileDescriptor
var file_status_proto_rawDesc = []byte{
0x0a, 0x0c, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x06,
0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x22, 0x5d, 0x0a, 0x11, 0x53, 0x65, 0x6e, 0x73, 0x6f, 0x72,
0x43, 0x68, 0x61, 0x6e, 0x6e, 0x65, 0x6c, 0x44, 0x61, 0x74, 0x61, 0x12, 0x18, 0x0a, 0x07, 0x76,
0x6f, 0x6c, 0x74, 0x61, 0x67, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x02, 0x52, 0x07, 0x76, 0x6f,
0x6c, 0x74, 0x61, 0x67, 0x65, 0x12, 0x18, 0x0a, 0x07, 0x63, 0x75, 0x72, 0x72, 0x65, 0x6e, 0x74,
0x18, 0x02, 0x20, 0x01, 0x28, 0x02, 0x52, 0x07, 0x63, 0x75, 0x72, 0x72, 0x65, 0x6e, 0x74, 0x12,
0x14, 0x0a, 0x05, 0x70, 0x6f, 0x77, 0x65, 0x72, 0x18, 0x03, 0x20, 0x01, 0x28, 0x02, 0x52, 0x05,
0x70, 0x6f, 0x77, 0x65, 0x72, 0x22, 0xd5, 0x01, 0x0a, 0x0a, 0x53, 0x65, 0x6e, 0x73, 0x6f, 0x72,
0x44, 0x61, 0x74, 0x61, 0x12, 0x2b, 0x0a, 0x03, 0x75, 0x73, 0x62, 0x18, 0x01, 0x20, 0x01, 0x28,
0x0b, 0x32, 0x19, 0x2e, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x2e, 0x53, 0x65, 0x6e, 0x73, 0x6f,
0x72, 0x43, 0x68, 0x61, 0x6e, 0x6e, 0x65, 0x6c, 0x44, 0x61, 0x74, 0x61, 0x52, 0x03, 0x75, 0x73,
0x62, 0x12, 0x2d, 0x0a, 0x04, 0x6d, 0x61, 0x69, 0x6e, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32,
0x19, 0x2e, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x2e, 0x53, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x43,
0x68, 0x61, 0x6e, 0x6e, 0x65, 0x6c, 0x44, 0x61, 0x74, 0x61, 0x52, 0x04, 0x6d, 0x61, 0x69, 0x6e,
0x12, 0x2b, 0x0a, 0x03, 0x76, 0x69, 0x6e, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x19, 0x2e,
0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x2e, 0x53, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x43, 0x68, 0x61,
0x6e, 0x6e, 0x65, 0x6c, 0x44, 0x61, 0x74, 0x61, 0x52, 0x03, 0x76, 0x69, 0x6e, 0x12, 0x21, 0x0a,
0x0c, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x5f, 0x6d, 0x73, 0x18, 0x04, 0x20,
0x01, 0x28, 0x04, 0x52, 0x0b, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x4d, 0x73,
0x12, 0x1b, 0x0a, 0x09, 0x75, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x5f, 0x6d, 0x73, 0x18, 0x05, 0x20,
0x01, 0x28, 0x04, 0x52, 0x08, 0x75, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x4d, 0x73, 0x22, 0x71, 0x0a,
0x0a, 0x57, 0x69, 0x66, 0x69, 0x53, 0x74, 0x61, 0x74, 0x75, 0x73, 0x12, 0x1c, 0x0a, 0x09, 0x63,
0x6f, 0x6e, 0x6e, 0x65, 0x63, 0x74, 0x65, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x08, 0x52, 0x09,
0x63, 0x6f, 0x6e, 0x6e, 0x65, 0x63, 0x74, 0x65, 0x64, 0x12, 0x12, 0x0a, 0x04, 0x73, 0x73, 0x69,
0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x73, 0x73, 0x69, 0x64, 0x12, 0x12, 0x0a,
0x04, 0x72, 0x73, 0x73, 0x69, 0x18, 0x03, 0x20, 0x01, 0x28, 0x05, 0x52, 0x04, 0x72, 0x73, 0x73,
0x69, 0x12, 0x1d, 0x0a, 0x0a, 0x69, 0x70, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x18,
0x04, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, 0x69, 0x70, 0x41, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73,
0x22, 0x1e, 0x0a, 0x08, 0x55, 0x61, 0x72, 0x74, 0x44, 0x61, 0x74, 0x61, 0x12, 0x12, 0x0a, 0x04,
0x64, 0x61, 0x74, 0x61, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, 0x64, 0x61, 0x74, 0x61,
0x22, 0x34, 0x0a, 0x0c, 0x4c, 0x6f, 0x61, 0x64, 0x53, 0x77, 0x53, 0x74, 0x61, 0x74, 0x75, 0x73,
0x12, 0x12, 0x0a, 0x04, 0x6d, 0x61, 0x69, 0x6e, 0x18, 0x01, 0x20, 0x01, 0x28, 0x08, 0x52, 0x04,
0x6d, 0x61, 0x69, 0x6e, 0x12, 0x10, 0x0a, 0x03, 0x75, 0x73, 0x62, 0x18, 0x02, 0x20, 0x01, 0x28,
0x08, 0x52, 0x03, 0x75, 0x73, 0x62, 0x22, 0xee, 0x01, 0x0a, 0x0d, 0x53, 0x74, 0x61, 0x74, 0x75,
0x73, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x12, 0x35, 0x0a, 0x0b, 0x73, 0x65, 0x6e, 0x73,
0x6f, 0x72, 0x5f, 0x64, 0x61, 0x74, 0x61, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x12, 0x2e,
0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x2e, 0x53, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x44, 0x61, 0x74,
0x61, 0x48, 0x00, 0x52, 0x0a, 0x73, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x44, 0x61, 0x74, 0x61, 0x12,
0x35, 0x0a, 0x0b, 0x77, 0x69, 0x66, 0x69, 0x5f, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x18, 0x02,
0x20, 0x01, 0x28, 0x0b, 0x32, 0x12, 0x2e, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x2e, 0x57, 0x69,
0x66, 0x69, 0x53, 0x74, 0x61, 0x74, 0x75, 0x73, 0x48, 0x00, 0x52, 0x0a, 0x77, 0x69, 0x66, 0x69,
0x53, 0x74, 0x61, 0x74, 0x75, 0x73, 0x12, 0x33, 0x0a, 0x09, 0x73, 0x77, 0x5f, 0x73, 0x74, 0x61,
0x74, 0x75, 0x73, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x14, 0x2e, 0x73, 0x74, 0x61, 0x74,
0x75, 0x73, 0x2e, 0x4c, 0x6f, 0x61, 0x64, 0x53, 0x77, 0x53, 0x74, 0x61, 0x74, 0x75, 0x73, 0x48,
0x00, 0x52, 0x08, 0x73, 0x77, 0x53, 0x74, 0x61, 0x74, 0x75, 0x73, 0x12, 0x2f, 0x0a, 0x09, 0x75,
0x61, 0x72, 0x74, 0x5f, 0x64, 0x61, 0x74, 0x61, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x10,
0x2e, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x2e, 0x55, 0x61, 0x72, 0x74, 0x44, 0x61, 0x74, 0x61,
0x48, 0x00, 0x52, 0x08, 0x75, 0x61, 0x72, 0x74, 0x44, 0x61, 0x74, 0x61, 0x42, 0x09, 0x0a, 0x07,
0x70, 0x61, 0x79, 0x6c, 0x6f, 0x61, 0x64, 0x42, 0x06, 0x5a, 0x04, 0x2e, 0x2f, 0x70, 0x62, 0x62,
0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var (
file_status_proto_rawDescOnce sync.Once
file_status_proto_rawDescData = file_status_proto_rawDesc
)
func file_status_proto_rawDescGZIP() []byte {
file_status_proto_rawDescOnce.Do(func() {
file_status_proto_rawDescData = protoimpl.X.CompressGZIP(file_status_proto_rawDescData)
})
return file_status_proto_rawDescData
}
var file_status_proto_msgTypes = make([]protoimpl.MessageInfo, 6)
var file_status_proto_goTypes = []interface{}{
(*SensorChannelData)(nil), // 0: status.SensorChannelData
(*SensorData)(nil), // 1: status.SensorData
(*WifiStatus)(nil), // 2: status.WifiStatus
(*UartData)(nil), // 3: status.UartData
(*LoadSwStatus)(nil), // 4: status.LoadSwStatus
(*StatusMessage)(nil), // 5: status.StatusMessage
}
var file_status_proto_depIdxs = []int32{
0, // 0: status.SensorData.usb:type_name -> status.SensorChannelData
0, // 1: status.SensorData.main:type_name -> status.SensorChannelData
0, // 2: status.SensorData.vin:type_name -> status.SensorChannelData
1, // 3: status.StatusMessage.sensor_data:type_name -> status.SensorData
2, // 4: status.StatusMessage.wifi_status:type_name -> status.WifiStatus
4, // 5: status.StatusMessage.sw_status:type_name -> status.LoadSwStatus
3, // 6: status.StatusMessage.uart_data:type_name -> status.UartData
7, // [7:7] is the sub-list for method output_type
7, // [7:7] is the sub-list for method input_type
7, // [7:7] is the sub-list for extension type_name
7, // [7:7] is the sub-list for extension extendee
0, // [0:7] is the sub-list for field type_name
}
func init() { file_status_proto_init() }
func file_status_proto_init() {
if File_status_proto != nil {
return
}
if !protoimpl.UnsafeEnabled {
file_status_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*SensorChannelData); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_status_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*SensorData); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_status_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*WifiStatus); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_status_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*UartData); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_status_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*LoadSwStatus); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_status_proto_msgTypes[5].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*StatusMessage); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
}
file_status_proto_msgTypes[5].OneofWrappers = []interface{}{
(*StatusMessage_SensorData)(nil),
(*StatusMessage_WifiStatus)(nil),
(*StatusMessage_SwStatus)(nil),
(*StatusMessage_UartData)(nil),
}
type x struct{}
out := protoimpl.TypeBuilder{
File: protoimpl.DescBuilder{
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
RawDescriptor: file_status_proto_rawDesc,
NumEnums: 0,
NumMessages: 6,
NumExtensions: 0,
NumServices: 0,
},
GoTypes: file_status_proto_goTypes,
DependencyIndexes: file_status_proto_depIdxs,
MessageInfos: file_status_proto_msgTypes,
}.Build()
File_status_proto = out.File
file_status_proto_rawDesc = nil
file_status_proto_goTypes = nil
file_status_proto_depIdxs = nil
}

View File

@@ -0,0 +1,49 @@
syntax = "proto3";
package status;
option go_package = "./pb";
// Represents data for a single sensor channel
message SensorChannelData {
float voltage = 1;
float current = 2;
float power = 3;
}
// Contains data for all sensor channels and system info
message SensorData {
SensorChannelData usb = 1;
SensorChannelData main = 2;
SensorChannelData vin = 3;
uint64 timestamp_ms = 4;
uint64 uptime_ms = 5;
}
// Contains WiFi connection status
message WifiStatus {
bool connected = 1;
string ssid = 2;
int32 rssi = 3;
string ip_address = 4;
}
// Contains raw UART data
message UartData {
bytes data = 1;
}
// Contains load sw status
message LoadSwStatus {
bool main = 1;
bool usb = 2;
}
// Top-level message for all websocket communication
message StatusMessage {
oneof payload {
SensorData sensor_data = 1;
WifiStatus wifi_status = 2;
LoadSwStatus sw_status = 3;
UartData uart_data = 4;
}
}