package main

import (
	"encoding/json"
	"flag"
	"fmt"
	"net"
	"net/http"
	_ "net/http/pprof"
	"os"
	"os/signal"
	"syscall"

	"github.com/Terry-Mao/goim/pkg/ip"
	"github.com/opentracing/opentracing-go"
	"github.com/rs/zerolog/log"
	"github.com/uber/jaeger-client-go"
	"github.com/uber/jaeger-client-go/config"
	xlog "gitlab.33.cn/chat/im-pkg/log"
	"gitlab.33.cn/chat/im-pkg/trace"
	"gitlab.33.cn/chat/im/logic"
	_ "gitlab.33.cn/chat/im/logic/auth/dtalk"
	_ "gitlab.33.cn/chat/im/logic/auth/zhaobi"
	"gitlab.33.cn/chat/im/logic/conf"
	"gitlab.33.cn/chat/im/logic/grpc"
	"gitlab.33.cn/chat/im/naming"
)

const (
	srvName = "logic"
)

var (
	debug bool
)

func init() {
	flag.BoolVar(&debug, "debug", false, "sets log level to debug")
}

var (
	// projectVersion 项目版本
	projectVersion = "3.1.4"
	// goVersion go版本
	goVersion = ""
	// gitCommit git提交commit id
	gitCommit = ""
	// buildTime 编译时间
	buildTime = ""

	isShowVersion = flag.Bool("version", false, "show project version")
)

// showVersion 显示项目版本信息
func showVersion(isShow bool) {
	if isShow {
		fmt.Printf("Project: %s\n", srvName)
		fmt.Printf(" Version: %s\n", projectVersion)
		fmt.Printf(" Go Version: %s\n", goVersion)
		fmt.Printf(" Git Commit: %s\n", gitCommit)
		fmt.Printf(" Build Time: %s\n", buildTime)
		os.Exit(0)
	}
}

func main() {
	flag.Parse()
	showVersion(*isShowVersion)

	if err := conf.Init(); err != nil {
		panic(err)
	}

	//log init
	var err error
	log.Logger, err = xlog.Init(conf.Conf.Log)
	if err != nil {
		panic(err)
	}
	log.Logger.With().Str("service", srvName)

	byte, _ := json.Marshal(conf.Conf)
	log.Info().Str("config", string(byte)).Send()

	// trace init
	tracer, tracerCloser := trace.Init(srvName, conf.Conf.Trace, config.Logger(jaeger.NullLogger))
	//不然后续不会有Jaeger实例
	opentracing.SetGlobalTracer(tracer)

	srv := logic.New(conf.Conf)
	rpcSrv := grpc.New(conf.Conf.RPCServer, srv)

	go func() {
		if err := http.ListenAndServe(":8001", nil); err != nil {
			panic(err)
		}
	}()

	// register logic
	_, port, _ := net.SplitHostPort(conf.Conf.RPCServer.Addr)
	addr := fmt.Sprintf("%s:%s", ip.InternalIP(), port)
	if err := naming.Register(conf.Conf.Reg.RegAddrs, conf.Conf.Reg.SrvName, addr, conf.Conf.Reg.Schema, 15); err != nil {
		panic(err)
	}
	fmt.Println("register ok")

	// signal
	c := make(chan os.Signal, 1)
	signal.Notify(c, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT)
	for {
		s := <-c
		log.Info().Str("signal", s.String()).Send()
		switch s {
		case syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT:
			if err := naming.UnRegister(conf.Conf.Reg.SrvName, addr, conf.Conf.Reg.Schema); err != nil {
				log.Error().Err(err).Msg("naming.UnRegister")
			}
			srv.Close()
			rpcSrv.GracefulStop()
			if err := tracerCloser.Close(); err != nil {
				log.Error().Err(err).Msg("tracer close failed")
			}
			//log.Flush()
			return
		case syscall.SIGHUP:
		default:
			return
		}
	}
}