mirror of
https://github.com/shtorm-7/sing-box-extended.git
synced 2026-09-15 21:00:27 +00:00
Add api command
This commit is contained in:
@@ -0,0 +1,63 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"io"
|
||||
|
||||
"github.com/sagernet/sing-box/daemon"
|
||||
E "github.com/sagernet/sing/common/exceptions"
|
||||
F "github.com/sagernet/sing/common/format"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/protobuf/types/known/emptypb"
|
||||
)
|
||||
|
||||
var commandAPIOpenConnect = &cobra.Command{
|
||||
Use: "openconnect",
|
||||
Short: "Manage OpenConnect authentication",
|
||||
}
|
||||
|
||||
func init() {
|
||||
commandAPIRoot.AddCommand(commandAPIOpenConnect)
|
||||
}
|
||||
|
||||
func subscribeOpenConnectStatus(ctx context.Context, client daemon.StartedServiceClient) (grpc.ServerStreamingClient[daemon.OpenConnectStatusUpdate], []*daemon.OpenConnectEndpointStatus, error) {
|
||||
stream, err := client.SubscribeOpenConnectStatus(ctx, &emptypb.Empty{})
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
endpoints, err := recvOpenConnectStatus(stream)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
return stream, endpoints, nil
|
||||
}
|
||||
|
||||
func recvOpenConnectStatus(stream grpc.ServerStreamingClient[daemon.OpenConnectStatusUpdate]) ([]*daemon.OpenConnectEndpointStatus, error) {
|
||||
update, err := stream.Recv()
|
||||
if err != nil {
|
||||
if errors.Is(err, io.EOF) {
|
||||
return nil, E.New("api service closed the status stream")
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return update.GetEndpoints(), nil
|
||||
}
|
||||
|
||||
func openConnectChallengeSummary(challenge *daemon.OpenConnectAuthChallenge) string {
|
||||
form := challenge.GetForm()
|
||||
if form != nil {
|
||||
return F.ToString("form (", len(form.GetFields()), " fields)")
|
||||
}
|
||||
browser := challenge.GetBrowser()
|
||||
if browser == nil {
|
||||
return "unknown"
|
||||
}
|
||||
mode, err := deriveOpenConnectBrowserMode(browser)
|
||||
if err != nil {
|
||||
return "browser (invalid)"
|
||||
}
|
||||
return "browser (" + mode + ")"
|
||||
}
|
||||
Reference in New Issue
Block a user