+package main
+
+import (
+ "fmt"
+ "log"
+ "os"
+ "os/exec"
+
+ "github.com/jezek/xgb"
+ "github.com/jezek/xgb/xinerama"
+ "github.com/jezek/xgb/xproto"
+)
+
+// TODO: Make a separate keysym file and fully populate it.
+const (
+ XK_q = 0x0071
+ XK_Return = 0xff0d
+)
+
+var xc *xgb.Conn
+var attachedScreens []xinerama.ScreenInfo
+var xroot xproto.ScreenInfo
+var keymap [256][]xproto.Keysym
+
+func main() {
+ xConn, err := xgb.NewConn()
+ if err != nil {
+ log.Fatal(err)
+ }
+ xc = xConn
+ defer xc.Close()
+
+ setup := xproto.Setup(xc)
+ if setup == nil || len(setup.Roots) < 1 {
+ log.Fatal("ILLI: Unable to parse received SetupInfo from X11 server.")
+ }
+
+ if err := xinerama.Init(xc); err != nil {
+ log.Fatal(err)
+ }
+
+ if r, err := xinerama.QueryScreens(xc).Reply(); err != nil {
+ log.Fatal(err)
+ } else {
+ if len(r.ScreenInfo) == 0 {
+ attachedScreens = []xinerama.ScreenInfo{
+ xinerama.ScreenInfo{
+ Width: setup.Roots[0].WidthInPixels,
+ Height: setup.Roots[0].HeightInPixels,
+ },
+ }
+ } else {
+ attachedScreens = r.ScreenInfo
+ }
+ }
+
+ connInfo := xproto.Setup(xc)
+ if connInfo == nil {
+ log.Fatal("ILLI: Unable to parse X connection information")
+ }
+ if len(connInfo.Roots) != 1 {
+ log.Fatal("ILLI: Inappropriate number of roots. Did xinerama initialize correctly?")
+ }
+ xroot = connInfo.Roots[0]
+
+ // Attempt to register as the window manager
+ if err := TakeWMOwnership(); err != nil {
+ if _, ok := err.(xproto.AccessError); ok {
+ log.Fatal("ILLI: Unable to register as window manager with X server. Perhaps another WM is already running?")
+ }
+ log.Fatal(err)
+ }
+
+ fmt.Println("ILLI: Successfully registered as WM with X server.")
+
+// -----------------------------------------------------------------------------
+
+ const (
+ loKey = 8
+ hiKey = 255
+ )
+
+ m := xproto.GetKeyboardMapping(xc, loKey, hiKey-loKey+1)
+ reply, err := m.Reply()
+ if err != nil {
+ log.Fatal(err)
+ }
+ if reply == nil {
+ log.Fatal("Could not load keyboard map")
+ }
+
+ for i := 0; i < hiKey-loKey+1; i++ {
+ keymap[loKey+i] = reply.Keysyms[i*int(reply.KeysymsPerKeycode) : (i+1)*int(reply.KeysymsPerKeycode)]
+ }
+ grabs := []struct {
+ sym xproto.Keysym
+ modifiers uint16
+ codes []xproto.Keycode
+ }{
+ {
+ sym: XK_q,
+ modifiers: xproto.ModMask1 | xproto.ModMaskShift,
+ },
+ {
+ sym: XK_Return,
+ modifiers: xproto.ModMask1 | xproto.ModMaskShift,
+ },
+ }
+
+ for i, syms := range keymap {
+ for _, sym := range syms {
+ for c := range grabs {
+ if grabs[c].sym == sym {
+ grabs[c].codes = append(grabs[c].codes, xproto.Keycode(i))
+ }
+ }
+ }
+ }
+ for _, grabbed := range grabs {
+ for _, code := range grabbed.codes {
+ if err := xproto.GrabKeyChecked(
+ xc,
+ false,
+ xroot.Root,
+ grabbed.modifiers,
+ code,
+ xproto.GrabModeAsync,
+ xproto.GrabModeAsync,
+ ).Check(); err != nil {
+ log.Print(err)
+ }
+
+ }
+ }
+
+// -----------------------------------------------------------------------------
+
+ tree, err := xproto.QueryTree(xc, xroot.Root).Reply()
+ if err != nil {
+ log.Fatal(err)
+ }
+ if tree != nil {
+ //workspaces = make(map[string]*Workspace)
+ //defaultw := &Workspace{mu: &sync.Mutex{}}
+ for _, c := range tree.Children {
+ if isMappedWindow(c) {
+// err := defaultw.Add(c)
+// if err != nil {
+// log.Println(err)
+// }
+ fmt.Println("ILLI: Found a client.")
+ }
+
+ }
+
+// if len(attachedScreens) > 0 {
+// defaultw.Screen = &attachedScreens[0]
+// }
+//
+// workspaces["default"] = defaultw
+//
+// if err := defaultw.TileWindows(); err != nil {
+// log.Println(err)
+// }
+
+ }
+
+eventloop:
+ for {
+ xevent, err := xc.WaitForEvent()
+ if err != nil {
+ log.Println(err)
+ continue
+ }
+ switch e := xevent.(type) {
+ case xproto.KeyPressEvent:
+ if err := handleKeyPressEvent(e); err != nil {
+ break eventloop
+ }
+ default:
+ log.Println(xevent)
+ }
+ }
+}
+
+func TakeWMOwnership() error {
+ return xproto.ChangeWindowAttributesChecked(
+ xc,
+ xroot.Root,
+ xproto.CwEventMask,
+ []uint32{
+ xproto.EventMaskKeyPress |
+ xproto.EventMaskKeyRelease |
+ xproto.EventMaskButtonPress |
+ xproto.EventMaskButtonRelease |
+ xproto.EventMaskStructureNotify |
+ xproto.EventMaskSubstructureRedirect,
+ }).Check()
+}
+
+func isMappedWindow(windowID xproto.Window) bool {
+ reply, err := xproto.GetWindowAttributes(xc, windowID).Reply()
+ if err != nil {
+ log.Fatal(err)
+ }
+ // TODO: Verify this.
+ // It appears that only windows with a `MapState` value of 2 should be 'viewable'.
+ // - https://github.com/BurntSushi/xgb/blob/master/xproto/xproto.go#L3772
+ // - https://github.com/BurntSushi/xgb/blob/master/xproto/xproto.go#L10287
+ if reply != nil && reply.MapState == 2 {
+ return true
+ }
+ return false
+}
+
+func handleKeyPressEvent(key xproto.KeyPressEvent) error {
+ switch keymap[key.Detail][0] {
+ case XK_q:
+ switch key.State {
+ case xproto.ModMask1 | xproto.ModMaskShift:
+ // TODO: Where and how do I want to handle exit/restart?
+ // -------------------------
+ // We must manually close the X connection since we used
+ // `defer` when setting it up and os.Exit() short-circuits
+ // that deferral.
+ xc.Close()
+ os.Exit(0)
+ }
+ case XK_Return:
+ switch key.State {
+ case xproto.ModMask1 | xproto.ModMaskShift:
+ cmd := exec.Command("xterm")
+ err := cmd.Start()
+ if err != nil {
+ log.Fatal(err)
+ }
+ }
+ default:
+ return nil
+ }
+ return nil // What do I actually want to return here?
+}
projects / illi / commitdiff