sb.go

package main

import (
	"context"
	"log"
	"os"
	"path/filepath"
	runtimeDebug "runtime/debug"

	"github.com/tidwall/jsonc"

	"github.com/pkg/errors"
	box "github.com/sagernet/sing-box"
	_ "github.com/sagernet/sing-box/include"
	"github.com/sagernet/sing-box/option"
)

type SingBox struct {
	Running  bool
	ConfPath string
	instance *box.Box
	cancel   context.CancelFunc
}

func (s *SingBox) Close() {
	if s.instance != nil {
		_ = s.instance.Close()
	}
	if s.cancel != nil {
		s.cancel()
	}
	s.Running = false
}

func (s *SingBox) Start(basePath, configPath string) error {
	defer func() {
		if err := recover(); err != nil {
			log.Println(err)
		}
	}()

	instance, cancel, err := create(filepath.Join(basePath, configPath))
	if err != nil {
		return err
	}
	runtimeDebug.FreeOSMemory()

	s.Running = true
	s.instance = instance
	s.cancel = cancel

	return nil
}

func create(configPath string) (*box.Box, context.CancelFunc, error) {
	options, err := readConfig(configPath)
	if err != nil {
		return nil, nil, err
	}
	options.Experimental.ClashAPI.ExternalUI = filepath.Join(ConfDir, "ui")
	options.Log.DisableColor = true
	//options.Log.Output = filepath.Join(ConfDir, "singbox.log")

	ctx, cancel := context.WithCancel(context.Background())
	instance, err := box.New(box.Options{
		Options:           options,
		Context:           ctx,
		PlatformInterface: nil,
	})
	if err != nil {
		cancel()
		return nil, nil, errors.Wrap(err, "sing-box core create service")
	}
	err = instance.Start()
	if err != nil {
		cancel()
		return nil, nil, errors.Wrap(err, "sing-box core start service")
	}
	return instance, cancel, nil
}

func readConfig(configPath string) (option.Options, error) {
	var (
		configContent []byte
		err           error
	)
	configContent, err = os.ReadFile(configPath)
	if err != nil {
		return option.Options{}, errors.Wrap(err, "read config")
	}

	var options option.Options
	err = options.UnmarshalJSON(jsonc.ToJSON(configContent))
	if err != nil {
		return option.Options{}, errors.Wrap(err, "decode config")
	}

	if options.Log == nil {
		options.Log = &option.LogOptions{}
	}

	if options.Experimental == nil {
		options.Experimental = &option.ExperimentalOptions{}
	}
	if options.Experimental.ClashAPI == nil {
		options.Experimental.ClashAPI = &option.ClashAPIOptions{}
	}

	return options, nil
}