Tulir Asokan
65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
package updates
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.opentelemetry.io/otel/trace"
|
|
"go.uber.org/zap"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
)
|
|
|
|
// API is the interface which contains
|
|
// Telegram RPC methods used by manager for internalState synchronization.
|
|
type API interface {
|
|
UpdatesGetState(ctx context.Context) (*tg.UpdatesState, error)
|
|
UpdatesGetDifference(ctx context.Context, request *tg.UpdatesGetDifferenceRequest) (tg.UpdatesDifferenceClass, error)
|
|
UpdatesGetChannelDifference(ctx context.Context, request *tg.UpdatesGetChannelDifferenceRequest) (tg.UpdatesChannelDifferenceClass, error)
|
|
ChannelsGetParticipant(ctx context.Context, request *tg.ChannelsGetParticipantRequest) (*tg.ChannelsChannelParticipant, error)
|
|
}
|
|
|
|
// Config of the manager.
|
|
type Config struct {
|
|
// Handler where updates will be passed.
|
|
Handler telegram.UpdateHandler
|
|
// Callback called if manager cannot
|
|
// recover channel gap (optional).
|
|
OnChannelTooLong func(channelID int64) error
|
|
OnNotChannelMember func(ctx context.Context, channelID int64) error
|
|
// State storage.
|
|
// In-mem used if not provided.
|
|
Storage StateStorage
|
|
// Channel access hash storage.
|
|
// In-mem used if not provided.
|
|
AccessHasher AccessHasher
|
|
// Logger (optional).
|
|
Logger *zap.Logger
|
|
// TracerProvider (optional).
|
|
TracerProvider trace.TracerProvider
|
|
}
|
|
|
|
func (cfg *Config) setDefaults() {
|
|
if cfg.Handler == nil {
|
|
panic("Handler is nil")
|
|
}
|
|
if cfg.AccessHasher == nil {
|
|
cfg.AccessHasher = newMemAccessHasher()
|
|
}
|
|
if cfg.Logger == nil {
|
|
cfg.Logger = zap.NewNop()
|
|
}
|
|
if cfg.TracerProvider == nil {
|
|
cfg.TracerProvider = trace.NewNoopTracerProvider()
|
|
}
|
|
if cfg.Storage == nil {
|
|
cfg.Storage = newMemStorage()
|
|
}
|
|
if cfg.OnChannelTooLong == nil {
|
|
cfg.OnChannelTooLong = func(channelID int64) error {
|
|
cfg.Logger.Error("Difference too long", zap.Int64("channel_id", channelID))
|
|
return nil
|
|
}
|
|
}
|
|
}
|