rpi-vsts-ci-led/main.go

package main

import (
	"fmt"
	"log"
	"os"
	"os/signal"
	"rpi-ci-led/pkg/config"
	"rpi-ci-led/pkg/database"
	"rpi-ci-led/pkg/led"
	"rpi-ci-led/pkg/websocket"
	"rpi-ci-led/pkg/ws2812b"
	"syscall"

	"git.trj.tw/golang/argparse"
	"github.com/joho/godotenv"
)

func main() {
	fmt.Println("GO ci hook")

	godotenv.Load()

	var configFile string
	var dbPath string

	// argument parser
	argParser := argparse.New()
	argParser.Help("h", "help")
	argParser.StringVar(&configFile, "", "f", "config", "config yaml file path", nil)
	argParser.StringVar(&dbPath, "", "d", "db", "database file path", nil)
	if err := argParser.Parse(os.Args); err != nil {
		log.Fatal(err)
	}

	if err := config.Load(configFile); err != nil {
		log.Fatal(err)
	}

	if err := database.New(dbPath); err != nil {
		log.Fatal(err)
	}

	conf := config.Get()

	socket, err := websocket.NewClient(conf.Remote.WSLoc)
	if err != nil {
		log.Fatal(err)
	}

	ledsvc, err := led.Init(conf.LED.Pin, conf.LED.Count)
	if err != nil {
		log.Fatal(err)
	}

	lock := make(chan os.Signal)
	signal.Notify(lock, syscall.SIGINT, syscall.SIGTERM)

	// main
	go func() {
		if err := socket.Listen(); err != nil {
			log.Fatal(err)
		}
		lock <- syscall.SIGQUIT
	}()

	go func() { ledsvc.Run() }()

	// start ws client connect to server

	<-lock

	fmt.Printf("Before process exit, close all connection\n")
	socket.Close()
	ws2812b.Close()
}