From ed2b9374599cc00467b1c27121318748cc0d3f49 Mon Sep 17 00:00:00 2001 From: Stefan Majewsky Date: Sat, 28 Mar 2020 13:53:10 +0100 Subject: rename pkg/ to internal/ --- internal/i3status/main.go | 169 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 169 insertions(+) create mode 100644 internal/i3status/main.go (limited to 'internal/i3status/main.go') diff --git a/internal/i3status/main.go b/internal/i3status/main.go new file mode 100644 index 0000000..4af726a --- /dev/null +++ b/internal/i3status/main.go @@ -0,0 +1,169 @@ +/******************************************************************************* +* +* Copyright 2017 Stefan Majewsky +* +* 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 . +* +*******************************************************************************/ + +package i3status + +import ( + "encoding/json" + "fmt" + "os" + "sort" + "time" +) + +//Exec executes the i3status applet and returns an exit code (0 for +//success, >0 for error). +func Exec(args []string) int { + //start protocol (we handle errors here for once because if stdout is + //working, we have nothing left to do) + _, err := os.Stdout.Write([]byte("{\"version\":1}\n[\n")) + if err != nil { + fmt.Fprintf(os.Stderr, err.Error()) + return 1 + } + + //main loop + currentBlocks := make(map[string][]Block) + for { + //prepare clock (this is inline instead of split into a separate function + //because the wallclock also drives the main loop's clock, see below) + now := time.Now() + currentBlocks["clock"] = []Block{ + { + Name: "clock", + Instance: "date", + Position: PositionClock, + FullText: now.Format("2006-01-02"), + ShortText: " ", + Color: "#AAAAAA", + SeparatorBlockWidth: 6, + }, + { + Name: "clock", + Instance: "time", + Position: PositionClock, + FullText: now.Format("15:04:05"), + }, + } + + //prepare other blocks + currentBlocks["battery"] = getBatteryStatus() + currentBlocks["network"] = getNetworkStatus() + + //put blocks in rendering order + var allBlocks []Block + for _, blocks := range currentBlocks { + allBlocks = append(allBlocks, blocks...) + } + sort.Sort(byPositionAndInstance(allBlocks)) + + //write output + buf, err := json.Marshal(allBlocks) + if err == nil { + os.Stdout.Write(append(buf, ',', '\n')) + } else { + //this should not happen, but if it does, fall back to just encoding the + //error string (which always works, otherwise wtf) + buf, _ = json.Marshal(err.Error()) + fmt.Printf( + `[{"name":"error","urgent":true,"color":"#FF0000","full_text":%s}],`, + string(buf), + ) + } + + //sleep until the next second starts, so that the clock is always on time + nsec := 1000000000 - now.Nanosecond() + time.Sleep(time.Duration(nsec) * time.Nanosecond) + } + + return 0 +} + +//Block is a block of text as used in the i3status protocol. +//(Not all attributes are represented.) +type Block struct { + Position Position `json:"-"` + Name string `json:"name"` //REQUIRED + Instance string `json:"instance,omitempty"` + FullText string `json:"full_text"` //REQUIRED + ShortText string `json:"short_text"` + Color string `json:"color,omitempty"` //CSS hex syntax, e.g. #123456 + BackgroundColor string `json:"background,omitempty"` + MinWidth uint `json:"min_width,omitempty"` + Alignment Alignment `json:"align,omitempty"` //only plausible with MinWidth + Urgent bool `json:"urgent,omitempty"` + Separator bool `json:"separator"` + SeparatorBlockWidth int `json:"separator_block_width,omitempty"` +} + +type byPositionAndInstance []Block + +func (b byPositionAndInstance) Len() int { return len(b) } +func (b byPositionAndInstance) Swap(i, j int) { b[i], b[j] = b[j], b[i] } +func (b byPositionAndInstance) Less(i, j int) bool { + if b[i].Position == b[j].Position { + if b[i].Instance == "_caption" { + return true + } + if b[j].Instance == "_caption" { + return false + } + return b[i].Instance < b[j].Instance + } + return b[i].Position < b[j].Position +} + +//Alignment is the alignment of a Block. +type Alignment string + +//Acceptable values for Alignment. +const ( + AlignmentLeft Alignment = "left" + AlignmentCenter Alignment = "center" + AlignmentRight Alignment = "right" +) + +//Position defines how blocks are ordered. +type Position int + +//Acceptable values for Position, from left to right. +const ( + PositionNone Position = iota + PositionNetwork + PositionBattery + PositionClock +) + +//Order the given blocks byPositionAndInstance, then add a separator to the last one, then add a caption block of the same style in front. +func section(caption string, blocks ...Block) []Block { + if len(blocks) == 0 { + return nil + } + sort.Sort(byPositionAndInstance(blocks)) + last := len(blocks) - 1 + blocks[last].Separator = true + blocks[last].SeparatorBlockWidth = 15 + + return append([]Block{{ + Name: blocks[0].Name, + Position: blocks[0].Position, + Instance: "_caption", + FullText: caption, + Color: blocks[0].Color, + }}, blocks...) +} -- cgit v1.3.1