mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-12-15 03:21:34 +03:00
84 lines
2.4 KiB
Go
84 lines
2.4 KiB
Go
//go:build linux
|
|
|
|
package home
|
|
|
|
import (
|
|
"github.com/AdguardTeam/AdGuardHome/internal/aghos"
|
|
"github.com/kardianos/service"
|
|
)
|
|
|
|
// chooseSystem checks the current system detected and substitutes it with local
|
|
// implementation if needed.
|
|
func chooseSystem() {
|
|
sys := service.ChosenSystem()
|
|
// By default, package service uses the SysV system if it cannot detect
|
|
// anything other, but the update-rc.d fix should not be applied on OpenWrt,
|
|
// so exclude it explicitly.
|
|
//
|
|
// See https://github.com/AdguardTeam/AdGuardHome/issues/4480 and
|
|
// https://github.com/AdguardTeam/AdGuardHome/issues/4677.
|
|
if sys.String() == "unix-systemv" && !aghos.IsOpenWrt() {
|
|
service.ChooseSystem(sysvSystem{System: sys})
|
|
}
|
|
}
|
|
|
|
// sysvSystem is a wrapper for service.System that wraps the service.Service
|
|
// while creating a new one.
|
|
//
|
|
// TODO(e.burkov): File a PR to github.com/kardianos/service.
|
|
type sysvSystem struct {
|
|
// System is expected to have an unexported type
|
|
// *service.linuxSystemService.
|
|
service.System
|
|
}
|
|
|
|
// New returns a wrapped service.Service.
|
|
func (sys sysvSystem) New(i service.Interface, c *service.Config) (s service.Service, err error) {
|
|
s, err = sys.System.New(i, c)
|
|
if err != nil {
|
|
return s, err
|
|
}
|
|
|
|
return sysvService{
|
|
Service: s,
|
|
name: c.Name,
|
|
}, nil
|
|
}
|
|
|
|
// sysvService is a wrapper for a service.Service that also calls update-rc.d in
|
|
// a proper way on installing and uninstalling.
|
|
type sysvService struct {
|
|
// Service is expected to have an unexported type *service.sysv.
|
|
service.Service
|
|
// name stores the name of the service to call updating script with it.
|
|
name string
|
|
}
|
|
|
|
// Install wraps service.Service.Install call with calling the updating script.
|
|
func (svc sysvService) Install() (err error) {
|
|
err = svc.Service.Install()
|
|
if err != nil {
|
|
// Don't wrap an error since it's informative enough as is.
|
|
return err
|
|
}
|
|
|
|
_, _, err = aghos.RunCommand("update-rc.d", svc.name, "defaults")
|
|
|
|
// Don't wrap an error since it's informative enough as is.
|
|
return err
|
|
}
|
|
|
|
// Uninstall wraps service.Service.Uninstall call with calling the updating
|
|
// script.
|
|
func (svc sysvService) Uninstall() (err error) {
|
|
err = svc.Service.Uninstall()
|
|
if err != nil {
|
|
// Don't wrap an error since it's informative enough as is.
|
|
return err
|
|
}
|
|
|
|
_, _, err = aghos.RunCommand("update-rc.d", svc.name, "remove")
|
|
|
|
// Don't wrap an error since it's informative enough as is.
|
|
return err
|
|
}
|