package warewulfd

import (
	"log/syslog"
	"os"
	"os/exec"
	"strconv"
	"syscall"
	"time"

	"github.com/pkg/errors"
	warewulfconf "github.com/warewulf/warewulf/internal/pkg/config"
	"github.com/warewulf/warewulf/internal/pkg/util"
	"github.com/warewulf/warewulf/internal/pkg/wwlog"
)

const (
	WAREWULFD_PIDFILE = "/var/run/warewulfd.pid"
	WAREWULFD_LOGFILE = "/var/log/warewulfd.log"
)

var loginit bool

// allow to run without daemon for tests
var nodaemon bool

func init() {
	nodaemon = false
}

// run without daemon
func SetNoDaemon() {
	nodaemon = true
}

func DaemonFormatter(logLevel int, rec *wwlog.LogRecord) string {
	return "[" + rec.Time.Format(time.UnixDate) + "] " + wwlog.DefaultFormatter(logLevel, rec)
}

func DaemonInitLogging() error {
	if loginit {
		return nil
	}

	wwlog.SetLogFormatter(DaemonFormatter)

	level_str, ok := os.LookupEnv("WAREWULFD_LOGLEVEL")
	if ok {
		level, err := strconv.Atoi(level_str)
		if err == nil {
			wwlog.SetLogLevel(level)
		}
	} else {
		wwlog.SetLogLevel(wwlog.INFO)
	}

	conf := warewulfconf.Get()

	if conf.Warewulf.Syslog {

		wwlog.Debug("Changing log output to syslog")

		logwriter, err := syslog.New(syslog.LOG_NOTICE, "warewulfd")
		if err != nil {
			return errors.Wrap(err, "Could not create syslog writer")
		}

		wwlog.SetLogFormatter(wwlog.DefaultFormatter)
		wwlog.SetLogWriter(logwriter)

	}

	loginit = true

	return nil
}

func DaemonStatus() error {
	if nodaemon {
		return nil
	}

	if !util.IsFile(WAREWULFD_PIDFILE) {
		return errors.New("Warewulf server is not running")
	}

	dat, err := os.ReadFile(WAREWULFD_PIDFILE)
	if err != nil {
		return errors.Wrap(err, "could not read Warewulfd PID file")
	}

	pid, _ := strconv.Atoi(string(dat))
	process, err := os.FindProcess(pid)
	if err != nil {
		return errors.Wrap(err, "failed to find running PID")
	} else {
		err := process.Signal(syscall.Signal(0))
		if err != nil {
			return errors.Wrap(err, "failed to send process SIGCONT")
		} else {
			wwlog.Serv("Warewulf server is running at PID: %d", pid)
		}
	}

	return nil
}

func DaemonReload() error {
	if nodaemon {
		return nil
	}
	cmd := exec.Command("/usr/sbin/service", "warewulfd", "reload")
	err := cmd.Start()
	if err != nil {
		return errors.Wrap(err, "failed to reload warewulfd")
	}
	err = cmd.Wait()
	if err != nil {
		return errors.Wrap(err, "failed to reload warewulfd")
	}
	return nil
}