package main

import (
	"context"
	"database/sql"
	"flag"
	"fmt"
	"math/rand"
	"net/http"
	_ "net/http/pprof"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/pingcap/tidb-test/stability_test/cluster"
	"github.com/pingcap/tidb-test/stability_test/config"
	"github.com/pingcap/tidb-test/stability_test/nemesis"
	"github.com/pingcap/tidb-test/stability_test/suite"
	"github.com/pingcap/tidb-test/util"
	"github.com/pingcap/tidb/util/logutil"
	log "github.com/sirupsen/logrus"
	zaplog "github.com/pingcap/log"
)

var (
	configFile  = flag.String("c", "config.toml", "stability test config file")
	clusterType = flag.String("cluster", "ssh", "cluster type: [ssh, docker-compose, k8s]")
	pprofAddr   = flag.String("pprof", "0.0.0.0:16060", "pprof address")
	logFile     = flag.String("log-file", "", "log file")
	logLevel    = flag.String("L", "info", "log level: info, debug, warn, error, fatal")
)

func openDB(cfg *config.Config) (*sql.DB, error) {
	dbDSN := fmt.Sprintf("%s:%s@tcp(%s:%d)/test", cfg.User, cfg.Password, cfg.Host, cfg.Port)
	db, err := util.OpenDBWithRetry("mysql", dbDSN)
	if err != nil {
		return nil, err
	}
	db.SetMaxIdleConns(cfg.Suite.Concurrency)
	log.Info("DB opens successfully")
	return db, nil
}

func main() {
	// Initialize the default random number source.
	rand.Seed(time.Now().UTC().UnixNano())

	flag.Parse()

	zapFileConfig := zaplog.FileLogConfig{
		Filename:  *logFile,
		LogRotate: true,
	}
	fileConfig := logutil.FileLogConfig{
		FileLogConfig: zapFileConfig,
	}
	logutil.InitLogger(logutil.NewLogConfig(*logLevel, logutil.DefaultLogFormat, "",fileConfig, false))

	go func() {
		if err := http.ListenAndServe(*pprofAddr, nil); err != nil {
			log.Fatal(err)
		}
	}()

	cfg, err := config.ParseConfig(*configFile)
	if err != nil {
		log.Fatal(err)
	}

	log.Infof("%#v", cfg)

	// Prometheus metrics
	PushMetrics(cfg)

	// Create the cluster with cluster config file.
	// Start all services.

	// Create the database
	db, err := openDB(cfg)
	if err != nil {
		log.Fatal(err)
	}

	// Use "select 1" to check whether cluster is started OK or not.

	// Run all cases, and nemeses
	// Capture signal

	ctx, cancel := context.WithCancel(context.Background())

	sc := make(chan os.Signal, 1)
	signal.Notify(sc,
		syscall.SIGHUP,
		syscall.SIGINT,
		syscall.SIGTERM,
		syscall.SIGQUIT)

	go func() {
		sig := <-sc
		log.Infof("Got signal [%d] to exit.", sig)

		db.Close()
		cancel()
	}()

	// Initialize suites.
	suiteCases := suite.InitSuite(ctx, cfg, db)

	// Run all nemeses in background.
	nemesis.RunNemeses(ctx, &cfg.Nemeses, cluster.NewCluster(&cfg.Cluster))

	// Run suites.
	suite.RunSuite(ctx, suiteCases, cfg.Suite.Concurrency, db)
}