lucifer/cmd/lucifer-server/main.go

package main

import (
	"context"
	"crypto/rand"
	"database/sql"
	"encoding/hex"
	"flag"
	"fmt"
	"log"
	"net/http"
	"os"
	"os/signal"
	"path"
	"syscall"
	"time"

	"git.aiterp.net/lucifer/lucifer/controllers"
	"git.aiterp.net/lucifer/lucifer/database/sqlite"
	"git.aiterp.net/lucifer/lucifer/internal/config"
	"git.aiterp.net/lucifer/lucifer/light"
	"git.aiterp.net/lucifer/lucifer/middlewares"
	"git.aiterp.net/lucifer/lucifer/models"
	"github.com/gorilla/mux"

	_ "git.aiterp.net/lucifer/lucifer/light/hue"
)

func main() {
	// Flag Variables
	var uiDir string

	// Parse Flags
	flag.StringVar(&uiDir, "uidir", "/var/www/", "UI directory")
	flag.Parse()

	// Configuration
	conf, err := config.Load("./config.yaml", "/etc/lucifer/lucifer.yaml")
	if err != nil {
		log.Fatalln("Failed to load configuration:", err)
	}

	// Database
	err = sqlite.Initialize(conf.DB.FileName)
	if err != nil {
		log.Fatalln("Failed to set up database:", err)
	}

	// Initialize
	setupAdmin(sqlite.UserRepository, sqlite.GroupRepository)

	// Services
	lightService := light.NewService(sqlite.BridgeRepository, sqlite.LightRepository, sqlite.GroupRepository, sqlite.ButtonRepository)

	// Controllers
	userController := controllers.NewUserController(sqlite.UserRepository, sqlite.SessionRepository)
	groupController := controllers.NewGroupController(sqlite.GroupRepository, sqlite.UserRepository, sqlite.LightRepository)
	lightController := controllers.NewLightController(lightService, sqlite.GroupRepository, sqlite.UserRepository, sqlite.LightRepository)
	bridgeController := controllers.NewBridgeController(lightService, sqlite.GroupRepository)

	// Router
	router := mux.NewRouter()
	router.Use(middlewares.Session(sqlite.SessionRepository, sqlite.UserRepository))
	groupController.Mount(router, "/api/group/")
	userController.Mount(router, "/api/user/")
	lightController.Mount(router, "/api/light/")
	bridgeController.Mount(router, "/api/bridge/")

	// Background Tasks
	go lightService.SyncLoop(context.TODO())

	// Server UI in production
	stat, err := os.Stat(uiDir)
	if err != nil || !stat.IsDir() {
		log.Println("ui directory not found, skipping UI routes")
	} else {
		// Static files
		router.PathPrefix("/static/").Handler(http.FileServer(http.Dir(uiDir)))
		router.PathPrefix("/favicon.ico").Handler(http.FileServer(http.Dir(uiDir)))
		router.PathPrefix("/manifest.json").Handler(http.FileServer(http.Dir(uiDir)))
		router.PathPrefix("/index.html").Handler(http.FileServer(http.Dir(uiDir)))

		// Serve index on any other path
		router.PathPrefix("/").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			http.ServeFile(w, r, path.Join(uiDir, "index.html"))
		})
	}

	// Setup webserver
	server := http.Server{Addr: conf.Server.Address, Handler: router}
	webserverErrCh := make(chan error)
	go func() {
		log.Println("Listening on", conf.Server.Address)
		webserverErrCh <- server.ListenAndServe()
	}()

	// Setup quit signal listener
	signalCh := make(chan os.Signal)
	signal.Notify(signalCh, os.Interrupt, os.Kill, syscall.SIGTERM)

	// Run until interrupt or server death
	select {
	case signal := <-signalCh:
		{
			log.Println("Stopping due to signal:", signal)

			timeout, cancel := context.WithTimeout(context.Background(), time.Second*5)
			defer cancel()

			err := server.Shutdown(timeout)
			if err != nil {
				log.Println("Warning: graceful shutdown timed out.")
			}

			os.Exit(0)
		}
	case err := <-webserverErrCh:
		{
			log.Println("Listening failed:", err)
			os.Exit(1)
		}
	}
}

func setupAdmin(users models.UserRepository, groups models.GroupRepository) {
	ctx, cancel := context.WithTimeout(context.Background(), time.Second*15)
	defer cancel()

	admin, err := users.FindByName(ctx, "Admin")
	if err != nil {
		if err != sql.ErrNoRows {
			log.Fatalln("Could not check for admin user:", err)
		}

		admin = models.User{Name: "Admin"}
		admin.SetPassword("123456")

		admin, err = users.Insert(ctx, admin)
		if err != nil {
			fmt.Println("Failed to insert admin username:", err)
		}
	}

	buf := make([]byte, 16)
	_, err = rand.Read(buf)
	if err != nil {
		log.Fatalln("Could not get random bytes:", err)
	}

	password := hex.EncodeToString(buf)
	admin.SetPassword(password)

	err = users.Update(ctx, admin)
	if err != nil {
		log.Println("Could not update admin password:", err)
	} else {
		log.Println("Administrator: Admin /", password)
	}

	groups.UpdatePermissions(ctx, models.GroupPermission{
		UserID:  admin.ID,
		GroupID: 0,
		Read:    true,
		Write:   true,
		Create:  true,
		Delete:  true,
		Manage:  true,
	})
}