caddy/service_windows.go

// Copyright 2015 Matthew Holt and The Caddy Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package caddy

import (
	"os"
	"path/filepath"

	"golang.org/x/sys/windows/svc"

	"github.com/caddyserver/caddy/v2/notify"
)

func init() {
	isService, err := svc.IsWindowsService()
	if err != nil || !isService {
		return
	}

	// Windows services always start in the system32 directory, try to
	// switch into the directory where the caddy executable is.
	execPath, err := os.Executable()
	if err == nil {
		_ = os.Chdir(filepath.Dir(execPath))
	}

	go func() {
		_ = svc.Run("", runner{})
	}()
}

type runner struct{}

func (runner) Execute(args []string, request <-chan svc.ChangeRequest, status chan<- svc.Status) (bool, uint32) {
	// All status updates go through notify, which publishes the current
	// status now: StartPending, or whatever Caddy reported before the
	// service started. Detach before returning, since the dispatcher
	// stops receiving status updates once Execute returns.
	notify.SetGlobalStatus(status)
	defer notify.SetGlobalStatus(nil)

	for req := range request {
		// Unsupported service controls are ignored.
		//exhaustive:ignore
		switch req.Cmd {
		case svc.Interrogate:
			notify.Interrogate()
		case svc.Stop, svc.Shutdown:
			_ = notify.Stopping()
			exitProcessFromSignal("SIGINT")
			return false, 0
		}
	}
	return false, 0
}