77 lines
1.4 KiB
Go
77 lines
1.4 KiB
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()
|
|
}
|