trojan-go/stat/stat.go

package stat

import (
	"context"
	"io"

	"github.com/p4gefau1t/trojan-go/common"
	"github.com/p4gefau1t/trojan-go/conf"
)

type TrafficMeter interface {
	io.Closer
	Hash() string
	Count(sent, recv int)
	Get() (sent, recv uint64)
	Reset()
	GetAndReset() (sent, recv uint64)
	GetSpeed() (sent, recv uint64)
	LimitSpeed(send, recv int)
	GetSpeedLimit() (send, recv int)
}

type Authenticator interface {
	io.Closer
	AuthUser(hash string) (valid bool, meter TrafficMeter)
	AddUser(hash string) error
	DelUser(hash string) error
	ListUsers() []TrafficMeter
}

type AuthCreator func(ctx context.Context, config *conf.GlobalConfig) (Authenticator, error)

var authCreators = map[string]AuthCreator{}

func RegisterAuthCreator(name string, creator AuthCreator) {
	authCreators[name] = creator
}

func NewAuth(ctx context.Context, name string, config *conf.GlobalConfig) (Authenticator, error) {
	creator, found := authCreators[name]
	if !found {
		return nil, common.NewError("driver name " + name + " not found")
	}
	return creator(ctx, config)
}