main.go

package main

import (
	"fmt"
	"os"
	"runtime"

	"github.com/afreakk/go-i3"
	"github.com/afreakk/i3statusbear/internal/activewindow"
	"github.com/afreakk/i3statusbear/internal/command"
	"github.com/afreakk/i3statusbear/internal/config"
	"github.com/afreakk/i3statusbear/internal/cpu"
	"github.com/afreakk/i3statusbear/internal/datetime"
	"github.com/afreakk/i3statusbear/internal/memory"
	"github.com/afreakk/i3statusbear/internal/protocol"
	"github.com/afreakk/i3statusbear/internal/pulseaudio"
	"github.com/afreakk/i3statusbear/internal/readfile"
	"github.com/robfig/cron"
)

func errorExit(str string, code int) {
	fmt.Fprintln(os.Stderr, str)
	os.Exit(code)
}

func main() {
	osArgsLen := len(os.Args)

	if osArgsLen < 2 {
		errorExit("Please provide config as argument", 1)
	}
	configFilePath := os.Args[1]

	cfg, err := config.GetConfigFromPath(configFilePath)
	if err != nil {
		panic(err)
	}

	if osArgsLen > 2 {
		cfg.WMClient = os.Args[2]
	}

	// not implemented, havent really had the use for it
	//go protocol.HandleInput()

	if cfg.WMClient == "sway" {
		i3.WMClient = i3.WMTypeSway
	}

	output := protocol.Output{}
	err = output.Init(cfg)
	if err != nil {
		panic(err)
	}

	c := cron.New()
	for _, module := range cfg.Modules {
		var err error
		switch module.Name {
		case "datetime":
			err = c.AddFunc(module.Cron, datetime.Datetime(&output, module))
		case "pulseaudio":
			err = pulseaudio.Pulseaudio(&output, module)
		case "readfile":
			err = c.AddFunc(module.Cron, readfile.Readfile(&output, module))
		case "memory":
			err = c.AddFunc(module.Cron, memory.Memory(&output, module))
		case "cpu":
			err = c.AddFunc(module.Cron, cpu.Cpu(&output, module))
		case "command":
			err = c.AddFunc(module.Cron, command.Command(&output, module))
		case "activewindow":
			activewindow.ActiveWindow(&output, module)
		}
		if err != nil {
			fmt.Fprintf(os.Stderr, "\nError when initializing module: %s, \nerror: %s\n", module.Name, err.Error())
		}
	}
	output.PrintMsgs()
	c.Start()
	runtime.Goexit()
}