WindyMadman/tdl
1
0
Fork 0
mirror of https://github.com/iyear/tdl synced 2025-01-05 10:27:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(extension): pass namespace and pool config

Browse Source
This commit is contained in:
iyear 2024-11-23 14:42:01 +08:00
parent eee2d1c546
commit 20aec50413
2 changed files with 30 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
cmd/extension.go
View File

@ -122,14 +122,16 @@ func NewExtensionCmd(em *extensions.Manager, ext extensions.Extension, stdin io.
}
env := &extbase.Env{
Name: ext.Name(),
AppID: app.AppID,
AppHash: app.AppHash,
Session: session,
DataDir: dataDir,
NTP: opts.NTP,
Proxy: opts.Proxy,
Debug: viper.GetBool(consts.FlagDebug),
Name: ext.Name(),
AppID: app.AppID,
AppHash: app.AppHash,
Session: session,
Namespace: viper.GetString(consts.FlagNamespace),
DataDir: dataDir,
NTP: opts.NTP,
Proxy: opts.Proxy,
Pool: viper.GetInt64(consts.FlagPoolSize),
Debug: viper.GetBool(consts.FlagDebug),
}
if err = em.Dispatch(ext, args, env, stdin, stdout, stderr); err != nil {

34
extension/extension.go
View File

@ -21,14 +21,16 @@ import (
const EnvKey = "TDL_EXTENSION"
type Env struct {
Name string `json:"name"`
AppID int `json:"app_id"`
AppHash string `json:"app_hash"`
Session []byte `json:"session"`
DataDir string `json:"data_dir"`
NTP string `json:"ntp"`
Proxy string `json:"proxy"`
Debug bool `json:"debug"`
Name string `json:"name"`
Namespace string `json:"namespace"`
AppID int `json:"app_id"`
AppHash string `json:"app_hash"`
Session []byte `json:"session"`
DataDir string `json:"data_dir"`
NTP string `json:"ntp"`
Proxy string `json:"proxy"`
Pool int64 `json:"pool"`
Debug bool `json:"debug"`
}
type Options struct {
@ -50,9 +52,11 @@ type Extension struct {
}
type Config struct {
DataDir string // data directory for extension
Proxy string // proxy URL
Debug bool // debug mode enabled
Namespace string // tdl namespace
DataDir string // data directory for extension
Proxy string // proxy URL
Pool int64 // pool size
Debug bool // debug mode enabled
}
func (e *Extension) Name() string {
@ -136,9 +140,11 @@ func buildExtension(ctx context.Context, o Options) (*Extension, *telegram.Clien
client: client,
log: o.Logger,
config: &Config{
DataDir: env.DataDir,
Proxy: env.Proxy,
Debug: env.Debug,
Namespace: env.Namespace,
DataDir: env.DataDir,
Proxy: env.Proxy,
Pool: env.Pool,
Debug: env.Debug,
},
}, client, nil
}