ctrld/internal/router/openwrt/openwrt.go

package openwrt

import (
	"fmt"
	"os"
	"os/exec"

	"github.com/kardianos/service"

	"github.com/Control-D-Inc/ctrld"
	"github.com/Control-D-Inc/ctrld/internal/router/dnsmasq"
)

const (
	Name                     = "openwrt"
	openwrtDNSMasqConfigPath = "/tmp/dnsmasq.d/ctrld.conf"
)

type Openwrt struct {
	cfg *ctrld.Config
}

// New returns a router.Router for configuring/setup/run ctrld on Openwrt routers.
func New(cfg *ctrld.Config) *Openwrt {
	return &Openwrt{cfg: cfg}
}

func (o *Openwrt) ConfigureService(svc *service.Config) error {
	svc.Option["SysvScript"] = openWrtScript
	return nil
}

func (o *Openwrt) Install(config *service.Config) error {
	return exec.Command("/etc/init.d/ctrld", "enable").Run()
}

func (o *Openwrt) Uninstall(config *service.Config) error {
	return nil
}

func (o *Openwrt) PreRun() error {
	return nil
}

func (o *Openwrt) Setup() error {
	if o.cfg.FirstListener().IsDirectDnsListener() {
		return nil
	}
	data, err := dnsmasq.ConfTmpl(dnsmasq.ConfigContentTmpl, o.cfg)
	if err != nil {
		return err
	}
	if err := os.WriteFile(openwrtDNSMasqConfigPath, []byte(data), 0600); err != nil {
		return err
	}
	// Restart dnsmasq service.
	if err := restartDNSMasq(); err != nil {
		return err
	}
	return nil
}

func (o *Openwrt) Cleanup() error {
	if o.cfg.FirstListener().IsDirectDnsListener() {
		return nil
	}
	// Remove the custom dnsmasq config
	if err := os.Remove(openwrtDNSMasqConfigPath); err != nil {
		return err
	}
	// Restart dnsmasq service.
	if err := restartDNSMasq(); err != nil {
		return err
	}
	return nil
}

func restartDNSMasq() error {
	if out, err := exec.Command("/etc/init.d/dnsmasq", "restart").CombinedOutput(); err != nil {
		return fmt.Errorf("%s: %w", string(out), err)
	}
	return nil
}