60 lines
1.7 KiB
Go
60 lines
1.7 KiB
Go
|
package handler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/xtls/xray-core/app/proxyman/command"
|
||
|
|
"github.com/xtls/xray-core/common/protocol"
|
||
|
|
"github.com/xtls/xray-core/common/serial"
|
||
|
|
)
|
||
|
|
|
||
|
|
func RemoveInbound(ctx context.Context, client command.HandlerServiceClient, tag string) error {
|
||
|
|
_, err := client.RemoveInbound(ctx, &command.RemoveInboundRequest{Tag: tag})
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func RemoveOutbound(ctx context.Context, client command.HandlerServiceClient, tag string) error {
|
||
|
|
_, err := client.RemoveOutbound(ctx, &command.RemoveOutboundRequest{Tag: tag})
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func ListInboundTags(ctx context.Context, client command.HandlerServiceClient) ([]string, error) {
|
||
|
|
resp, err := client.ListInbounds(ctx, &command.ListInboundsRequest{IsOnlyTags: true})
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
tags := make([]string, 0, len(resp.GetInbounds()))
|
||
|
|
for _, inbound := range resp.GetInbounds() {
|
||
|
|
tags = append(tags, inbound.GetTag())
|
||
|
|
}
|
||
|
|
return tags, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetInboundUsers(ctx context.Context, client command.HandlerServiceClient, inboundTag string) ([]*protocol.User, error) {
|
||
|
|
resp, err := client.GetInboundUsers(ctx, &command.GetInboundUserRequest{
|
||
|
|
Tag: inboundTag,
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return resp.GetUsers(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetInboundUsersCount(ctx context.Context, client command.HandlerServiceClient, inboundTag string) (int64, error) {
|
||
|
|
resp, err := client.GetInboundUsersCount(ctx, &command.GetInboundUserRequest{
|
||
|
|
Tag: inboundTag,
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
return resp.GetCount(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func AlterOutbound(ctx context.Context, client command.HandlerServiceClient, tag string, operation *serial.TypedMessage) error {
|
||
|
|
_, err := client.AlterOutbound(ctx, &command.AlterOutboundRequest{
|
||
|
|
Tag: tag,
|
||
|
|
Operation: operation,
|
||
|
|
})
|
||
|
|
return err
|
||
|
|
}
|