package clickhouse

import (
	"context"
	"fmt"
	"os"

	"github.com/ClickHouse/clickhouse-go/v2"
)

// public API
func NewClickHouseConn() (clickhouse.Conn, error) {
	host := os.Getenv("CLICKHOUSE_HOST")
	port := os.Getenv("CLICKHOUSE_PORT")
	return clickhouse.Open(&clickhouse.Options{
		Addr: []string{fmt.Sprintf("%v:%v", host, port)},
		Auth: clickhouse.Auth{
			Database: os.Getenv("CLICKHOUSE_DATABASE"),
			Username: os.Getenv("CLICKHOUSE_USERNAME"),
			Password: os.Getenv("CLICKHOUSE_PASSWORD"),
		},
	})
}

func GetClientDbTbl() string {
	return fmt.Sprintf("%v.%v", os.Getenv("CLICKHOUSE_DATABASE"), os.Getenv("CLICKHOUSE_CLIENT_LOG_TABLE"))
}

func GetSystemDbTbl() string {
	return fmt.Sprintf("%v.%v", os.Getenv("CLICKHOUSE_DATABASE"), os.Getenv("CLICKHOUSE_SYSTEM_LOG_TABLE"))
}

func GetSwgReportDbTbl() string {
	return fmt.Sprintf("%v.%v", os.Getenv("CLICKHOUSE_DATABASE"), os.Getenv("CLICKHOUSE_SWG_URL_COLLECT_TABLE"))
}

func GetSwgCfsCategoryDbTbl() string {
	return fmt.Sprintf("%v.%v", os.Getenv("CLICKHOUSE_DATABASE"), os.Getenv("CLICKHOUSE_CFS_CATEGORY_TABLE"))
}

func GetSwgCfsDomainsDbTbl() string {
	return fmt.Sprintf("%v.%v", os.Getenv("CLICKHOUSE_DATABASE"), os.Getenv("CLICKHOUSE_CFS_DOMAINS_TABLE"))
}

func GetSwgAppCtrlAppActivityDbTbl() string {
	return fmt.Sprintf("%v.%v", os.Getenv("CLICKHOUSE_DATABASE"), os.Getenv("CLICKHOUSE_APP_CTRL_APP_ACTIVITY_TABLE"))
}

func GetSwgAppCtrlAppInfoDbTbl() string {
	return fmt.Sprintf("%v.%v", os.Getenv("CLICKHOUSE_DATABASE"), os.Getenv("CLICKHOUSE_APP_CTRL_APP_INFO_TABLE"))
}

func GetSwgAppCtrlAppActivityInfoDbTbl() string {
	return fmt.Sprintf("%v.%v", os.Getenv("CLICKHOUSE_DATABASE"), os.Getenv("CLICKHOUSE_APP_CTRL_APP_ACTIVITY_INFO_TABLE"))
}

func ClickHouseCount(dql string, asName string) (ret uint64, err error) {
	conn, err := NewClickHouseConn()
	if err != nil {
		return 0, err
	}
	var countResult struct {
		LogNum uint64 `ch:"logNum"`
	}
	// countResult := make([]CountResult, 0)
	ctx := context.TODO()
	if err = conn.QueryRow(ctx, dql).ScanStruct(&countResult); err != nil {
		return 0, err
	}

	ret = countResult.LogNum

	return ret, nil
}