mtg/cli/proxy.go

package cli

import (
	"net"
	"os"
	"time"

	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"

	"github.com/9seconds/mtg/antireplay"
	"github.com/9seconds/mtg/config"
	"github.com/9seconds/mtg/ntp"
	"github.com/9seconds/mtg/obfuscated2"
	"github.com/9seconds/mtg/proxy"
	"github.com/9seconds/mtg/stats"
	"github.com/9seconds/mtg/telegram"
	"github.com/9seconds/mtg/utils"
)

func Proxy() error {
	ctx := utils.GetSignalContext()

	atom := zap.NewAtomicLevel()
	switch {
	case config.C.Debug:
		atom.SetLevel(zapcore.DebugLevel)
	case config.C.Verbose:
		atom.SetLevel(zapcore.InfoLevel)
	default:
		atom.SetLevel(zapcore.ErrorLevel)
	}

	encoderCfg := zap.NewProductionEncoderConfig()
	logger := zap.New(zapcore.NewCore(
		zapcore.NewJSONEncoder(encoderCfg),
		zapcore.Lock(os.Stderr),
		atom,
	))
	zap.ReplaceGlobals(logger)
	defer logger.Sync() // nolint: errcheck

	if err := config.InitPublicAddress(ctx); err != nil {
		Fatal(err.Error())
	}
	zap.S().Debugw("Configuration", "config", config.C)

	if len(config.C.AdTag) > 0 {
		zap.S().Infow("Use middle proxy connection to Telegram")
		diff, err := ntp.Fetch()
		if err != nil {
			Fatal("Cannot fetch time data from NTP")
		}
		if diff > time.Second {
			Fatal("Your local time is skewed and drift is bigger than a second. Please sync your time.")
		}
		go ntp.AutoUpdate()
	} else {
		zap.S().Infow("Use direct connection to Telegram")
	}

	PrintJSONStdout(config.GetURLs())

	if err := antireplay.Init(); err != nil {
		Fatal(err.Error())
	}
	if err := stats.Init(ctx); err != nil {
		Fatal(err.Error())
	}
	proxyListener, err := net.Listen("tcp", config.C.ListenAddr.String())
	if err != nil {
		Fatal(err.Error())
	}
	go func() {
		<-ctx.Done()
		proxyListener.Close()
	}()

	app := &proxy.Proxy{
		Logger:  zap.S().Named("proxy"),
		Context: ctx,
	}
	if len(config.C.AdTag) == 0 {
		app.TelegramProtocolMaker = obfuscated2.MakeTelegramProtocol
		app.TelegramDialer = telegram.NewDirectTelegram()
	}
	if config.C.SecretMode != config.SecretModeTLS {
		app.ClientProtocolMaker = obfuscated2.MakeClientProtocol
	}

	app.Serve(proxyListener)

	return nil
}