main.go

package main

import (
	"flag"
	"fmt"
	"io"
	"log"
	"os"

	"github.com/hashicorp/logutils"
	shabik "github.com/sadeemio/shabik/libshabik"
)

var debug, version bool

func SetLogging(level string, f io.Writer) *logutils.LevelFilter {
	filter := &logutils.LevelFilter{
		Levels:   []logutils.LogLevel{"DEBUG", "INFO", "WARN", "ERROR", "FATAL"},
		MinLevel: logutils.LogLevel(level),
		Writer:   f,
	}
	log.SetOutput(filter)
	return filter
}

func init() {
	flag.BoolVar(&debug, "d", false, "run in debug mode")
	flag.BoolVar(&version, "v", false, "print version and exit (shorthand)")
	flag.Parse()
}

func main() {
	// Set logging output, append to file and create if not exist
	f, err := os.OpenFile(shabik.LOGFILE, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
	if err != nil {
		log.Fatalf("Error opening logfile: %v", err)
	}
	defer f.Close()

	if debug {
		SetLogging("DEBUG", f)
	} else {
		SetLogging("INFO", f)
	}
	if version {
		fmt.Printf("Shabik Version: %v \n", shabik.VERSION)
		os.Exit(0)
	}
	// Connect to node.sadeem.io
	go shabik.FetchData()
	// Serve DHCP
	dhcp := shabik.NewDHCPHandler()
	dhcp.RunServer()
}