monitor.go

package bspc

import (
	"bytes"
	"errors"
	"fmt"
	"os/exec"
	"strconv"
)

type Monitor struct {
	Name     string
	Desktops []*Desktop
}

func Monitors() ([]*Monitor, error) {
	out, err := exec.Command(CommandName, "query", "--monitors").Output()
	if err != nil {
		return nil, err
	}
	names := bytes.Split(out, []byte("\n"))
	// Last entry is blank
	monitors := make([]*Monitor, len(names)-1)

	for i, name := range names[:len(names)-1] {
		monitors[i] = &Monitor{Name: string(name)}
		err = monitors[i].loadDesktops()
		if err != nil {
			return nil, err
		}
	}
	return monitors, nil
}

func (m *Monitor) loadDesktops() error {
	out, err := exec.Command(CommandName, "query", "--monitor", m.Name, "--desktops").Output()

	if err != nil {
		return err
	}

	names := bytes.Split(out, []byte("\n"))
	// Last entry is blank
	desktops := make([]*Desktop, len(names)-1)

	for i, name := range names[:len(names)-1] {
		desktops[i] = &Desktop{Name: string(name)}
		err = desktops[i].loadWindows()
		if err != nil {
			return err
		}
	}
	m.Desktops = desktops
	return nil
}

func (m *Monitor) Rename(newName string) error {
	err := exec.Command(CommandName, "monitor", m.Name, "--rename", newName).Run()
	if err != nil {
		return err
	}
	m.Name = newName
	return nil
}

func (m *Monitor) AddDesktops(desktops ...string) error {
	// TODO: Refactor into one shell call
	for _, desktop := range desktops {
		err := exec.Command(CommandName, "monitor", m.Name, "--add-desktops", desktop).Run()
		if err != nil {
			return err
		}
		m.Desktops = append(m.Desktops, &Desktop{Name: desktop})
	}
	return nil
}

func (m *Monitor) DesktopByIndex(index int) (*Desktop, error) {
	if index < 0 {
		return nil, errors.New("Need to enter a non-negative index")
	}

	if index >= len(m.Desktops) {
		return nil, errors.New("Invalid index")
	}
	return m.Desktops[index], nil
}

func (m *Monitor) RemoveEmptyDesktops() error {
	for i, desktop := range m.Desktops {
		if len(desktop.Windows) == 0 {
			err := m.RemoveDesktopByIndex(i + 1)
			if err != nil {
				return err
			}
		}
	}
	return nil
}

func (m *Monitor) RemoveDesktopByIndex(index int) error {
	fmt.Println("Removing desktop", index)
	err := exec.Command(CommandName, "desktop", m.Name+":^"+strconv.Itoa(index), "--remove").Run()
	return err
}

func (m *Monitor) DefragDesktops() error {
	for i := range m.Desktops {
		err := exec.Command(CommandName, "desktop", m.Name+":^"+strconv.Itoa(i), "--rename", string(i+1)).Run()
		if err != nil {
			return err
		}
	}
	return nil
}

func (m *Monitor) FocusDesktopByIndex(index int) error {
	err := exec.Command(CommandName, "desktop", m.Name+":^"+strconv.Itoa(index), "--focus").Run()
	return err
}