add_client.go

/* Copyright 2025 İrem Kuyucu <irem@digilol.net>
 * Copyright 2025 Laurynas Četyrkinas <laurynas@digilol.net>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package client3xui

import (
	"context"
	"encoding/json"
	"fmt"
	"net/http"
)

type AddClientRequest struct {
	// Inbound ID
	ID uint `json:"id"`

	// See ClientSettings. This is it marshaled into an escaped JSON string.
	Settings string `json:"settings"`
}

type ClientSettings struct {
	Clients []ClientInterface `json:"clients"`
}
type ClientInterface interface {
    isClient()
}

const (
	FlowXtlsRprxVision       = "xtls-rprx-vision"
	FlowXtlsRprxVisionUDP443 = "xtls-rprx-vision-udp443"
)

type XrayClient struct {
	ID         string `json:"id"`
	AlterID    uint   `json:"alter_id,omitempty"`
	Email      string `json:"email"`
	LimitIP    uint   `json:"limitIp"`
	TotalGB    uint64 `json:"totalGB"`
	ExpiryTime uint64 `json:"expiryTime"`
	Enable     bool   `json:"enable"`
	TgID       uint   `json:"tgId"`
	SubID      string `json:"subId"`
	Flow       string `json:"flow"`
}

func (XrayClient) isClient() {}

type TrojanClient struct {
	Password   string `json:"password"`
	Email      string `json:"email"`
	LimitIP    uint   `json:"limitIp"`
	TotalGB    uint64 `json:"totalGB"`
	ExpiryTime uint64 `json:"expiryTime"`
	Enable     bool   `json:"enable"`
	TgID       uint   `json:"tgId"`
	SubID      string `json:"subId"`
}

func (TrojanClient) isClient() {}

// Add client to an inbound.
func (c *Client) AddAnyClient(ctx context.Context, inboundId uint, clients []ClientInterface) (*ApiResponse, error) {
	settings := &ClientSettings{Clients: clients}
	settingsBytes, err := json.Marshal(settings)
	if err != nil {
		return nil, err
	}

	req := &AddClientRequest{ID: inboundId, Settings: string(settingsBytes)}
	resp := &ApiResponse{}
	err = c.Do(ctx, http.MethodPost, "/panel/api/inbounds/addClient", req, resp)
	if err != nil {
		return nil, err
	}
	if !resp.Success {
		return resp, fmt.Errorf(resp.Msg)
	}
	return resp, err
}

// Backward-compatible: only XrayClient
func (c *Client) AddClient(ctx context.Context, inboundId uint, clients []XrayClient) (*ApiResponse, error) {
    generic := make([]ClientInterface, len(clients))
    for i, cl := range clients {
        generic[i] = cl
    }
    return c.AddAnyClient(ctx, inboundId, generic)
}