504 lines
11 KiB
Go
504 lines
11 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"encoding/xml"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"sort"
|
|
"time"
|
|
|
|
"fyne.io/fyne/v2"
|
|
"fyne.io/fyne/v2/app"
|
|
"fyne.io/fyne/v2/container"
|
|
"fyne.io/fyne/v2/data/binding"
|
|
"fyne.io/fyne/v2/driver/desktop"
|
|
"fyne.io/fyne/v2/theme"
|
|
"fyne.io/fyne/v2/widget"
|
|
"golang.org/x/crypto/ssh"
|
|
"golang.org/x/crypto/ssh/knownhosts"
|
|
)
|
|
|
|
var commitCache = make(map[string]commit)
|
|
|
|
type commit struct {
|
|
hash string
|
|
date time.Time
|
|
message string
|
|
}
|
|
|
|
type xinStatus struct {
|
|
tabs *container.AppTabs
|
|
cards []fyne.CanvasObject
|
|
boundStrings []binding.ExternalString
|
|
boundBools []binding.ExternalBool
|
|
log *widget.TextGrid
|
|
repoCommit commit
|
|
config Config
|
|
upgradeProgress *widget.ProgressBar
|
|
hasReboot bool
|
|
}
|
|
|
|
type Status struct {
|
|
card *widget.Card
|
|
commit commit
|
|
client *ssh.Client
|
|
clientEstablished bool
|
|
upToDate bool
|
|
|
|
ConfigurationRevision string `json:"configurationRevision"`
|
|
NeedsRestart bool `json:"needs_restart"`
|
|
NixosVersion string `json:"nixosVersion"`
|
|
NixpkgsRevision string `json:"nixpkgsRevision"`
|
|
Host string `json:"host"`
|
|
Port int32 `json:"port"`
|
|
}
|
|
|
|
type Config struct {
|
|
Statuses []*Status `json:"statuses"`
|
|
Repo string `json:"repo"`
|
|
PrivKeyPath string `json:"priv_key_path"`
|
|
FlakeRSS string `json:"flake_rss"`
|
|
}
|
|
|
|
func (c *commit) getInfo(repo string) error {
|
|
msgCmd := exec.Command("git", "log", "--format=%B", "-n", "1", c.hash)
|
|
msgCmd.Dir = repo
|
|
msg, err := msgCmd.Output()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c.message = trim(msg)
|
|
|
|
dateCmd := exec.Command("git", "log", "--format=%ci", c.hash)
|
|
dateCmd.Dir = repo
|
|
d, err := dateCmd.Output()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
dateStr := trim(d)
|
|
date, err := time.Parse("2006-01-02 15:04:05 -0700", dateStr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.date = date
|
|
|
|
return nil
|
|
}
|
|
|
|
func NewCommit(c string) *commit {
|
|
return &commit{
|
|
hash: c,
|
|
}
|
|
}
|
|
|
|
func trim(b []byte) string {
|
|
head := bytes.Split(b, []byte("\n"))
|
|
return string(head[0])
|
|
}
|
|
|
|
func (x *xinStatus) aliveCount() float64 {
|
|
alive := 0
|
|
x.hasReboot = false
|
|
for _, s := range x.config.Statuses {
|
|
if s.clientEstablished {
|
|
alive = alive + 1
|
|
if s.NeedsRestart {
|
|
x.hasReboot = true
|
|
}
|
|
}
|
|
}
|
|
return float64(alive)
|
|
}
|
|
|
|
func (x *xinStatus) uptodateCount() float64 {
|
|
utd := 0
|
|
for _, s := range x.config.Statuses {
|
|
if s.upToDate {
|
|
utd = utd + 1
|
|
}
|
|
}
|
|
return float64(utd)
|
|
}
|
|
|
|
func (x *xinStatus) getCommit(c string) (*commit, error) {
|
|
commit := &commit{
|
|
hash: c,
|
|
}
|
|
if c == "DIRTY" {
|
|
return commit, nil
|
|
}
|
|
|
|
if commit, ok := commitCache[c]; ok {
|
|
return &commit, nil
|
|
} else {
|
|
commit := NewCommit(c)
|
|
err := commit.getInfo(x.config.Repo)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
commitCache[c] = *commit
|
|
}
|
|
|
|
return commit, nil
|
|
}
|
|
|
|
func (x *xinStatus) updateRepoInfo() error {
|
|
switch {
|
|
case (x.config.Repo != "" && x.config.FlakeRSS == ""):
|
|
revCmd := exec.Command("git", "rev-parse", "HEAD")
|
|
revCmd.Dir = x.config.Repo
|
|
currentRev, err := revCmd.Output()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
commit, err := x.getCommit(trim(currentRev))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
x.repoCommit = *commit
|
|
default:
|
|
resp := &Feed{}
|
|
res, err := http.Get(x.config.FlakeRSS)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer res.Body.Close()
|
|
|
|
if err = xml.NewDecoder(res.Body).Decode(&resp); err != nil {
|
|
return err
|
|
}
|
|
cmit, err := resp.LatestHash()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
x.repoCommit = *cmit
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (x *xinStatus) updateHostInfo() error {
|
|
khFile := path.Clean(path.Join(os.Getenv("HOME"), ".ssh/known_hosts"))
|
|
hostKeyCB, err := knownhosts.New(khFile)
|
|
if err != nil {
|
|
return fmt.Errorf("can't parse %q: %q", khFile, err)
|
|
}
|
|
|
|
key, err := os.ReadFile(x.config.PrivKeyPath)
|
|
if err != nil {
|
|
return fmt.Errorf("can't load key %q: %q", x.config.PrivKeyPath, err)
|
|
}
|
|
|
|
signer, err := ssh.ParsePrivateKey(key)
|
|
if err != nil {
|
|
return fmt.Errorf("can't parse key: %q", err)
|
|
}
|
|
|
|
sshConf := &ssh.ClientConfig{
|
|
User: "root",
|
|
HostKeyAlgorithms: []string{"ssh-ed25519"},
|
|
Auth: []ssh.AuthMethod{
|
|
ssh.PublicKeys(signer),
|
|
},
|
|
Timeout: 2 * time.Second,
|
|
HostKeyCallback: hostKeyCB,
|
|
}
|
|
|
|
upToDateCount := len(x.config.Statuses)
|
|
for _, s := range x.config.Statuses {
|
|
var err error
|
|
ds := fmt.Sprintf("%s:%d", s.Host, s.Port)
|
|
if !s.clientEstablished {
|
|
s.client, err = ssh.Dial("tcp", ds, sshConf)
|
|
if err != nil {
|
|
s.card.Subtitle = "can't connect"
|
|
upToDateCount = upToDateCount - 1
|
|
x.Log(fmt.Sprintf("can't Dial host %q (%q): %q", s.Host, ds, err))
|
|
s.card.Refresh()
|
|
continue
|
|
}
|
|
s.clientEstablished = true
|
|
}
|
|
|
|
session, err := s.client.NewSession()
|
|
if err != nil {
|
|
x.Log(fmt.Sprintf("can't create session: %q", err))
|
|
upToDateCount = upToDateCount - 1
|
|
s.clientEstablished = false
|
|
continue
|
|
}
|
|
defer session.Close()
|
|
|
|
output, err := session.Output("xin-status")
|
|
if err != nil {
|
|
x.Log(fmt.Sprintf("can't run command: %q", err))
|
|
upToDateCount = upToDateCount - 1
|
|
continue
|
|
}
|
|
|
|
err = json.Unmarshal(output, s)
|
|
if err != nil {
|
|
x.Log(err.Error())
|
|
upToDateCount = upToDateCount - 1
|
|
continue
|
|
}
|
|
|
|
if s.ConfigurationRevision != x.repoCommit.hash {
|
|
s.card.Subtitle = fmt.Sprintf("%.8s", s.ConfigurationRevision)
|
|
upToDateCount = upToDateCount - 1
|
|
} else {
|
|
s.card.Subtitle = ""
|
|
}
|
|
s.card.Refresh()
|
|
|
|
commit, err := x.getCommit(s.ConfigurationRevision)
|
|
if err != nil {
|
|
x.Log(err.Error())
|
|
continue
|
|
}
|
|
s.commit = *commit
|
|
|
|
s.upToDate = false
|
|
if s.commit.hash == x.repoCommit.hash {
|
|
s.upToDate = true
|
|
}
|
|
}
|
|
|
|
x.upgradeProgress.SetValue(float64(upToDateCount))
|
|
|
|
return nil
|
|
}
|
|
|
|
func (x *xinStatus) Log(s string) {
|
|
log.Println(s)
|
|
/*
|
|
text := x.log.Text()
|
|
now := time.Now()
|
|
log.Println(s)
|
|
x.log.SetText(strings.Join([]string{
|
|
fmt.Sprintf("%s: %s", now.Format(time.RFC822), s),
|
|
text,
|
|
}, "\n"))
|
|
*/
|
|
}
|
|
|
|
func (c *Config) Load(file string) error {
|
|
data, err := ioutil.ReadFile(file)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return json.Unmarshal(data, &c)
|
|
}
|
|
|
|
func (s *Status) ToTable() *widget.Table {
|
|
t := widget.NewTable(
|
|
func() (int, int) {
|
|
return 4, 2
|
|
},
|
|
func() fyne.CanvasObject {
|
|
return widget.NewLabel("")
|
|
},
|
|
func(i widget.TableCellID, o fyne.CanvasObject) {
|
|
if i.Col == 0 {
|
|
switch i.Row {
|
|
case 0:
|
|
o.(*widget.Label).SetText("NixOS Version")
|
|
case 1:
|
|
o.(*widget.Label).SetText("Nixpkgs Revision")
|
|
case 2:
|
|
o.(*widget.Label).SetText("Configuration Revision")
|
|
case 3:
|
|
o.(*widget.Label).SetText("Restart?")
|
|
}
|
|
}
|
|
if i.Col == 1 {
|
|
switch i.Row {
|
|
case 0:
|
|
o.(*widget.Label).SetText(s.NixosVersion)
|
|
case 1:
|
|
o.(*widget.Label).SetText(s.NixpkgsRevision)
|
|
case 2:
|
|
o.(*widget.Label).SetText(s.ConfigurationRevision)
|
|
case 3:
|
|
str := "No"
|
|
if s.NeedsRestart {
|
|
str = "Yes"
|
|
}
|
|
o.(*widget.Label).SetText(str)
|
|
}
|
|
}
|
|
},
|
|
)
|
|
|
|
t.Refresh()
|
|
|
|
t.SetColumnWidth(0, 200.0)
|
|
t.SetColumnWidth(1, 33.0)
|
|
|
|
return t
|
|
}
|
|
|
|
func buildCards(stat *xinStatus) fyne.CanvasObject {
|
|
var cards []fyne.CanvasObject
|
|
sort.Slice(stat.config.Statuses, func(i, j int) bool {
|
|
return stat.config.Statuses[i].Host < stat.config.Statuses[j].Host
|
|
})
|
|
for _, s := range stat.config.Statuses {
|
|
commitBStr := binding.BindString(&s.commit.message)
|
|
bsl := widget.NewLabelWithData(commitBStr)
|
|
|
|
verBStr := binding.BindString(&s.NixosVersion)
|
|
bvl := widget.NewLabelWithData(verBStr)
|
|
|
|
restartBBool := binding.BindBool(&s.NeedsRestart)
|
|
bbl := widget.NewCheckWithData("Reboot", restartBBool)
|
|
bbl.Disable()
|
|
|
|
stat.boundStrings = append(stat.boundStrings, commitBStr)
|
|
stat.boundStrings = append(stat.boundStrings, verBStr)
|
|
stat.boundBools = append(stat.boundBools, restartBBool)
|
|
|
|
card := widget.NewCard(s.Host, "",
|
|
container.NewVBox(
|
|
container.NewHBox(bvl),
|
|
container.NewHBox(bbl),
|
|
container.NewHBox(bsl),
|
|
),
|
|
)
|
|
|
|
s.card = card
|
|
cards = append(cards, card)
|
|
stat.cards = append(stat.cards, card)
|
|
}
|
|
|
|
stat.upgradeProgress = widget.NewProgressBar()
|
|
stat.upgradeProgress.Min = 0
|
|
stat.upgradeProgress.Max = stat.aliveCount()
|
|
stat.upgradeProgress.TextFormatter = func() string {
|
|
return fmt.Sprintf("%.0f of %.0f hosts up-to-date",
|
|
stat.upgradeProgress.Value, stat.upgradeProgress.Max)
|
|
}
|
|
|
|
bsCommitMsg := binding.BindString(&stat.repoCommit.message)
|
|
bsCommitHash := binding.BindString(&stat.repoCommit.hash)
|
|
|
|
stat.boundStrings = append(stat.boundStrings, bsCommitMsg)
|
|
stat.boundStrings = append(stat.boundStrings, bsCommitHash)
|
|
|
|
statusCard := widget.NewCard("Xin Status", "", container.NewVBox(
|
|
widget.NewLabelWithData(bsCommitMsg),
|
|
stat.upgradeProgress,
|
|
))
|
|
stat.cards = append(cards, statusCard)
|
|
return container.NewVBox(
|
|
statusCard,
|
|
container.NewGridWithColumns(3, cards...),
|
|
)
|
|
}
|
|
|
|
func main() {
|
|
status := &xinStatus{}
|
|
dataPath := path.Clean(path.Join(os.Getenv("HOME"), ".xin.json"))
|
|
err := status.config.Load(dataPath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
a := app.New()
|
|
w := a.NewWindow("xintray")
|
|
|
|
ctrlQ := &desktop.CustomShortcut{KeyName: fyne.KeyQ, Modifier: fyne.KeyModifierControl}
|
|
ctrlW := &desktop.CustomShortcut{KeyName: fyne.KeyW, Modifier: fyne.KeyModifierControl}
|
|
w.Canvas().AddShortcut(ctrlQ, func(shortcut fyne.Shortcut) {
|
|
a.Quit()
|
|
})
|
|
w.Canvas().AddShortcut(ctrlW, func(shortcut fyne.Shortcut) {
|
|
w.Hide()
|
|
})
|
|
|
|
tabs := container.NewAppTabs(
|
|
container.NewTabItemWithIcon("Status", theme.ComputerIcon(), buildCards(status)),
|
|
)
|
|
|
|
status.tabs = tabs
|
|
status.log = widget.NewTextGrid()
|
|
|
|
err = status.updateRepoInfo()
|
|
if err != nil {
|
|
status.log.SetText(err.Error())
|
|
}
|
|
|
|
go func() {
|
|
for {
|
|
err = status.updateRepoInfo()
|
|
if err != nil {
|
|
status.log.SetText(err.Error())
|
|
}
|
|
|
|
err = status.updateHostInfo()
|
|
if err != nil {
|
|
status.log.SetText(err.Error())
|
|
}
|
|
for _, s := range status.boundStrings {
|
|
s.Reload()
|
|
}
|
|
for _, s := range status.boundBools {
|
|
s.Reload()
|
|
}
|
|
time.Sleep(3 * time.Second)
|
|
status.upgradeProgress.Max = status.aliveCount()
|
|
}
|
|
}()
|
|
|
|
for _, s := range status.config.Statuses {
|
|
tabs.Append(container.NewTabItem(s.Host, s.ToTable()))
|
|
}
|
|
|
|
tabs.SetTabLocation(container.TabLocationLeading)
|
|
|
|
iconImg := buildImage(status)
|
|
a.SetIcon(iconImg)
|
|
|
|
if desk, ok := a.(desktop.App); ok {
|
|
iconImg := buildImage(status)
|
|
m := fyne.NewMenu("xintray",
|
|
fyne.NewMenuItem("Show", func() {
|
|
w.Show()
|
|
}))
|
|
desk.SetSystemTrayMenu(m)
|
|
desk.SetSystemTrayIcon(iconImg)
|
|
a.SetIcon(iconImg)
|
|
go func() {
|
|
for {
|
|
img := buildImage(status)
|
|
desk.SetSystemTrayIcon(img)
|
|
a.SetIcon(img)
|
|
time.Sleep(3 * time.Second)
|
|
}
|
|
}()
|
|
}
|
|
|
|
status.log.SetText("starting...")
|
|
|
|
w.SetContent(container.NewAppTabs(
|
|
container.NewTabItem("Hosts", tabs),
|
|
container.NewTabItem("Config", container.NewMax(widget.NewCard("Config", "", nil))),
|
|
container.NewTabItem("Logs", container.NewMax(status.log)),
|
|
))
|
|
w.SetCloseIntercept(func() {
|
|
w.Hide()
|
|
})
|
|
w.ShowAndRun()
|
|
}
|