Alireza Ahmadi
108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package core
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"s-ui/config"
|
|
"s-ui/logger"
|
|
|
|
sb "github.com/sagernet/sing-box"
|
|
"github.com/sagernet/sing-box/adapter"
|
|
_ "github.com/sagernet/sing-box/experimental/clashapi"
|
|
_ "github.com/sagernet/sing-box/experimental/v2rayapi"
|
|
"github.com/sagernet/sing-box/log"
|
|
"github.com/sagernet/sing-box/option"
|
|
_ "github.com/sagernet/sing-box/transport/v2rayquic"
|
|
_ "github.com/sagernet/sing-dns/quic"
|
|
"github.com/sagernet/sing/service"
|
|
)
|
|
|
|
var (
|
|
globalCtx context.Context
|
|
inbound_manager adapter.InboundManager
|
|
outbound_manager adapter.OutboundManager
|
|
endpoint_manager adapter.EndpointManager
|
|
router adapter.Router
|
|
connTracker *ConnTracker
|
|
factory log.Factory
|
|
)
|
|
|
|
type Core struct {
|
|
isRunning bool
|
|
instance *Box
|
|
}
|
|
|
|
func NewCore() *Core {
|
|
globalCtx = context.Background()
|
|
globalCtx = sb.Context(globalCtx, inboundRegistry(), outboundRegistry(), EndpointRegistry())
|
|
return &Core{
|
|
isRunning: false,
|
|
instance: nil,
|
|
}
|
|
}
|
|
|
|
func (c *Core) GetCtx() context.Context {
|
|
return globalCtx
|
|
}
|
|
|
|
func (c *Core) GetInstance() *Box {
|
|
return c.instance
|
|
}
|
|
|
|
func (c *Core) Start() error {
|
|
filepath := config.GetBinFolderPath() + "/config.json"
|
|
configFile, err := os.ReadFile(filepath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var opt option.Options
|
|
err = opt.UnmarshalJSONContext(globalCtx, configFile)
|
|
if err != nil {
|
|
logger.Error("Unmarshal config err:", err.Error())
|
|
}
|
|
|
|
c.instance, err = NewBox(Options{
|
|
Context: globalCtx,
|
|
Options: opt,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = c.instance.Start()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
globalCtx = service.ContextWith(globalCtx, c)
|
|
inbound_manager = service.FromContext[adapter.InboundManager](globalCtx)
|
|
outbound_manager = service.FromContext[adapter.OutboundManager](globalCtx)
|
|
endpoint_manager = service.FromContext[adapter.EndpointManager](globalCtx)
|
|
router = service.FromContext[adapter.Router](globalCtx)
|
|
|
|
c.isRunning = true
|
|
return nil
|
|
}
|
|
|
|
func (c *Core) Stop() error {
|
|
if c.isRunning {
|
|
c.isRunning = false
|
|
return c.instance.Close()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Core) Restart() error {
|
|
err := c.Stop()
|
|
if err != nil {
|
|
logger.Error("stop sing-box err:", err.Error())
|
|
return err
|
|
}
|
|
logger.Info("sing-box stopped")
|
|
return c.Start()
|
|
}
|
|
|
|
func (c *Core) IsRunning() bool {
|
|
return c.isRunning
|
|
}
|