api.go

package agent

import (
	"errors"
	"net"
	"net/http"
	"os"
	"time"
)

func (a *Agent) ListenAndServe(apiSocket string) error {
	if _, err := os.Stat(apiSocket); !errors.Is(err, os.ErrNotExist) {
		// TODO: add check if another agent is running
		return errors.New("found api socket")
	}

	listener, err := net.Listen("unix", apiSocket)
	if err != nil {
		return err
	}

	defer listener.Close()

	if err := os.Chmod(apiSocket, 0660); err != nil {
		return err
	}

	handler := a.apiRouterEngine()

	a.apiServer = &http.Server{
		Handler:           handler,
		ReadTimeout:       10 * time.Second,
		ReadHeaderTimeout: 10 * time.Second,
		IdleTimeout:       60 * time.Second,
	}

	return a.apiServer.Serve(listener)
}

func (a *Agent) apiRouterEngine() *http.ServeMux {
	r := http.NewServeMux()

	r.HandleFunc("/info", a.apiInfo)

	return r
}